# get uom list filename = 'C:/Users/xcxg109/NonDriveFiles/reference/UOM_data_sheet.csv' uom_df = pd.read_csv(filename) # create df of the lovs and their concat values filename = 'C:/Users/xcxg109/NonDriveFiles/reference/LOV_list.csv' lov_df, lov_list = q.get_LOVs(filename) data_type = 'grainger_query' print('working...') search_level = fd.WS_search_type() if search_level == 'cat.CATEGORY_ID': search_data = fd.data_in(data_type, settings.directory_name) while True: try: query_level = input( "Query Level: \n1. ETL query (L1)\n2. ALL (no restrictions) Query " ) if query_level in ['1']: query_level = 'etl' break elif query_level in ['2']: query_level = 'all' break except ValueError: print('Invalid search type')
def generate_data(): gcom = GraingerQuery() #request the type of data to pull: blue or yellow, SKUs or node, single entry or read from file data_type = fd.search_type() search_level = 'cat.CATEGORY_ID' #if Blue is chosen, determine the level to pull L1 (segment), L2 (family), or L1 (category) if data_type == 'node': search_level = fd.blue_search_level() #ask user for node number/SKU or pull from file if desired search_data = fd.data_in(data_type, settings.directory_name) sku_status = skus_to_pull( ) #determine whether or not to include discontinued items in the data pull grainger_df = pd.DataFrame() if data_type == 'node': for k in search_data: if sku_status == 'filtered': temp_df = gcom.grainger_q(grainger_basic_query, search_level, k) temp_df = gcom.grainger_q(grainger_basic_query, search_level, k) elif sku_status == 'all': temp_df = gcom.grainger_q(grainger_discontinued_query, search_level, k) grainger_df = pd.concat([grainger_df, temp_df], axis=0) print(k) elif data_type == 'yellow': for k in search_data: if isinstance(k, int): #k.isdigit() == True: pass else: k = "'" + str(k) + "'" if sku_status == 'filtered': temp_df = gcom.grainger_q(grainger_basic_query, 'yellow.PROD_CLASS_ID', k) elif sku_status == 'all': temp_df = gcom.grainger_q(grainger_discontinued_query, 'yellow.PROD_CLASS_ID', k) grainger_df = pd.concat([grainger_df, temp_df], axis=0) print(k) elif data_type == 'sku': sku_str = ", ".join("'" + str(i) + "'" for i in search_data) if sku_status == 'filtered': grainger_df = gcom.grainger_q(grainger_basic_query, 'item.MATERIAL_NO', sku_str) elif sku_status == 'all': grainger_df = gcom.grainger_q(grainger_discontinued_query, 'item.MATERIAL_NO', sku_str) elif data_type == 'supplier': for k in search_data: if sku_status == 'filtered': temp_df = gcom.grainger_q(grainger_basic_query, 'supplier.SUPPLIER_NO', k) elif sku_status == 'all': temp_df = gcom.grainger_q(grainger_discontinued_query, 'supplier.SUPPLIER_NO', k) grainger_df = pd.concat([grainger_df, temp_df], axis=0) print(k) return [grainger_df, search_level]