Refactor code structure for improved readability and maintainability

core/queue.py

@@ -0,0 +1,61 @@
+import logging
+from pydantic import BaseModel
+from typing import Any, Optional
+from gitea.models import IssueModel, PullRequestModel
+
+logger: logging.Logger = logging.getLogger("work-queue")
+
+
+class WorkItem(BaseModel):
+    repo_full_name: str
+    task_type: str  # 'issue' or 'pr'
+    task_number: int
+    task_info: IssueModel | PullRequestModel
+    notification_id: Optional[int] = None
+    priority: int = 0
+
+
+class WorkQueue:
+    """Thread-safe work queue grouped by repo."""
+
+    def __init__(self) -> None:
+        self._queue: list[WorkItem] = []
+        self._enqueued_repos: set[str] = set()
+
+    def enqueue(self, item: WorkItem) -> None:
+        self._queue.append(item)
+        self._enqueued_repos.add(item.repo_full_name)
+        logger.info(f"Enqueued work item: {item.task_type} #{item.task_number} for {item.repo_full_name}")
+
+    def enqueue_batch(self, items: list[WorkItem]) -> None:
+        for item in items:
+            self.enqueue(item)
+
+    def get_repo_work(self, repo: str) -> list[WorkItem]:
+        """Get all work items for a specific repo."""
+        items: list[WorkItem] = [
+            item for item in self._queue if item.repo_full_name == repo
+        ]
+        logger.info(f"Retrieved {len(items)} work items for repository: {repo}")
+        return items
+
+    def remove_repo_work(self, repo: str) -> None:
+        """Remove all work items for a specific repo."""
+        self._queue = [
+            item for item in self._queue if item.repo_full_name != repo
+        ]
+        self._enqueued_repos.discard(repo)
+        logger.info(f"Removed all work items for repository: {repo}")
+
+    def get_next_repo(self) -> str | None:
+        """Get the next repo with work, or None if empty."""
+        if not self._enqueued_repos:
+            return None
+        return next(iter(self._enqueued_repos))
+
+    @property
+    def is_empty(self) -> bool:
+        return len(self._queue) == 0
+
+    def __len__(self) -> int:
+        return len(self._queue)