# 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]
