def add_user():
"""
添加新用户
:return:
"""
username = request.forms.username
password = request.forms.password
nickname = request.forms.nickname
status = request.forms.status
create_time = datetime.datetime.now()
# 加密密码
password = hashlib.md5(password.encode("UTF-8")).hexdigest()
# 检测用户名是否存在
if UserModel.select().where(UserModel.username == username,
UserModel.status != 2).count():
return {"code": -1}
user = UserModel(username=username,
password=password,
nickname=nickname,
status=status,
create_time=create_time)
user.save()
return {"code": 0}
def post(self):
data = arguments.parse_args()
if UserModel.findByLogin(data['login']):
return BadRequest('The informed login already exists.')
if UserModel.findByEmail(data['email']):
return BadRequest('The informed email already exists.')
user = UserModel(**data)
user.activated = False
user.save()
user.send_confirmation_email()
return user.json(), 201
def post(self):
data = User.parser.parse_args()
user = UserModel.find(data["username"])
if user:
return {
"message": "User {} already exists".format(data["username"])
}, 400
user = UserModel(**data)
user.save()
return {"message": "User added"}, 201
def on_post(self, req, resp):
email = req.media['email']
password = req.media['password']
if not password:
resp.status = falcon.HTTP_400
resp.body = "Password cannot be emptied!"
if not UserModel.objects(email=email):
hashPassword = bcrypt.hashpw(bytes(password, encoding='utf-8'),
bcrypt.gensalt())
newUser = UserModel(email=email, password=hashPassword)
newUser.save()
resp.status = falcon.HTTP_201
resp.body = "Your account has been created!"
else:
resp.status = falcon.HTTP_400
resp.body = "The email has been used!"
def add_user():
"""
添加新用户
:return:
"""
username = request.forms.username
password = request.forms.password
nickname = request.forms.nickname
status = request.forms.status
create_time = datetime.datetime.now()
# 加密密码
password = hashlib.md5(password.encode("UTF-8")).hexdigest()
# 检测用户名是否存在
if UserModel.select().where(UserModel.username == username, UserModel.status != 2).count():
return {"code": -1}
user = UserModel(username=username, password=password, nickname=nickname, status=status, create_time=create_time)
user.save()
return {"code": 0}
