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

🔧 chore(utils.py): refactor setup_superuser and teardown_superuser functions to improve readability and remove unnecessary code duplication

Browse source 
🔧 chore(utils.py): update teardown_superuser function to only remove default superuser if AUTO_LOGIN is set to False
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-20 18:25:42 -03:00
commit f060155fe6
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/backend/langflow/services/utils.py
View file

@ -102,6 +102,9 @@ def get_or_create_super_user(session: Session, username, password, is_default):
def setup_superuser(settings_service, session: Session): def setup_superuser(settings_service, session: Session):
if settings_service.auth_settings.AUTO_LOGIN: if settings_service.auth_settings.AUTO_LOGIN:
logger.debug("AUTO_LOGIN is set to True. Creating default superuser.") logger.debug("AUTO_LOGIN is set to True. Creating default superuser.")
else:
# Remove the default superuser if it exists
teardown_superuser(settings_service, session)
username = settings_service.auth_settings.SUPERUSER username = settings_service.auth_settings.SUPERUSER
password = settings_service.auth_settings.SUPERUSER_PASSWORD password = settings_service.auth_settings.SUPERUSER_PASSWORD
@ -132,10 +135,12 @@ def teardown_superuser(settings_service, session):
# If AUTO_LOGIN is True, we will remove the default superuser # If AUTO_LOGIN is True, we will remove the default superuser
# from the database. # from the database.
if settings_service.auth_settings.AUTO_LOGIN: if not settings_service.auth_settings.AUTO_LOGIN:
try: try:
logger.debug("AUTO_LOGIN is set to True. Removing default superuser.") logger.debug(
username = settings_service.auth_settings.SUPERUSER "AUTO_LOGIN is set to False. Removing default superuser if exists."
)
username = DEFAULT_SUPERUSER
from langflow.services.database.models.user.user import User from langflow.services.database.models.user.user import User
user = session.query(User).filter(User.username == username).first() user = session.query(User).filter(User.username == username).first()