mailman-sync/mailman_sync.py

91 lines
2.4 KiB
Python
Executable File

#!/usr/bin/env python3
"""
Update Mailman 2 lists via a json API of the form {"LIST": ["ADDRESS", ...]}
"""
import argparse
import os
from pathlib import Path
import subprocess
import requests
def sync_members(
mailman_bin: Path, mailing_list: str, members: list[str], dry_run: bool
):
command = [
mailman_bin / "sync_members",
"--welcome-msg=no",
"--goodbye-msg=no",
"--notifyadmin=no",
"--file",
"-",
mailing_list,
]
if dry_run:
command.append("--no-change")
members_data = "\n".join(members)
output = subprocess.run(
command,
input=members_data,
encoding="ascii",
capture_output=True,
check=True,
)
print(output.stdout)
def main(mailman_bin: Path, api: str, api_auth: str, list_suffix: str, dry_run: bool):
r = requests.get(api, headers={"Authorization": api_auth})
if not r.ok:
print(f"Failed to get mailing list data from api: {r.status_code} {r.text}")
return
existing_lists = subprocess.run(
[mailman_bin / "list_lists", "-b"],
encoding="ascii",
capture_output=True,
check=True,
).stdout.split("\n")
certification_lists = r.json()
for name, members in certification_lists.items():
list_name = name + list_suffix
if list_name in existing_lists:
sync_members(mailman_bin, list_name, members, dry_run)
else:
print(f"Skipping {list_name}, as it does not exist in Mailman")
if __name__ == "__main__":
argp = argparse.ArgumentParser(description=__doc__)
argp.add_argument(
"--bin",
default="/usr/local/mailman",
type=Path,
help="Path to Mailman site admin scripts (default %(default)s)",
)
argp.add_argument("--api", required=True, help="API endpoint to retrieve JSON from")
argp.add_argument("--list-suffix", help="Suffix for mailing lists")
argp.add_argument(
"-n",
"--dry-run",
action="store_true",
help="Don't make changes, just print what would happen",
)
args = argp.parse_args()
if "API_AUTH" in os.environ:
api_auth = os.environ
else:
print("Missing API_AUTH environment variable")
exit(-1)
try:
main(args.bin, args.api, api_auth, args.list_suffix, args.dry_run)
except subprocess.CalledProcessError as e:
print(e.stderr)
raise