def add_lib(user_id, lib_name):
"""
添加新的分类列表
"""
new_lib = Library(owner_id=user_id, lib_name=lib_name)
try:
new_lib.save()
User.objects(id=user_id).update_one(inc__lib_amount=1)
code = 200
new_lib_id = new_lib.id
except Exception as e:
print(str(e))
new_lib_id = ''
code = 403
return new_lib_id, code
def RegisterHandler(username, password_hash):
userQuery = User.objects(username=username).first()
if userQuery == None:
# 创建新的用户记录
new_user = User(
username=username,
password_hash=password_hash,
user_type='normal'
)
new_user.save()
# 为新用户创建待读列表
new_lib = Library(
owner_id=new_user.id,
lib_name='待读列表'
)
new_lib.save()
return 'Register Success', 200
else:
msg = 'Username exists!'
return msg, 403