From 4b27b45e4ca5c0ffabe292c9cbffc18bbb393672 Mon Sep 17 00:00:00 2001
From: JeLuF <jf@mormo.org>
Date: Tue, 25 Jul 2023 20:30:35 +0200
Subject: [PATCH] sqlalchemy

---
 ui/easydiffusion/bucket_manager.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)
 create mode 100644 ui/easydiffusion/bucket_manager.py

diff --git a/ui/easydiffusion/bucket_manager.py b/ui/easydiffusion/bucket_manager.py
new file mode 100644
index 00000000..e24b0cc0
--- /dev/null
+++ b/ui/easydiffusion/bucket_manager.py
@@ -0,0 +1,14 @@
+import os
+from easydiffusion import app
+
+from sqlalchemy import create_engine
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import sessionmaker
+
+SQLALCHEMY_DATABASE_URL = "sqlite:///"+os.path.join(app.BUCKET_DIR, "bucket.db")
+print("## SQLALCHEMY_DATABASE_URL = ", SQLALCHEMY_DATABASE_URL)
+
+engine = create_engine(SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False})
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+
+Base = declarative_base()