def parse_total_xls_file(xls_file,db,pricelist_name, wizard_obj): user = wizard_obj.postgres_config_id.db_user host = wizard_obj.postgres_config_id.host_name password = wizard_obj.postgres_config_id.db_user_pass utility_type = wizard_obj.categ_id and wizard_obj.categ_id.name or False reader = csv.DictReader(open(xls_file, 'r'), delimiter=",") final_lst = [] data_lst = [] file_gas = addons.get_module_resource('import_xls/csv_data', 'master_sheet_cng_gas.csv') if utility_type == 'Gas': product_lst,sheet_1_data_gas = first_sheet.sheet_1_total_gas_price(file_gas,reader,pricelist_name,utility_type) product_import = pricelist.import_pricelist_total_product(db,user,host,password,product_lst,utility_type) pricelist.import_pricelist_total_gas(db,user,host,password,sheet_1_data_gas,utility_type) elif utility_type == 'Electricity': product_lst,sheet_1_data_gas = first_sheet.sheet_1_total_gas_price(file_gas,reader,pricelist_name,utility_type) product_import = pricelist.import_pricelist_total_product(db,user,host,password,product_lst,utility_type) pricelist.import_total_pricelist_ele(db,user,host,password,sheet_1_data_gas,utility_type) final_lst += sheet_1_data_gas return final_lst
def parse_sse_xls_file(xls_file,db,pricelist_name, wizard_obj): user = wizard_obj.postgres_config_id.db_user host = wizard_obj.postgres_config_id.host_name password = wizard_obj.postgres_config_id.db_user_pass utility_type = wizard_obj.categ_id and wizard_obj.categ_id.name or False rb = xlrd.open_workbook(xls_file,formatting_info=False,on_demand=False, ragged_rows=False) final_lst = [] data_lst = [] worksheet = rb.sheet_by_index(0) col_values = worksheet.row_values(0) # worksheet = rb.sheet_by_name(sheet) final_lst = [] data_lst = [] file_gas = addons.get_module_resource('import_xls/csv_data', 'master_sheet_cng_gas.csv') if utility_type == 'Gas': sheet_1_data_gas = first_sheet.sheet_1_sse_gas_price(file_gas,col_values,worksheet,pricelist_name,utility_type,rb) pricelist.import_pricelist_sse_gas(db,user,host,password,sheet_1_data_gas,utility_type) elif utility_type == 'Electricity': stop product_lst,sheet_1_data_gas = first_sheet.sheet_1_total_gas_price(file_gas,reader,pricelist_name,utility_type) product_import = pricelist.import_pricelist_total_product(db,user,host,password,product_lst,utility_type) pricelist.import_total_pricelist_ele(db,user,host,password,sheet_1_data_gas,utility_type) final_lst += sheet_1_data_gas return final_lst
def parse_opus_xls_file(xls_file,db,pricelist_name, wizard_obj): user = wizard_obj.postgres_config_id.db_user host = wizard_obj.postgres_config_id.host_name password = wizard_obj.postgres_config_id.db_user_pass utility_type = wizard_obj.categ_id and wizard_obj.categ_id.name or False rb = xlrd.open_workbook(xls_file,formatting_info=False,on_demand=False, ragged_rows=False) final_lst = [] product_final_lst = [] if utility_type == 'Electricity': file_dic = { '10': '10_opus_master.csv', '11': '11_opus_master.csv', '12': '12_opus_master.csv', '13': '13_opus_master.csv', '14': '14_opus_master.csv', '15': '15_opus_master.csv', '16': '16_opus_master.csv', '17': '17_opus_master.csv', '18': '18_opus_master.csv', '19': '19_opus_master.csv', '20': '20_opus_master.csv', '21': '21_opus_master.csv', '22': '22_opus_master.csv', '23': '23_opus_master.csv' } for sheet in ['10','11','12','13','14','15','16','17','18','19','20','21','22','23']: # for sheet in ['10']: print "________sheet__________",sheet # worksheet = rb.sheet_by_index(sheet) worksheet = rb.sheet_by_name(sheet) # print "________adsadsa_________",rb.sheet_by_name('10') # stop # col_values = worksheet.row_values(0) # region_values = worksheet.row_values(1) file_gas = addons.get_module_resource('import_xls/csv_data', file_dic[sheet]) product_values = {'ELECTRICITY CONTRACT': { '1':{'Baserate': 14, }, '2':{'Nightsaver': 15, 'RHT':16, }, '3':{'Baserate': 18, 'Flexirate 2':19 }, '4':{'Nightsaver': 20, 'Flexirate 3':21, 'RHT':22, 'extra': 23, }, '5':{'Baserate': 25, 'Nightsaver': 26, # 'Flexirate 3':27, }, '6':{'Baserate': 25, 'Nightsaver': 26, # 'Flexirate 3':27, }, '7':{'Baserate': 25, 'Nightsaver': 26, # 'Flexirate 3':27, }, '8':{'Baserate': 25, 'Nightsaver': 26, # 'Flexirate 3':27, } } } sheet_1_data_gas = first_sheet.sheet_1_opus_price(file_gas,worksheet,product_values,pricelist_name,utility_type,sheet) final_lst += sheet_1_data_gas pricelist.import_opus_pricelist_ele(db,user,host,password,final_lst) elif utility_type == 'Gas': # for sheet in ['1 Yr EB w SC','1 Yr EB no SC']: for sheet in [0,1]: product_values = { 'GAS 1 CONTRACT': {'start_col': 1,'end_col':5,'min': 0.0, 'max': 73200.0}, 'GAS 2 CONTRACT': {'start_col': 6,'end_col':10,'min': 73200.0, 'max': 293000.0}, 'GAS 3 CONTRACT': {'start_col': 11,'end_col':15,'min': 293000.0, 'max': 73200.0} } # worksheet = rb.sheet_by_name(sheet) worksheet = rb.sheet_by_index(sheet) col_values = worksheet.row_values(5,start_colx=1, end_colx=5) file_gas = addons.get_module_resource('import_xls/csv_data', 'master_bg_ele.csv') sheet_1_data_gas,product_lst = first_sheet.sheet_1_opus_gas_price(file_gas,col_values,worksheet,pricelist_name,product_values,sheet) final_lst += sheet_1_data_gas product_final_lst += product_lst product_import = pricelist.import_pricelist_total_product(db,user,host,password,product_final_lst,utility_type) pricelist.import_pricelist_opus_gas(db,user,host,password,final_lst,utility_type) print "__________final_lst___________",final_lst return final_lst