jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

🔧 chore(utils.py): refactor create_super_user function to accept optional username and password parameters for flexibility

Browse source 
🔒 fix(utils.py): use the provided username and password parameters if available when creating the super user, instead of relying solely on the settings manager
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-27 19:55:20 -03:00
commit 19653858be
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/backend/langflow/services/auth/utils.py
View file

@ -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() settings_manager = get_settings_manager()
super_user = get_user_by_username( 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: if not super_user:
super_user = User( super_user = User(
username=settings_manager.auth_settings.FIRST_SUPERUSER, username=username or settings_manager.auth_settings.FIRST_SUPERUSER,
password=get_password_hash( password=get_password_hash(
settings_manager.auth_settings.FIRST_SUPERUSER_PASSWORD password or settings_manager.auth_settings.FIRST_SUPERUSER_PASSWORD
), ),
is_superuser=True, is_superuser=True,
is_active=True, is_active=True,