create gloable sqlite conn

deepattic · Nov 23, 2024 · c05d08d · c05d08d
1 parent 399fde5
commit c05d08d
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/libwardenpy/db.py b/libwardenpy/db.py
@@ -0,0 +1,13 @@
+import sqlite3
+from contextlib import contextmanager
+
+
+### This just create a sqlite connection to pass around the other functions
+### https://stackoverflow.com/questions/67436362/decorator-for-sqlite3/67436763#67436763
+@contextmanager
+def get_connection(DB_PATH="data/db.sqlite3"):
+    conn = sqlite3.connect(DB_PATH)
+    try:
+        yield conn
+    finally:
+        conn.close()