init

backend/database/connect.py

@@ -0,0 +1,23 @@
+import os
+import sys
+import redis
+from sqlalchemy import create_engine
+from sqlalchemy.orm import Session
+from config.main import config
+
+RedisPool = redis.ConnectionPool.from_url(config.REDIS_URL)
+
+def get_redis():
+    with redis.Redis(connection_pool=RedisPool) as redis_conn:
+        yield redis_conn
+
+def get_db_engine(version='0'):
+    try:
+        os.chdir(os.path.dirname(sys.argv[0]))
+    except Exception:
+        pass
+    return create_engine(f'sqlite:///{os.path.join("./", "dataFiles", f"dcalc_{version}.db")}')
+
+def get_db():
+    with Session(get_db_engine()) as session:
+        yield session