WebSVN – SmartDukaan – Diff – //trunk/PyProj/src/shop2020/model/v1/catalog/impl/DataAcessors.py

 import datetime
 import sys
 from shop2020.utils.Utils import log_entry, to_py_date, log_risky_flag
 from sqlalchemy import desc, asc
 from shop2020.model.v1.catalog.impl.CategoryManager import CategoryManager
-from sqlalchemy.sql.expression import and_
+from sqlalchemy.sql.expression import and_, distinct, func
 def initialize():
     DataService.initialize()
         raise InventoryServiceException(101, "bad item")
     warehouses = item.currentInventory.warehouse
     return warehouses
+def get_child_categories(category):
+    cm = CategoryManager()
+    return cm.getCategory(category).children_category_ids
 def get_best_sellers(start_index, stop_index, category=-1):
+    '''
+    Returns the Best Sellers between the start and the stop index in the given category
+    '''
     query = get_best_sellers_query(category, None)
     best_sellers = query.all()[start_index:stop_index]
     return get_thrift_item_list(best_sellers)
 def get_best_sellers_count(category=-1):
+    '''
+    Returns the number of best sellers in the given category
+    '''
     count = get_best_sellers_query(category, None).count()
     if count is None:
         count = 0
     return count
 def get_best_sellers_catalog_ids(start_index, stop_index, brand, category=-1):
+    '''
+    Returns the Best sellers for the given brand and category between the start and the stop index.
+    Ignores the category if it's passed as -1 and the brand if it's passed as None.
+    '''
     query = get_best_sellers_query(category, brand)
     best_sellers = query.all()[start_index:stop_index]
     return [item.catalog_item_id for item in best_sellers]
-def get_child_categories(category):
-    cm = CategoryManager()
-    return cm.getCategory(category).children_category_ids
 def get_best_sellers_query(category, brand):
+    '''
+    Returns the query to be used for getting Best Sellers.
+    Ignores the category if it's passed as -1 and the brand if it's passed as None.
+    '''
     query = Item.query.filter_by(status=status.ACTIVE).filter(Item.bestSellingRank != None)
     if category != -1:
         all_categories = [category]
         child_categories = get_child_categories(category)
         if child_categories is not None:
         query = query.filter_by(brand=brand)
     query = query.order_by(asc(Item.bestSellingRank))
     return query
 def get_best_deals(category=-1):
+    '''
+    Returns the Best deals in the given category. Ignores the category if it's passed as -1.
+    '''
-    query = get_best_deals_query(category, None)
+    query = get_best_deals_query(Item, category, None)
     items = query.all()
     return get_thrift_item_list(items)
 def get_best_deals_count(category=-1):
+    '''
+    Returns the count of best deals in the given category.
+    Ignores the category if it's -1.
+    '''
-    count = get_best_deals_query(category, None).count()
+    count = get_best_deals_counting_query(func.count(distinct(Item.catalog_item_id)), category, None).scalar()
     if count is None:
         count = 0
     return count
 def get_best_deals_catalog_ids(start_index, stop_index, brand, category=-1):
+    '''
+    Returns the catalog_item_ids of best deal items for the given brand and category.
+    Ignores the category if it's passed as -1 and the brand if it's passed as None.
+    '''
-    query = get_best_deals_query(category, brand)
+    query = get_best_deals_query(Item, category, brand)
     best_deal_items = query.all()[start_index:stop_index]
     return [item.catalog_item_id for item in best_deal_items]
-def get_best_deals_query(category, brand):
+def get_best_deals_counting_query(obj, category, brand):
+    '''
+    Returns the query to be used to select the best deals in the given brand and category.
+    Ignores the category if it's passed as -1 and the brand if it's passed as None.
+    '''
-    query = Item.query.filter_by(status=status.ACTIVE).filter(Item.bestDealValue != None)
+    query = session.query(obj).filter_by(status=status.ACTIVE).filter(Item.bestDealValue != None)
     if category != -1:
         all_categories = [category]
         child_categories = get_child_categories(category)
         if child_categories is not None:
             all_categories = all_categories + child_categories
         query = query.filter(Item.category.in_(all_categories))
     if brand is not None:
         query = query.filter_by(brand=brand)
+    return query
+def get_best_deals_query(obj, category, brand):
+    '''
+    Returns the query to be used to get the best deals in the given category and brand.
+    Ignores the category if it's passed as -1 and the brand if it's passed as None.
+    '''
+    query = get_best_deals_counting_query(obj, category, brand)
     query = query.group_by(Item.catalog_item_id).order_by(desc(Item.bestDealValue))
     return query
 def get_latest_arrivals(limit, category=-1):
+    '''
+    Returns up to limit number of Latest Arrivals in the given category.
+    '''
-    query = get_latest_arrivals_query(category, None)
+    query = get_latest_arrivals_query(Item, category, None)
     items = query.all()[0:limit]
     return get_thrift_item_list(items)
 def get_latest_arrivals_count(limit, category=-1):
+    '''
+    Returns the number of latest arrivals which will be displayed on the website.
+    Ignores the category if it's passed as -1.
+    '''
-    count = get_latest_arrivals_query(category, None).count()
+    count = get_latest_arrivals_counting_query(func.count(distinct(Item.catalog_item_id)), category, None).scalar()
     if count is None:
         count = 0
     count = min(count, limit)
     return count
 def get_latest_arrivals_catalog_ids(start_index, stop_index, brand, category=-1):
+    '''
+    Returns the catalog_item_ids of the latest arrivals between the start and the stop index
+    Ignores the category if it's passed as -1 and the brand if it's passed as None.
+    '''
-    query = get_latest_arrivals_query(category, brand)
+    query = get_latest_arrivals_query(Item, category, brand)
     latest_arrivals = query.all()[start_index:stop_index]
     return [item.catalog_item_id for item in latest_arrivals]
-def get_latest_arrivals_query(category, brand):
+def get_latest_arrivals_counting_query(obj, category, brand):
+    '''
+    Returns the query to be used to count Latest arrivals.
+    Ignores the category if it's passed as -1 and the brand if it's passed as None.
+    '''
-    query = Item.query.filter_by(status=status.ACTIVE)
+    query = session.query(obj).filter_by(status=status.ACTIVE)
     if category != -1:
         all_categories = [category]
         child_categories = get_child_categories(category)
         if child_categories is not None:
             all_categories = all_categories + child_categories
         query = query.filter(Item.category.in_(all_categories))
     if brand is not None:
         query = query.filter_by(brand=brand)
+    return query
+def get_latest_arrivals_query(obj, category, brand):
+    '''
+    Returns the query to be used to retrieve Latest Arrivals.
+    Ignores the category if it's passed as -1 and the brand if it's passed as None.
+    '''
+    query = get_latest_arrivals_counting_query(obj, category, brand)
     query = query.group_by(Item.catalog_item_id).order_by(desc(Item.startDate))
     return query
 def get_thrift_item_list(items):
     return [to_t_item(item) for item in items if item != None]

Subversion Repositories SmartDukaan

(root)//trunk/PyProj/src/shop2020/model/v1/catalog/impl/DataAcessors.py – Rev 2403 → 2404