From 19653858bec79629d93c46fef624ea14ff14add2 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sun, 27 Aug 2023 19:55:20 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20chore(utils.py):=20refactor=20cr?= =?UTF-8?q?eate=5Fsuper=5Fuser=20function=20to=20accept=20optional=20usern?= =?UTF-8?q?ame=20and=20password=20parameters=20for=20flexibility=20?= =?UTF-8?q?=F0=9F=94=92=20fix(utils.py):=20use=20the=20provided=20username?= =?UTF-8?q?=20and=20password=20parameters=20if=20available=20when=20creati?= =?UTF-8?q?ng=20the=20super=20user,=20instead=20of=20relying=20solely=20on?= =?UTF-8?q?=20the=20settings=20manager?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/services/auth/utils.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/services/auth/utils.py b/src/backend/langflow/services/auth/utils.py index f89bd6c88..0582ee382 100644 --- a/src/backend/langflow/services/auth/utils.py +++ b/src/backend/langflow/services/auth/utils.py @@ -134,18 +134,20 @@ def create_token(data: dict, expires_delta: timedelta): ) -def create_super_user(db: Session = Depends(get_session)) -> User: +def create_super_user( + db: Session = Depends(get_session), username: str = None, password: str = None +) -> User: settings_manager = get_settings_manager() super_user = get_user_by_username( - db, settings_manager.auth_settings.FIRST_SUPERUSER + db, username or settings_manager.auth_settings.FIRST_SUPERUSER ) if not super_user: super_user = User( - username=settings_manager.auth_settings.FIRST_SUPERUSER, + username=username or settings_manager.auth_settings.FIRST_SUPERUSER, password=get_password_hash( - settings_manager.auth_settings.FIRST_SUPERUSER_PASSWORD + password or settings_manager.auth_settings.FIRST_SUPERUSER_PASSWORD ), is_superuser=True, is_active=True,