anzu 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

anzu/__init__.py

@@ -1 +1,5 @@
-from .logger import (logger, redis_logger)
+import anzu.logger
+import anzu.websocket
+import anzu.worker
+
+__version__ = "2.3.0"

anzu/logger/__init__.py

@@ -0,0 +1 @@
+from .logger import redis_logger, logger, AnzuLogger

anzu/logger/logger.py

@@ -0,0 +1,45 @@
+import logging
+
+
+class AnzuLogger:
+    _instance = None
+
+    def __new__(cls, level: str = logging.INFO, name: str = __name__):
+        # Ensures that only one instance of the logger is created
+        if cls._instance is None:
+            cls._instance = super(AnzuLogger, cls).__new__(cls)
+            cls._instance._logger = logging.getLogger(name)
+            cls._instance._logger.setLevel(logging.INFO)  # Default log level
+            handler = logging.StreamHandler()  # You can also use FileHandler
+            formatter = logging.Formatter(
+                "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+            )
+            handler.setFormatter(formatter)
+            cls._instance._logger.addHandler(handler)
+
+        return cls._instance._logger
+
+    def get_logger(self):
+        """Returns the logger instance."""
+        return self._instance._logger
+
+    def set_name(self, name: str):
+        """Sets the name of the logger."""
+        self._instance._logger = logging.getLogger(name)
+
+    def add_file_handler(self, file_name: str):
+        """Adds a file handler to the logger."""
+        handler = logging.FileHandler(file_name)
+        formatter = logging.Formatter(
+            "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+        )
+        handler.setFormatter(formatter)
+        self._instance._logger.addHandler(handler)
+
+    def remove_handler(self, handler):
+        """Removes a handler from the logger."""
+        self._instance._logger.removeHandler(handler)
+
+
+logger = AnzuLogger()
+redis_logger = AnzuLogger(name="redis_logger")

anzu/websocket/__init__.py

@@ -0,0 +1 @@
+from .sockets import send_socket_response, emit_socket_event

anzu/websocket/sockets.py

@@ -0,0 +1,14 @@
+import socketio
+
+# Create client
+sio = socketio.Client(ssl_verify=False)
+sio_server = 'https://django:9500'
+
+def emit_socket_event(event, data):
+    if sio.connected is False:
+        sio.connect(sio_server)
+
+    sio.emit(event, data)
+
+def send_socket_response(event, data):
+    emit_socket_event(event, data)

anzu/worker/__init__.py

@@ -0,0 +1,2 @@
+from .client import update_queue_item
+from .constants import WAITING, PROCESSING, FAILURE, SUCCESS

anzu/worker/client.py

@@ -0,0 +1,112 @@
+import json
+import requests
+from requests.auth import HTTPBasicAuth
+from .exceptions import QueueUpdateError
+from .utils import get_env_var, setup_logger
+
+logger = setup_logger()
+
+
+class QueueClient:
+    """Client for updating queue items in a Django service."""
+
+    def __init__(self, django_url=None, username=None, password=None, service_endpoint=None, verify_ssl=None):
+        """
+        Initialize the queue client with connection parameters.
+
+        Args:
+            django_url (str, optional): URL of the Django service. Defaults to DJANGO_URL env var.
+            username (str, optional): Django superuser username. Defaults to DJANGO_SUPERUSER_USERNAME env var.
+            password (str, optional): Django superuser password. Defaults to DJANGO_SUPERUSER_PASSWORD env var.
+            service_endpoint (str, optional): Service endpoint. Defaults to SERVICE_ENDPOINT env var or '/'.
+            verify_ssl (bool, optional): Whether to verify SSL certificates. Defaults to True unless URL contains 'https'.
+        """
+        self.django_url = django_url or get_env_var('DJANGO_URL', required=True)
+        self.username = username or get_env_var('DJANGO_SUPERUSER_USERNAME', required=True)
+        self.password = password or get_env_var('DJANGO_SUPERUSER_PASSWORD', required=True)
+        self.service_endpoint = service_endpoint or get_env_var('SERVICE_ENDPOINT', '/')
+
+        # Handle the verify_ssl parameter
+        if verify_ssl is None:
+            self.verify_ssl = False if 'https' in self.django_url else True
+        else:
+            self.verify_ssl = verify_ssl
+
+        self.headers = {
+            'Content-Type': 'application/json'
+        }
+
+    def update_queue_item(self, qi_hash, status, data=None, error=None, timeout=10):
+        """
+        Update a queue item with the given status and optional data or error.
+
+        Args:
+            qi_hash (str): Hash of the queue item to update.
+            status (str): New status for the queue item.
+            data (dict, optional): Additional data to update. Defaults to None.
+            error (str, optional): Error message if applicable. Defaults to None.
+            timeout (int, optional): Request timeout in seconds. Defaults to 10.
+
+        Returns:
+            dict: Response data from the API
+
+        Raises:
+            TypeError: If data is not a dictionary.
+            QueueUpdateError: If the update request fails.
+        """
+        if not qi_hash:
+            logger.warning("No queue item to update")
+            return
+
+        payload = {
+            "status": status
+        }
+
+        if data is not None:
+            if not isinstance(data, dict):
+                raise TypeError(f"Expected dict, got {type(data)} with value: {data}")
+            payload.update(data)
+
+        if error is not None:
+            payload['error'] = error
+
+        url = f'{self.django_url}{self.service_endpoint}{qi_hash}/'
+
+        response = requests.request(
+            "PATCH",
+            url,
+            headers=self.headers,
+            data=json.dumps(payload),
+            auth=HTTPBasicAuth(self.username, self.password),
+            verify=self.verify_ssl,
+            timeout=timeout
+        )
+
+        if response.status_code not in [200, 204]:
+            response_text = None
+            if 'text/html' in response.headers.get('Content-Type', ''):
+                logger.error(f"Failed to update queue item. Status Code: {response.status_code}")
+            else:
+                response_text = response.text
+                logger.error(
+                    f"Failed to update queue item. Status Code: {response.status_code}, Response: {response_text}")
+
+            raise QueueUpdateError(response.status_code, response_text)
+
+        try:
+            return response.json()
+        except (ValueError, json.JSONDecodeError):
+            # Return empty dict if no JSON in response
+            return {}
+
+
+# For backwards compatibility
+def update_queue_item(qi_hash, status, data=None, error=None):
+    """
+    Legacy function to update a queue item. Uses environment variables for connection details.
+
+    This function exists for backwards compatibility with the original code.
+    New code should use the QueueClient class instead.
+    """
+    client = QueueClient()
+    return client.update_queue_item(qi_hash, status, data, error)

anzu/worker/constants.py

@@ -0,0 +1,4 @@
+WAITING = "0"
+PROCESSING = "1"
+SUCCESS = "2"
+FAILURE = "3"

anzu/worker/exceptions.py

@@ -0,0 +1,14 @@
+class QueueUpdaterError(Exception):
+    """Base exception for queue updater."""
+    pass
+
+
+class QueueUpdateError(QueueUpdaterError):
+    """Exception raised when updating a queue item fails."""
+    def __init__(self, status_code, response_text=None):
+        self.status_code = status_code
+        self.response_text = response_text
+        message = f"Failed to update queue item. Status Code: {status_code}"
+        if response_text:
+            message += f", Response: {response_text}"
+        super().__init__(message)

anzu/worker/utils.py

@@ -0,0 +1,25 @@
+import logging
+import os
+
+
+def get_env_var(name, default=None, required=False):
+    """Get environment variable with proper error handling."""
+    value = os.environ.get(name, default)
+    if required and value is None:
+        raise ValueError(f"Required environment variable '{name}' is not set")
+    return value
+
+
+def setup_logger(name='queue_updater', level=logging.INFO):
+    """Setup and return a logger with the given name and level."""
+    logger = logging.getLogger(name)
+
+    # Only add handler if not already added to avoid duplicate logs
+    if not logger.handlers:
+        handler = logging.StreamHandler()
+        formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+        handler.setFormatter(formatter)
+        logger.addHandler(handler)
+
+    logger.setLevel(level)
+    return logger

{anzu-0.1.4.dist-info → anzu-0.1.5.dist-info}/METADATA

@@ -1,35 +1,19 @@
 Metadata-Version: 2.4
 Name: anzu
-Version: 0.1.4
-Summary: A package for updating queue items in a Django service
-Home-page: https://github.com/yourusername/queue_updater
-Author: Your Name
+Version: 0.1.5
+Summary: Anzu package for workers
 Author-email: Your Name <your.email@example.com>
 License: MIT
-Project-URL: Homepage, https://github.com/yourusername/queue_updater
-Project-URL: Bug Tracker, https://github.com/yourusername/queue_updater/issues
+License-File: LICENSE
 Keywords: django,queue,update
-Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
-Requires-Python: >=3.6
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: requests>=2.25.0
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.8
 Requires-Dist: aiohttp>=3.7.3
 Requires-Dist: python-socketio>=5.12.1
-Provides-Extra: langfuse
-Requires-Dist: langfuse>=2.0.0; extra == "langfuse"
-Provides-Extra: dev
-Requires-Dist: pytest>=6.0.0; extra == "dev"
-Requires-Dist: black>=22.0.0; extra == "dev"
-Requires-Dist: isort>=5.0.0; extra == "dev"
-Requires-Dist: mypy>=0.900; extra == "dev"
-Requires-Dist: flake8>=4.0.0; extra == "dev"
-Dynamic: author
-Dynamic: home-page
-Dynamic: license-file
-Dynamic: requires-python
+Requires-Dist: requests>=2.25.0
+Description-Content-Type: text/markdown
 
 # Queue Updater
 
@@ -104,4 +88,4 @@ update_queue_item(
 
 ## License
 
-MIT
+MIT

anzu-0.1.5.dist-info/RECORD

@@ -0,0 +1,16 @@
+anzu/__init__.py,sha256=rNEjTqQzvpIw6-SKRv1Vy68XnX99GUoSYSKqCCFEHXM,83
+anzu/__main__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+anzu/logger/__init__.py,sha256=kb2kDG1QFvRU48_Seod3RHVCrpq1udgzACOdmy7-Rg8,52
+anzu/logger/logger.py,sha256=M6oVFVUFD053BXWKEkj0Mz--d3FbvFGcKWFexsYsbco,1584
+anzu/websocket/__init__.py,sha256=vJVQgpYZZeJtIt9KrlhP049l_Xx_eFpSQrZlLxyluuQ,60
+anzu/websocket/sockets.py,sha256=VOGePsUkBMvP-54gabY6nPW5zAWMOujPoBenbIwiPPE,309
+anzu/worker/__init__.py,sha256=6SGGl1IOaSfJP0lOHEYI-8ENv_Ed8FvpNzCc0UMYkGw,98
+anzu/worker/client.py,sha256=SBBPAxJu9n1vfoqTSLu-SvjsARx0JvB4NhHEJ4bTLks,4347
+anzu/worker/constants.py,sha256=Z_w0Hr-PtZ9mLDJwEkrfMKebCdm7GJidNXtjnJwQzkc,59
+anzu/worker/exceptions.py,sha256=At2boea4ABjyph9b59pAyUIaSF5C0zl8Scqjxmjr-R8,524
+anzu/worker/utils.py,sha256=mcUMb3J4MhsLrJnL57_ULRd4END2_MYXZ-6lIwZJjzE,835
+anzu-0.1.5.dist-info/METADATA,sha256=Z8C3LKs5mwtVDjFLRiryghRzEFIbXXnB1lW_acjl-kU,2150
+anzu-0.1.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+anzu-0.1.5.dist-info/entry_points.txt,sha256=ZzQxwCwDcp-JpCDaU6LiacEGNk7MhbeQgE1xhQqBUsQ,35
+anzu-0.1.5.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+anzu-0.1.5.dist-info/RECORD,,

{anzu-0.1.4.dist-info → anzu-0.1.5.dist-info}/WHEEL

@@ -1,5 +1,4 @@
 Wheel-Version: 1.0
-Generator: setuptools (78.1.0)
+Generator: hatchling 1.27.0
 Root-Is-Purelib: true
 Tag: py3-none-any
-

anzu-0.1.5.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+anzu = anzu:main

anzu-0.1.4.dist-info/RECORD

@@ -1,6 +0,0 @@
-anzu/__init__.py,sha256=yapPQwS18DmiLOYcWJc2tK5KUTVs5lrsW_zmwUcb9zo,43
-anzu-0.1.4.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-anzu-0.1.4.dist-info/METADATA,sha256=nRyxc6NCUk0g-LWouq22y_KfRK2vj9NlBiVujheN0DI,2805
-anzu-0.1.4.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-anzu-0.1.4.dist-info/top_level.txt,sha256=ScsUMREYhumuqwuPGmSlaT4aifa-g8khMiFr7-h-xQ0,5
-anzu-0.1.4.dist-info/RECORD,,

anzu-0.1.4.dist-info/top_level.txt

@@ -1 +0,0 @@
-anzu