Add create_members and create_bans

import_mailman3_list_config.py

@@ -9,7 +9,9 @@ from pathlib import Path
 from requests.auth import HTTPBasicAuth
 
 REST_USER="restadmin"
-REST_PATH="http://127.0.0.1:8001/3.0/lists"
+REST_PATH="http://127.0.0.1:8001/3.0"
+REST_PASS = Path("/run/secrets/mailman_rest_password").read_text().strip()
+REST_AUTH = HTTPBasicAuth(REST_USER, REST_PASS)
 
 BANNER="""
             _) |                 __ /                _|_)
@@ -19,6 +21,55 @@ _|_|_|\__,_|_|_|_|_|_|\__,_|_| _|___/\__|\___/_| _|_|  _|\__, |
 """
 
 
+# def create_user(email):
+#     result = requests.get(f"{REST_PATH}/users/{email}")
+#     print(result)
+#     print(result.text)
+
+#     if not result.ok:
+#         print(f"user {email} not found, creating")
+#         result = requests.post(f"{REST_PATH}/users", auth=REST_AUTH, json={"email": email})
+#         if (result.ok):
+#             print("success")
+#             print(result)
+#             print(result.text)
+#         else:
+#             print("failed")
+#             print(result)
+#             print(result.reason)
+#             return False
+#     return True
+
+def create_member(email, mlist, role):
+    result = requests.post(f"{REST_PATH}/members", auth=REST_AUTH, json={"list_id": mlist.replace('@', '.'), "subscriber": email, "role": role})
+    if (result.ok):
+        print(f"{email} <- {role} success")
+        print(result)
+        print(result.text)
+        return True
+    else:
+        print(f"{email} <- {role} failed")
+        print(result)
+        print(result.reason)
+        print(result.text)
+        return False
+
+
+def create_ban(email, mlist):
+    result = requests.post(f"{REST_PATH}/lists/{mlist}/bans", auth=REST_AUTH, json={"email": email})
+    if (result.ok):
+        print(f"{email} <- BAN success")
+        print(result)
+        print(result.text)
+        return True
+    else:
+        print(f"{email} <- BAN failed")
+        print(result)
+        print(result.reason)
+        print(result.text)
+        return False
+
+
 def parse_args(args):
     parser = argparse.ArgumentParser("load a json config into local mailman configuration")
 
@@ -37,16 +88,13 @@ def main(args):
     with pargs.patch.open() as inf:
         p = json.load(inf)
 
-    rest_pass = Path("/run/secrets/mailman_rest_password").read_text().strip()
-    rest_auth = HTTPBasicAuth(REST_USER, rest_pass)
-
     # determine if list exists, if not create with defaults
-    result = requests.get(REST_PATH, auth=rest_auth)
+    result = requests.get(REST_PATH + '/lists', auth=REST_AUTH)
     if result.ok:
         if (not (pargs.list in [x['fqdn_listname'] for x in result.json()['entries']])):
             # create list
             print("list not found: creating")
-            result = requests.post(f"{REST_PATH}", auth=rest_auth, json={"fqdn_listname": pargs.list})
+            result = requests.post(f"{REST_PATH}/lists", auth=REST_AUTH, json={"fqdn_listname": pargs.list})
             if (result.ok):
                 print("success")
                 print(result)
@@ -55,6 +103,7 @@ def main(args):
                 print("failed")
                 print(result)
                 print(result.reason)
+                print(result.text)
                 return 1
             # list exists
     else:
@@ -65,11 +114,17 @@ def main(args):
 
 
     # fixme pull out owner, moderator keys, we'll poke them in separately
+    owners = []
+    moderators = []
+    ban_list = []
     if "owner" in p:
+        owners = p["owner"]
         del p["owner"]
     if "moderator" in p:
+        moderators = p["moderator"]
         del p["moderator"]
     if "ban_list" in p:
+        ban_list = p["ban_list"]
         del p["ban_list"]
 
     # patch config
@@ -77,7 +132,7 @@ def main(args):
     for k, v in p.items():
         if (type(v) == bool):
             v = str(v)
-        result = requests.patch(f"{REST_PATH}/{pargs.list}/config", auth=rest_auth, json={k: v})
+        result = requests.patch(f"{REST_PATH}/lists/{pargs.list}/config", auth=REST_AUTH, json={k: v})
         if result.ok:
             cnt += 1
             print(f"*** [OK] {k} success")
@@ -93,5 +148,16 @@ def main(args):
 
     print(f"- {cnt} / {len(p) - 1} succeeded.")
 
+    print("assigning roles")
+    for user in owners:
+        create_member(user, pargs.list, "owner")
+
+    for user in moderators:
+        create_member(user, pargs.list, "moderator")
+
+    print("banning")
+    for user in ban_list:
+        create_ban(user, pargs.list)
+
 if __name__ == "__main__":
     sys.exit(main(sys.argv))