def put(self, store_id, user_id, name_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(400, u"店铺不存在")
user = store.users.filter_by(id=user_id).first()
if not user:
return tool.fail(404, u"用户不存在")
user_name = user.names.filter_by(id=name_id).first()
if not user_name:
return tool.fail(404, u"用户姓名不存在")
self.parser.add_argument('name', type=unicode)
self.parser.add_argument('remark', type=unicode)
self.parser.add_argument('address', type=unicode)
args = self.parser.parse_args()
if args.name:
user_name.name = args.name
if args.remark is not None:
user_name.remark = args.remark
if args.address is not None:
user_name.address = args.address
user_name.save()
result = schemas.UserNameSchema().dump(user_name)
return tool.success(result.data)
def delete(self, store_id):
store = current_user.stores.filter_by(id=store_id).first_or_404()
if store.companies.count() > 0:
return tool.fail(407, u"存在关联,无法删除")
if store.users.count() > 0:
return tool.fail(407, u"存在关联,无法删除")
current_user.stores.remove(store)
db.session.delete(store)
current_user.save()
return tool.success("")
def get(self, store_id, user_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(400, u"店铺不存在")
user = store.users.filter_by(id=user_id).first()
if not user:
return tool.fail(404, u"用户不存在")
result = schemas.UserSchema(exclude=("orders",)).dump(user)
return tool.success(result.data)
def delete(self, store_id, user_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(400, u"店铺不存在")
user = store.users.filter_by(id=user_id).first()
if not user:
return tool.fail(404, u"用户不存在")
if user.orders.count() > 0:
return tool.fail(407, u"存在关联,无法删除")
models.db.session.delete(user)
models.db.session.commit()
return tool.success(None)
def get(self, store_id, order_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(404, u"店铺不存在")
order = models.Order.query.join(models.Company).filter(
models.Company.store == store,
models.Order.id == order_id).first()
if not order:
return tool.fail(404, u"快递单不存在")
result = schemas.OrderSchema().dump(order)
return tool.success(result.data)
def get(self, store_id, user_id, name_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(400, u"店铺不存在")
user = store.users.filter_by(id=user_id).first()
if not user:
return tool.fail(404, u"用户不存在")
user_name = user.names.filter_by(id=name_id).first()
if not user_name:
return tool.fail(404, u"用户姓名不存在")
result = schemas.UserNameSchema().dump(user_name)
return tool.success(result.data)
def delete(self, store_id, order_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(404, u"店铺不存在")
order = models.Order.query.join(models.Company).filter(
models.Company.store == store,
models.Order.id == order_id).first()
if not order:
return tool.fail(404, u"快递单不存在")
models.db.session.delete(order)
models.db.session.commit()
return tool.success(None, u"删除成功")
def delete(self, store_id, user_id, name_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(400, u"店铺不存在")
user = store.users.filter_by(id=user_id).first()
if not user:
return tool.fail(404, u"用户不存在")
user_name = user.names.filter_by(id=name_id).first()
if not user_name:
return tool.fail(404, u"用户姓名不存在")
models.db.session.delete(user_name)
models.db.session.commit()
return tool.success(None)
def delete(self, store_id, order_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(404, u"店铺不存在")
order = models.Order.query.join(models.Company).filter(
models.Company.store == store,
models.Order.id == order_id
).first()
if not order:
return tool.fail(404, u"快递单不存在")
models.db.session.delete(order)
models.db.session.commit()
return tool.success(None, u"删除成功")
def get(self, store_id, order_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(404, u"店铺不存在")
order = models.Order.query.join(models.Company).filter(
models.Company.store == store,
models.Order.id == order_id
).first()
if not order:
return tool.fail(404, u"快递单不存在")
result = schemas.OrderSchema().dump(order)
return tool.success(result.data)
def post(self, store_id):
self.parser.add_argument('mobile', type=unicode, required=True, help="手机号码必填")
args = self.parser.parse_args()
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(400, u"店铺不存在")
user = store.users.filter_by(mobile=args.mobile).first()
if user:
return tool.fail(406, u"手机号码已存在")
user = models.User(mobile=args.mobile, store=store)
models.db.session.add(user)
models.db.session.commit()
result = schemas.UserSchema(exclude=("orders",)).dump(user)
return tool.success(result.data)
def get(self, store_id):
self.parser.add_argument('page', type=int, location="args", default=1)
self.parser.add_argument('per_page', type=int, location="args", default=10)
self.parser.add_argument('mobile', type=unicode, location="args")
self.parser.add_argument('name', type=unicode, location="args")
args = self.parser.parse_args()
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(400, u"店铺不存在")
order_query = models.User.query.filter(models.User.store == store)
if args.mobile:
order_query = order_query.filter(
models.User.mobile.like(u'%{}%'.format(args.mobile))
)
if args.name:
order_query = order_query.join(models.UserName).filter(
models.UserName.name.like(u'%{}%'.format(args.name))
)
per_page = args['per_page'] if args['per_page'] <= 100 else 100
paginate = order_query.paginate(args.page, per_page, False)
result = schemas.UserSchema(many=True, exclude=("orders", )).dump(paginate.items)
return tool.success({
'items': result.data,
'paginate': tool.paginate_to_json(paginate)
})
def delete(self, store_id, company_id):
store = current_user.stores.filter_by(id=store_id).first_or_404()
company = store.companies.filter_by(id=company_id).first_or_404()
if company.orders.count() > 0:
return tool.fail(407, u"存在关联,无法删除")
db.session.delete(company)
db.session.commit()
return tool.success('')
def put(self, store_id, order_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(404, u"店铺不存在")
order = models.Order.query.join(models.Company).filter(
models.Company.store == store,
models.Order.id == order_id
).first()
if not order:
return tool.fail(404, u"快递单不存在")
self.parser.add_argument('company_id', type=int, help="公司")
self.parser.add_argument('no', type=int, default=None)
self.parser.add_argument('remark', type=unicode)
self.parser.add_argument('status', type=int)
self.parser.add_argument('user_id', type=int)
args = self.parser.parse_args()
if args.company_id:
company = models.Company.get_by_user_id_and_company_id(current_user.id, args.company_id)
if not company:
return tool.fail(400, u"公司不存在")
if order.company_id != args.company_id:
order.company_id = args.company_id
if args.user_id:
user = order.company.store.users.filter_by(id=args.user_id).first()
if not user:
return tool.fail(400, u"用户不存在")
if order.user_id != user.id:
order.user_id = user.id
if args.no:
if args.no < 1:
return tool.fail(407, u"编号不能小于 1")
if args.status:
order.status = args.status
order.updated_at = tool.now()
if args.no:
order.no = args.no
if args.remark:
order.remark = args.remark
order.save()
result = schemas.OrderSchema().dump(order)
return tool.success(result.data)
def put(self, store_id, order_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(404, u"店铺不存在")
order = models.Order.query.join(models.Company).filter(
models.Company.store == store,
models.Order.id == order_id).first()
if not order:
return tool.fail(404, u"快递单不存在")
self.parser.add_argument('company_id', type=int, help="公司")
self.parser.add_argument('no', type=int, default=None)
self.parser.add_argument('remark', type=unicode)
self.parser.add_argument('status', type=int)
self.parser.add_argument('user_id', type=int)
args = self.parser.parse_args()
if args.company_id:
company = models.Company.get_by_user_id_and_company_id(
current_user.id, args.company_id)
if not company:
return tool.fail(400, u"公司不存在")
if order.company_id != args.company_id:
order.company_id = args.company_id
if args.user_id:
user = order.company.store.users.filter_by(id=args.user_id).first()
if not user:
return tool.fail(400, u"用户不存在")
if order.user_id != user.id:
order.user_id = user.id
if args.no:
if args.no < 1:
return tool.fail(407, u"编号不能小于 1")
if args.status:
order.status = args.status
order.updated_at = tool.now()
if args.no:
order.no = args.no
if args.remark:
order.remark = args.remark
order.save()
result = schemas.OrderSchema().dump(order)
return tool.success(result.data)
def put(self, store_id, user_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(400, u"店铺不存在")
user = store.users.filter_by(id=user_id).first()
if not user:
return tool.fail(404, u"用户不存在")
self.parser.add_argument('mobile', type=unicode, required=True, help="手机号码必填")
args = self.parser.parse_args()
_user = store.users.filter_by(mobile=args.mobile).first()
if _user:
if _user.id != user.id:
return tool.fail(406, u"手机号码已存在")
user.mobile = args.mobile
user.save()
result = schemas.UserSchema(exclude=("orders",)).dump(user)
return tool.success(result.data)
def post(self, store_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(404, u"店铺不存在")
self.parser.add_argument('company_id',
type=int,
required=True,
help="公司必填")
self.parser.add_argument('number',
type=unicode,
required=True,
help="快递单号必填")
self.parser.add_argument('no', type=int, default=None)
args = self.parser.parse_args()
company = store.companies.filter_by(id=args.company_id).first()
if not company:
return tool.fail(400, u"公司不存在")
order = company.orders.filter_by(number=args.number).first()
if order:
return tool.fail(406, u"快递单号已存在")
order = models.Order()
if args.no:
order.no = args.no
else:
# 计算出 no
# 1 此用户已有未领取的快递时,编同一个号
no = models.Order.get_no(store.id)
order.no = no
order.status = 0
order.company = company
order.number = args.number
order.is_quick = True
models.db.session.add(order)
models.db.session.commit()
result = schemas.OrderSchema().dump(order)
return tool.success(result.data)
def post(self, store_id, user_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(400, u"店铺不存在")
user = store.users.filter_by(id=user_id).first()
if not user:
return tool.fail(404, u"用户不存在")
self.parser.add_argument('name', type=unicode, required=True, help="用户姓名必填")
self.parser.add_argument('remark', type=unicode)
self.parser.add_argument('address', type=unicode)
args = self.parser.parse_args()
user_name = models.UserName()
user_name.name = args.name
user_name.remark = args.remark
user_name.address = args.address
user_name.user = user
models.db.session.add(user_name)
models.db.session.commit()
result = schemas.UserNameSchema().dump(user_name)
return tool.success(result.data)
def put(self):
self.parser.add_argument('name', type=unicode)
self.parser.add_argument('password', type=unicode)
self.parser.add_argument('password1', type=unicode)
self.parser.add_argument('password2', type=unicode)
args = self.parser.parse_args()
if args.name:
current_user.name = args.name
if args.password1:
password2 = args.password2
password = args.password
if password2 != args.password1:
return tool.fail(407, '两次密码不同')
if not current_user.check_password(password):
return tool.fail(408, '原密码不正确')
current_user.password = password2
current_user.save()
member_schema = schemas.MemberSchema(exclude=("password",))
return tool.success(member_schema.dump(current_user).data)
def post(self, store_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(404, u"店铺不存在")
self.parser.add_argument('company_id', type=int, required=True, help="公司必填")
self.parser.add_argument('number', type=unicode, required=True, help="快递单号必填")
self.parser.add_argument('no', type=int, default=None)
args = self.parser.parse_args()
company = store.companies.filter_by(id=args.company_id).first()
if not company:
return tool.fail(400, u"公司不存在")
order = company.orders.filter_by(number=args.number).first()
if order:
return tool.fail(406, u"快递单号已存在")
order = models.Order()
if args.no:
order.no = args.no
else:
# 计算出 no
# 1 此用户已有未领取的快递时,编同一个号
no = models.Order.get_no(store.id)
order.no = no
order.status = 0
order.company = company
order.number = args.number
order.is_quick = True
models.db.session.add(order)
models.db.session.commit()
result = schemas.OrderSchema().dump(order)
return tool.success(result.data)
def get(self, store_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(404, u"店铺不存在")
self.parser.add_argument('page', type=int, location="args", default=1)
self.parser.add_argument('per_page',
type=int,
location="args",
default=10)
self.parser.add_argument('company_id',
type=int,
help="公司 ID",
location="args")
self.parser.add_argument('number', type=unicode, location="args")
args = self.parser.parse_args()
per_page = args['per_page'] if args['per_page'] <= 100 else 100
order_query = models.Order.query. \
join(models.Company).filter(
models.Company.store == store,
models.Order.status == 0,
models.Order.is_quick == True
)
if args.company_id:
order_query = order_query.filter(
models.Order.company_id == args.company_id)
if args.number:
order_query = order_query.filter(
models.Order.number == args.number)
paginate = order_query.order_by(
models.Order.no,
models.Order.created_at.desc()).paginate(args.page, per_page,
False)
order_schema = schemas.OrderSchema(many=True)
return tool.success({
"items": order_schema.dump(paginate.items).data,
"paginate": tool.paginate_to_json(paginate)
})
def get(self, store_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(404, u"店铺不存在")
self.parser.add_argument('page', type=int, location="args", default=1)
self.parser.add_argument('per_page', type=int, location="args", default=10)
self.parser.add_argument('company_id', type=int, help="公司 ID", location="args")
self.parser.add_argument('number', type=unicode, location="args")
args = self.parser.parse_args()
per_page = args['per_page'] if args['per_page'] <= 100 else 100
order_query = models.Order.query. \
join(models.Company).filter(
models.Company.store == store,
models.Order.status == 0,
models.Order.is_quick == True
)
if args.company_id:
order_query = order_query.filter(
models.Order.company_id == args.company_id
)
if args.number:
order_query = order_query.filter(
models.Order.number == args.number
)
paginate = order_query.order_by(models.Order.no, models.Order.created_at.desc()).paginate(
args.page,
per_page,
False
)
order_schema = schemas.OrderSchema(many=True)
return tool.success({
"items": order_schema.dump(paginate.items).data,
"paginate": tool.paginate_to_json(paginate)
})
def get(self, store_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(404, u"店铺不存在")
# order_query = models.Order.query.join(models.Company).filter(
# models.Order.status < 10,
# models.Company.store == store,
# ).order_by(models.Order.no)
# all_no = [i.no for i in order_query]
# if not all_no:
# return tool.success(1)
#
# number = all_no[0] - 1
# if number > 0:
# return tool.success(1)
# for index, no in enumerate(map(lambda x: x-number, all_no)):
# if no != index + 1:
# return tool.success(index + 1 + number)
#
# return tool.success(all_no[-1] + 1)
return tool.success(models.Order.get_no(store.id))
def get(self, store_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(404, u"店铺不存在")
# order_query = models.Order.query.join(models.Company).filter(
# models.Order.status < 10,
# models.Company.store == store,
# ).order_by(models.Order.no)
# all_no = [i.no for i in order_query]
# if not all_no:
# return tool.success(1)
#
# number = all_no[0] - 1
# if number > 0:
# return tool.success(1)
# for index, no in enumerate(map(lambda x: x-number, all_no)):
# if no != index + 1:
# return tool.success(index + 1 + number)
#
# return tool.success(all_no[-1] + 1)
return tool.success(models.Order.get_no(store.id))
def put(self, store_id, company_id):
store = current_user.stores.filter_by(id=store_id).first_or_404()
company = store.companies.filter_by(id=company_id).first_or_404()
self.parser.add_argument('name', type=unicode)
self.parser.add_argument('remark', type=unicode)
self.parser.add_argument('web_site', type=unicode)
self.parser.add_argument('store_id', type=int)
args = self.parser.parse_args()
if args.get('store_id'):
store = current_user.stores.filter_by(id=args['store_id']).first()
if not store:
return tool.fail(404, u"店铺不存在"), 404
company.store = store
if args.get('name'):
company.name = args['name']
company.remark = args.get('remark', company.remark)
company.web_site = args.get('web_site', company.web_site)
company.save()
company_schema = schemas.CompanySchema()
result = company_schema.dump(company)
return tool.success(result.data)
def get(self, store_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(404, u"店铺不存在")
self.parser.add_argument('page', type=int, location="args", default=1)
self.parser.add_argument('per_page',
type=int,
location="args",
default=10)
self.parser.add_argument('start_at',
type=lib.arrow_datetime,
help="开始时间必填",
required=True,
location="args")
self.parser.add_argument('end_at',
type=lib.arrow_datetime,
help="结束时间",
location="args")
self.parser.add_argument('user_id',
type=int,
help="用户 ID",
location="args")
self.parser.add_argument('company_id',
type=int,
help="公司 ID",
location="args")
self.parser.add_argument('number', type=unicode, location="args")
self.parser.add_argument('status',
type=int,
location="args",
default=None)
self.parser.add_argument('key', type=unicode, location="args")
self.parser.add_argument('is_not_sign',
type=unicode,
location="args",
default=None)
args = self.parser.parse_args()
per_page = args['per_page'] if args['per_page'] <= 100 else 100
end_at = args.end_at or tool.now(False)
order_query = models.Order.query.\
join(models.Company).filter(
models.Company.store == store,
models.Order.created_at >= args.start_at.naive,
models.Order.created_at <= end_at.naive,
).join(models.UserName).join(models.User)
if args.user_id:
order_query = order_query.filter(
models.Order.user_id == args.user_id, )
if args.company_id:
order_query = order_query.filter(
models.Order.company_id == args.company_id, )
if args.number:
order_query = order_query.filter(
models.Order.number.like(u'%{}%'.format(args.number)))
if args.key:
order_query = order_query.filter(
or_(
models.Order.number.like(u'%{}%'.format(args.key)),
models.User.mobile.like(u'%{}%'.format(args.key)),
models.UserName.name.like(u'%{}%'.format(args.key)),
))
if args.status is not None:
if args.is_not_sign is None:
order_query = order_query.filter(
models.Order.status == args.status)
if args.is_not_sign is not None:
order_query = order_query.filter(models.Order.status < 10)
paginate = order_query.order_by(
models.Order.no,
models.Order.created_at.desc()).paginate(args.page, per_page,
False)
order_schema = schemas.OrderSchema(many=True)
return tool.success({
"items": order_schema.dump(paginate.items).data,
"paginate": tool.paginate_to_json(paginate)
})
def post(self, store_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(404, u"店铺不存在")
self.parser.add_argument('company_id', type=int, required=True, help="公司必填")
self.parser.add_argument('number', type=unicode, required=True, help="快递单号必填")
self.parser.add_argument('no', type=int, default=None)
self.parser.add_argument('remark', type=unicode)
self.parser.add_argument('plan_at', type=lib.arrow_datetime)
self.parser.add_argument('status', type=int, default=0)
self.parser.add_argument('user', type=dict, required=True, help="用户必填")
args = self.parser.parse_args()
company = store.companies.filter_by(id=args.company_id).first()
if not company:
return tool.fail(400, u"公司不存在")
order = company.orders.filter_by(number=args.number).first()
if order:
return tool.fail(406, u"快递单号已存在")
if args.no:
if args.no < 1:
return tool.fail(407, u"编号不能小于 1")
if "id" in args.user:
user = store.users.filter_by(id=args.user['id']).first()
if "user_name_id" in args.user:
user_name = user.names.filter_by(id=args.user['user_name_id']).first()
if not user_name:
return tool.fail(400, u"用户名字不存在")
else:
user_name = user.names.first()
# # 用户编号都存在时,看是否为他自己的
# order = models.Order.query.join(models.Company).join(models.Store).filter(
# models.Store.id == store.id,
# models.Order.no == args.no,
# models.Order.user != user,
# ).first()
# if order:
# return tool.fail(406, u"编号已存在")
if not user:
return tool.fail(400, u"用户不存在")
else:
# order = models.Order.query.join(models.Company).join(models.Store).filter(
# models.Store.id == store.id,
# models.Order.no == args.no,
# ).first()
# if order:
# return tool.fail(406, u"编号已存在")
if "mobile" not in args.user:
return tool.fail(400, u"联系方式必填")
if "name" not in args.user:
return tool.fail(400, u"姓名必填")
if store.get_user_by_mobile(args.user["mobile"]):
return tool.fail(406, u"手机号码 {} 已存在".format(args.user["mobile"]))
user = models.User()
user.mobile = args.user["mobile"]
user_name = models.UserName(name=args.user["name"])
user.store = store
models.db.session.add(user)
user_name.user = user
if "address" in args.user:
user_name.address = args.user["address"]
if "remark" in args.user:
user_name.remark = args.user["remark"]
models.db.session.add(user_name)
order = models.Order()
if args.no:
order.no = args.no
else:
# 计算出 no
# 1 此用户已有未领取的快递时,编同一个号
no = models.Order.get_no(store.id)
order.no = no
order.user_name = user_name
order.company = company
order.number = args.number
order.status = args.status
if args.plan_at:
order.plan_at = args.plan_at.naive
models.db.session.add(order)
models.db.session.commit()
result = schemas.OrderSchema().dump(order)
return tool.success(result.data)
def get(self, store_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(404, u"店铺不存在")
self.parser.add_argument('page', type=int, location="args", default=1)
self.parser.add_argument('per_page', type=int, location="args", default=10)
self.parser.add_argument('start_at', type=lib.arrow_datetime, help="开始时间必填", required=True, location="args")
self.parser.add_argument('end_at', type=lib.arrow_datetime, help="结束时间", location="args")
self.parser.add_argument('user_id', type=int, help="用户 ID", location="args")
self.parser.add_argument('company_id', type=int, help="公司 ID", location="args")
self.parser.add_argument('number', type=unicode, location="args")
self.parser.add_argument('status', type=int, location="args", default=None)
self.parser.add_argument('key', type=unicode, location="args")
self.parser.add_argument('is_not_sign', type=unicode, location="args", default=None)
args = self.parser.parse_args()
per_page = args['per_page'] if args['per_page'] <= 100 else 100
end_at = args.end_at or tool.now(False)
order_query = models.Order.query.\
join(models.Company).filter(
models.Company.store == store,
models.Order.created_at >= args.start_at.naive,
models.Order.created_at <= end_at.naive,
).join(models.UserName).join(models.User)
if args.user_id:
order_query = order_query.filter(
models.Order.user_id == args.user_id,
)
if args.company_id:
order_query = order_query.filter(
models.Order.company_id == args.company_id,
)
if args.number:
order_query = order_query.filter(
models.Order.number.like(u'%{}%'.format(args.number))
)
if args.key:
order_query = order_query.filter(
or_(
models.Order.number.like(u'%{}%'.format(args.key)),
models.User.mobile.like(u'%{}%'.format(args.key)),
models.UserName.name.like(u'%{}%'.format(args.key)),
)
)
if args.status is not None:
if args.is_not_sign is None:
order_query = order_query.filter(
models.Order.status == args.status
)
if args.is_not_sign is not None:
order_query = order_query.filter(
models.Order.status < 10
)
paginate = order_query.order_by(models.Order.no, models.Order.created_at.desc()).paginate(
args.page,
per_page,
False
)
order_schema = schemas.OrderSchema(many=True)
return tool.success({
"items": order_schema.dump(paginate.items).data,
"paginate": tool.paginate_to_json(paginate)
})
def post(self, store_id):
store = current_user.stores.filter_by(id=store_id).first()
if not store:
return tool.fail(404, u"店铺不存在")
self.parser.add_argument('company_id',
type=int,
required=True,
help="公司必填")
self.parser.add_argument('number',
type=unicode,
required=True,
help="快递单号必填")
self.parser.add_argument('no', type=int, default=None)
self.parser.add_argument('remark', type=unicode)
self.parser.add_argument('plan_at', type=lib.arrow_datetime)
self.parser.add_argument('status', type=int, default=0)
self.parser.add_argument('user', type=dict, required=True, help="用户必填")
args = self.parser.parse_args()
company = store.companies.filter_by(id=args.company_id).first()
if not company:
return tool.fail(400, u"公司不存在")
order = company.orders.filter_by(number=args.number).first()
if order:
return tool.fail(406, u"快递单号已存在")
if args.no:
if args.no < 1:
return tool.fail(407, u"编号不能小于 1")
if "id" in args.user:
user = store.users.filter_by(id=args.user['id']).first()
if "user_name_id" in args.user:
user_name = user.names.filter_by(
id=args.user['user_name_id']).first()
if not user_name:
return tool.fail(400, u"用户名字不存在")
else:
user_name = user.names.first()
# # 用户编号都存在时,看是否为他自己的
# order = models.Order.query.join(models.Company).join(models.Store).filter(
# models.Store.id == store.id,
# models.Order.no == args.no,
# models.Order.user != user,
# ).first()
# if order:
# return tool.fail(406, u"编号已存在")
if not user:
return tool.fail(400, u"用户不存在")
else:
# order = models.Order.query.join(models.Company).join(models.Store).filter(
# models.Store.id == store.id,
# models.Order.no == args.no,
# ).first()
# if order:
# return tool.fail(406, u"编号已存在")
if "mobile" not in args.user:
return tool.fail(400, u"联系方式必填")
if "name" not in args.user:
return tool.fail(400, u"姓名必填")
if store.get_user_by_mobile(args.user["mobile"]):
return tool.fail(406,
u"手机号码 {} 已存在".format(args.user["mobile"]))
user = models.User()
user.mobile = args.user["mobile"]
user_name = models.UserName(name=args.user["name"])
user.store = store
models.db.session.add(user)
user_name.user = user
if "address" in args.user:
user_name.address = args.user["address"]
if "remark" in args.user:
user_name.remark = args.user["remark"]
models.db.session.add(user_name)
order = models.Order()
if args.no:
order.no = args.no
else:
# 计算出 no
# 1 此用户已有未领取的快递时,编同一个号
no = models.Order.get_no(store.id)
order.no = no
order.user_name = user_name
order.company = company
order.number = args.number
order.status = args.status
if args.plan_at:
order.plan_at = args.plan_at.naive
models.db.session.add(order)
models.db.session.commit()
result = schemas.OrderSchema().dump(order)
return tool.success(result.data)