Update auths.py

backend/apps/webui/models/auths.py

@@ -102,6 +102,8 @@ class AuthsTable:
         role: str = "pending",
         oauth_sub: Optional[str] = None,
     ) -> Optional[UserModel]:
+        with get_db() as db:
+
             log.info("insert_new_auth")
 
             id = str(uuid.uuid4())
@@ -127,6 +129,8 @@ class AuthsTable:
     def authenticate_user(self, email: str, password: str) -> Optional[UserModel]:
         log.info(f"authenticate_user: {email}")
         try:
+            with get_db() as db:
+
                 auth = db.query(Auth).filter_by(email=email, active=True).first()
                 if auth:
                     if verify_password(password, auth.password):
@@ -154,6 +158,7 @@ class AuthsTable:
     def authenticate_user_by_trusted_header(self, email: str) -> Optional[UserModel]:
         log.info(f"authenticate_user_by_trusted_header: {email}")
         try:
+            with get_db() as db:
                 auth = db.query(Auth).filter(email=email, active=True).first()
                 if auth:
                     user = Users.get_user_by_id(auth.id)
@@ -163,13 +168,19 @@ class AuthsTable:
 
     def update_user_password_by_id(self, id: str, new_password: str) -> bool:
         try:
-            result = db.query(Auth).filter_by(id=id).update({"password": new_password})
+            with get_db() as db:
+
+                result = (
+                    db.query(Auth).filter_by(id=id).update({"password": new_password})
+                )
                 return True if result == 1 else False
         except:
             return False
 
     def update_email_by_id(self, id: str, email: str) -> bool:
         try:
+            with get_db() as db:
+
                 result = db.query(Auth).filter_by(id=id).update({"email": email})
                 return True if result == 1 else False
         except:
@@ -177,6 +188,8 @@ class AuthsTable:
 
     def delete_auth_by_id(self, id: str) -> bool:
         try:
+            with get_db() as db:
+
                 # Delete User
                 result = Users.delete_user_by_id(id)