baseline: 批次A-D 成果 + membership 半成品（测试红）

2026-09-11 23:11:35 +08:00
commit b3f3095d53
311 changed files with 40540 additions and 0 deletions
@@ -0,0 +1,39 @@
+"""Dramatiq 任务队列配置（broker 选择 + Django 集成）。
+
+- 生产 / 开发默认：RedisBroker（REDIS_URL，默认指向内网 192.168.5.7:6379）
+- 测试环境：通过环境变量 TASK_BROKER=stub 切换 StubBroker（进程内、可断言）
+- dramatiq-crontab 在 setup 模块导入后接管周期调度
+"""
+
+import os
+
+
+def get_broker():
+    broker_kind = os.environ.get("TASK_BROKER", "redis")
+
+    if broker_kind == "stub":
+        import dramatiq
+        from dramatiq.brokers.stub import StubBroker
+
+        broker = StubBroker(middleware=[
+            dramatiq.middleware.AgeLimit(),
+            dramatiq.middleware.TimeLimit(),
+            dramatiq.middleware.Callbacks(),
+            dramatiq.middleware.Retries(),
+        ])
+        dramatiq.set_broker(broker)
+        return broker
+
+    import dramatiq
+    from dramatiq.brokers.redis import RedisBroker
+
+    redis_url = os.environ.get(
+        "REDIS_URL",
+        "redis://192.168.5.7:6379/2",
+    )
+    broker = RedisBroker(url=redis_url)
+    dramatiq.set_broker(broker)
+    return broker
+
+
+broker = get_broker()