diff --git a/edisgo/io/generators_import.py b/edisgo/io/generators_import.py index 20d8c7b4c..c84d5ca27 100644 --- a/edisgo/io/generators_import.py +++ b/edisgo/io/generators_import.py @@ -65,20 +65,20 @@ def _import_conv_generators(session): orm_conv_generators.columns.id, orm_conv_generators.columns.subst_id, orm_conv_generators.columns.la_id, - orm_conv_generators.columns.capacity, + (orm_conv_generators.columns.capacity).label("electrical_capacity"), orm_conv_generators.columns.type, orm_conv_generators.columns.voltage_level, - orm_conv_generators.columns.fuel, + (orm_conv_generators.columns.fuel).label("generation_type"), func.ST_AsText( func.ST_Transform(orm_conv_generators.columns.geom, srid) - ), + ).label("geom"), ) .filter( orm_conv_generators.columns.subst_id == edisgo_object.topology.mv_grid.id ) .filter( - orm_conv_generators.columns.voltage_level.in_([4, 5, 6, 7]) + orm_conv_generators.columns.voltage_level.in_([4, 5]) ) .filter(orm_conv_generators_version) ) @@ -110,11 +110,6 @@ def _import_res_generators(session): """ - # Create filter for generation technologies - # ToDo: This needs to be removed when all generators can be imported - types_filter = orm_re_generators.columns.generation_type.in_( - ["solar", "wind"] - ) # build basic query generators_sqla = ( @@ -142,7 +137,6 @@ def _import_res_generators(session): == edisgo_object.topology.mv_grid.id ) .filter(orm_re_generators_version) - .filter(types_filter) ) # extend basic query for MV generators and read data from db @@ -183,11 +177,6 @@ def _import_res_generators(session): return generators_mv, generators_lv - logging.warning( - "Right now only solar and wind generators can be " - "imported from the oedb." - ) - def _validate_generation(): """ Validate generation capacity in updated grids. @@ -203,8 +192,7 @@ def _validate_generation(): capacity_imported = ( generators_res_mv["electrical_capacity"].sum() + generators_res_lv["electrical_capacity"].sum() - ) # + \ - # generators_conv_mv['capacity'].sum() + ) + generators_conv_mv['electrical_capacity'].sum() if p_target is not None: capacity_imported = p_target @@ -327,10 +315,10 @@ def _validate_sample_geno_location(): # get conventional and renewable generators with session_scope() as session: - # generators_conv_mv = _import_conv_generators(session) + generators_conv_mv = _import_conv_generators(session) generators_res_mv, generators_res_lv = _import_res_generators(session) - # generators_mv = generators_conv_mv.append(generators_res_mv) + generators_mv = generators_conv_mv.append(generators_res_mv) # validate that imported generators are located inside the grid district _validate_sample_geno_location() @@ -338,7 +326,7 @@ def _validate_sample_geno_location(): update_grids( edisgo_object=edisgo_object, # generators_mv=generators_mv, - imported_generators_mv=generators_res_mv, + imported_generators_mv=generators_mv, imported_generators_lv=generators_res_lv, p_target=p_target )