pengjunjie/langflow
1
0
Fork 1
mirror of https://github.com/langflow-ai/langflow.git synced 2026-07-24 03:21:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

🐛 fix(manager.py): refactor create_db_and_tables method to check if database and tables already exist before creating them

Browse Source
This commit is contained in:
Gabriel Luiz Freitas Almeida
2023-10-03 17:39:50 -03:00
parent a8f444656b
commit dc559e61ae
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/backend/langflow/services/database/manager.py
View File

@ -154,6 +154,14 @@ class DatabaseService(Service):
return results
def create_db_and_tables(self):
from sqlalchemy import inspect
inspector = inspect(self.engine)
current_tables = ["flow", "user", "apikey"]
table_names = inspector.get_table_names()
if table_names and all(table in table_names for table in current_tables):
logger.debug("Database and tables already exist")
return
logger.debug("Creating database and tables")
try:
SQLModel.metadata.create_all(self.engine)
@ -163,10 +171,6 @@ class DatabaseService(Service):
# Now check if the table "flow" exists, if not, something went wrong
# and we need to create the tables again.
from sqlalchemy import inspect
inspector = inspect(self.engine)
current_tables = ["flow", "user", "apikey"]
table_names = inspector.get_table_names()
for table in current_tables:
if table not in table_names: