Advertisement
JessRoseE

models.py

May 17th, 2017
86
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.25 KB | None | 0 0
  1. rom app import db, models
  2. from werkzeug.security import generate_password_hash, check_password_hash
  3.  
  4. class User:
  5.     def register(username, password, email, nickname):
  6.         u = models.user_structure(UserName = username, PassWord = password, Email = email, NickName = nickname, Active = 1)
  7.         db.session.add(u)
  8.         db.session.commit()
  9.        
  10.     def getUserId(id):
  11.         return models.user_structure.query.get(id)
  12.        
  13.     def Verify_password(username, password):
  14.         u = models.user_structure.query.filter_by(UserName=username).first()
  15.         if u.PassWord == password :
  16.             return True
  17.         else:
  18.             return False
  19.            
  20.     def getUser(username):
  21.         return models.user_structure.query.filter_by(UserName=username).first()
  22.        
  23.     def getUserByNickName(nickname):
  24.         return models.user_structure.query.filter_by(NickName=nickname).first()
  25.    
  26.     def checkNickName(nickname):
  27.         return models.user_structure.query.filter_by(NickName=nickname).first() == None
  28.        
  29.     def checkUserName(username):
  30.         return models.user_structure.query.filter_by(UserName=username).first() == None
  31.        
  32.     def UpdateUser(User):
  33.         db.session.add(User)
  34.         db.session.commit()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement