def update(cls, admin, data):
user_id = data.get('user_id')
now = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
if user_id:
del data['user_id']
data.update({'updated_at': now})
UserModel.update(**data).where(
UserModel.id == user_id,
UserModel.deleted_at == None).execute()
return cls.success_with_result({'updated_at': now})
else:
raise UserModel.LackOfFieldError(u'请传入用户ID')
def decorator(*args, **kwargs):
data = request.json or request.form or request.args
user_id = data.get('user_id')
if user_id:
try:
user = UserModel.get(UserModel.id == user_id,
UserModel.deleted_at == None)
kwargs.update({'user': user})
return f(*args, **kwargs)
except UserModel.DoesNotExist as e:
raise UserModel.NotFoundError(u'该用户不存在')
else:
raise UserModel.LackOfFieldError(u'请传入用户ID')