Merge pull request #2 from devdaydresden/pretix-orders-address-source

Pretix orders address source

Author: jandd

README.md

@@ -46,6 +46,18 @@ pretix://:<pretix_token>@pretix.eu/<organizer>
 
 `<organizer>` is the short name of the organizer. You can find valid organizer names at https://pretix.eu/control/organizers/.
 
+A variant of the Pretix address source that uses addresses from orders is available via the
+`pretix-orders` address scheme:
+
+Pretix orders URI Example:
+```
+pretix-orders://:<pretix_token>@pretix.eu/<organizer>/<event>
+```
+
+The `<pretix_token>` and `<organizer>` fields are the same as for the `pretix` URI scheme.
+
+`<event>` is the technical identifier for the event.
+
 ## License
 
 This program is free software: you can redistribute it and/or modify

devday_mailbot.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 #
-# Copyright 2024 Dev Day Dresden e.V.
+# Copyright Dev Day Dresden e.V.
 #
 # This program is free software: you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
@@ -59,10 +59,16 @@ def get_addresses(self) -> list[str]:
             return sourcefile.readlines()
 
 
-class PretixAddressSource(AddressSource):
+class BasePretixAddressSource(AddressSource):
     def __init__(self, token, netloc, path):
-        self.token, self.netloc, self.organizer = token, netloc, path[1:]
+        if len(path) < 2:
+            raise ValueError("path is too short")
+        if path.count("/") < 1:
+            raise ValueError("invalid path")
+        self.token, self.netloc, self.organizer = token, netloc, path[1:].split("/")[0]
 
+
+class PretixCustomerAddressSource(BasePretixAddressSource):
     def get_addresses(self) -> list[str]:
         addresses = []
         url = f"https://{self.netloc}/api/v1/organizers/{self.organizer}/customers/"
@@ -81,14 +87,44 @@ def get_addresses(self) -> list[str]:
         return addresses
 
 
+class PretixOrderAddressSource(BasePretixAddressSource):
+    def __init__(self, token, netloc, path):
+        if len(path) < 4:
+            raise ValueError("path is too short")
+        if path.count("/") != 2:
+            raise ValueError("invalid path")
+        super().__init__(token, netloc, path)
+        self.event = path[1:].split("/")[1]
+
+    def get_addresses(self) -> list[str]:
+        addresses = []
+        url = f"https://{self.netloc}/api/v1/organizers/{self.organizer}/events/{self.event}/orders/?status=p"
+        while True:
+            r = get(url, headers={"Authorization": f"Token {self.token}"})
+            r.raise_for_status()
+            json_data = r.json()
+            for order in json_data["results"]:
+                for position in order["positions"]:
+                    if position["attendee_email"]:
+                        addresses.append(position["attendee_email"])
+                    addresses.append(order["email"])
+            if not json_data["next"]:
+                break
+            url = json_data["next"]
+        return list(sorted(set(addresses)))
+
+
 def determine_address_source(src_url: str) -> AddressSource:
     url = urlparse(src_url)
 
     if url.scheme == "file":
         return FileAddressSource(url.path)
 
     if url.scheme == "pretix":
-        return PretixAddressSource(url.password, url.hostname, url.path)
+        return PretixCustomerAddressSource(url.password, url.hostname, url.path)
+
+    if url.scheme == "pretix-orders":
+        return PretixOrderAddressSource(url.password, url.hostname, url.path)
 
     raise ValueError(f"unsupported URL scheme {url.scheme}")