modify: uv.lock

src/queue_sqlite/__init__.py

@@ -33,7 +33,7 @@ error_log_handler.setFormatter(
 logging.getLogger().addHandler(error_log_handler)
 
 __title__ = "queue_sqlite"
-__version__ = "0.2.1"
+__version__ = "0.2.2"
 __author__ = "chakcy"
 __email__ = "947105045@qq.com"
 __license__ = "MIT"

src/queue_sqlite/scheduler/__init__.py

@@ -12,11 +12,14 @@ from .standard import StandardQueueScheduler
 from ._async import AsyncQueueScheduler
 from .base import BaseScheduler
 from ..model import MessageItem, SchedulerConfig
-from typing import Callable, Literal
+from typing import Callable, Literal, Dict
 import logging
 
 
-SCHEDULER_TYPES = {"standard": StandardQueueScheduler, "async": AsyncQueueScheduler}
+SCHEDULER_TYPES: Dict[str, type[BaseScheduler]] = {
+    "standard": StandardQueueScheduler,
+    "async": AsyncQueueScheduler,
+}
 
 try:
     from .qt import QtQueueScheduler
@@ -39,12 +42,12 @@ class QueueScheduler(BaseScheduler):
         config: SchedulerConfig = SchedulerConfig(),
     ):
         scheduler_class = SCHEDULER_TYPES.get(scheduler_type, None)
-        if not scheduler_class:
+        if scheduler_class is None:
             raise ValueError(f"Invalid scheduler type: {scheduler_type}")
         self.scheduler = scheduler_class(config)
 
-        if self.scheduler:
-            self.queue_operation = self.scheduler.queue_operation
+        # if self.scheduler:
+        #     self.queue_operation = self.scheduler.queue_operation
 
     def send_message(self, message: MessageItem, callback: Callable):
         if not self.scheduler:

src/queue_sqlite/scheduler/base/base_scheduler.py

@@ -1,11 +1,15 @@
 from abc import ABC, abstractmethod
 from typing import Callable
 from ...model import MessageItem, SchedulerConfig
+from queue_sqlite_core import ShardedQueueOperation
 
 
 class BaseScheduler(ABC):
     """调度器抽象类"""
 
+    @property
+    def queue_operation(self) -> ShardedQueueOperation: ...
+
     def __init__(self, config: SchedulerConfig = SchedulerConfig()):
         pass
 

src/queue_sqlite/scheduler/cleanup_scheduler.py

@@ -21,7 +21,7 @@ class CleanupScheduler:
     def __init__(
         self,
         queue_operation: ShardedQueueOperation,
-        interval_minutes=60,
+        interval_minutes=1,
         remove_days=30,
     ):
         self.queue_operation = queue_operation

src/queue_sqlite_core/Cargo.lock

@@ -486,7 +486,7 @@ dependencies = [
 
 [[package]]
 name = "queue_sqlite_core"
-version = "0.2.0"
+version = "0.2.1"
 dependencies = [
  "chrono",
  "pyo3",

src/queue_sqlite_core/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "queue_sqlite_core"
-version = "0.2.0"
+version = "0.2.1"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html