.gitignore

@@ -1,5 +1,3 @@
 __pycache__/
 out/
 .venv
-.idea/
-.vscode/

src/config.py

@@ -3,6 +3,7 @@ import random
 vfyd_tags = [
     "vfyd accounting",
     "vfyd discuss",
+    "vfyd easy",
     "vfyd editor",
     "vfyd js",
     "vfyd marketing",
@@ -16,8 +17,6 @@ vfyd_tags = [
     "vfyd website",
     "vfyd industry",
     "vfyd voip",
-    "vfyd ai",
-    "vfyd iot",
 ]
 
 team = [
@@ -28,7 +27,6 @@ team = [
     "sbel",
     "wasa",
     "huvw",
-    "artn",
 ]
 
 random.shuffle(team)

src/main.py

@@ -34,7 +34,9 @@ def dispatch():
     )
     leaves = client.web_search_read(Leave, domain)
 
-    team_availability = {employee: (True, True) for employee in team}
+    team_availability = dict()
+    for employee in team:
+        team_availability[employee] = (True, True)
 
     for leave in leaves:
         employee = re.search(r"\((.*?)\)", leave.employee_id).group(1).lower()
@@ -47,14 +49,10 @@ def dispatch():
 
     available_employees = [
         employee
-        for employee, (morning, afternoon) in team_availability.items()
-        if morning or afternoon
+        for employee, availability in team_availability.items()
+        if availability[0] or availability[1]
     ]
 
-    if not available_employees:
-        print("[red]No available employees for dispatch.[/red]")
-        return
-
     previous_dispatch = DispatchStorage.load_week_to_date()
     previous_dispatch_ids = [
         task_id for tasks in previous_dispatch.values() for task_id in tasks

src/model.py

@@ -1,12 +1,12 @@
-from typing import Self, Any, get_origin
+from typing import Dict, Self, Any, get_origin
 from dataclasses import fields, Field
 from datetime import datetime
 
 
 class OdooModel:
     @classmethod
-    def specification(cls) -> dict[str, Any]:
-        def field_spec(f: Field) -> dict[str, Any]:
+    def specification(cls) -> Dict[str, Any]:
+        def field_spec(f: Field) -> Dict[str, Any]:
             field_type = f.type
             if f.name.endswith("_id") or f.name.endswith("_ids"):
                 return {
@@ -29,7 +29,7 @@ class OdooModel:
         return spec
 
     @classmethod
-    def from_record(cls, record: dict) -> Self:
+    def from_record(cls, record: Dict) -> Self:
         init_args = {}
         for f in fields(cls):
             if f.init and f.name in record:

src/odoo_client.py

@@ -1,7 +1,7 @@
 import os
 import requests
 import random
-from typing import Any, TypeVar
+from typing import Any, Dict, List, TypeVar, Type
 from model import OdooModel
 
 
@@ -22,10 +22,10 @@ class OdooClient:
     def rpc(
         self, model: str, method: str, *args: Any, **kwargs: Any
     ) -> requests.Response:
-        body: dict[str, Any] = {
+        body: Dict[str, Any] = {
             "jsonrpc": "2.0",
             "method": method,
-            "id": random.randint(1, 1_000_000),
+            "id": random.randint(1, 1000000),
             "params": {
                 "model": model,
                 "method": method,
@@ -34,27 +34,23 @@ class OdooClient:
             },
         }
 
-        response = self.session.post(
+        return self.session.post(
             f"{self.base_url}/web/dataset/call_kw",
             json=body,
         )
-        response.raise_for_status()
-        return response
 
     def web_search_read(
         self,
-        model: type[T],
-        domain: list[tuple[Any, ...]],
+        model: Type[T],
+        domain: List[tuple],
         *args: Any,
         **kwargs: Any,
-    ) -> list[T]:
+    ) -> List[T]:
         args = [domain, model.specification(), *args]
         res = self.rpc(model._name, "web_search_read", *args, **kwargs)
         res.raise_for_status()
         json = res.json()
-
-        if error := json.get("error"):
-            raise RuntimeError(f"Odoo RPC Error: {error}")
-
+        if json.get("error"):
+            raise Exception(json.get("error"))
         records = json.get("result", {}).get("records", [])
         return [model.from_record(record) for record in records]

src/storage.py

@@ -1,33 +1,27 @@
-from typing import Optional
+from typing import Dict, List, Optional
 from datetime import datetime, timedelta
-from pathlib import Path
 import json
 
 
 class DispatchStorage:
-    @staticmethod
-    def path(date: Optional[str] = None) -> Path:
+    @classmethod
+    def path(cls, date: Optional[str] = None) -> str:
         if date is None:
             date = datetime.now().strftime("%Y-%m-%d")
-        return Path("out", f"{date}_dispatch.json")
+        return f"out/{date}_dispatch.json"
 
-    @staticmethod
-    def load(date: Optional[str] = None) -> dict[str, list[int]]:
-        try:
-            with open(DispatchStorage.path(date), "r") as f:
-                return json.load(f)
-        except FileNotFoundError:
-            return {}
+    @classmethod
+    def load(cls, date: Optional[str] = None) -> Dict[str, List[int]]:
+        with open(cls.path(date), "r") as f:
+            return json.load(f)
 
-    @staticmethod
-    def save(dispatch: dict[str, list[int]], date: Optional[str] = None) -> None:
-        path = DispatchStorage.path(date)
-        path.parent.mkdir(exist_ok=True)
-        with open(path, "w") as f:
+    @classmethod
+    def save(cls, dispatch: Dict[str, List[int]], date: Optional[str] = None) -> None:
+        with open(cls.path(date), "w") as f:
             json.dump(dispatch, f)
 
-    @staticmethod
-    def load_week_to_date() -> dict[str, list[int]]:
+    @classmethod
+    def load_week_to_date(cls) -> Dict[str, List[int]]:
         """
         Loads and combines dispatch data from the start of the current week up to today.
         """
@@ -35,10 +29,10 @@ class DispatchStorage:
         days_since_monday = today.weekday()
         current_date = today - timedelta(days=days_since_monday)
 
-        combined_dispatch: dict[str, list[int]] = {}
+        combined_dispatch: Dict[str, List[int]] = {}
         while current_date.date() < today.date():
             date_str = current_date.strftime("%Y-%m-%d")
-            daily_dispatch = DispatchStorage.load(date_str)
+            daily_dispatch = cls.load(date_str)
             for key, values in daily_dispatch.items():
                 if key in combined_dispatch:
                     combined_dispatch[key].extend(values)

src/tasks.py

@@ -1,4 +1,5 @@
 from dataclasses import dataclass, field
+from typing import List
 from model import OdooModel
 
 
@@ -8,7 +9,7 @@ class Task(OdooModel):
 
     id: int
     name: str
-    tag_ids: list[str]
+    tag_ids: List[str]
     stage_id: str
     priority: bool
     url: str = field(init=False)