Release 2025-05-19

pyworker/pyworker/tasks/base.py

@@ -1,64 +0,0 @@
-"""
-Base task module for the pyworker package.
-"""
-
-from abc import ABC, abstractmethod
-from contextlib import AbstractContextManager
-from typing import Any, Optional, Type
-
-from pyworker.database import get_db_connection
-from pyworker.utils.app_logging import setup_logging
-
-logger = setup_logging(__name__)
-
-
-class Task(ABC):
-    """Base class for all worker tasks."""
-
-    def __init__(self, name: str):
-        """
-        Initialize a task.
-
-        Args:
-            name: The name of the task.
-        """
-        self.name = name
-        self.logger = setup_logging(f"pyworker.task.{name}")
-        self.conn: Optional[Any] = None
-        self._context: Optional[AbstractContextManager[Any]] = None
-
-    def __enter__(self):
-        """Enter context manager, acquiring a database connection."""
-        self._context = get_db_connection()
-        self.conn = self._context.__enter__()
-        return self
-
-    def __exit__(
-        self,
-        exc_type: Optional[Type[BaseException]],
-        exc_val: Optional[BaseException],
-        exc_tb: Optional[Any],
-    ) -> Optional[bool]:
-        """Exit context manager, releasing the database connection."""
-        if self._context:
-            return self._context.__exit__(exc_type, exc_val, exc_tb)
-        return None
-
-    @abstractmethod
-    def run(self, *args: Any, **kwargs: Any) -> Any:
-        """
-        Run the task.
-
-        This method must be implemented by subclasses.
-
-        Args:
-            *args: Variable length argument list.
-            **kwargs: Arbitrary keyword arguments.
-
-        Returns:
-            The result of the task.
-        """
-        pass
-
-    def __str__(self) -> str:
-        return f"{self.__class__.__name__}(name={self.name})"