WebSVN – SmartDukaan – Diff – /trunk/PyProj/src/shop2020/model/v1/catalog/script/ItemLoader.py

 from shop2020.model.v1.catalog.impl import DataService
 from shop2020.model.v1.catalog.impl.DataService import Item, EntityIDGenerator,\
     ItemChangeLog, Vendor, VendorItemPricing, VendorItemMapping
 from elixir import *
-def load_item_data(filename, vendorId, category, full_update, dry_run, supplied_product_group):
+def load_item_data(filename, vendorId, full_update, dry_run, supplied_product_group):
     DataService.initialize('catalog')
     vendor = Vendor.get_by(id=vendorId)
     if vendor is None:
         raise Exception("No vendor found for the id: " + str(vendorId))
     updatedOn = datetime.datetime.now()
     new_items = []
     updated_items = []
     not_created_items = []
-    existing_vendor_item_mappings = VendorItemMapping.query.filter_by(vendor=vendor, vendor_category=category).all()
+    existing_vendor_item_mappings = VendorItemMapping.query.filter_by(vendor=vendor).all()
     existing_vendor_item_mappings_set = set([mapping.item_key for mapping in existing_vendor_item_mappings])
     for rownum in range(1, num_rows):
         print sheet.row_values(rownum)
         if supplied_product_group != None and supplied_product_group != '':
             our_brand, our_model_number, our_model_name, our_color,\
             brand, model_number, model_name, color,\
             dp, mrp, mop, sp, xfer_price,\
             comments, weight, start_date, deal_text, deal_value,\
-            warranty_period, defaultWarehouse, preferredVendor = sheet.row_values(rownum)[0:21]
+            warranty_period, defaultWarehouse, preferredWarehouse, preferredVendor = sheet.row_values(rownum)[0:22]
             product_group = supplied_product_group
         else:
             our_brand, our_model_number, our_model_name, our_color,\
             product_group, brand, model_number, model_name, color,\
             dp, mrp, mop, sp, xfer_price,\
             comments, weight, start_date, deal_text, deal_value,\
-            warranty_period, defaultWarehouse, preferredVendor = sheet.row_values(rownum)[0:22]
+            warranty_period, defaultWarehouse, preferredWarehouse, preferredVendor = sheet.row_values(rownum)[0:23]
         print product_group
         if isinstance(model_number, float):
             model_number = str(int(model_number))
         if item is None:
             #print "[ADDING:]{0} {1} {2} {3} to our catalogue.".format(brand, model_number, model_name, color)
             new_items.append(rownum)
             item = Item()
-            item.product_group = product_group.strip()
             item.brand = our_brand.strip()
+            item.product_group = product_group.strip()
             item.model_number = our_model_number.strip()
             item.model_name = our_model_name.strip()
             item.color = our_color.strip()
             item.status = status.IN_PROCESS
             item.status_description = "This item is in process"
             item.addedOn = updatedOn
-            item.hotspotCategory = category
-            if category == 'Handsets':
-                item.preferredWarehouse = 1
-            else:
-                item.preferredWarehouse = 2
-            vendor_item_mapping = VendorItemMapping(vendor=vendor, item=item, item_key=key, vendor_category=category)
+            vendor_item_mapping = VendorItemMapping(vendor=vendor, item=item, item_key=key)
             vendor_item_pricing = VendorItemPricing(vendor=vendor, item=item)
             session.add(item)
             session.add(vendor_item_mapping)
             session.add(vendor_item_pricing)
                     similar_item.catalog_item_id = entity_id.id
             else:
                 #If a similar item already exists for a product group, brand and model_number, set it as same.
                 item.catalog_item_id = similar_item.catalog_item_id
                 item.category = similar_item.category
+                item.product_group = similar_item.product_group
                 item.status = similar_item.status
                 item.status_description = similar_item.status_description
                 #Use the same brand, model name and model number as in similar item in database
                 item.brand = similar_item.brand
                 item.model_name = similar_item.model_name
             if check_color_valid(our_color) and (item.color == None or item.color.strip().lower() != our_color.strip().lower()):
                 item.color = our_color.strip()
             vendor_item_mapping = VendorItemMapping.get_by(vendor=vendor, item=item)
             if vendor_item_mapping is None:
-                vendor_item_mapping = VendorItemMapping(vendor=vendor, item=item, item_key=key, vendor_category=category)
+                vendor_item_mapping = VendorItemMapping(vendor=vendor, item=item, item_key=key)
                 session.add(vendor_item_mapping)
             else:
                 vendor_item_mapping.item_key = key
-                vendor_item_mapping.vendor_category = category
             vendor_item_pricing = VendorItemPricing.get_by(vendor=vendor, item=item)
             if vendor_item_pricing is None:
                 vendor_item_pricing = VendorItemPricing(vendor=vendor, item=item)
                 session.add(vendor_item_pricing)
             if item.mrp != mrp or vendor_item_pricing.dealerPrice != dp or vendor_item_pricing.transfer_price != xfer_price or item.sellingPrice != sp:
-                updated_items.append(sheet.row_values(rownum)[0:19] + [item.mrp, item.dealerPrice, item.transfer_price])
+                updated_items.append(sheet.row_values(rownum)[0:19] + [item.mrp, vendor_item_pricing.dealerPrice, vendor_item_pricing.transfer_price])
         if dp != "":
-            item.dealerPrice = dp
             item.sellingPrice = dp
             vendor_item_pricing.dealerPrice = dp
         if mrp != "" and mop != "" and mrp <  mop:
             raise Exception("[BAD MRP and MOP:] for {0} {1} {2} {3}. MRP={4}. MOP={5}".format(brand, model_number, model_name, color, mrp, mop))
         if warranty_period != "":
             try:
                 item.warranty_period = int(warranty_period)
             except:
                 pass
+        if preferredWarehouse != "":
+            try:
+                item.preferredWarehouse = int(preferredWarehouse)
+            except:
+                pass
         if mrp != "":
             item.mrp = mrp
         if mop != "":
-            item.mop = mop
             vendor_item_pricing.mop = mop
         if xfer_price !="":
-            item.transfer_price = xfer_price
             vendor_item_pricing.transfer_price = xfer_price
         if start_date is not None and start_date != '':
             #If a start date has been specified, it takes precedence.
             item.startDate = datetime.datetime(*xlrd.xldate_as_tuple(start_date, workbook.datemode))
     phased_out_items = [key.split('|') for key in list(existing_vendor_item_mappings_set)]
     write_report("phased_out_items.csv", phased_out_items, sheet, True)
     if (not dry_run) and full_update:
         query_string =  "UPDATE " + str(Item.table) + " SET status=" + str(status.PHASED_OUT) + ", status_description='This item has been phased out'" + ", updatedOn='"+ str(updatedOn) +"'"+\
-                        " WHERE updatedOn <> '" + str(updatedOn) + "' AND hotspotCategory='" + category +"'";
+                        " WHERE updatedOn <> '" + str(updatedOn) + "'";
         session.execute(query_string, mapper=Item)
         session.commit()
     print "Successfully updated the item list information."
     parser = optparse.OptionParser()
     parser.add_option("-f", "--file", dest="filename",
                    default="ItemList.xls", type="string",
                    help="Read the item list from FILE",
                    metavar="FILE")
-    parser.add_option("-c", "--category", dest="category",
-                      type="string",
-                      help="Update the list only for the products belonging to CATEGORY",
-                      metavar="CATEGORY")
     parser.add_option("-v", "--vendor", dest="vendor",
                       type="int",
                       help="Update the pricing information for VENDOR",
                       metavar="VENDOR")
     parser.add_option("-d", "--dry-run", dest="dry_run",
                       help="In a partial update, older items are left as is. Also see -fm.")
     parser.add_option("-g", "--product-group", dest="product_group",
                       type="string",
                       help="Set GROUP as the product group for all items added/updated during this run.",
                       metavar="GROUP")
-    parser.set_defaults(full_update=True, dry_run=False, category="Handsets", vendor=1)
+    parser.set_defaults(full_update=True, dry_run=False, vendor=1)
     (options, args) = parser.parse_args()
     if len(args) != 0:
         parser.error("You've supplied extra arguments. Are you sure you want to run this program?")
     filename = options.filename
-    load_item_data(filename, options.vendor, options.category, options.full_update, options.dry_run, options.product_group)
+    load_item_data(filename, options.vendor, options.full_update, options.dry_run, options.product_group)
 if __name__ == '__main__':
     main()

Subversion Repositories SmartDukaan

(root)/trunk/PyProj/src/shop2020/model/v1/catalog/script/ItemLoader.py – Rev 4739 → 4762