from well_search import well_search
from OPGEE_defaults import OPGEE_defaults
from general_well_data_analysis import OPGEE_well_data, general_well_data_analysis
from dates_array import dates_array
from OPGEE_input_sensitivity import OPGEE_input_sensitivity
#from LNG_well_search import LNG_well_search
#well_data_function = get_formation_well_data() # MONTNEY
well_data_function = well_search()
#well_data_function = get_tight_oil_wells()
#well_data_function = LNG_well_search()
well_data_headings = well_data_function[0]
well_data = well_data_function[1]
#well_data = []
OPGEE_data = OPGEE_defaults()
OPGEE_data = OPGEE_well_data(well_data, well_data_headings, OPGEE_data)
dates_array = dates_array = dates_array('2017-01', '2017-12')
#dates_array = ['2017-06']
OPGEE_data, province_facility_total, count_AB_wells, count_AB_facilities = AB_facility_analysis(
well_data, well_data_headings, OPGEE_data, dates_array)
#for well in OPGEE_data:
#print(OPGEE_data['assessed field'][OPGEE_data['headings'].index('Flaring-to-oil ratio')])
#print(OPGEE_data['defaults'][OPGEE_data['headings'].index('Flaring-to-oil ratio')])
#OPGEE_input_sensitivity(OPGEE_data, well_data)
from get_all_post_2005_well_data import get_tight_oil_wells, get_all_post_2005_well_data
#from LNG_well_search import LNG_well_search
print('Importing General Well Data') #MONTN1EY
#well_data_function = get_formation_well_data() # MONTNEY
#well_data_function = get_tight_oil_wells()
#well_data_function = LNG_well_search()
well_data_function = well_search()
#well_data_function = get_all_post_2005_well_data()
general_well_data_headings = well_data_function[0] # MONTNEY
general_well_data = well_data_function[1] # MONTNEY
field_name = 'Montney'
OPGEE_data = OPGEE_defaults()
OPGEE_data = OPGEE_well_data(general_well_data, general_well_data_headings, OPGEE_data)
production_data_headings, production_data, well_data_headings, well_header_data = search_production_data(general_well_data)
production_analysis(general_well_data, general_well_data_headings, production_data, production_data_headings, OPGEE_data)
#value = sumaverage_production_value('100141503010W500', production_data, production_data_headings, ['2016-01','2016-02'], 'PRD Monthly WTR m3', 'AVERAGE')
#print(value)
OPGEE_input_sensitivity(OPGEE_data, general_well_data)
#for heading in range(0,len(OPGEE_data['100131108118W600'])):
# print(OPGEE_data['headings'][heading], OPGEE_data['100131108118W600'][heading], OPGEE_data['assessed field'][heading])