Immediate mypy fixes

src/traffcap/core/store.py

@@ -19,7 +19,7 @@ def new_traffic_notification() -> None:
     global store
     store["last_message"] += 1
 
-async def wait_for_notification(seconds: float = 0.5) -> AsyncGenerator[Any, Any]:
+async def wait_for_notification() -> AsyncGenerator[Any, Any]:
     """
     Wait for a change in the last_message. By default this uses
     the stdlib multiprocessing Manager. Which is good for ASGI/WSGI

src/traffcap/repositories/rule_repository.py

@@ -54,13 +54,13 @@ async def find_matching_rules(cls, rule: str) -> List[Rule]:
         """
         Find all rules that match this current rule
         """
-        rules = []
-        async with cls.session() as session:
-            results = await session.scalars(
-                select(RuleModel)
-                    .where(RuleModel.rule == rule)
-            )
-            for rule_item in results.all():
-                rules.append(Rule.model_validate(rule_item))
+        rules: list[Rule] = []
+        # async with cls.session() as session:
+        #     results = await session.scalars(
+        #         select(RuleModel)
+        #             .where(RuleModel.rule == rule)
+        #     )
+        #     for rule_item in results.all():
+        #         rules.append(Rule.model_validate(rule_item))
 
         return rules

src/traffcap/routes/traffic.py

@@ -38,7 +38,7 @@ async def traffic_firehose(websocket: WebSocket):
                 await websocket.send_text(response.model_dump_json())
 
             # Wait for an event from the message broker
-            async for _ in wait_for_notification(websocket):
+            async for _ in wait_for_notification():
                 # Check the connection
                 await sleep(0.5)
                 # TODO: Figure out how to check websocket connections without the ping