def init_db(self, args):
DB.create_tables([
Shop,
ShopPhoto,
User,
Administrator,
Product,
ProductLabelList,
ProductSceneList,
ProductPhoto,
Repertory,
RepertoryEntry,
SaleOrder,
SaleOrderDetail,
SaleOrderTrace,
Poster,
SessionDB
])
def test_dal(self):
DB.connect()
c = Chateau.create(chateau_id=uuid.uuid4(), chateau_name='who', address='hello', geo='11,22', description='world', created_at=datetime.today())
c.save()
DB.close()
def add_one_order(self, order_data, user):
order_detail_data = order_data["order"]
order_obj = SaleOrder(
soid=str(uuid.uuid4()),
addr_level1=order_data["addr_level1"],
addr_level2=order_data["addr_level2"],
addr_level3=order_data["addr_level3"],
shipping_cost=order_data["shipping_cost"],
receiver=order_data["receiver"],
phone=order_data["phone"],
receipt_sort=order_data["receipt_sort"],
receipt_content=order_data["receipt_content"],
receipt_title=order_data["receipt_title"],
pay_sort=order_data["pay_sort"],
)
if user:
order_obj.user = User(**user)
pay_sort_ll = PaySortLL()
order_state_ll = OrderStateLL()
if order_obj.pay_sort == pay_sort_ll.value(COD):
order_obj.order_state = order_state_ll.value(HANDLING)
elif order_obj.pay_sort == pay_sort_ll.value(ALIPAY):
order_obj.order_state = order_state_ll.value(NOPAY)
product_ll = ProductLL()
product_details = product_ll.get_product_obj_by_pid_list(order_detail_data.keys())
order_detail_obj = []
product_count = 0
product_sum_price = 0.0
for pd in product_details:
sod = SaleOrderDetail(
sod_id=str(uuid.uuid4()),
sale_order=order_obj,
product=pd,
purchase_count=order_detail_data[pd.pid],
price=pd.price,
)
product_count += sod.purchase_count
product_sum_price += sod.purchase_count * sod.price
order_detail_obj.append(sod)
order_obj.product_count = product_count
order_obj.product_sum_price = product_sum_price
trace = SaleOrderTrace(sot_id=str(uuid.uuid4()), sale_order=order_obj, state=order_obj.order_state)
reper_ll = RepertoryLL()
product_reper_list = reper_ll.product_repertory_by_pid_list(order_detail_data.keys())
with DB.transaction() as trans:
try:
for order_detail in order_detail_obj:
product = order_detail.product
product_reper_list = filter(lambda x: x.product.pid == product.pid, product_reper_list)
if not product_reper_list:
product_reper = Repertory(product=product)
# raise NoProductInRepertory(pid)
else:
product_reper = product_reper_list[0]
# if product_reper.store_count < order_detail.purchase_count:
# raise NoEnoughStore(product.pid)
product_reper.sale_count += order_detail.purchase_count
product_reper.store_count -= order_detail.purchase_count
product_reper.save(force_insert=True)
order_obj.save(force_insert=True)
for d in order_detail_obj:
d.save(force_insert=True)
trace.save(force_insert=True)
trans.commit()
except NoProductInRepertory as e:
self.err[e.product_id] = str(e)
trans.rollback()
except NoEnoughStore as e:
self.err[e.product_id] = str(e)
trans.rollback()
except Exception, e:
self.err["all"] = str(e)
trans.rollback()
