From c56245ee6952ec195571847c6af04fc62602519c Mon Sep 17 00:00:00 2001 From: Valentin CZERYBA Date: Tue, 17 Oct 2023 22:01:59 +0200 Subject: [PATCH] add condition for add user --- app/main.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/app/main.py b/app/main.py index a7e42e0..55aaf51 100644 --- a/app/main.py +++ b/app/main.py @@ -14,10 +14,14 @@ app.include_router(mail.router) @app.on_event("startup") async def startup_event(): - if len(os.environ["USERNAME_ADMIN"]) > 0 and len(os.environ["PASSWORD_ADMIN"]) > 0: + if os.environ.get("USERNAME_ADMIN") is not None and os.environ.get("PASSWORD_ADMIN") is not None: user_add.add(username=os.environ["USERNAME_ADMIN"], password=os.environ["PASSWORD_ADMIN"], roles="Admin") - if len(os.environ["USERNAME_TEST"]) > 0 and len(os.environ["PASSWORD_TEST"]) > 0: + else: + print("User admin not added") + if os.environ.get("USERNAME_TEST") is not None and os.environ.get("PASSWORD_TEST") is not None: user_add.add(username=os.environ["USERNAME_TEST"], password=os.environ["PASSWORD_TEST"], roles="User") + else: + print("User test not added") @app.get("/") async def root(): -- 2.47.1