pam-python 0.1.18__tar.gz → 0.1.20__tar.gz

{pam_python-0.1.18/pam_python.egg-info → pam_python-0.1.20}/PKG-INFO

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

{pam_python-0.1.18 → pam_python-0.1.20}/pam/service.py

@@ -97,7 +97,7 @@ class Service:
         
         return report_csv_path
 
-    def _request_sqlite(self, file_name: str = "", is_shared: bool = False) -> str | None:
+    def _request_sqlite(self, file_name: str = "", is_shared: bool = False) -> str:
         """Requests last sqlite file that this plugin has uploaded from the last time.
 
         Returns:
@@ -129,7 +129,7 @@ class Service:
             return output_path
         else:
             log(f"{self.request.service_name}: Failed to download SQLite file.")
-            return None
+            return output_path
 
 
     def _upload_sqlite(self, file_name: str = "", is_shared: bool = False, sqlite_file: str = "") -> str:

pam_python-0.1.20/pam/sqlite.py

@@ -0,0 +1,75 @@
+import sqlite3
+from pam.utils import log
+from typing import Any, List, Optional, Tuple, Union
+
+class SQLiteDB:
+    def __init__(self, db_path: str, row_as_dict: bool = False, timeout=30):
+        """
+        Initialize the SQLiteDB wrapper.
+
+        :param db_path: Path to the SQLite database file.
+        :param row_as_dict: If True, fetches rows as dictionaries.
+        """
+        self.db_path = db_path
+        self.row_as_dict = row_as_dict
+        self.timeout = timeout
+        self.conn = sqlite3.connect(db_path, timeout)
+        if row_as_dict:
+            self.conn.row_factory = sqlite3.Row
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.close()
+
+    def execute(self, query: str, params: Optional[Tuple[Any, ...]] = None) -> None:
+        """Execute a write query (INSERT/UPDATE/DELETE)."""
+        try:
+            with self.conn:
+                self.conn.execute(query, params or ())
+        except sqlite3.Error as e:
+            log(f"SQLite execute error: {e}")
+            raise
+
+    def executemany(self, query: str, data: List[Tuple[Any, ...]]) -> None:
+        """Execute a batch write query."""
+        try:
+            with self.conn:
+                self.conn.executemany(query, data)
+        except sqlite3.Error as e:
+            log(f"SQLite executemany error: {e}")
+            raise
+
+    def fetchall(self, query: str, params: Optional[Tuple[Any, ...]] = None) -> List[Union[Tuple, dict]]:
+        """Fetch all results from a SELECT query."""
+        try:
+            cursor = self.conn.execute(query, params or ())
+            return cursor.fetchall()
+        except sqlite3.Error as e:
+            log(f"SQLite fetchall error: {e}")
+            raise
+
+    def fetchone(self, query: str, params: Optional[Tuple[Any, ...]] = None) -> Optional[Union[Tuple, dict]]:
+        """Fetch the first result from a SELECT query."""
+        try:
+            cursor = self.conn.execute(query, params or ())
+            return cursor.fetchone()
+        except sqlite3.Error as e:
+            log(f"SQLite fetchone error: {e}")
+            raise
+
+    def create_table(self, create_sql: str) -> None:
+        """Create a table from a SQL string."""
+        try:
+            with self.conn:
+                self.conn.execute(create_sql)
+        except sqlite3.Error as e:
+            log(f"SQLite create_table error: {e}")
+            raise
+
+    def close(self) -> None:
+        """Close the connection."""
+        if self.conn:
+            self.conn.close()
+            log("SQLite connection closed.")

{pam_python-0.1.18 → pam_python-0.1.20/pam_python.egg-info}/PKG-INFO

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

{pam_python-0.1.18 → pam_python-0.1.20}/pam_python.egg-info/SOURCES.txt

@@ -7,6 +7,7 @@ pam/cli.py
 pam/interface_task_manager.py
 pam/server.py
 pam/service.py
+pam/sqlite.py
 pam/task_manager.py
 pam/temp_file_utils.py
 pam/tester_task.py

{pam_python-0.1.18 → pam_python-0.1.20}/setup.py

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