import pandas as pd

path1 = "C:\\Users\\Marek\\Desktop\\Diplomka\\Stáž\\Data_tmobile_porovnanie_cd\\obdobie4\\obdobie_4.csv"
path2 = "C:\\Users\\Marek\\Desktop\\Diplomka\\Stáž\\Data_tmobile_porovnanie_cd\\obdobie5\\obdobie_5.csv"
path3 = "C:\\Users\\Marek\\Desktop\\Diplomka\\Stáž\\Data_tmobile_porovnanie_cd\\obdobie7\\obdobie_7.csv"
path4 = "C:\\Users\\Marek\\Desktop\\Diplomka\\Stáž\\Data_tmobile_porovnanie_cd\\obdobie9\\obdobie_9.csv"

path_list = [path1, path2, path3, path4]

list_sign = [["atv > brq","atv_brq"],["atz > brq","atz_brq"],["atz > prg","atz_prg"],["brq > atv","brq_atv"],["brq > atz","brq_atz"],
                  ["brq > osr","brq_osr"],["brq > prg","brq_prg"],["brq > sk","brq_sk"],["dej > prg","dej_prg"],["des > prg","des_prg"],["osr > brq","osr_brq"],
                  ["osr > plv","osr_plv"],["osr > prg","osr_prg"],["osr > sk","osr_sk"],["plv > osr","plv_osr"],["plz > prg","plz_prg"],
                  ["prg > atz","prg_atz"],["prg > brq","prg_brq"],["prg > dej","prg_dej"],["prg > des","prg_des"],["prg > osr","prg_osr"],
                  ["prg > plz","prg_plz"],["sk > brq","sk_brq"],["sk > brq","sk_brq"]]


def filter_zony(file, zone, outfile):
    df = pd.read_csv(f'{file}', encoding='windows-1250')
    df = df[df["mod_dopravy"] == "rail"]
    df = df[df["signatura"] == zone]
    df = df[["hrana_z_datum[datum]", "pocet_cestujicich"]]
    df.to_csv(f'{file}\\..\\{outfile}.csv', encoding='windows-1250', index= False)

for path in path_list:
    for sign in list_sign:
        filter_zony(path, sign[0], sign[1])