'avatarrole''

api/main.py

@@ -249,6 +249,8 @@ async def register_old(request: Request):
     user_obj = first(db.query('SELECT * FROM users where username ="'+str(user.username)+'"'))
     if user_obj == None:
         id = user_register(user)
+        result = util.user.add_to_basic_role(id)
+        print(result)
         if type(id) is int:
             code = str(time.time()).replace('.','')
             db['register_veri_code'].insert({'code':code,'user_id':id})
@@ -268,6 +270,8 @@ async def register(request: util.models.register_req):
     
     if user_obj == None:
         id = user_register(request)
+        result = util.user.add_to_basic_role(id)
+        print(result)
         if type(id) is int:
             code = str(time.time()).replace('.','')
             db['register_veri_code'].insert({'code':code,'user_id':id})

api/util/user.py

@@ -91,17 +91,17 @@ 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 role_avatar.role_id, avatar_id, avatar.name,num from role_avatar'\
-            'INNER JOIN'\
-            '(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 '\
-            'on role_avatar.role_id = a.role_id'\
+    state = 'select role_avatar.role_id, avatar_id, avatar.name,num from role_avatar '\
+            'INNER JOIN '\
+            '(SELECT role_id FROM user_role '\
+            'INNER JOIN users on user_role.user_id= users.id '\
+            'INNER JOIN role on user_role.role_id = role.id '\
+            'WHERE users.id='+str(id)+') a '\
+            'on role_avatar.role_id = a.role_id '\
             'INNER JOIN avatar on role_avatar.avatar_id = avatar.id;'
     role_list = []
     for row in db.query(state):
-        print(row)
+        role_list.append({'role_id':row['role_id'],'avatar_id':row['avatar_id'],'name':row['name'],'num':row['num']})
     db.close()
     return role_list
 #def add_role( username,role_id):
@@ -109,6 +109,17 @@ def get_avatar_by_role(id):
     #user_role_table = db['user_role']
     #user_role_table.insert({'user_id':,'role_id':role_id})
 
+def add_to_basic_role(id):
+    db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/AI_anchor?charset=utf8mb4')
+    state ='insert into user_role (user_id,role_id) values('+str(id)+',5) '
+    print(state)
+    code = 'ok'
+    try :
+        db.query(state)
+    except:
+        code = 'not ok'
+    return code
+
 def get_user_id(token):
     db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/AI_anchor?charset=utf8mb4')
     credentials_exception = HTTPException(