Filter by scenario

openego · Dec 16, 2021 · e00ed30 · e00ed30
1 parent 465f635
commit e00ed30
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/egon/data/datasets/fill_etrago_gen.py b/src/egon/data/datasets/fill_etrago_gen.py
@@ -157,11 +157,12 @@ def fill_etrago_gen_time_table(
     return etrago_pp
 
 
-def load_tables(con, cfg):
+def load_tables(con, cfg, scenario='eGon2035'):
     sql = f"""
     SELECT * FROM
     {cfg['sources']['power_plants']['schema']}.
     {cfg['sources']['power_plants']['table']}
+    WHERE scenario = '{scenario}'
     """
     power_plants = gpd.GeoDataFrame.from_postgis(
         sql, con, crs="EPSG:4326", index_col="id"
@@ -185,13 +186,15 @@ def load_tables(con, cfg):
     SELECT * FROM
     {cfg['targets']['etrago_generators']['schema']}.
     {cfg['targets']['etrago_generators']['table']}
+    WHERE scn_name = '{scenario}'
     """
     etrago_gen_orig = pd.read_sql(sql, con)
 
     sql = f"""
     SELECT * FROM
     {cfg['targets']['etrago_gen_time']['schema']}.
     {cfg['targets']['etrago_gen_time']['table']}
+    WHERE scn_name = '{scenario}'
     """
     pp_time = pd.read_sql(sql, con)