add schema column #5 #14

ReeemProject · Nov 20, 2017 · dc43fa0 · dc43fa0
1 parent f6733c8
commit dc43fa0
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/database_adapter/reeem_adapter_plexos.py b/database_adapter/reeem_adapter_plexos.py
@@ -23,11 +23,11 @@
 pathway = 'Test_data'       # 'BASE', 'BASE_TI1_P1', 'BASE_TI1_P2', 'Test_data', 'Pilot'
 version = 'V1'              # 'V2', 'V3'
 
-file_name_input = 'REEEM_PLEXOS_Input.xlsx'
+file_name_input = 'REEEM_PLEXOS_Input_LH.xlsx'
 file_name_output = 'REEEM_PLEXOS_Output.xlsx'
 
-# regions = ['BG']
-regions = ['BG', 'HR', 'HU', 'RO', 'SI']
+regions = ['BG']
+# regions = ['BG', 'HR', 'HU', 'RO', 'SI']
 
 empty_rows = 4
 
@@ -51,20 +51,23 @@ def plexos_2_reeem_db(model, pathway, version, file_name, empty_rows, db_schema,
     ## make dataframe
     df.columns = ['indicator', 'unit', 
         '2030',
-        'field', 'category', 'aggregation', 'source']
+        'schema', 'field', 'category', 'aggregation', 'source']
     df.index.names = ['nid']
     # print(df.dtypes)
     # print(df.head())
 
     ## seperate columns
-    dfunit = df[['field', 'category', 'indicator', 'unit', 'aggregation', 'source']].copy().dropna()
+    dfunit = df[['schema', 'field', 'category', 
+        'indicator', 'unit', 'aggregation', 'source']].copy().dropna()
     dfunit.index.names = ['nid']
-    dfunit.columns = ['field', 'category', 'indicator', 'unit', 'aggregation', 'source']
+    dfunit.columns = ['schema', 'field', 'category', 
+        'indicator', 'unit', 'aggregation', 'source']
     # print(dfunit)
     # print(dfunit.dtypes)
 
     ## drop seperated columns
-    dfclean = df.drop(['field', 'category', 'indicator', 'unit', 'aggregation', 'source'],axis=1).dropna()
+    dfclean = df.drop(['schema', 'field', 'category', 
+        'indicator', 'unit', 'aggregation', 'source'],axis=1).dropna()
     # print(dfclean)
 
     ## stack dataframe