def _load_data(form_model, manager, questionnaire_code, request):
header_list = helper.get_headers(form_model.fields)
aggregation_type_list = json.loads(request.POST.get("aggregation-types"))
start_time = helper.get_formatted_time_string(request.POST.get("start_time").strip() + START_OF_DAY)
end_time = helper.get_formatted_time_string(request.POST.get("end_time").strip() + END_OF_DAY)
aggregates = helper.get_aggregate_list(header_list[1:], aggregation_type_list)
aggregates = [aggregate_module.aggregation_factory("latest", form_model.fields[0].name)] + aggregates
data_dictionary = aggregate_module.aggregate_by_form_code_python(manager, questionnaire_code,
aggregates=aggregates, starttime=start_time,
endtime=end_time)
return data_dictionary
def _format_data_for_presentation(data_dictionary, form_model):
header_list = helper.get_headers(form_model.fields)
type_list = helper.get_type_list(form_model.fields[1:])
header_list[0] = form_model.entity_type[0] + " Code"
if data_dictionary == {}:
return "[]", header_list, type_list
entity_question_description=form_model.entity_question.name
data_list = helper.get_values(data_dictionary, header_list,entity_question_description)
data_list = helper.to_report(data_list)
response_string = encode_json(data_list)
return response_string, header_list, type_list
def _format_data_for_presentation(data_dictionary, form_model):
header_list = helper.get_headers(form_model.fields)
type_list = helper.get_type_list(form_model.fields[1:])
header_list[0] = form_model.entity_type[0] + " Code"
if data_dictionary == {}:
return "[]", header_list, type_list
entity_question_description = form_model.entity_question.name
data_list = helper.get_values(data_dictionary, header_list,
entity_question_description)
data_list = helper.to_report(data_list)
response_string = encode_json(data_list)
return response_string, header_list, type_list
def _load_data(form_model, manager, questionnaire_code, request):
header_list = helper.get_headers(form_model.fields)
aggregation_type_list = json.loads(request.POST.get("aggregation-types"))
start_time = helper.get_formatted_time_string(
request.POST.get("start_time").strip() + START_OF_DAY)
end_time = helper.get_formatted_time_string(
request.POST.get("end_time").strip() + END_OF_DAY)
aggregates = helper.get_aggregate_list(header_list[1:],
aggregation_type_list)
aggregates = [
aggregate_module.aggregation_factory("latest",
form_model.fields[0].name)
] + aggregates
data_dictionary = aggregate_module.aggregate_by_form_code_python(
manager,
questionnaire_code,
aggregates=aggregates,
starttime=start_time,
endtime=end_time)
return data_dictionary
def clean(make_output, clean_data, orders_imported, verbose, run_pricing,
ordernum, orderid, email, customerid, product, productid, lineitem,
lineid, shipmentorder, shipmentid):
'''
param make_output <Boolean> output the data to different files, ie move from tsv to csv
param clean_data <Boolean> clean the data and apply the proper fields we expect
param orders_imported <Boolean> if the order record has been imported, flip to true so that
we can create the linkage between order : record
'''
# field names for swapping down below
ORDERC = 'Order__c'
ACCOUNTC = 'AccountId__c'
PRODUCTC = 'Product__c'
LINEITEMC = 'OrderProduct__c'
SHIPC = 'Shipment__c'
# make logs
logs = helper.Logger(verbose)
# the actual good stuff
dir_path = os.path.join(os.path.dirname(os.path.realpath(__file__)))
logs.write("Current file directory :: %s" % dir_path)
directories = [
d for d in os.listdir(dir_path)
if os.path.isdir(os.path.join(dir_path, d))
]
print("Please choose a directory")
for ix, dir in enumerate(directories):
print(ix, dir)
directoryindex = int(input(">> "))
row = None
if directoryindex < len(directories):
# get the directory that we want to clean/run through
clean_directory = directories[directoryindex]
# retrieve root, dirs, and files from walk, retain only filenames
_, _, filenames = next(
os.walk(os.path.join(dir_path, clean_directory, 'tsv')),
(None, None, []))
# get the pricing files that we need
if run_pricing:
pricingfile = pricingmap.create_pricing(clean_directory)
prices = pricingmap.return_pricing(pricingfile)
else:
pricingfile = os.path.join(pricingmap.make_dir(clean_directory),
'order_product_pricing.csv')
prices = pricingmap.return_pricing(pricingfile)
import_folder = os.path.join(dir_path, 'imports')
# generate the order map (commented out)
# order_path = os.path.join(import_folder, 'imported_orders')
# ordermap = helper.OrderMap(order_path, ordernum, orderid)
# ordermapping = ordermap.get_order_map()
# add a place to get customer mappings too
customer_directory = os.path.join(import_folder, 'imported_customers')
customermap = helper.CustomerMap(customer_directory, email, customerid)
customermapping = customermap.get_customer_map()
# and product mappings
product_directory = os.path.join(import_folder, 'imported_products')
productmap = helper.ProductMap(product_directory, product, productid)
productmapping = productmap.get_product_map()
# and finally order products
lineitem_directory = os.path.join(import_folder,
'imported_orderproducts')
linemap = helper.LineItemMap(lineitem_directory, lineitem, lineid)
lineitemmapping = linemap.get_lineitem_map()
# and shipments too
shipment_directory = os.path.join(import_folder, 'imported_shipments')
shipmentmap = helper.ShipmentMap(shipment_directory, shipmentorder,
shipmentid)
shipmentmapping = shipmentmap.get_shipment_map()
# for each file, let's clean out the tsvs and export to csv
for filename in filenames:
logs.write('Reading %s' % filename)
if filename[0] == '.' or filename[0] == '_':
logs.write('Skipping, invalid file %s' % filename)
else:
filetype = helper.chooser(filename)
# directory to read from, read to, and clean to
p_in, p_out, p_clean = helper.get_paths(
dir_path, clean_directory)
# get headers
order_type, header = helper.get_headers(
os.path.join(p_in, filename), logs)
# need to handle this encoding because we have weird bytes
with open(os.path.join(p_in, filename),
encoding='ISO-8859-1') as tsvfile:
reader = csv.DictReader(
tsvfile, delimiter='\t'
) # use a DictReader to preserve header names
if make_output:
# backup the stuff from tsv to csv
out_filename = '%s%s.csv' % (filename.split('.')[0],
order_type)
with open(os.path.join(p_out, out_filename),
'w') as csvfile:
csvwriter = csv.writer(csvfile,
delimiter=',',
quotechar='"',
quoting=csv.QUOTE_ALL)
try:
csvwriter.writerow(header)
for row in reader:
csvwriter.writerow(v
for k, v in row.items())
except Exception as e:
logs.write(row) if row else logs.write(
'Exception !!')
logs.write(e)
break
if clean_data:
logs.write('cleaning %s' % filename)
# make a new list for objects that dont have record
# missingorders = []
# missing_filename = '%s%s-Clean.csv' % (filename.split('.')[0], '_MissingOrderLink')
with open(os.path.join(p_in, filename),
encoding='ISO-8859-1') as tsvfile:
clean_reader = csv.DictReader(
tsvfile,
delimiter='\t') # and one for the clean file
if make_output:
# and create a new clean file for this
clean_filename = '%s%s-Clean.csv' % (
filename.split('.')[0], order_type)
with open(os.path.join(p_clean, clean_filename),
'w') as cleanfile:
cleanwriter = csv.writer(cleanfile,
delimiter=',',
quotechar='"',
quoting=csv.QUOTE_ALL)
try:
if filetype == 1 or filetype == 2:
header.append('Product_External_ID__c')
header.append('Order_External_ID__c')
elif filetype == 4:
header.append('Custom_Field_1__c')
header.append('Order_External_Id__c')
elif filetype == 5:
header.append(
'TLA_Shipment_Provider_Carrier__c')
header.append('Order_External_ID__c')
header.append('Order_Site_Id__c')
cleanwriter.writerow(header)
for crow in clean_reader:
if filetype == 3:
helper.order_cleaner(
crow, prices, ordernum)
helper.map__c(
crow, customermapping,
ACCOUNTC)
if orders_imported and filetype != 3:
if filetype == 1: # order product
helper.order_product_cleaner(
crow)
helper.map__c(
crow, productmapping,
PRODUCTC)
elif filetype == 2: # shipment product
helper.shipment_product_cleaner(
crow)
helper.map__c(
crow, lineitemmapping,
LINEITEMC)
helper.map__c(
crow, shipmentmapping,
SHIPC)
elif filetype == 4: # payment
helper.payment_cleaner(crow)
elif filetype == 5: # shipment
helper.shipment_cleaner(crow)
cleanwriter.writerow(
v for k, v in crow.items())
except Exception as e:
logs.write(crow) if crow else logs.write(
'Exception !!')
logs.write(e)
raise (e)
