changes

Author: marcoacierno

backend/association_membership/handlers/pretix/api.py

@@ -1,3 +1,4 @@
+import urllib.parse
 from typing import Literal
 
 import requests
@@ -6,7 +7,7 @@
 
 from conferences.models.conference import Conference
 
-METHODS = Literal["get"]
+METHODS = Literal["get", "post", "put", "patch", "delete"]
 
 
 class PretixAPI:
@@ -25,14 +26,19 @@ def for_conference(cls, conference: Conference):
         )
 
     def run_request(
-        self, url: str, *, method: METHODS = "get", qs: dict[str, str] = None
+        self,
+        url: str,
+        *,
+        method: METHODS = "get",
+        qs: dict[str, str] = None,
+        json: dict = None,
     ):
         headers = {"Authorization": f"Token {str(settings.PRETIX_API_TOKEN)}"}
 
         if qs:
             url = f"{url}?" + "&".join([f"{key}={value}" for key, value in qs.items()])
 
-        return getattr(requests, method)(url, headers=headers)
+        return getattr(requests, method)(url, headers=headers, json=json)
 
     def _request(self, endpoint: str, **kwargs):
         url = f"{self.base_url}/{endpoint}/"
@@ -54,6 +60,30 @@ def get_categories(self) -> dict:
 
     def get_all_attendee_tickets(self, attendee_email: str) -> list[dict]:
         response = self._request(
-            "tickets/attendee-tickets", qs={"attendee_email": attendee_email}
+            "tickets/attendee-tickets",
+            qs={"attendee_email": urllib.parse.quote_plus(attendee_email)},
         )
         return response.json()
+
+    def has_attendee_ticket(
+        self, attendee_email: str, additional_events: list[dict] = None
+    ) -> bool:
+        additional_events = additional_events or []
+        events = [
+            {
+                "organizer_slug": self.organizer,
+                "event_slug": self.event,
+            }
+        ] + additional_events
+
+        response = self._request(
+            "tickets/attendee-has-ticket",
+            method="post",
+            json={
+                "attendee_email": attendee_email,
+                "events": events,
+            },
+        )
+
+        data = response.json()
+        return data["user_has_admission_ticket"]

backend/visa/admin.py

@@ -1,6 +1,7 @@
 from django.utils.safestring import mark_safe
 
 from django.urls import path
+from association_membership.handlers.pretix.api import PretixAPI
 from ordered_model.admin import (
     OrderedInlineModelAdminMixin,
     OrderedModelAdmin,
@@ -81,6 +82,10 @@ def save_form(self, request, form, change):
         return obj
 
     def process_now(self, obj):
+        pretix_api = PretixAPI.for_conference(obj.conference)
+        if not pretix_api.has_attendee_ticket(obj.email):
+            return "No attendee ticket found! Can't generate invitation letter."
+
         return mark_safe(
             '<input type="submit" name="_process_now" value="Process now" />'
         )

backend/visa/models.py

@@ -150,6 +150,9 @@ def schedule(self):
     def get_config(self):
         return self.conference.invitation_letter_config
 
+    def __str__(self):
+        return f"{self.full_name} - {self.conference.name}"
+
     class Meta:
         constraints = [
             UniqueConstraint(

backend/visa/tests/test_admin.py

@@ -72,19 +72,31 @@ def test_edit_dynamic_document_button_is_empty_for_static_docs():
     assert html == ""
 
 
-def test_invitation_letter_request_admin():
+def test_invitation_letter_request_admin(mock_has_ticket):
     admin = InvitationLetterRequestAdmin(
         model=InvitationLetterRequest, admin_site=AdminSite()
     )
 
     invitation_letter_request = InvitationLetterRequestFactory()
+    mock_has_ticket(invitation_letter_request.conference, True)
 
     assert 'name="_process_now"' in admin.process_now(invitation_letter_request)
     assert "Generate the invitation letter first" == admin.send_via_email(
         invitation_letter_request
     )
 
 
+def test_generate_button_doesnt_appear_with_no_ticket(mock_has_ticket):
+    admin = InvitationLetterRequestAdmin(
+        model=InvitationLetterRequest, admin_site=AdminSite()
+    )
+
+    invitation_letter_request = InvitationLetterRequestFactory()
+    mock_has_ticket(invitation_letter_request.conference, False)
+
+    assert "No attendee ticket found!" in admin.process_now(invitation_letter_request)
+
+
 def test_invitation_letter_request_admin_post_processing_redirects_to_changelist(
     rf, admin_user
 ):