Mopo.py

import os
import shutil
import typing as ty

import pandas as pd
from ChaProEV import ChaProEV
from ETS_CookBook import ETS_CookBook as cook


def copy_select_files(case_name: str) -> None:
    desired_output_elements: ty.List[str] = [
        'profile_fleet',
        'weekly_consumption_table_fleet',
    ]
    select_output_folder: str = f'select_output/{case_name}/'
    cook.check_if_folder_exists(select_output_folder)
    for output_file in os.listdir(f'output/{case_name}'):
        if any(
            desired_output_element in output_file
            for desired_output_element in desired_output_elements
        ):
            if output_file.endswith('.csv'):
                if 'XX' not in output_file:
                    if 'driveway' not in output_file:
                        if 'street' not in output_file:
                            shutil.copy(
                                f'output/{case_name}/{output_file}',
                                select_output_folder,
                            )


def rename_files_with_country_codes(case_name: str) -> None:
    select_output_folder: str = f'select_output/{case_name}/'
    country_codes: pd.DataFrame = pd.read_csv('country_codes.csv').set_index(
        'Country'
    )
    print(country_codes.loc['United_Kingdom']['Code'])
    for country_name in country_codes.index:
        country_files: ty.List[str] = [
            data_file
            for data_file in os.listdir(select_output_folder)
            if data_file.startswith(country_name)
        ]
        country_files_with_code: ty.List[str] = [
            country_file.replace(
                country_name, country_codes.loc[country_name]['Code']
            )
            for country_file in country_files
        ]
        for country_file, country_file_with_code in zip(
            country_files, country_files_with_code
        ):
            os.rename(
                f'{select_output_folder}/{country_file}',
                f'{select_output_folder}/{country_file_with_code}',
            )


if __name__ == '__main__':
    case_name: str = 'Mopo'
    ChaProEV.run_ChaProEV(case_name)

    copy_select_files(case_name)
    rename_files_with_country_codes(case_name)
    print(
        ' line 81 in car home parking needs an if sessions and '
        'if generate profiles'
    )
# test use profile from net * effective effiicency (replace errors by zeros)