Source code for egon.data.datasets.low_flex_scenario

"""The central module to create low flex scenarios

"""
from airflow.operators.postgres_operator import PostgresOperator
from sqlalchemy.ext.declarative import declarative_base
import importlib_resources as resources

from egon.data.datasets import Dataset


Base = declarative_base()


class LowFlexScenario(Dataset):
    def __init__(self, dependencies):
        super().__init__(
            name="low_flex_scenario",
            version="0.0.1",
            dependencies=dependencies,
            tasks=(
                {
                    PostgresOperator(
                        task_id="low_flex_eGon2035",
                        sql=resources.read_text(
                            __name__, "low_flex_eGon2035.sql"
                        ),
                        postgres_conn_id="egon_data",
                        autocommit=True,
                    ),
                },
            ),
        )