add FastAPI

db/helpers.py

@@ -0,0 +1,51 @@
+import logging
+
+from sqlalchemy import create_engine, insert, update, select
+from sqlalchemy.orm import Session, sessionmaker
+from db.models import Base, Order, OrderStatus
+
+
+def get_engine():
+    return create_engine('postgresql+psycopg2://postgres:guardia123@192.168.1.202:5432/export-tray', echo=False)
+
+
+def create_tables():
+    engine = get_engine()
+
+    Base.metadata.create_all(engine)
+
+def drop_tables():
+    engine = get_engine()
+
+    tables = Base.metadata.tables
+    logging.debug(tables)
+
+    Session = sessionmaker(bind=engine)
+    session = Session()
+
+    for table in [Order, OrderStatus]:
+        session.query(table).delete()
+
+    session.commit()
+    session.close()
+
+
+    Base.metadata.drop_all(engine)
+def insert_into_table(values_map, table_name):
+    engine = get_engine()
+    stmt = insert(table_name).values(values_map)
+
+    with engine.connect() as conn:
+        result = conn.execute(stmt)
+        print(result)
+        conn.commit()
+
+
+def update_record(where_map, values_map, table_name):
+    engine = get_engine()
+    stmt = update(table_name).where(where_map).values(values_map)
+
+    with engine.connect() as conn:
+        result = conn.execute(stmt)
+        print(result)
+        conn.commit()