🔧 chore(__main__.py): refactor imports to improve readability and maintainability

 feat(__main__.py): add 'superuser' command to create a superuser with provided username and password
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-27 19:55:09 -03:00
commit 7d3a3db7e3

View file

@ -1,8 +1,9 @@
import sys
import time
import httpx
from langflow.services.manager import initialize_settings_manager
from langflow.services.utils import get_settings_manager
from langflow.services.database.utils import session_getter
from langflow.services.manager import initialize_services, initialize_settings_manager
from langflow.services.utils import get_db_manager, get_settings_manager
from langflow.utils.util import get_number_of_workers
from multiprocess import Process # type: ignore
import platform
@ -312,6 +313,24 @@ def run_langflow(host, port, log_level, options, app):
sys.exit(1)
@app.command()
def superuser(
username: str = typer.Option(..., prompt=True, help="Username for the superuser."),
password: str = typer.Option(
..., prompt=True, hide_input=True, help="Password for the superuser."
),
):
initialize_services()
db_manager = get_db_manager()
with session_getter(db_manager) as session:
from langflow.services.auth.utils import create_super_user
if create_super_user(session, username, password):
typer.echo("Superuser created successfully.")
else:
typer.echo("Superuser creation failed.")
def main():
app()