add_reset_pwd

api/main.py

@@ -273,7 +273,11 @@ async def send_reset_pwd(user_id,email):
     print(msg)
     msg =msg.encode(encoding='utf-8')
     print(msg)
-    user_dict = next(iter(db.query('SELECT * FROM users where id ="'+str(user_id)+'"')))
+    if user_id != -1:
+        user_dict = next(iter(db.query('SELECT * FROM users where id ="'+str(user_id)+'"')))
+    else:
+        user_id = user.get_id_by_email(email)
+        user_dict = next(iter(db.query('SELECT * FROM users where id ="'+str(user_id)+'"')))
     mailer.send(msg, user_dict['email'])
     return {'msg':'ok'}
 
@@ -283,7 +287,7 @@ async def reset_password(req :models.reset_pwd):
     veri_obj = next(iter(db.query('SELECT * FROM reset_pw_code where code ="'+str(req.code)+'"')))
     if veri_obj != None:
         db['reset_pw_code'].delete(code=req.code)
-    if veri_obj['user_id'] == -1:
+    if veri_obj['user_id'] != -1:
         val = db.query('UPDATE users SET password = '+'"'+get_password_hash(req.password)+'" where id ='+str(veri_obj['user_id']))
     else:
         user_id = user.get_id_by_email(veri_obj['email'])