def _post_update(self):
""" 将Customer的类型进行更新 """
# argument的校验与获取
self.parser.add_argument('customer_id', type=int, required=True)
self.parser.add_argument('customer_type',
type=int,
required=True,
choices=CustomerType.choices(),
help='请传入正确的customer_type参数')
args = self.parser.parse_args()
customer_id = args.get('customer_id')
customer_type = args.get('customer_type')
# 获取具体的customer，并更新的客户类型
try:
customer = Customer.query.filter_by(id=customer_id).first()
customer.customer_type = customer_type
customer.update()
except AttributeError:
return Response.error('请求出错', 'custoemr_id无法定位到一个具体的customer对象')
except Exception as e:
return Response.error('请求出错', str(e))
return Response.success('customer的状态修改成功')
def _post_delete(self):
""" 将指定的customer对象删除 """
# argument的校验与获取
self.parser.add_argument('customer_id', type=int, required=True)
args = self.parser.parse_args()
customer_id = args.get('customer_id')
try:
customer = Customer.query.filter_by(id=customer_id).first()
Customer.delete(customer)
except AttributeError:
return Response.error('请求出错', 'custoemr_id无法定位到一个具体的customer对象')
except Exception as e:
return Response.error('请求出错', str(e))
return Response.success('删除成功')
def _post_add(self):
""" 新增一个customer对象 """
# argument的校验与获取
self.parser.add_argument('short_name', type=str, required=True)
self.parser.add_argument('full_name', type=str, required=True)
self.parser.add_argument('telephone', type=str, required=True)
self.parser.add_argument('customer_type',
type=int,
required=True,
choices=CustomerType.choices(),
help='请传入正确的customer_type参数')
args = self.parser.parse_args()
short_name = args.get('short_name')
full_name = args.get('full_name')
telephone = args.get('telephone')
customer_type = args.get('customer_type')
if Customer.query.filter_by(short_name=short_name).first():
return Response.error('请求出错', f'{short_name}已经添加在customer列表中了')
if Customer.query.filter_by(full_name=full_name).first():
return Response.error('请求出错', f'{full_name}已经添加在customer列表中了')
if not Validator.validate_phone(telephone):
return Response.error('请求出错', 'telephone不是一个有效的电话号码')
try:
customer = Customer(short_name=short_name,
full_name=full_name,
telephone=telephone,
customer_type=customer_type)
Customer.add(obj=customer)
except Exception as e:
return Response.error('请求出错', str(e))
return Response.success('添加成功')
