anzu 0.1.4__tar.gz → 0.1.5__tar.gz

anzu-0.1.5/.gitignore

@@ -0,0 +1,6 @@
+*.pyc
+__pycache__/
+*.pkl
+dist/
+build/
+.venv

{anzu-0.1.4 → anzu-0.1.5}/PKG-INFO

@@ -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/anzu/__init__.py

@@ -0,0 +1,5 @@
+import anzu.logger
+import anzu.websocket
+import anzu.worker
+
+__version__ = "2.3.0"

anzu-0.1.5/anzu/logger/__init__.py

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

anzu-0.1.5/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-0.1.5/anzu/websocket/__init__.py

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

anzu-0.1.5/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-0.1.5/anzu/worker/__init__.py

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

anzu-0.1.5/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-0.1.5/anzu/worker/constants.py

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

anzu-0.1.5/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-0.1.5/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.5/pyproject.toml

@@ -0,0 +1,38 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "anzu"
+version = "0.1.5"
+description = "Anzu package for workers"
+readme = "README.md"
+requires-python = ">=3.8"
+authors = [
+    {name = "Your Name", email = "your.email@example.com"}
+]
+license = {text = "MIT"}
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+]
+keywords = ["django", "queue", "update"]
+dependencies = [
+    "requests>=2.25.0",
+    "aiohttp>=3.7.3",
+    "python-socketio>=5.12.1"
+]
+
+[project.scripts]
+anzu = "anzu:main"
+
+[tool.hatch.version]
+path = "anzu/__init__.py"
+
+[tool.hatch.build.targets.sdist]
+include = [
+    "/anzu",
+]
+
+

anzu-0.1.4/anzu/__init__.py

@@ -1 +0,0 @@
-from .logger import (logger, redis_logger)

anzu-0.1.4/anzu.egg-info/PKG-INFO

@@ -1,107 +0,0 @@
-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
-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
-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
-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
-
-# Queue Updater
-
-A simple Python package for updating queue items in a Django service.
-
-## Installation
-
-```bash
-pip install anzu
-```
-
-## Usage
-
-### Using the QueueClient class (recommended)
-
-```python
-from anzu import QueueClient
-
-# Initialize with explicit parameters
-client = QueueClient(
-    django_url="https://your-django-service.com",
-    username="admin",
-    password="password",
-    service_endpoint="/api/queue/"
-)
-
-# Update a queue item
-client.update_queue_item(
-    qi_hash="abc123",
-    status="completed",
-    data={"result": "success"}
-)
-
-# Or initialize using environment variables
-import os
-
-os.environ["DJANGO_URL"] = "https://your-django-service.com"
-os.environ["DJANGO_SUPERUSER_USERNAME"] = "admin"
-os.environ["DJANGO_SUPERUSER_PASSWORD"] = "password"
-os.environ["SERVICE_ENDPOINT"] = "/api/queue/"
-
-client = QueueClient()
-client.update_queue_item("abc123", "completed")
-```
-
-### Using the legacy function (for backward compatibility)
-
-```python
-import os
-from anzu import update_queue_item
-
-# Set required environment variables
-os.environ["DJANGO_URL"] = "https://your-django-service.com"
-os.environ["DJANGO_SUPERUSER_USERNAME"] = "admin"
-os.environ["DJANGO_SUPERUSER_PASSWORD"] = "password"
-os.environ["SERVICE_ENDPOINT"] = "/api/queue/"
-
-# Update a queue item
-update_queue_item(
-    qi_hash="abc123",
-    status="completed",
-    data={"result": "success"}
-)
-```
-
-## Environment Variables
-
-- `DJANGO_URL`: URL of the Django service
-- `DJANGO_SUPERUSER_USERNAME`: Django superuser username
-- `DJANGO_SUPERUSER_PASSWORD`: Django superuser password
-- `SERVICE_ENDPOINT`: Service endpoint (defaults to '/')
-
-## License
-
-MIT

anzu-0.1.4/anzu.egg-info/SOURCES.txt

@@ -1,10 +0,0 @@
-LICENSE
-README.md
-pyproject.toml
-setup.py
-anzu/__init__.py
-anzu.egg-info/PKG-INFO
-anzu.egg-info/SOURCES.txt
-anzu.egg-info/dependency_links.txt
-anzu.egg-info/requires.txt
-anzu.egg-info/top_level.txt

anzu-0.1.4/anzu.egg-info/dependency_links.txt

@@ -1 +0,0 @@
-

anzu-0.1.4/anzu.egg-info/requires.txt

@@ -1,13 +0,0 @@
-requests>=2.25.0
-aiohttp>=3.7.3
-python-socketio>=5.12.1
-
-[dev]
-pytest>=6.0.0
-black>=22.0.0
-isort>=5.0.0
-mypy>=0.900
-flake8>=4.0.0
-
-[langfuse]
-langfuse>=2.0.0

anzu-0.1.4/anzu.egg-info/top_level.txt

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

anzu-0.1.4/pyproject.toml

@@ -1,61 +0,0 @@
-[build-system]
-requires = ["setuptools>=42", "wheel"]
-build-backend = "setuptools.build_meta"
-
-[project]
-name = "anzu"
-version = "0.1.4"
-description = "A package for updating queue items in a Django service"
-readme = "README.md"
-authors = [
-    {name = "Your Name", email = "your.email@example.com"}
-]
-license = {text = "MIT"}
-classifiers = [
-    "Programming Language :: Python :: 3",
-    "License :: OSI Approved :: MIT License",
-    "Operating System :: OS Independent",
-]
-keywords = ["django", "queue", "update"]
-dependencies = [
-    "requests>=2.25.0",
-    "aiohttp>=3.7.3",
-    "python-socketio>=5.12.1"
-]
-requires-python = ">=3.6"
-
-[project.optional-dependencies]
-langfuse = ["langfuse>=2.0.0"]
-dev = [
-    "pytest>=6.0.0",
-    "black>=22.0.0",
-    "isort>=5.0.0",
-    "mypy>=0.900",
-    "flake8>=4.0.0",
-]
-
-[project.urls]
-"Homepage" = "https://github.com/yourusername/queue_updater"
-"Bug Tracker" = "https://github.com/yourusername/queue_updater/issues"
-
-[tool.setuptools]
-packages = ["anzu"]
-
-[tool.black]
-line-length = 88
-target-version = ["py36", "py37", "py38", "py39", "py310"]
-
-[tool.isort]
-profile = "black"
-line_length = 88
-
-[tool.mypy]
-python_version = "3.6"
-warn_return_any = true
-warn_unused_configs = true
-disallow_untyped_defs = true
-disallow_incomplete_defs = true
-
-[tool.pytest.ini_options]
-testpaths = ["tests"]
-python_files = "test_*.py"

anzu-0.1.4/setup.cfg

@@ -1,4 +0,0 @@
-[egg_info]
-tag_build = 
-tag_date = 0
-

anzu-0.1.4/setup.py

@@ -1,27 +0,0 @@
-import setuptools
-
-with open("README.md", "r", encoding="utf-8") as fh:
-    long_description = fh.read()
-
-setuptools.setup(
-    name="anzu",
-    version="0.1.4",
-    author="Your Name",
-    author_email="your.email@example.com",
-    description="A package for updating queue items in a Django service",
-    long_description=long_description,
-    long_description_content_type="text/markdown",
-    url="https://github.com/yourusername/queue_updater",
-    packages=setuptools.find_packages(),
-    classifiers=[
-        "Programming Language :: Python :: 3",
-        "License :: OSI Approved :: MIT License",
-        "Operating System :: OS Independent",
-    ],
-    python_requires=">=3.6",
-    install_requires=[
-        "requests>=2.25.0",
-        "aiohttp>=3.7.3",
-        "python-socketio>=5.12.1",
-    ],
-)