def delete_user(username):
user = User.objects(user_name=username).first()
if user:
user.delete()
else:
raise DatabaseError('No such user!')
def update_user(username, password):
user = User.objects(user_name=username).first()
if user:
user.update(set__user_passwd=password)
else:
raise DatabaseError('No such user!')
def create_user(username, password):
user = User.objects(user_name=username).first()
if user:
raise DatabaseError('User already exists!')
else:
User(user_name=username, user_passwd=password).save()
def get_user(username):
user = User.objects(user_name=username).first()
if user:
return {'username': user.user_name, 'password': user.user_passwd}
else:
raise DatabaseError('No such user!')
def logout(username):
user = User.objects(user_name=username).first()
if not user:
raise DatabaseError('No such user!')
def login(username, password):
user = User.objects(user_name=username, user_passwd=password).first()
if user:
return {'username': user.user_name, 'password': user.user_passwd}
else:
raise DatabaseError('Username or password error!')
