Browse Source

add edit profile api and limit user_profile

ming 3 years ago
parent
commit
a705dca991
2 changed files with 3 additions and 3 deletions
  1. 2 2
      api/main.py
  2. 1 1
      api/models.py

+ 2 - 2
api/main.py

@@ -147,7 +147,7 @@ async def user_profile(token: str = Depends(oauth2_scheme)):
     left_sec = user_obj['left_time']
 
     video_info_list = []
-    statement = 'SELECT * FROM history_input WHERE user_id='+str(user_obj['id']+' ORDER BY timestamp DESC LIMIT 50')
+    statement = 'SELECT * FROM history_input WHERE user_id='+str(user_obj['id'])+' ORDER BY timestamp DESC LIMIT 50')
     for row in db.query(statement):
         video_info_list.append({'id':row['id'],'title':row['name'],'duration':row['duration'],'url':row['link'],'time_stamp':row['timestamp'].strftime("%m/%d/%Y, %H:%M:%S")})
     dic_return = {'user_info':{'id':user_id,'userName':user_obj['username'],'email':user_obj['email'],'video_num':video_num,'total_sec':total_sec,'left_sec':user_obj['left_time']},'video_info':video_info_list}
@@ -161,7 +161,7 @@ async def edit_profile(userModel : models.UserProfile ,token: str = Depends(oaut
     user_id = get_user_id(token)
     print(user_id)
     user_obj = next(iter(db.query('SELECT * FROM users where id ="'+str(user_id)+'"')))
-    user_obj['email'] = 'asdadasd'
+    user_obj['email'] = userModel.email
     db['users'].update(dict(user_obj), ['id'])
     return {'msg':'ok'}
 

+ 1 - 1
api/models.py

@@ -67,4 +67,4 @@ class reset_pwd(BaseModel):
     password: str
 
 class UserProfile(BaseModel):
-    email:str
+    email: str