Merge pull request #1 from devdaydresden/add-pretix-pagination-support

Add pagination support for pretix source

Author: jandd

.gitignore

@@ -3,3 +3,4 @@
 /.env
 /.idea/
 /.venv/
+__pycache__/

devday_mailbot.py

@@ -64,16 +64,21 @@ def __init__(self, token, netloc, path):
         self.token, self.netloc, self.organizer = token, netloc, path[1:]
 
     def get_addresses(self) -> list[str]:
-        r = get(
-            f"https://{self.netloc}/api/v1/organizers/{self.organizer}/customers/",
-            headers={"Authorization": f"Token {self.token}"},
-        )
-        r.raise_for_status()
-        return [
-            c["email"].lower()
-            for c in r.json()["results"]
-            if c["is_active"] and c["is_verified"]
-        ]
+        addresses = []
+        url = f"https://{self.netloc}/api/v1/organizers/{self.organizer}/customers/"
+        while True:
+            r = get(url, headers={"Authorization": f"Token {self.token}"})
+            r.raise_for_status()
+            json_data = r.json()
+            addresses.extend([
+                c["email"].lower()
+                for c in json_data["results"]
+                if c["is_active"] and c["is_verified"]
+            ])
+            if not json_data["next"]:
+                break
+            url = json_data["next"]
+        return addresses
 
 
 def determine_address_source(src_url: str) -> AddressSource:
@@ -89,13 +94,13 @@ def determine_address_source(src_url: str) -> AddressSource:
 
 
 def process_mail(
-    smtp_conn: SMTP,
-    address_source: AddressSource,
-    sender_address: str,
-    default_to: str,
-    default_reply_to: str,
-    valid_sender_patterns: list[str],
-    mail: bytes,
+        smtp_conn: SMTP,
+        address_source: AddressSource,
+        sender_address: str,
+        default_to: str,
+        default_reply_to: str,
+        valid_sender_patterns: list[str],
+        mail: bytes,
 ):
     email_parser = BytesParser()
     mail_data = email_parser.parsebytes(mail)

get_addresses.py

@@ -0,0 +1,15 @@
+#!/usr/bin/env python3
+import os
+
+from devday_mailbot import determine_address_source
+
+if __name__ == "__main__":
+    address_source_url = os.getenv("MAILBOT_ADDRESS_SOURCE")
+    if not address_source_url:
+        print("set MAILBOT_ADDRESS_SOURCE environment variable")
+        exit(1)
+
+    source = determine_address_source(address_source_url)
+
+    for address in source.get_addresses():
+        print(address)

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "devday-mailbot"
-version = "0.1.1"
+version = "0.1.2"
 description = "Dev Day Mail bot"
 readme = "README.md"
 requires-python = ">=3.11"