pretix-map 0.0.7__tar.gz → 0.1.1__tar.gz

{pretix_map-0.0.7/pretix_map.egg-info → pretix_map-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pretix-map
-Version: 0.0.7
+Version: 0.1.1
 Summary: An overview map of the catchment area of previous orders. Measured by postcode
 Author-email: MarkenJaden <jjsch1410@gmail.com>
 Maintainer-email: MarkenJaden <jjsch1410@gmail.com>
@@ -140,10 +140,12 @@ This command is essential for processing orders that were placed *before* the ma
 **Example Workflow:**
 
 1.  **Test with Dry Run (All Organizers):**
+
     .. code-block:: bash
 
         python manage.py geocode_existing_orders --dry-run
 2.  **(If satisfied) Run for Real (All Organizers):**
+
     .. code-block:: bash
 
         python manage.py geocode_existing_orders

{pretix_map-0.0.7 → pretix_map-0.1.1}/README.rst

@@ -125,10 +125,12 @@ This command is essential for processing orders that were placed *before* the ma
 **Example Workflow:**
 
 1.  **Test with Dry Run (All Organizers):**
+
     .. code-block:: bash
 
         python manage.py geocode_existing_orders --dry-run
 2.  **(If satisfied) Run for Real (All Organizers):**
+
     .. code-block:: bash
 
         python manage.py geocode_existing_orders

{pretix_map-0.0.7 → pretix_map-0.1.1/pretix_map.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pretix-map
-Version: 0.0.7
+Version: 0.1.1
 Summary: An overview map of the catchment area of previous orders. Measured by postcode
 Author-email: MarkenJaden <jjsch1410@gmail.com>
 Maintainer-email: MarkenJaden <jjsch1410@gmail.com>
@@ -140,10 +140,12 @@ This command is essential for processing orders that were placed *before* the ma
 **Example Workflow:**
 
 1.  **Test with Dry Run (All Organizers):**
+
     .. code-block:: bash
 
         python manage.py geocode_existing_orders --dry-run
 2.  **(If satisfied) Run for Real (All Organizers):**
+
     .. code-block:: bash
 
         python manage.py geocode_existing_orders

pretix_map-0.1.1/pretix_mapplugin/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.1"

pretix_map-0.1.1/pretix_mapplugin/tasks.py

@@ -0,0 +1,90 @@
+import logging
+from django.db import transaction
+from django.core.exceptions import ObjectDoesNotExist
+
+# --- Import django-scopes ---
+from django_scopes import scope
+
+# --- Use Pretix Celery app instance ---
+from pretix.celery_app import app
+# --- Import necessary Pretix models ---
+from pretix.base.models import Order, Organizer  # Import Organizer
+
+# --- Import your Geocode model and geocoding functions ---
+from .models import OrderGeocodeData
+from .geocoding import (
+    get_formatted_address_from_order,
+    geocode_address,
+    DEFAULT_NOMINATIM_USER_AGENT
+)
+
+logger = logging.getLogger(__name__)
+
+
+@app.task(bind=True, max_retries=3, default_retry_delay=60, ignore_result=True)
+def geocode_order_task(self, order_pk: int, nominatim_user_agent: str | None = None):
+    """
+    Celery task to geocode the address for a given order PK.
+    Accepts the Nominatim User-Agent as an argument.
+    Activates django-scopes organizer scope before querying scoped models.
+    """
+    order = None  # Initialize order to None
+    try:
+        try:
+            order = Order.objects.select_related('event__organizer', 'invoice_address').get(pk=order_pk)
+            organizer = order.event.organizer  # Get organizer from the fetched order
+        except ObjectDoesNotExist:
+            logger.error(f"Order with PK {order_pk} not found in geocode_order_task.")
+            return
+        with scope(organizer=organizer):
+            logger.info(
+                f"Starting geocoding task for Order {order.code} (PK: {order_pk}) within scope of Organizer '{organizer.slug}'")
+
+            relation_name = 'geocode_data'  # Ensure this matches your model
+            # Check existence within scope (safer)
+            if OrderGeocodeData.objects.filter(order_id=order_pk).exists():
+                logger.info(f"Geocode data already exists for Order {order.code} (checked within scope). Skipping.")
+                return
+
+            address_str = get_formatted_address_from_order(order)
+            if not address_str:
+                logger.info(f"Order {order.code} has no address suitable for geocoding. Storing null coordinates.")
+                # Store null within scope
+                with transaction.atomic():
+                    OrderGeocodeData.objects.update_or_create(
+                        order=order, defaults={'latitude': None, 'longitude': None}
+                    )
+                return
+
+            logger.debug(f"Attempting to geocode address for Order {order.code}: '{address_str}'")
+            coordinates = geocode_address(address_str, nominatim_user_agent=nominatim_user_agent)
+
+            # Store result within scope
+            with transaction.atomic():
+                if coordinates:
+                    latitude, longitude = coordinates
+                    obj, created = OrderGeocodeData.objects.update_or_create(
+                        order=order, defaults={'latitude': latitude, 'longitude': longitude}
+                    )
+                    log_level = logging.INFO if created else logging.DEBUG
+                    logger.log(log_level,
+                               f"Saved{' new' if created else ' updated'} geocode data for Order {order.code}: ({latitude}, {longitude})")
+                else:
+                    logger.warning(f"Geocoding failed for Order {order.code}. Storing null coordinates.")
+                    obj, created = OrderGeocodeData.objects.update_or_create(
+                        order=order, defaults={'latitude': None, 'longitude': None}
+                    )
+                    log_level = logging.INFO if created else logging.DEBUG
+                    logger.log(log_level,
+                               f"Saved{' new' if created else ' updated'} null geocode data for Order {order.code} after failed attempt.")
+        # --- Scope deactivated automatically here ---
+
+    # Keep outer exception handling
+    except ObjectDoesNotExist:
+        # This case is now handled earlier, but keep for safety
+        logger.error(f"Order with PK {order_pk} not found outside scope handling.")
+    except Exception as e:
+        logger.exception(
+            f"Unexpected error in geocode_order_task for Order PK {order_pk} (potentially outside scope): {e}")
+        # Retry on unexpected errors before scope activation
+        raise self.retry(exc=e)

pretix_map-0.0.7/pretix_mapplugin/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.0.7"

pretix_map-0.0.7/pretix_mapplugin/tasks.py

@@ -1,90 +0,0 @@
-import logging
-from django.db import transaction
-from django.core.exceptions import ObjectDoesNotExist
-
-# --- Use Pretix Celery app instance ---
-from pretix.celery_app import app
-# --- Import necessary Pretix models ---
-from pretix.base.models import Order
-
-# --- Import your Geocode model and geocoding functions ---
-from .models import OrderGeocodeData
-from .geocoding import (
-    get_formatted_address_from_order,
-    geocode_address,
-    DEFAULT_NOMINATIM_USER_AGENT  # Import default for safety/logging
-)
-
-logger = logging.getLogger(__name__)
-
-
-# Define the Celery task
-# bind=True gives access to self (the task instance) for retrying
-# ignore_result=True as we don't need the return value stored in Celery backend
-@app.task(bind=True, max_retries=3, default_retry_delay=60, ignore_result=True)
-def geocode_order_task(self, order_pk: int,
-                       nominatim_user_agent: str | None = None):  # Added nominatim_user_agent kwarg
-    """
-    Celery task to geocode the address for a given order PK.
-    Accepts the Nominatim User-Agent as an argument.
-    """
-    try:
-        # Fetch order with related address and country data efficiently
-        order = Order.objects.select_related(
-            'invoice_address',
-        ).get(pk=order_pk)
-        logger.info(f"Starting geocoding task for Order {order.code} (PK: {order_pk})")
-
-        # Check if already geocoded to prevent redundant work
-        # Replace 'geocode_data' if your related_name is different
-        relation_name = 'geocode_data'  # Ensure this matches your OrderGeocodeData.order related_name
-        if hasattr(order, relation_name) and getattr(order, relation_name) is not None:
-            logger.info(f"Geocode data already exists for Order {order.code}. Skipping.")
-            return  # Exit successfully
-
-        # 1. Get formatted address string
-        address_str = get_formatted_address_from_order(order)
-        if not address_str:
-            logger.info(f"Order {order.code} has no address suitable for geocoding. Storing null coordinates.")
-            # Store null to prevent reprocessing
-            with transaction.atomic():
-                OrderGeocodeData.objects.update_or_create(
-                    order=order,
-                    defaults={'latitude': None, 'longitude': None}
-                )
-            return  # Exit successfully, nothing to geocode
-
-        # 2. Perform geocoding, passing the user agent received by the task
-        logger.debug(f"Attempting to geocode address for Order {order.code}: '{address_str}'")
-        coordinates = geocode_address(address_str, nominatim_user_agent=nominatim_user_agent)
-
-        # 3. Store result (or null if failed) using atomic transaction
-        with transaction.atomic():
-            if coordinates:
-                latitude, longitude = coordinates
-                obj, created = OrderGeocodeData.objects.update_or_create(
-                    order=order,
-                    defaults={'latitude': latitude, 'longitude': longitude}
-                )
-                log_level = logging.INFO if created else logging.DEBUG  # Be less noisy on updates
-                logger.log(log_level,
-                           f"Saved{' new' if created else ' updated'} geocode data for Order {order.code}: ({latitude}, {longitude})")
-            else:
-                logger.warning(f"Geocoding failed for Order {order.code}. Storing null coordinates.")
-                # Store nulls to indicate an attempt was made and failed
-                obj, created = OrderGeocodeData.objects.update_or_create(
-                    order=order,
-                    defaults={'latitude': None, 'longitude': None}
-                )
-                log_level = logging.INFO if created else logging.DEBUG
-                logger.log(log_level,
-                           f"Saved{' new' if created else ' updated'} null geocode data for Order {order.code} after failed attempt.")
-
-    except ObjectDoesNotExist:  # More specific exception
-        logger.error(f"Order with PK {order_pk} not found in geocode_order_task.")
-        # Don't retry if the order doesn't exist
-    except Exception as e:
-        # Catch any other unexpected errors
-        logger.exception(f"Unexpected error in geocode_order_task for Order PK {order_pk}: {e}")
-        # Retry on potentially temporary errors (database, network issues etc.)
-        raise self.retry(exc=e)  # Let Celery handle retry logic