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

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

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

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

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

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

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

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

@@ -17,6 +17,7 @@ def table(name: str, columns: list[str]):
     Create a table. Columns should be formatted like `column_name TYPE` e.g. `server_id BIGINT PRIMARY KEY` or `data JSONB`
     """
     run(f"CREATE TABLE IF NOT EXISTS {SCHEMA.as_string()}.{name} ({", ".join(columns)});")
+    logger.info(f"Created table {name} if it didn't already exist")
 
 def check_connection():
     cursor.execute("SELECT version();")
@@ -131,3 +132,4 @@ def start(schema: str, host: str, name: str, user: str, password: str, port: int
     check_connection()
     logger.info("Connected to database.")
     run(f"CREATE SCHEMA IF NOT EXISTS {SCHEMA.as_string()};")
+    logger.info(f"Created schema {schema} if it didn't already exist")

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

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