add display for MetaData

2025-05-04 12:27:03 +02:00
parent 4cf1941f44
commit 13dad3961c
33 changed files with 609 additions and 44 deletions
@@ -0,0 +1,28 @@
+import databases
+from src.db.session import SQLALCHEMY_DATABASE_URL
+
+
+async def check_db_connected():
+    try:
+        if not str(SQLALCHEMY_DATABASE_URL).__contains__("sqlite"):
+            database = databases.Database(SQLALCHEMY_DATABASE_URL)
+            if not database.is_connected:
+                await database.connect()
+                await database.execute("SELECT 1")
+        print("Database is connected (^_^)")
+    except Exception as e:
+        print(
+            "Looks like db is missing or is there is some problem in connection,see below traceback"
+        )
+        raise e
+
+
+async def check_db_disconnected():
+    try:
+        if not str(SQLALCHEMY_DATABASE_URL).__contains__("sqlite"):
+            database = databases.Database(SQLALCHEMY_DATABASE_URL)
+            if database.is_connected:
+                await database.disconnect()
+        print("Database is Disconnected (-_-) zZZ")
+    except Exception as e:
+        raise e