drifTech-lib 0.2.2__tar.gz → 0.2.4__tar.gz

{driftech_lib-0.2.2 → driftech_lib-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: drifTech-lib
-Version: 0.2.2
+Version: 0.2.4
 Summary: Common files/functions across most drifTech software
 Author: driftbluestone
 Author-email: driftbluestone <driftbluestone@gmail.com>

{driftech_lib-0.2.2 → driftech_lib-0.2.4}/pyproject.toml

@@ -8,7 +8,7 @@ build-backend = "uv_build"
 
 [project]
 name = "drifTech-lib"
-version = "0.2.2"
+version = "0.2.4"
 description = "Common files/functions across most drifTech software"
 readme = "README.md"
 requires-python = ">=3.14"

{driftech_lib-0.2.2 → driftech_lib-0.2.4}/pyproject.toml.orig

@@ -8,7 +8,7 @@ build-backend = "uv_build"
 
 [project]
 name = "drifTech-lib"
-version = "0.2.2"
+version = "0.2.4"
 authors = [{name = "driftbluestone", email = "driftbluestone@gmail.com"}]
 description = "Common files/functions across most drifTech software"
 readme = "README.md"

{driftech_lib-0.2.2 → driftech_lib-0.2.4}/src/driftech_lib/db.py

@@ -4,6 +4,7 @@ Module for interacting with PostgreSQL
 import psycopg, logging
 from typing import Any
 from psycopg import sql
+logger = logging.getLogger("db")
 
 __all__ = ["SCHEMA", "start", "run", "single", "multiple", "insert", "delete", "get", "table"]
 
@@ -21,13 +22,13 @@ def check_connection():
     cursor.execute("SELECT version();")
     db_version = cursor.fetchone()
     
-    logging.info("Connection Successful")
-    logging.info(f"PostgreSQL version: {db_version[0]}")
+    logger.info("Connection Successful")
+    logger.info(f"PostgreSQL version: {db_version[0]}")
 
 def close_connection():
     cursor.close()
     connection.close()
-    logging.info("Database connection closed.")
+    logger.info("Database connection closed.")
 
 def run(*args):
     cursor.execute(*args)
@@ -118,7 +119,7 @@ def get(table: str, value: tuple[Any], key: tuple[str], column: tuple[str]):
 def start(schema: str, host: str, name: str, user: str, password: str, port: int):
     global SCHEMA, connection, cursor
     SCHEMA = sql.Identifier(schema)
-    logging.info("Connecting to database...")
+    logger.info("Connecting to database...")
     connection = psycopg.connect(
         host=host,
         dbname=name,
@@ -128,5 +129,5 @@ def start(schema: str, host: str, name: str, user: str, password: str, port: int
     )
     cursor = connection.cursor()
     check_connection()
-    logging.info("Connected to database.")
-    connection.execute(f"CREATE SCHEMA IF NOT EXISTS {SCHEMA.as_string()};")
+    logger.info("Connected to database.")
+    run(f"CREATE SCHEMA IF NOT EXISTS {SCHEMA.as_string()};")

{driftech_lib-0.2.2 → driftech_lib-0.2.4}/src/driftech_lib/ipc/client.py

@@ -1,4 +1,5 @@
 import asyncio, logging
+logger = logging.getLogger("ipc:client")
 __all__ = ["Connector"]
 
 class Connector:
@@ -13,11 +14,11 @@ class Connector:
         self.command_queue: asyncio.Queue[bytes] = asyncio.Queue(10)
 
     async def start(self):
-        logging.info(f"Connecting to server at {self.HOST}:{self.PORT}...")
+        logger.info(f"Connecting to server at {self.HOST}:{self.PORT}...")
         
         # Establish the asynchronous socket connection
         reader, writer = await asyncio.open_connection(self.HOST, self.PORT)
-        logging.info("Connected successfully!")
+        logger.info("Connected successfully!")
         
         # Start the background listening loop as a concurrent task
         listen_task = asyncio.create_task(self._listen(reader))
@@ -36,13 +37,13 @@ class Connector:
             while True:
                 data = await reader.read(1024)
                 if not data:
-                    logging.error("[DISCONNECTED] Server closed the connection.")
+                    logger.error("[DISCONNECTED] Server closed the connection.")
                     break
                 await self.on_message(data)
         except asyncio.CancelledError:
             pass
         except Exception as e:
-            logging.error(f"[ERROR] Reading error: {e}")
+            logger.error(f"[ERROR] Reading error: {e}")
 
     async def on_message(self, message: bytes):
         """
@@ -66,12 +67,12 @@ class Connector:
                     continue
                     
                 if message.lower() == b"exit":
-                    logging.info("Closing connection...")
+                    logger.info("Closing connection...")
                     break
                     
                 writer.write(message)
                 await writer.drain()
                 
         except Exception as e:
-            logging.error(f"Writing error: {e}")
+            logger.error(f"Writing error: {e}")
     

{driftech_lib-0.2.2 → driftech_lib-0.2.4}/src/driftech_lib/ipc/server.py

@@ -1,28 +1,29 @@
 import asyncio, logging
+logger = logging.getLogger("ipc:server")
 __all__ = ["start"]
 
 async def start(HOST, PORT):
     server = await asyncio.start_server(interface, HOST, PORT)
-    logging.info(f"[LISTENING] Server is running on {HOST}:{PORT}")
+    logger.info(f"[LISTENING] Server is running on {HOST}:{PORT}")
     async with server:
         await server.serve_forever()
 
 async def interface(reader: asyncio.StreamReader, writer: asyncio.StreamWriter):
     addr = writer.get_extra_info("peername")
-    logging.info(f"[CONNECTION]: Connection opened for {addr}")
+    logger.info(f"[CONNECTION]: Connection opened for {addr}")
     try:
         while True:
             data = await reader.read(1024)
             if not data:
                 break
-            logging.info(f"[{addr}] Received: {data.decode()}")
+            logger.info(f"[{addr}] Received: {data.decode()}")
             writer.write(b"Message processed")
             await writer.drain()  # Ensure data is flushed to the network buffer
     except (asyncio.CancelledError, ConnectionResetError, BrokenPipeError, OSError):
-        logging.info(f"[DISCONNECTED] Connection closed for {addr}")
+        logger.info(f"[DISCONNECTED] Connection closed for {addr}")
         writer.close()
         await writer.wait_closed()
     finally:
-        logging.info(f"[DISCONNECTED] Connection closed for {addr}")
+        logger.info(f"[DISCONNECTED] Connection closed for {addr}")
         writer.close()
         await writer.wait_closed()