data.datasets.power_plants.assign_weather_data.write_power_plants_table()

Complexity

Conditions

6

Size

Total Lines	49
Code Lines	21

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	data.datasets.power_plants.assign_weather_data.find_weather_id()	0	19	1
A	data.datasets.power_plants.assign_weather_data.weatherId_and_busId()	0	3	1

import geopandas as gpd

from egon.data import db
from egon.data.datasets.power_plants.pv_rooftop_buildings import timer_func
import egon.data.config

def assign_bus_id(power_plants, cfg):

This code seems to be duplicated in your project.

    """Assigns bus_ids to power plants according to location and voltage level

    Parameters
    ----------
    power_plants : pandas.DataFrame
        Power plants including voltage level

    Returns
    -------
    power_plants : pandas.DataFrame
        Power plants including voltage level and bus_id

    """

    mv_grid_districts = db.select_geodataframe(
        f"""
        SELECT * FROM {cfg['sources']['egon_mv_grid_district']}
        """,
        epsg=4326,
    )

    ehv_grid_districts = db.select_geodataframe(
        f"""
        SELECT * FROM {cfg['sources']['ehv_voronoi']}
        """,
        epsg=4326,
    )

    # Assign power plants in hv and below to hvmv bus
    power_plants_hv = power_plants[power_plants.voltage_level >= 3].index
    if len(power_plants_hv) > 0:
        power_plants.loc[power_plants_hv, "bus_id"] = gpd.sjoin(
            power_plants[power_plants.index.isin(power_plants_hv)],
            mv_grid_districts,
        ).bus_id

    # Assign power plants in ehv to ehv bus
    power_plants_ehv = power_plants[power_plants.voltage_level < 3].index

    if len(power_plants_ehv) > 0:
        ehv_join = gpd.sjoin(
            power_plants[power_plants.index.isin(power_plants_ehv)],
            ehv_grid_districts,
        )

        if "bus_id_right" in ehv_join.columns:
            power_plants.loc[power_plants_ehv, "bus_id"] = gpd.sjoin(
                power_plants[power_plants.index.isin(power_plants_ehv)],
                ehv_grid_districts,
            ).bus_id_right

        else:
            power_plants.loc[power_plants_ehv, "bus_id"] = gpd.sjoin(
                power_plants[power_plants.index.isin(power_plants_ehv)],
                ehv_grid_districts,
            ).bus_id

    # Assert that all power plants have a bus_id
    assert power_plants.bus_id.notnull().all(), f"""Some power plants are
    not attached to a bus: {power_plants[power_plants.bus_id.isnull()]}"""

    return power_plants


@timer_func
def add_missing_bus_ids(scn_name):
    """Assign busses by spatal intersection of mvgrid districts or ehv voronois."""

    sql = f"""
                -- Assign missing buses to mv grid district buses for HV and below
                UPDATE supply.egon_power_plants AS epp
                SET bus_id = (
                    SELECT emgd.bus_id
                    FROM grid.egon_mv_grid_district AS emgd
                    WHERE ST_Intersects(ST_Transform(epp.geom, 4326), ST_Transform(emgd.geom, 4326))
                    ORDER BY ST_Transform(emgd.geom, 4326) <-> ST_Transform(epp.geom, 4326)
                    LIMIT 1
                )
                WHERE (epp.carrier = 'solar'
                    OR epp.carrier = 'wind_onshore'
                    OR epp.carrier = 'solar_rooftop'
                    OR epp.carrier = 'wind_offshore')
                AND epp.scenario = '{scn_name}'
                AND epp.bus_id is null
                AND epp.voltage_level >= 3; -- HV and below


                -- Assign missing buses to EHV buses for EHV
                UPDATE supply.egon_power_plants AS epp
                SET bus_id = (
                    SELECT eesv.bus_id
                    FROM grid.egon_ehv_substation_voronoi AS eesv
                    WHERE ST_Intersects(ST_Transform(epp.geom, 4326), ST_Transform(eesv.geom, 4326))
                    ORDER BY ST_Transform(eesv.geom, 4326) <-> ST_Transform(epp.geom, 4326)
                    LIMIT 1
                )
                WHERE (epp.carrier = 'solar'
                    OR epp.carrier = 'wind_onshore'
                    OR epp.carrier = 'solar_rooftop'
                    OR epp.carrier = 'wind_offshore')
                AND epp.scenario = '{scn_name}'
                AND epp.bus_id is null
                AND epp.voltage_level < 3; --EHV


        """

    db.execute_sql(sql)


@timer_func
def find_weather_id(scn_name):

    sql = f"""UPDATE supply.egon_power_plants AS epp
                SET weather_cell_id = (
                    SELECT eewc.w_id
                    FROM supply.egon_era5_weather_cells AS eewc
                    WHERE ST_Intersects(epp.geom, eewc.geom)
                    ORDER BY eewc.geom <-> epp.geom
                    LIMIT 1
                )
                WHERE (epp.carrier = 'solar'
                    OR epp.carrier = 'solar_rooftop'
                    OR epp.carrier = 'wind_onshore'
                    OR epp.carrier = 'wind_offshore')
                AND epp.scenario = '{scn_name}';
        """

    db.execute_sql(sql)


def weatherId_and_busId(scn_name):
    find_weather_id(scn_name)
    add_missing_bus_ids(scn_name)