Jelajahi Sumber

role_avatar relation tmp

ming 3 tahun lalu
induk
melakukan
df62022532
2 mengubah file dengan 10 tambahan dan 6 penghapusan
  1. 2 1
      api/main.py
  2. 8 5
      api/util/user.py

+ 2 - 1
api/main.py

@@ -300,7 +300,8 @@ async def get_role(token: str = Depends(oauth2_scheme)):
 @app.post('/get_avatar_by_role')
 async def get_role(token: str = Depends(oauth2_scheme)):
     db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/AI_anchor?charset=utf8mb4')
-    user_id = get_user_id(token)
+    user_id = util.user.get_user_id(token)
+    roles = util.user.get_user_role(user_id)
     
     return user_id
 

+ 8 - 5
api/util/user.py

@@ -81,12 +81,15 @@ def get_user_role(id):
         role_list.append({'id':row['role_id'],'name':row['name']})
     return role_list
 
-def get_user_role(id):
+def get_avatar_by_role(id):
     db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/AI_anchor?charset=utf8mb4')
-    state = 'SELECT * FROM user_role '\
-    'INNER JOIN users on user_role.user_id= users.id '\
-    'INNER JOIN role on user_role.role_id = role.id '\
-    'WHERE username="ming"'
+    state = 'SELECT * FROM (
+'SELECT role_id FROM AI_anchor.user_role '\
+'INNER JOIN users on user_role.user_id= users.id '\
+'INNER JOIN role on user_role.role_id = role.id '\
+'WHERE AI_anchor.users.id=36) a'\
+'INNER JOIN role on a.role_id = role.id'\
+'INNER JOIN avatar on role_avatar.id = avatar.id'\
     role_list = []
     for row in db.query(state):
         role_list.append({'id':row['role_id'],'name':row['name']})