added first db client

mhorst00 · Oct 14, 2022 · 3cb0c5f · 3cb0c5f
1 parent 9f41fa8
commit 3cb0c5f
Showing 1 changed file with 49 additions and 15 deletions.
diff --git a/mining/db.py b/mining/db.py
@@ -1,49 +1,73 @@
 from datetime import datetime
 import functools
+import sqlalchemy.exc
 from sqlalchemy import create_engine, Integer, JSON, Column, Sequence
 from sqlalchemy.orm import sessionmaker
 from sqlalchemy.ext.declarative import declarative_base
 
-EntityBase = declarative_base()
-
 CURRENT_DATE = datetime.now().date()
 ENGINE = create_engine(f"sqlite:///db{str(CURRENT_DATE)}.db", echo=True)
 SESSION = sessionmaker(bind=ENGINE)()
+ENTITY_BASE = declarative_base()
 
 
-class Trip(EntityBase):
+class Trip(ENTITY_BASE):
     __tablename__ = "trips"
-    id = Column(
-        Integer, Sequence("trip_id_seq"), primary_key=True, nullable=False
-    )
+    id = Column(Integer, Sequence("trip_id_seq"), primary_key=True, nullable=False)
     information = Column(JSON, nullable=True)
 
 
 # Create all tables derived from the EntityBase object
-EntityBase.metadata.create_all(ENGINE)
+ENTITY_BASE.metadata.create_all(ENGINE)
 
 
 def daily_db(func):
     @functools.wraps(func)
     def wrapper(*args, **kwargs):
+        global CURRENT_DATE
+        global ENGINE
+        global SESSION
+        global ENTITY_BASE
         new_date = datetime.now().date()
         if new_date > CURRENT_DATE:
-            ENGINE = create_engine(
-                f"sqlite:///db{str(new_date)}.db", echo=True
-            )
+            ENGINE = create_engine(f"sqlite:///db{str(new_date)}.db", echo=True)
             SESSION = sessionmaker(bind=ENGINE)()
-            CURRENT_DAY = new_date
-            EntityBase.metadata.create_all(ENGINE)
+            CURRENT_DATE = new_date
+            ENTITY_BASE.metadata.create_all(ENGINE)
         return func(*args, **kwargs)
 
     return wrapper
 
 
 @daily_db
 def new_entry(trip):
-    SESSION.add(trip)
-    SESSION.commit()
-    return True
+    try:
+        SESSION.add(trip)
+        SESSION.commit()
+        return True
+    except sqlalchemy.exc.SQLAlchemyError as e:
+        error = str(e.__dict__["orig"])
+        return error
+
+
+@daily_db
+def del_entry(trip):
+    try:
+        SESSION.delete(trip)
+        SESSION.commit()
+        return True
+    except sqlalchemy.exc.SQLAlchemyError as e:
+        error = str(e.__dict__["orig"])
+        return error
+
+
+@daily_db
+def get_all_entries():
+    try:
+        return SESSION.query(Trip).all()
+    except sqlalchemy.exc.SQLAlchemyError as e:
+        error = str(e.__dict__["orig"])
+        return error
 
 
 # Declare a new row
@@ -53,6 +77,16 @@ def new_entry(trip):
 new_entry(first_item)
 
 
+# Get all saved items from the database
+for item in SESSION.query(Trip).all():
+    print(type(item.information))
+    # <class 'dict'>
+    print(item.id, item.information)
+
+trips = get_all_entries()
+for i in trips:
+    del_entry(i)
+
 # Get all saved items from the database
 for item in SESSION.query(Trip).all():
     print(type(item.information))