def create_db():
try:
User.drop_table()
Member.drop_table()
Commodity.drop_table()
Category.drop_table()
UserOrder.drop_table()
OrderItem.drop_table()
CommodityImage.drop_table()
CommodityComment.drop_table()
Address.drop_table()
OrderStatus.drop_table()
Announcement.drop_table()
Feedback.drop_table()
except:
pass
User.create_table()
Member.create_table()
Commodity.create_table()
Category.create_table()
UserOrder.create_table()
OrderItem.create_table()
CommodityImage.create_table()
CommodityComment.create_table()
Address.create_table()
OrderStatus.create_table()
Announcement.create_table()
Feedback.create_table()
def random_oi(order):
global co_list
co = random.choice(co_list)
num = random.randint(1, 4)
price = co.price
oi = OrderItem.create(commodity=co, num=num, price=price, order=order)
return oi
def put(self, order_id, item_id):
args = self.reqparse.parse_args()
filtered_args = {k: v for k, v in args.items() if v is not None}
args.clear()
args.update(filtered_args)
query = OrderItem.update(**args).where(OrderItem.item_id == item_id,
OrderItem.order_id == order_id)
num = query.execute()
return {'isSuccess': True, 'rowsUpdated': str(num)}
def get(self, order_id):
result = OrderItem.select().where(OrderItem.order_id == order_id)
order_items = []
for order_item in result:
row = model_to_dict(
order_item,
exclude=[OrderItem.order_id, OrderItem.product_id.create_time])
order_items.append(row)
return {'result': order_items}
def additem():
json = request.json
order = get_object_or_404(UserOrder, id=json["order"])
commodity = Commodity.select().get(id=json["commodity"])
OrderItem.create(commodity=commodity, order=order, num=json["num"], price=commodity.price)
return jsonify(success=True)
def add_fruit(self, db: Session, order: Order, item: Fruit, quantity: int):
order.order_items.append(OrderItem(item=item, quantity=quantity))
db.add(order)
db.commit()
db.refresh(order)
return order
def delete(self, order_id, item_id):
query = OrderItem.delete().where(OrderItem.item_id == item_id,
OrderItem.order_id == order_id)
num = query.execute()
return {'isSuccess': True, 'rowsUpdated': str(num)}
def post(self, order_id):
args = self.reqparse.parse_args()
# with DB.atomic():
order_item = OrderItem.create(**args, order_id=order_id)
return {'isSuccess': True, 'product_id': order_item.item_id}
