示例#1
0
文件: manager.py 项目: xsauce/wineprj
 def init_db(self, args):
     DB.create_tables([
         Shop,
         ShopPhoto,
         User,
         Administrator,
         Product,
         ProductLabelList,
         ProductSceneList,
         ProductPhoto,
         Repertory,
         RepertoryEntry,
         SaleOrder,
         SaleOrderDetail,
         SaleOrderTrace,
         Poster,
         SessionDB
     ])
示例#2
0
 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()
示例#3
0
 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()