ready to deployment

backend/odoo_api/client.py

@@ -0,0 +1,128 @@
+import requests
+import json
+import logging
+import calendar
+from datetime import datetime, timedelta, timezone
+from fastapi import FastAPI, Depends, HTTPException, status
+
+logger = logging.getLogger("uvicorn")
+
+
+class OdooAPIClient:
+    def __init__(self, odoo_url: str, db_name: str, secret_key: str, algorithm: str):
+        self.odoo_url = odoo_url
+        self.db_name = db_name
+        self.session = requests.Session()
+
+    def login(self, username: str, password: str) -> dict:
+        auth_url = f"{self.odoo_url}/web/session/authenticate"
+        payload = {
+            "jsonrpc": "2.0",
+            "method": "call",
+            "params": {
+                "db": self.db_name,
+                "login": username,
+                "password": password
+            },
+            "id": 1
+        }
+        headers = {'Content-Type': 'application/json'}
+        response = self.session.post(auth_url, json=payload, headers=headers)
+
+        if response.status_code != 200:
+            # logger.error(
+            #     f"Authentication failed with status code {response.status_code}")
+            raise Exception(
+                f"Authentication failed with status code {response.status_code}")
+
+        # for cookie in response.cookies:
+        #     self._session_cookies[cookie.name] = cookie.value
+
+        data = response.json()
+        if "error" in data:
+            raise HTTPException(
+                status_code=502,
+                detail=f"Odoo RPC error: {data['error'].get('message', 'Unknown error')}"
+            )
+        return data["result"]
+
+    def get_hr_attendance_data(self, month=None, year=None, limit=None, domain=None):
+        if month is None:
+            month = datetime.now(timezone.utc).month
+        if year is None:
+            year = datetime.now(timezone.utc).year
+        if domain is None:
+            domain = []  # Default to no domain filter
+
+        url = f"{self.odoo_url}/hr.attendance"
+        first_day_of_month = datetime(year, month, 1)
+        last_day_of_month = datetime(
+            year, month, calendar.monthrange(year, month)[1], 23, 59, 59)
+
+        start_date_str = first_day_of_month.strftime('%Y-%m-%d %H:%M:%S')
+        end_date_str = last_day_of_month.strftime('%Y-%m-%d %H:%M:%S')
+
+        domain.extend([
+            ('check_in', '>=', start_date_str),
+            ('check_in', '<=', end_date_str)
+        ])
+
+        kwargs = {
+            "domain": domain,
+            "fields": ["id", "employee_id", "check_in", "check_out", "worked_hours", "attendance_reason_ids"],
+        }
+
+        if limit is not None:
+            kwargs["limit"] = limit
+
+        result = self.call_odoo_method(
+            "hr.attendance", "web_search_read", kwargs=kwargs)
+
+        if "records" not in result:
+            # logger.error("Unexpected response format: 'records' not found")
+            raise HTTPException(
+                status_code=502,
+                detail="Unexpected response format: 'records' not found"
+            )
+        
+        sorted_data = sorted(result["records"], key=lambda x: x["check_in"])
+        return sorted_data
+
+    def call_odoo_method(self, model: str, method: str, args: list = None, kwargs: dict = None):
+        if args is None:
+            args = []
+        if kwargs is None:
+            kwargs = {}
+
+        url = f"{self.odoo_url}/web/dataset/call_kw/{model}/{method}"
+        headers = {'Content-Type': 'application/json'}
+        payload = {
+            "jsonrpc": "2.0",
+            "method": "call",
+            "params": {
+                "model": model,
+                "method": method,
+                "args": args,
+                "kwargs": kwargs
+            },
+            "id": 1
+        }
+        response = self.session.post(url, json=payload, headers=headers)
+        # response = requests.post(url, json=payload, headers=headers, cookies=self._session_cookies)
+
+        if response.status_code != 200:
+            # logger.error(
+            #     f"Failed to call Odoo method {method} on model {model}: {response.status_code}")
+            raise HTTPException(
+                status_code=response.status_code,
+                detail=f"Failed to call Odoo method {method} on model {model}"
+            )
+        
+        data = response.json()
+        if "error" in data:
+            raise HTTPException(
+                status_code=502,
+                detail=f"Odoo RPC error: {data['error'].get('message', 'Unknown error')}"
+            )
+        return data["result"]
+