From 08c8e8c9a4d65677de6a493446a605d70efee631 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 10 十二月 2024 16:32:07 +0800
Subject: [PATCH] 12.10 16
---
app/service/auth.py | 33 ++++++++++++++++++---------------
1 files changed, 18 insertions(+), 15 deletions(-)
diff --git a/app/service/auth.py b/app/service/auth.py
index d843adb..d0436f8 100644
--- a/app/service/auth.py
+++ b/app/service/auth.py
@@ -77,7 +77,7 @@
db.refresh(db_user)
user_id = db_user.id
for k, v in register_dict.items():
- UserAppDao(db).update_and_insert_token(v.get("name"), pwd, v.get("email"), user_id, str(v.get("id")), k)
+ await UserAppDao(db).update_and_insert_data(v.get("name"), pwd, v.get("email"), user_id, str(v.get("id")), k)
except Exception as e:
logger.error(e)
@@ -89,11 +89,10 @@
async def update_user_token(db, user_id, token_dict):
-
try:
for k, v in token_dict.items():
- UserAppDao(db).update_user_app_data({"user_id": user_id, "app_type": k},
- {"access_token": v, "token_at": datetime.now()})
+ await UserAppDao(db).update_user_app_data({"user_id": user_id, "app_type": k},
+ {"access_token": v, "token_at": datetime.now()})
except Exception as e:
logger.error(e)
@@ -105,11 +104,11 @@
def __init__(self, db: Session):
self.db = db
- def get_data_by_id(self, user_id: int, app_type: int) -> Type[UserAppModel] | None:
+ async def get_data_by_id(self, user_id: int, app_type: int) -> Type[UserAppModel] | None:
session = self.db.query(UserAppModel).filter_by(user_id=user_id, app_type=app_type).first()
return session
- def update_user_app_data(self, query: int, update_data: str):
+ async def update_user_app_data(self, query: dict, update_data: dict):
logger.error("鏇存柊鏁版嵁df update_app_data---------------------------")
try:
@@ -120,7 +119,8 @@
self.db.rollback()
raise Exception("鏇存柊澶辫触锛�")
- def insert_user_app_data(self, username: str, password: str, email: str, user_id: int, app_id: str, app_type: int):
+ async def insert_user_app_data(self, username: str, password: str, email: str, user_id: int, app_id: str,
+ app_type: int):
logger.error("鏂板鏁版嵁df insert_user_app_data---------------------------")
new_session = UserAppModel(
username=username,
@@ -135,15 +135,18 @@
self.db.refresh(new_session)
return new_session
- def update_and_insert_token(self, username: str, password: str, email: str, user_id: int, app_id: str,
- app_type: int):
+ async def update_and_insert_data(self, username: str, password: str, email: str, user_id: int, app_id: str,
+ app_type: int):
logger.error("鏇存柊鎴栬�呮坊鍔犳暟鎹� update_and_insert_token---------------------------")
- token_boj = self.get_data_by_id(user_id, app_type)
+ token_boj = await self.get_data_by_id(user_id, app_type)
if token_boj:
- self.update_user_app_data({"id": token_boj.id}, {"username": username,
- "password": password, "email": email, "username": username,
- "updated_at": datetime.now(),
- })
+ await self.update_user_app_data({"id": token_boj.id}, {"username": username,
+ "password": password, "email": email,
+ "updated_at": datetime.now(),
+ })
else:
- self.insert_user_app_data(username, password, email, user_id, app_id, app_type)
+ await self.insert_user_app_data(username, password, email, user_id, app_id, app_type)
+
+ async def get_user_datas(self, user_id: int):
+ return self.db.query(UserAppModel).filter_by(user_id=user_id).all()
--
Gitblit v1.8.0