From ad406541f2b948d422f371121020c8d4d2278de0 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 3 Oct 2023 17:08:38 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(manager.py):=20improve=20rea?= =?UTF-8?q?dability=20by=20splitting=20a=20long=20line=20into=20multiple?= =?UTF-8?q?=20lines=20=F0=9F=90=9B=20fix(utils.py):=20add=20comment=20to?= =?UTF-8?q?=20explain=20the=20reason=20for=20the=20check?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/services/database/manager.py | 4 +++- src/backend/langflow/services/utils.py | 3 +++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/services/database/manager.py b/src/backend/langflow/services/database/manager.py index 0521ef583..8b9dde7bd 100644 --- a/src/backend/langflow/services/database/manager.py +++ b/src/backend/langflow/services/database/manager.py @@ -66,7 +66,9 @@ class DatabaseService(Service): if settings_service.auth_settings.AUTO_LOGIN: with Session(self.engine) as session: flows = ( - session.query(models.Flow).filter(models.Flow.user_id == None).all() + session.query(models.Flow) + .filter(models.Flow.user_id == None) # noqa + .all() ) if flows: logger.debug("Migrating flows to default superuser") diff --git a/src/backend/langflow/services/utils.py b/src/backend/langflow/services/utils.py index 86cfed92e..b7d93184f 100644 --- a/src/backend/langflow/services/utils.py +++ b/src/backend/langflow/services/utils.py @@ -58,6 +58,9 @@ def get_or_create_super_user(session: Session, username, password, is_default): return create_super_user(username, password, db=session) except Exception as exc: if "UNIQUE constraint failed: user.username" in str(exc): + # This is to deal with workers running this + # at startup and trying to create the superuser + # at the same time. logger.debug("Superuser already exists.") return None