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')
def migrate():
if UserAddress.table_exists():
UserAddress.drop_table()
UserAddress.create_table()
user = User.get(User.phone_number == '13632324433')
UserAddress.create(user_id=user.id,
region_code='440606',
street='勒流镇众涌村仁厚街2号',
is_default=True,
consignee='卢炳成',
contact='13632324433')
def create(cls, admin, data):
import backend_common.constants.sex as sex
phone_number = data.get('phone_number')
user_data = {
'nick': data.get('nick'),
'sex': data.get('sex', sex.UNKNOWN),
'avatar': data.get('avatar'),
'register_type': data.get('register_type')
}
user, is_new_created = UserModel.get_or_create(
phone_number=phone_number, defaults=user_data)
return cls.success_with_result(user.format('id,created_at'))
def migrate():
if Order.table_exists():
Order.drop_table()
Order.create_table()
user = User.get(User.phone_number == '13632324433')
user_address = UserAddress.get(UserAddress.user_id == user.id,
UserAddress.is_default == True)
Order.create(user_id=user.id,
logistics_user_address_id=user_address.id,
status=order_status.COMPLETED,
remark='这是测试的订单',
total_price=597,
paid_price=577)
def migrate():
if User.table_exists():
User.drop_table()
User.create_table()
import bcrypt
import backend_common.constants.sex as sex
import backend_common.constants.register_type as register_type
password = bcrypt.hashpw(u'123456'.encode('utf-8'), bcrypt.gensalt())
User.insert(phone_number=u'13632324433',
password=password,
nick=u'avicha',
sex=sex.MAN,
register_type=register_type.SYSTEM).execute()
def list(cls, admin, data):
page_number = int(data.get('page_number', 1))
page_size = int(data.get('page_size', 100))
phone_number = data.get('phone_number')
q = UserModel.select().where(UserModel.deleted_at == None)
if phone_number:
q = q.where(UserModel.phone_number == phone_number)
total_rows = q.count()
result = []
for x in q.order_by(UserModel.created_at.desc()).paginate(
page_number, page_size):
user = x.format(
'id,phone_number,nick,sex,avatar,register_type,created_at')
result.append(user)
return cls.success_with_list_result(total_rows, result)
def decorator(*args, **kwargs):
from flask import request, current_app
from itsdangerous import TimedJSONWebSignatureSerializer, SignatureExpired, BadSignature
from backend_common.models.user import User as UserModel
data = request.json or request.form or request.args
if data.get('token'):
token = data.get('token')
s = TimedJSONWebSignatureSerializer(current_app.secret_key)
try:
user_id = s.loads(token)
user = UserModel.get(UserModel.id == user_id,
UserModel.deleted_at == None)
tokens = user.tokens()
if token in tokens:
kwargs.update({'user': user})
return f(*args, **kwargs)
else:
raise UserModel.TokenError(u'请重新登录')
except SignatureExpired:
raise UserModel.TokenError(u'登录已经过期')
except BadSignature:
raise UserModel.PasswordError()
except UserModel.DoesNotExist, e:
raise UserModel.NotFoundError(u'该用户不存在')
def delete(cls, admin, data):
user_id = data.get('user_id')
now = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
UserModel.update(deleted_at=now).where(
UserModel.id == user_id).execute()
return cls.success_with_result({'deleted_at': now})
def decorator(*args, **kwargs):
if kwargs.get('user'):
return f(*args, **kwargs)
raise UserModel.NotAuthError()
def user():
from backend_common.models.user import User as UserModel
user = UserModel.select().where(UserModel.deleted_at == None).order_by(
UserModel.created_at.desc()).first()
return user