synmax-api-python-client 0.0.44__py3-none-any.whl → 0.0.45__py3-none-any.whl

test/hyperion_test.py

@@ -1,188 +1,188 @@
-import logging
-import multiprocessing
-
-import pandas
-import pandas as pd
-from tqdm import tqdm
-import os 
-import sys
-
-sys.path.append(os.path.abspath(os.path.join(os.path.dirname(os.path.dirname(__file__)))))
-
-from synmax.hyperion import HyperionApiClient, ApiPayload, add_daily, get_fips
-
-logging.basicConfig(level=logging.INFO)
-
-# Test - GET 
-
-def fetch_region():
-    regions = client.fetch_regions()
-    print(regions)
-
-def fetch_long_term_forecast():
-    long_term_forecast = client.fetch_long_term_forecast()
-    print(long_term_forecast)
-
-def fetch_operator_classification():
-    operator_classification = client.fetch_operator_classification()
-    print(operator_classification)
-
-def fetch_daily_fracked_feet():
-    daily_fracked_feet = client.fetch_daily_fracked_feet()
-    print(daily_fracked_feet)
-
-
-# Test - POST 
-
-def well_completion():
-    payload = ApiPayload(start_date='2021-05-1', end_date='2022-12-25', state_code='CO', operator_name='GREAT WESTERN OPERATING COMPANY LLC')
-
-    # result_df = client.wells(payload)
-    result_df = client.well_completion(payload)
-    print(result_df.count())
-
-
-def test_rigs():
-    states = ['CO', 'LA', 'ND', 'NM', 'OH', 'OK', 'PA', 'TX', 'WV', 'WY']
-    df_list = []
-    for _state in states:
-        payload = ApiPayload(start_date='2016-01-01', end_date='2022-01-31', state_code=_state)
-        result_df = client.rigs(payload)
-        print(result_df.count())
-        df_list.append(result_df)
-
-    print('Got all stats result')
-    if df_list:
-        df = pandas.concat(df_list)
-        print(df.count())
-
-
-def test_ducs_by_operator():
-    payload = ApiPayload(start_date='2021-01-01', end_date='2021-01-31', aggregate_by='operator', operator='LIME ROCK RESOURCES LP')
-
-    result_df = client.ducs_by_operator(payload)
-    print(result_df.count())
-
-
-def test_production_by_county_and_operator():
-    payload = ApiPayload(start_date='1929-04-01', end_date='1934-01-01', operator_name='Stephens Production Company', state_code='AR')
-    result_df = client.production_by_county_and_operator(payload)
-    print(result_df.count())
-
-
-def test_frac_crews():
-    payload = ApiPayload(start_date='2021-10-01', end_date='2021-12-31', state_code='CO', sub_region='Colorado wo SJ', region='west', operator='BILL BARRETT CORPORATION')
-
-    result_df = client.frac_crews(payload)
-    print(result_df.count())
-
-
-def test_production_by_well():
-    # payload = ApiPayload(start_date='2016-01-01', end_date='2016-01-31', production_month=529)
-    payload = ApiPayload(state_code='WY', start_date='2017-01-01', end_date='2017-12-31', operator_name='CITATION OIL & GAS CORP', region='west', sub_region='Wyoming')
-    # payload = ApiPayload(state_code='LA', start_date='2021-01-01', end_date='2021-01-01', production_month=2)
-    # result_df = client.production_by_well(payload)
-    # print(result_df.count())
-    with multiprocessing.Pool(processes=5) as pool:
-        data_list = [payload for _ in range(0, 5)]
-        message = 'API query progress'
-        list(tqdm(pool.imap(client.production_by_well, data_list), desc=message, total=len(data_list),
-                  dynamic_ncols=True, miniters=0))
-
-
-def test_short_term_forecast():
-    # payload = ApiPayload(start_date='2021-08-29', end_date='2022-09-29')
-    payload = ApiPayload(start_date='2016-01-01')
-    result_df = client.short_term_forecast(payload)
-    result_df.to_csv('df_data.csv', index=False)
-    print(result_df.count())
-    sum_df = result_df[['date', 'gas_monthly']].groupby(by=['date']).sum()
-    sum_df.to_csv('sum.csv')
-
-
-def test_short_term_forecast_history():
-    payload = ApiPayload(start_date='2021-08-29', end_date='2021-09-10')
-    result_df = client.short_term_forecast_history(payload)
-    print(result_df.count())
-
-    payload = ApiPayload(start_date='2020-01-01')
-    result_df = client.short_term_forecast(payload)
-    
-def test_pipeline_scrapes():
-    payload = ApiPayload(start_date='2024-01-01', end_date='2024-01-10')
-    print(client.__dir__())
-    result_df = client.pipeline_scrapes(payload)
-    print(result_df.count())
-    
-def test_fetch_pipeline_scrape_status():
-    result_df = client.fetch_pipeline_scrape_status()
-    print(result_df.count())
-
-# TEST HELPERS
-
-def test_daily_func():
-    df = pd.DataFrame({'date': ['2022-01-01', '2022-02-01', '2022-03-01'],
-                       'gas_monthly': [1000, 2000, 3000],
-                       'oil_monthly': [2000, 3000, 4000],
-                       'water_monthly': [8000, 9000, 10000]})
-
-    df = add_daily(df)
-    print(df)
-
-
-def test_add_fips():
-    df = get_fips()
-    print(df)
-
-
-def compare_df():
-    cols = ['api', 'gas_monthly']
-    df1 = pandas.read_csv('df_data.csv')
-    df2 = pandas.read_csv('GEOML_dbo_v_current_forecast.csv')
-    df1 = df1[cols]
-    df2 = df2[cols]
-    print(df1.count())
-    print(df2.count())
-
-    merged_df = pd.merge(df1, df2, how="inner", on=["api"])
-    print(merged_df.count())
-
-
-
-def main():
-
-    # Test GET
-    # fetch_region()
-    # fetch_operator_classification()
-    # fetch_long_term_forecast()
-    # fetch_daily_fracked_feet()
-
-    # Test POST
-    #well_completion()
-    #test_production_by_well()
-    #test_add_fips()
-    #test_frac_crews()
-    #test_rigs()
-    #test_short_term_forecast()
-    #test_short_term_forecast_history()
-    #test_ducs_by_operator()
-    #compare_df()
-    #test_pipeline_scrapes()
-    test_fetch_pipeline_scrape_status()
-
-
-if __name__ == '__main__':
-
-    # enable debug if required.
-    
-    #logging.basicConfig(level=logging.DEBUG)
-
-    access_token = ''    
-    client = HyperionApiClient(access_token=access_token, local_server=True)
-    
-    import synmax.hyperion as hyperion_api
-    print(hyperion_api.__file__)
-    
-    print(client.__le__)
-    print(dir(client))
+import logging
+import multiprocessing
+
+import pandas
+import pandas as pd
+from tqdm import tqdm
+import os 
+import sys
+
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(os.path.dirname(__file__)))))
+
+from synmax.hyperion import HyperionApiClient, ApiPayload, add_daily, get_fips
+
+logging.basicConfig(level=logging.INFO)
+
+# Test - GET 
+
+def fetch_region():
+    regions = client.fetch_regions()
+    print(regions)
+
+def fetch_long_term_forecast():
+    long_term_forecast = client.fetch_long_term_forecast()
+    print(long_term_forecast)
+
+def fetch_operator_classification():
+    operator_classification = client.fetch_operator_classification()
+    print(operator_classification)
+
+def fetch_daily_fracked_feet():
+    daily_fracked_feet = client.fetch_daily_fracked_feet()
+    print(daily_fracked_feet)
+
+
+# Test - POST 
+
+def well_completion():
+    payload = ApiPayload(start_date='2021-05-1', end_date='2022-12-25', state_code='CO', operator_name='GREAT WESTERN OPERATING COMPANY LLC')
+
+    # result_df = client.wells(payload)
+    result_df = client.well_completion(payload)
+    print(result_df.count())
+
+
+def test_rigs():
+    states = ['CO', 'LA', 'ND', 'NM', 'OH', 'OK', 'PA', 'TX', 'WV', 'WY']
+    df_list = []
+    for _state in states:
+        payload = ApiPayload(start_date='2016-01-01', end_date='2022-01-31', state_code=_state)
+        result_df = client.rigs(payload)
+        print(result_df.count())
+        df_list.append(result_df)
+
+    print('Got all stats result')
+    if df_list:
+        df = pandas.concat(df_list)
+        print(df.count())
+
+
+def test_ducs_by_operator():
+    payload = ApiPayload(start_date='2021-01-01', end_date='2021-01-31', aggregate_by='operator', operator='LIME ROCK RESOURCES LP')
+
+    result_df = client.ducs_by_operator(payload)
+    print(result_df.count())
+
+
+def test_production_by_county_and_operator():
+    payload = ApiPayload(start_date='1929-04-01', end_date='1934-01-01', operator_name='Stephens Production Company', state_code='AR')
+    result_df = client.production_by_county_and_operator(payload)
+    print(result_df.count())
+
+
+def test_frac_crews():
+    payload = ApiPayload(start_date='2021-10-01', end_date='2021-12-31', state_code='CO', sub_region='Colorado wo SJ', region='west', operator='BILL BARRETT CORPORATION')
+
+    result_df = client.frac_crews(payload)
+    print(result_df.count())
+
+
+def test_production_by_well():
+    # payload = ApiPayload(start_date='2016-01-01', end_date='2016-01-31', production_month=529)
+    payload = ApiPayload(state_code='WY', start_date='2017-01-01', end_date='2017-12-31', operator_name='CITATION OIL & GAS CORP', region='west', sub_region='Wyoming')
+    # payload = ApiPayload(state_code='LA', start_date='2021-01-01', end_date='2021-01-01', production_month=2)
+    # result_df = client.production_by_well(payload)
+    # print(result_df.count())
+    with multiprocessing.Pool(processes=5) as pool:
+        data_list = [payload for _ in range(0, 5)]
+        message = 'API query progress'
+        list(tqdm(pool.imap(client.production_by_well, data_list), desc=message, total=len(data_list),
+                  dynamic_ncols=True, miniters=0))
+
+
+def test_short_term_forecast():
+    # payload = ApiPayload(start_date='2021-08-29', end_date='2022-09-29')
+    payload = ApiPayload(start_date='2016-01-01')
+    result_df = client.short_term_forecast(payload)
+    result_df.to_csv('df_data.csv', index=False)
+    print(result_df.count())
+    sum_df = result_df[['date', 'gas_monthly']].groupby(by=['date']).sum()
+    sum_df.to_csv('sum.csv')
+
+
+def test_short_term_forecast_history():
+    payload = ApiPayload(start_date='2021-08-29', end_date='2021-09-10')
+    result_df = client.short_term_forecast_history(payload)
+    print(result_df.count())
+
+    payload = ApiPayload(start_date='2020-01-01')
+    result_df = client.short_term_forecast(payload)
+    
+def test_pipeline_scrapes():
+    payload = ApiPayload(start_date='2024-01-01', end_date='2024-01-10')
+    print(client.__dir__())
+    result_df = client.pipeline_scrapes(payload)
+    print(result_df.count())
+    
+def test_fetch_pipeline_scrape_status():
+    result_df = client.fetch_pipeline_scrape_status()
+    print(result_df.count())
+
+# TEST HELPERS
+
+def test_daily_func():
+    df = pd.DataFrame({'date': ['2022-01-01', '2022-02-01', '2022-03-01'],
+                       'gas_monthly': [1000, 2000, 3000],
+                       'oil_monthly': [2000, 3000, 4000],
+                       'water_monthly': [8000, 9000, 10000]})
+
+    df = add_daily(df)
+    print(df)
+
+
+def test_add_fips():
+    df = get_fips()
+    print(df)
+
+
+def compare_df():
+    cols = ['api', 'gas_monthly']
+    df1 = pandas.read_csv('df_data.csv')
+    df2 = pandas.read_csv('GEOML_dbo_v_current_forecast.csv')
+    df1 = df1[cols]
+    df2 = df2[cols]
+    print(df1.count())
+    print(df2.count())
+
+    merged_df = pd.merge(df1, df2, how="inner", on=["api"])
+    print(merged_df.count())
+
+
+
+def main():
+
+    # Test GET
+    # fetch_region()
+    # fetch_operator_classification()
+    # fetch_long_term_forecast()
+    # fetch_daily_fracked_feet()
+
+    # Test POST
+    #well_completion()
+    #test_production_by_well()
+    #test_add_fips()
+    #test_frac_crews()
+    #test_rigs()
+    #test_short_term_forecast()
+    #test_short_term_forecast_history()
+    #test_ducs_by_operator()
+    #compare_df()
+    #test_pipeline_scrapes()
+    test_fetch_pipeline_scrape_status()
+
+
+if __name__ == '__main__':
+
+    # enable debug if required.
+    
+    #logging.basicConfig(level=logging.DEBUG)
+
+    access_token = ''    
+    client = HyperionApiClient(access_token=access_token, local_server=True)
+    
+    import synmax.hyperion as hyperion_api
+    print(hyperion_api.__file__)
+    
+    print(client.__le__)
+    print(dir(client))
     main()

test/ip_rate_example.py

@@ -1,51 +1,51 @@
-"""Install SynMax Python Client"""
-
-# pip install --upgrade synmax-api-python-client
-
-"""Install other Dependencies"""
-
-# pip install plotly-geo plotly geopandas==0.8.1 pyshp shapely
-
-"""Query the production data"""
-
-from synmax.hyperion import HyperionApiClient, ApiPayload, add_daily, get_fips
-import pandas as pd
-import plotly.express as px
-import plotly.figure_factory as ff
-
-access_token = '************************************'
-client = HyperionApiClient(access_token=access_token)
-
-payload = ApiPayload(start_date='2021-01-01', end_date='2022-09-01', state_code='TX', production_month=2)
-df = client.production_by_well(payload)
-# df.to_csv(r'C:\Users\eric\Desktop\TEMP\prod_ip.csv', index=False)
-# df = pd.read_csv(r'C:\Users\eric\Desktop\TEMP\prod_ip.csv')
-
-"""convert to mcf/d"""
-df = add_daily(df)
-
-"""examine county-level histograms"""
-
-fig = px.histogram(df[df.county=='Midland'], x='gas_daily')
-fig.show()
-
-"""calculate average per county"""
-
-county_df = df[['state_ab', 'county', 'gas_daily']].groupby(['state_ab', 'county'], as_index=False).mean()
-
-"""Get FIPS code Lookup Table"""
-
-fips_df = get_fips()
-
-"""merge fips_df with county_df"""
-county_df = county_df.merge(fips_df[fips_df.state_ab == 'TX'], how='right', on=['state_ab','county'])
-county_df['gas_daily'] = county_df['gas_daily'].fillna(0)
-
-"""Create county map"""
-
-fig = ff.create_choropleth(fips=county_df.fips.tolist(),
-                           values=county_df.gas_daily.tolist(),
-                           county_outline={'color': 'rgb(255,255,255)', 'width': 0.5},
-                           scope=['TX'],
-                           binning_endpoints=[75, 125, 250, 500, 1000, 2000, 4000, 8000, 16000, 32000])
+"""Install SynMax Python Client"""
+
+# pip install --upgrade synmax-api-python-client
+
+"""Install other Dependencies"""
+
+# pip install plotly-geo plotly geopandas==0.8.1 pyshp shapely
+
+"""Query the production data"""
+
+from synmax.hyperion import HyperionApiClient, ApiPayload, add_daily, get_fips
+import pandas as pd
+import plotly.express as px
+import plotly.figure_factory as ff
+
+access_token = '************************************'
+client = HyperionApiClient(access_token=access_token)
+
+payload = ApiPayload(start_date='2021-01-01', end_date='2022-09-01', state_code='TX', production_month=2)
+df = client.production_by_well(payload)
+# df.to_csv(r'C:\Users\eric\Desktop\TEMP\prod_ip.csv', index=False)
+# df = pd.read_csv(r'C:\Users\eric\Desktop\TEMP\prod_ip.csv')
+
+"""convert to mcf/d"""
+df = add_daily(df)
+
+"""examine county-level histograms"""
+
+fig = px.histogram(df[df.county=='Midland'], x='gas_daily')
+fig.show()
+
+"""calculate average per county"""
+
+county_df = df[['state_ab', 'county', 'gas_daily']].groupby(['state_ab', 'county'], as_index=False).mean()
+
+"""Get FIPS code Lookup Table"""
+
+fips_df = get_fips()
+
+"""merge fips_df with county_df"""
+county_df = county_df.merge(fips_df[fips_df.state_ab == 'TX'], how='right', on=['state_ab','county'])
+county_df['gas_daily'] = county_df['gas_daily'].fillna(0)
+
+"""Create county map"""
+
+fig = ff.create_choropleth(fips=county_df.fips.tolist(),
+                           values=county_df.gas_daily.tolist(),
+                           county_outline={'color': 'rgb(255,255,255)', 'width': 0.5},
+                           scope=['TX'],
+                           binning_endpoints=[75, 125, 250, 500, 1000, 2000, 4000, 8000, 16000, 32000])
 fig.show()

synmax_api_python_client-0.0.44.dist-info/RECORD

@@ -1,17 +0,0 @@
-DATA/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-DATA/fips_lookup.csv,sha256=Ui1Nibt0RT9nfoccFv2DA9s_UemSldQVJpsqwWYUUsU,57771
-synmax/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-synmax/common/__init__.py,sha256=9lTQMNYvpSLKn54EzpWCsrvnztUOB820qGl3Wb79ppg,106
-synmax/common/api_client.py,sha256=tfbQn3qPB3KdKpK7oOnYchDxQlZJ6axrOahAF9qM7DA,12065
-synmax/common/model.py,sha256=g2w9i1PyUEMQXoxxRk8ntr54pTOfDQ3RF0NqxQruCMY,1317
-synmax/config/__init__.py,sha256=5jSxVXV43wK-OTc9xCdO2XaksoiMgC2Oy42HDXS8MoE,143
-synmax/hyperion/__init__.py,sha256=0pz5W71GgJIy8DkdTIzqWnvlcTqnItAEBJnr7unDHFs,2434
-synmax/hyperion/hyperion_client.py,sha256=_K7iyCrg-cNpIqDfaEd9pnz7Ty4aQtqzGgl7kt_gk-4,7528
-synmax/theia/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-test/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-test/hyperion_test.py,sha256=z0SWZ3HDilSxez73_-us7QmmdSzi6-UcZQWkR8XvrsM,5929
-test/ip_rate_example.py,sha256=XPjf6rqxNBU6d4SjSnoceyVPBrcS0ywbpqn1c3rUf90,1754
-synmax_api_python_client-0.0.44.dist-info/METADATA,sha256=mSGjc7ME-4VruY3L35GgOLV9PQk7m0B4zNcNGX_oOe8,3732
-synmax_api_python_client-0.0.44.dist-info/WHEEL,sha256=OqRkF0eY5GHssMorFjlbTIq072vpHpF60fIQA6lS9xA,92
-synmax_api_python_client-0.0.44.dist-info/top_level.txt,sha256=SAjmDfHlJzUmjoGCT1SnS4qGSLN_ZF-p58UN1yW5kB0,17
-synmax_api_python_client-0.0.44.dist-info/RECORD,,