DB Connection Registry Validation, Better Inheritance, Table Methods

Author: bjthres1

stat_log_db/pyproject.toml

@@ -7,7 +7,7 @@ name = "stat-log-db"
 version = "0.0.1"
 description = ""
 readme = "README.md"
-requires-python = "==3.12.10"
+requires-python = ">=3.12.10"
 dependencies = [
 ]
 

stat_log_db/src/stat_log_db/cli.py

@@ -2,7 +2,7 @@
 import sys
 
 from .parser import create_parser
-from .db import Database, BaseConnection
+from .db import Database, MemDB, FileDB, BaseConnection
 
 
 def main():
@@ -18,19 +18,18 @@ def main():
 
     # print(f"{args=}")
 
-    db_filename = 'sl_db.sqlite'
-    sl_db = Database(db_filename)
-    con = sl_db.init_db(False)
-    if isinstance(con, BaseConnection):
-        con.execute("CREATE TABLE IF NOT EXISTS logs (id INTEGER PRIMARY KEY, message TEXT);")
-        con.execute("INSERT INTO logs (message) VALUES (?);", ("Hello, world!",))
-        con.commit()
-        con.execute("SELECT * FROM logs;")
-        sql_logs = con.fetchall()
-        print(sql_logs)
-
+    sl_db = MemDB(":memory:", True, True)
+    con = sl_db.init_db(True)
+    con.create_table("test", [('notes', 'TEXT')], False, True)
+    con.execute("INSERT INTO test (notes) VALUES (?);", ("Hello world!",))
+    con.commit()
+    con.execute("SELECT * FROM test;")
+    sql_logs = con.fetchall()
+    print(sql_logs)
+    con.drop_table("test", True)
     sl_db.close_db()
-    # os.remove(db_filename)
+    if sl_db.is_file:
+        os.remove(sl_db.file_name)
 
 if __name__ == "__main__":
     main()