import zipfile
import json
import csv
import pandas as pd


def main():
    read_zip_file()




# viide https://codeyarns.com/2013/10/03/how-to-read-contents-of-zip-file-in-python/
def read_zip_file():
    csv_file = list()
    columns1 = ["tekst1","tekst2","tekst3_1","tekst3_2" , "tekst3_3","tekst3_4","tekst3_5","tekst3_6","tekst3_7","tekst3_8","tekst3_9","tekst3_10","tekst3_11","tekst3_12","tekst3_13","tekst3_14","tekst3_15","tekst4","tekst5", "tekst6_1", "tekst6_2" "tekst7", "tekst8", "tekst9", "sugu", "vanus", "faili_nimi"]
    FILE_PATH = "C:\\Users\\raunol\\Desktop\\logikataloog.zip"
    zfile = zipfile.ZipFile(FILE_PATH)
    key_logger_att = 'klogdict'
    for zipInfo in zfile.infolist():
        json_string = get_json(zfile, zipInfo)
        klogdict = json_string[key_logger_att]
        new_dict = {k: v[-1][-1] - v[0][-1] for (k, v) in klogdict.items()}
        new_dict['sugu'] = json_string['sugu']
        new_dict['vanus'] = json_string['vanus']
        new_dict['faili_nimi'] = zipInfo.filename
        csv_file.append(new_dict)
    df = pd.DataFrame(csv_file, columns=columns1, dtype=int)
    print(df)
    df.to_csv("C:\\Users\\raunol\Desktop\\file1.csv", index=False, columns=columns1)


def get_json(zfile, zipInfo):
    file_content = get_lines(zfile, zipInfo)[0]
    json_string = json.loads(file_content)
    return json_string


def get_lines(zfile, zip_info):
    file = zfile.open(zip_info)
    line_list = file.readlines()
    return line_list


def get_computed_results(csv_file, json_string, klogdict, result_list, zipInfo):
    sugu_att = 'sugu'
    vanus_att = 'vanus'
    for att in klogdict.keys():
        get_duration(att, klogdict, result_list)
    result_list.append(json_string[sugu_att])
    result_list.append(json_string[vanus_att])
    result_list.append(zipInfo.filename)
    csv_file.append(result_list)


def get_duration(i, klogdict, result_list):
    att = klogdict[i]
    start = att[0][-1]
    end = att[-1][-1]
    duration = end - start
    result = str(duration)
    result_list.append(i + "_" + result)


def write_results_to_csv(csv_file):
    with open("output.csv", "a") as f:
        writer = csv.writer(f)
        writer.writerows(csv_file)


if __name__ == '__main__':
    main()