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

🔧 chore(utils.py): add type hinting to the get_or_create_super_user function to improve code readability and maintainability

Browse source 
🔧 chore(utils.py): add type hinting to the session parameter in the get_or_create_super_user function to improve code readability and maintainability
🔧 chore(utils.py): add type hinting to the session parameter in the setup_superuser function to improve code readability and maintainability
🔧 chore(utils.py): add type hinting to the session parameter in the setup_superuser function to improve code readability and maintainability
🔧 chore(utils.py): add type hinting to the session parameter in the setup_superuser function to improve code readability and maintainability
🔧 chore(utils.py): add type hinting to the session parameter in the setup_superuser function to improve code readability and maintainability
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-03 14:27:12 -03:00
commit 7390c91028
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -6,11 +6,12 @@ from langflow.services.settings.constants import (
DEFAULT_SUPERUSER, DEFAULT_SUPERUSER,
DEFAULT_SUPERUSER_PASSWORD, DEFAULT_SUPERUSER_PASSWORD,
) )
from sqlmodel import Session
from .getters import get_session, get_settings_service from .getters import get_session, get_settings_service
from loguru import logger from loguru import logger
def get_or_create_super_user(session, username, password, is_default): def get_or_create_super_user(session: Session, username, password, is_default):
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()
@ -54,10 +55,10 @@ def get_or_create_super_user(session, username, password, is_default):
else: else:
logger.debug("Creating superuser.") logger.debug("Creating superuser.")
return create_super_user(session, username, password) return create_super_user(username, password, db=session)
def setup_superuser(settings_service, 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.")
@ -69,7 +70,9 @@ def setup_superuser(settings_service, session):
) )
try: try:
user = get_or_create_super_user(session, username, password, is_default) user = get_or_create_super_user(
session=session, username=username, password=password, is_default=is_default
)
if user is not None: if user is not None:
logger.debug("Superuser created successfully.") logger.debug("Superuser created successfully.")
except Exception as exc: except Exception as exc: