WebSVN – SmartDukaan – Diff – /trunk/PyProj/src/shop2020/model/v1/user/impl/CartDataAccessors.py

 from shop2020.utils.Utils import to_py_date, to_java_date
 from shop2020.thriftpy.model.v1.order.ttypes import Transaction as TTransaction,\
     TransactionStatus as TTransactionStatus, Order as TOrder, LineItem as TLineItem, OrderStatus
-from shop2020.thriftpy.model.v1.catalog.ttypes import Item as TItem
-from shop2020.thriftpy.logistics.ttypes import LogisticsInfo as TLogisticsInfo,\
+from shop2020.thriftpy.logistics.ttypes import LogisticsServiceException, DeliveryType
-    LogisticsServiceException, DeliveryType
 from shop2020.clients.TransactionClient import TransactionClient
 from shop2020.clients.CatalogClient import CatalogClient
 from shop2020.clients.LogisticsClient import LogisticsClient
 from shop2020.model.v1 import user
 from shop2020.clients.PromotionClient import PromotionClient
-from shop2020.model.v1.user.impl import UserDataAccessors
-def get_cart(user_id):
+def get_cart(userId):
-    query = Cart.query.filter_by(user_id=user_id)
+    user = User.get_by(id=userId)
-    query = query.filter_by(cart_status = CartStatus.ACTIVE)
-    try:
-        cart = query.first()
-    except:
-        return None
-#        raise ShoppingCartException(101, "Cart does not exist")
-    return cart
+    return user.active_cart
 def get_cart_by_id(id):
     cart = Cart.get_by(id=id)
     return cart
-def create_cart(user_id):
+def create_cart():
-    cart = get_cart(user_id)
-    if not cart:
-        cart = Cart()
+    cart = Cart()
-        cart.user_id = user_id
-        cart.created_on = datetime.datetime.now()
+    cart.created_on = datetime.datetime.now()
-        cart.updated_on = datetime.datetime.now()
+    cart.updated_on = datetime.datetime.now()
-        cart.cart_status = CartStatus.ACTIVE
+    cart.cart_status = CartStatus.ACTIVE
-        session.commit()
     return cart
-def get_cart_by_user_id_and_status(user_id, status):
-    query = Cart.query.filter_by(user_id=user_id)
-    if status:
-        query = query.filter_by(cart_status=status)
-    carts = query.all()
-    return carts
-def get_carts_by_status(status):
-    return Cart.query.filter_by(cart_status=status).all()
 def get_carts_between(start_time, end_time, status):
     init_time = to_py_date(start_time)
     finish_time = to_py_date(end_time)
     query = Cart.query
         else:
             return query.all()
     except:
         return None
-def change_cart_status(cart_id, status):
-    cart = get_cart_by_id(id)
-    if not cart:
-        raise ShoppingCartException(101, "no cart attached to this id")
-    if not status:
-        raise ShoppingCartException(101, "invalid status")
-    cart.cart_status = status
-    session.commit()
 def add_item_to_cart(cart_id, item_id, quantity, sourceId):
     if not item_id:
         raise ShoppingCartException(101, "item_id cannot be null")
     if not quantity:
                 discount.line = line
                 discount.discount = t_discount.discount
                 discount.quantity = t_discount.quantity
                 session.commit()
-def change_item_status(cart_id, item_id, status):
-    if not status:
-        raise ShoppingCartException(101, "Status cannot be made null")
-    line = get_line(item_id, cart_id, None, True)
-    if line:
-        line.line_status = status
-        current_time = datetime.datetime.now()
-        line.updated_on = current_time
-        line.cart.updated_on = current_time
-        session.commit()
-    else:
-        raise ShoppingCartException(101, "Unable to probe the line you desired")
 def add_address_to_cart(cart_id, address_id):
     if not cart_id:
         raise ShoppingCartException(101, "cart id cannot be made null")
     if not address_id:
     delete_discounts_from_cart(cart.id, cart=cart)
     cart.discounted_price = None
     cart.coupon_code = None
     session.commit()
-def commit_cart(cart_id, sessionSource, sessionTime, firstSource, firstSourceTime):
+def commit_cart(cart_id, sessionSource, sessionTime, firstSource, firstSourceTime, userId):
     cart = get_cart_by_id(cart_id)
     #now we have a cart. Need to create a transaction with it
     txn = TTransaction()
     txn.shoppingCartid = cart_id
-    txn.customer_id = cart.user_id
+    txn.customer_id = userId
     txn.createdOn = to_java_date(datetime.datetime.now())
     txn.transactionStatus = TTransactionStatus.INIT
     txn.statusDescription = "New Order"
     txn.coupon_code = cart.coupon_code
     txn.sessionSource = sessionSource
     txn.sessionStartTime = sessionTime
     txn.firstSource = firstSource
     txn.firstSourceTime = firstSourceTime
-    txn.orders = create_orders(cart)
+    txn.orders = create_orders(cart, userId)
     transaction_client = TransactionClient().get_client()
     txn_id = transaction_client.createTransaction(txn)
     session.commit()
-    #new_cart = create_cart(cart.user_id)
-    #user = User.get_by(id=cart.user_id)
-    #user.active_cart_id = new_cart.id
-    #session.commit()
     return txn_id
-def create_orders(cart):
+def create_orders(cart, userId):
     cart_lines = cart.lines
     orders = []
     for line in cart_lines:
         if line.line_status == LineStatus.LINE_ACTIVE:
             for discount in line.discounts:
                 i = 0
                 while i < discount.quantity:
                     t_line_item = create_line_item(line.item_id, line.actual_price - discount.discount)
-                    t_order = create_order(cart.user_id, cart.address_id, t_line_item)
+                    t_order = create_order(userId, cart.address_id, t_line_item)
                     orders.append(t_order)
                     i += 1
                 quantity_remaining_for_order -= discount.quantity
             i = 0
             while i < quantity_remaining_for_order:
                 t_line_item = create_line_item(line.item_id, line.actual_price)
-                t_order = create_order(cart.user_id, cart.address_id, t_line_item)
+                t_order = create_order(userId, cart.address_id, t_line_item)
                 orders.append(t_order)
                 i += 1
     return orders
-def create_order(user_id, address_id, t_line_item):
+def create_order(userId, address_id, t_line_item):
-    user = User.get_by(id=user_id)
+    user = User.get_by(id=userId)
     address = Address.get_by(id=address_id)
     t_order = TOrder()
     t_order.customer_id = user.id
     t_order.customer_email = user.email

Subversion Repositories SmartDukaan

(root)/trunk/PyProj/src/shop2020/model/v1/user/impl/CartDataAccessors.py – Rev 5293 → 5326