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

 @author: ashish
 '''
 from elixir import *
 from functools import partial
+from shop2020.clients.CatalogClient import CatalogClient
 from shop2020.clients.TransactionClient import TransactionClient
 from shop2020.model.v1.inventory.impl import DataService
+from shop2020.model.v1.inventory.impl.Convertors import to_t_warehouse, \
+    to_t_itemidwarehouseid
 from shop2020.model.v1.inventory.impl.DataService import Warehouse, \
     ItemInventoryHistory, CurrentInventorySnapshot, VendorItemPricing, \
     VendorItemMapping, Vendor, MissedInventoryUpdate, BadInventorySnapshot, \
-    VendorItemProcurementDelay, VendorHolidays, ItemAvailabilityCache,\
+    VendorItemProcurementDelay, VendorHolidays, ItemAvailabilityCache, \
-    CurrentReservationSnapshot,Ignoredinventoryupdateitems
+    CurrentReservationSnapshot, IgnoredInventoryUpdateItems
-from shop2020.thriftpy.model.v1.inventory.ttypes import InventoryServiceException, \
+from shop2020.thriftpy.model.v1.inventory.ttypes import \
-    HolidayType, InventoryType, WarehouseType
+    InventoryServiceException, HolidayType, InventoryType, WarehouseType
 from shop2020.thriftpy.model.v1.order.ttypes import AlertType
+from shop2020.thriftpy.purchase.ttypes import PurchaseServiceException
 from shop2020.utils import EmailAttachmentSender
 from shop2020.utils.EmailAttachmentSender import mail
+from shop2020.utils.Utils import to_py_date
 from sqlalchemy.orm.exc import MultipleResultsFound, NoResultFound
-from sqlalchemy.sql.expression import and_, func
+from sqlalchemy.sql.expression import and_, func, distinct
+from sqlalchemy.sql.functions import count
 import calendar
 import datetime
 import sys
 import threading
-from shop2020.clients.CatalogClient import CatalogClient
-from shop2020.utils.Utils import to_py_date
 to_addresses = ["chaitnaya.vats@shop2020.in", "chandan.kumar@shop2020.in"]
 mail_user = "cnc.center@shop2020.in"
 mail_password = "5h0p2o2o"
 skippedItems = { 175 : [27, 2160, 2175, 2163, 2158, 7128, 26, 2154],
         ds_warehouse.vendorString = warehouse.vendorString
     session.commit()
     return ds_warehouse.id
 def get_ignored_items(warehouse_id):
-    Ignored_inventory_items = Ignoredinventoryupdateitems.query.filter_by(warehouse_id=warehouse_id).all()
+    Ignored_inventory_items = IgnoredInventoryUpdateItems.query.filter_by(warehouse_id=warehouse_id).all()
     negativeItems = []
     for Ignored_inventory_item in Ignored_inventory_items:
         try:
             item_id = Ignored_inventory_item.item_id
             negativeItems.append(item_id)
     for item_key, quantity in availability.iteritems():
         try:
             vendor_item_mapping = VendorItemMapping.query.filter_by(vendor=vendor, item_key=item_key).one();
             item_id = vendor_item_mapping.item_id
         except:
             continue
         try:
             item_inventory_history = ItemInventoryHistory()
             item_inventory_history.warehouse = warehouse
             item_inventory_history.item_id = item_id
             item_inventory_history.timestamp = time
     if source_id == 1:
         client = CatalogClient("catalog_service_server_host_master", "catalog_service_server_port").get_client()
         return client.getItem(item_id)
     if source_id == 2:
         client = CatalogClient("catalog_service_server_host_hotspot", "catalog_service_server_port").get_client()
-        return client.getItem(item_id)
         return client.getItem(item_id)
+def get_monitored_warehouses_for_vendors(vendorIds):
+    w = []
+    for wh in Warehouse.query.filter_by(isAvailabilityMonitored = 1).all():
+        if wh.vendor.id in (vendorIds):
+            w.append(to_t_warehouse(wh).id)
+    return w
+def get_ignored_warehouseids_and_itemids():
+    iw = []
+    for i in IgnoredInventoryUpdateItems.query.all():
+        iw.append(to_t_itemidwarehouseid(i))
+    return iw
+def insert_item_to_ignore_inventory_update_list(item_id,warehouse_id):
+    try:
+        ds_warehouse=IgnoredInventoryUpdateItems()
+        ds_warehouse.item_id=item_id
+        ds_warehouse.warehouse_id=warehouse_id
+        session.commit()
+        return True
+    except:
+        return False
+def delete_item_from_ignore_inventory_update_list(item_id,warehouse_id):
+    try:
+        session.query(IgnoredInventoryUpdateItems).filter_by(item_id=item_id,warehouse_id=warehouse_id).delete()
+        session.commit()
+        return True
+    except:
+        return False
+def get_all_ignored_inventoryupdate_items_count():
+    return  session.query(func.count(distinct(IgnoredInventoryUpdateItems.item_id))).scalar()
+def get_ignored_inventoryupdate_itemids(offset=0,limit=None):
+    itemIds = session.query(distinct(IgnoredInventoryUpdateItems.item_id))
+    '''if limit is not None:
+        itemIds = itemIds.limit(limit)'''
+    print itemIds.all()
+    return [id for (id, ) in itemIds.all()]

Subversion Repositories SmartDukaan

(root)//trunk/PyProj/src/shop2020/model/v1/inventory/impl/DataAcessors.py – Rev 6524 → 6531