pam-python 0.1.40__tar.gz → 0.1.41__tar.gz

{pam_python-0.1.40/pam_python.egg-info → pam_python-0.1.41}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pam-python
-Version: 0.1.40
+Version: 0.1.41
 Summary: Pam Python Library
 Home-page: https://github.com/heart/pam-python
 Author: Narongrit Kanhanoi

pam_python-0.1.41/pam/logger.py

@@ -0,0 +1,96 @@
+from __future__ import annotations
+
+from datetime import datetime, timezone
+from threading import Lock
+from typing import Any
+
+from pam.utils import log as stdout_log
+from service import Service
+
+
+class Logger:
+    def __init__(self, service: Service, buffer_size: int = 20, source: str = "worker"):
+        self.service = service
+        self.log_api = service.request.log_api
+        self.token = service.request.token
+        self.source = source
+        self.buffer_size = max(1, int(buffer_size))
+        self._buffer: list[dict[str, Any]] = []
+        self._lock = Lock()
+
+    def debug(self, message: Any, stage: str = "") -> None:
+        self.log(message, level="debug", stage=stage)
+
+    def info(self, message: Any, stage: str = "") -> None:
+        self.log(message, level="info", stage=stage)
+
+    def warning(self, message: Any, stage: str = "") -> None:
+        self.log(message, level="warning", stage=stage)
+
+    def error(self, message: Any, stage: str = "") -> None:
+        self.log(message, level="error", stage=stage)
+
+    def log(self, message: Any, level: str = "info", stage: str = "") -> None:
+        message_str = self._stringify(message)
+        self._write_stdout(message_str, level)
+        should_flush = self._enqueue(self._build_log_item(message_str, level, stage))
+
+        if should_flush:
+            self.flush()
+
+    def flush(self) -> bool:
+        if not self.log_api or not self.token:
+            return False
+
+        with self._lock:
+            if not self._buffer:
+                return True
+            logs = list(self._buffer)
+
+        payload = {
+            "token": self.token,
+            "logs": logs,
+        }
+        response = self.service.task_manager.api.http_post(self.log_api, payload)
+        if response is None:
+            return False
+
+        with self._lock:
+            if self._buffer[:len(logs)] == logs:
+                del self._buffer[:len(logs)]
+            else:
+                self._buffer = self._buffer[len(logs):]
+        return True
+
+    def pending_count(self) -> int:
+        with self._lock:
+            return len(self._buffer)
+
+    def _write_stdout(self, message: str, level: str) -> None:
+        stdout_log(f"[{level}] {message}")
+
+    def _build_log_item(self, message: str, level: str, stage: str) -> dict[str, Any]:
+        log_item = {
+            "message": message,
+            "level": level,
+            "source": self.source,
+            "logged_at": self._utc_now(),
+        }
+        if stage:
+            log_item["stage"] = stage
+        return log_item
+
+    def _enqueue(self, log_item: dict[str, Any]) -> bool:
+        with self._lock:
+            self._buffer.append(log_item)
+            return len(self._buffer) >= self.buffer_size
+
+    @staticmethod
+    def _stringify(message: Any) -> str:
+        if isinstance(message, str):
+            return message
+        return str(message)
+
+    @staticmethod
+    def _utc_now() -> str:
+        return datetime.now(timezone.utc).isoformat(timespec="milliseconds")

{pam_python-0.1.40 → pam_python-0.1.41}/pam/models/request_command.py

@@ -43,6 +43,7 @@ class RequestCommand:
     next: str
     input_files: list[str]
     service_name: str
+    log_api: str
 
     def is_start_command(self):
         """Determine if this command is a start command."""
@@ -138,6 +139,9 @@ class RequestCommand:
         sqlite_download = params.pop("sqlite_download", "")
 
         response_api = params.pop("response", "")
+
+        log_api = params.pop("log", "")
+
         is_end = RequestCommand.__safe_str_to_bool(params.pop("is_end", "true"))
         next_page = params.pop("next", "")
 
@@ -167,4 +171,5 @@ class RequestCommand:
             next=next_page,
             input_files=input_files,
             service_name=service_name,
+            log_api=log_api
         ), error_message

{pam_python-0.1.40 → pam_python-0.1.41}/pam/utils.py

@@ -5,7 +5,6 @@ import re
 import pytz
 from typing import Any
 
-
 def get_adapter_id(url: str) -> str:
     """
     Extracts the adapter ID from a given URL.
@@ -55,6 +54,6 @@ def log(msg: str, level: str = "INFO"):
         tz = pytz.timezone('Asia/Bangkok')
         bangkok_time = utc_dt.astimezone(tz)  # Convert UTC time to Bangkok time
         date_str = bangkok_time.strftime("%m/%d/%Y %H:%M:%S")
-        print(f"[{date_str}] [{level}]: {msg}")
+        print(f"[{date_str}] [{level}]: {msg}", flush=True)
     except Exception as e:
-        print(f"Failed to log message: {msg}. Error: {e}")
+        print(f"Failed to log message: {msg}. Error: {e}", flush=True)

{pam_python-0.1.40 → pam_python-0.1.41/pam_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pam-python
-Version: 0.1.40
+Version: 0.1.41
 Summary: Pam Python Library
 Home-page: https://github.com/heart/pam-python
 Author: Narongrit Kanhanoi

{pam_python-0.1.40 → pam_python-0.1.41}/pam_python.egg-info/SOURCES.txt

@@ -5,6 +5,7 @@ pam/__init__.py
 pam/api.py
 pam/cli.py
 pam/interface_task_manager.py
+pam/logger.py
 pam/request_file_format.py
 pam/server.py
 pam/service.py

{pam_python-0.1.40 → pam_python-0.1.41}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="pam-python",
-    version="0.1.40",
+    version="0.1.41",
     author="Narongrit Kanhanoi",
     author_email="narongrit@pams.ai",
     description="Pam Python Library",