data.datasets.scenario_path.import_status2019.import_scn_status2019()

Complexity

Conditions

4

Size

Total Lines	72
Code Lines	41

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

"""
def import_scn_status2019():
    """
    Read and import the scenario status2019 and import it into db

    Parameters
    ----------
    *No parameters required

    """

    if "status2019" in config.settings()["egon-data"]["--scenarios"]:
        print("Scenario status2019 was created within the pipeline. "
              "It is not imported from zenodo.")
        return

    # Connect to the data base
    con = db.engine()

    # Clean existing data for status2019
    tables = pd.read_sql(
        """
        SELECT tablename FROM pg_catalog.pg_tables
        WHERE schemaname = 'grid'
        """,
        con,
    )

    tables = tables[
        ~tables["tablename"].isin(
            [
                "egon_etrago_carrier",
                "egon_etrago_temp_resolution",
            ]
        )
    ]

    for table in tables["tablename"]:
        db.execute_sql(
            f"""
        DELETE FROM grid.{table} WHERE scn_name = 'status2019';
        """
        )

    config_data = config.settings()["egon-data"]
    database = config_data["--database-name"]
    host = config_data["--database-host"]
    port = config_data["--database-port"]
    user = config_data["--database-user"]

    my_env = os.environ.copy()
    my_env["PGPASSWORD"] = config_data["--database-password"]

    for table in tables["tablename"]:
        subprocess.Popen(
            [
                "pg_restore",
                "-d",
                database,
                "--host",
                host,
                "--port",
                port,
                "-U",
                user,
                "-a",
                "--single-transaction",
                f"--table={table}",
                "PoWerD_status2019.backup",
            ],
            env=my_env,
        )
    return