#!/usr/bin/env python3

#####################################################
# Skript pro převod formátu OPTIcXNet (nová kamera) do starého OPTIc (stará kamera), který se dá importovat do AK2
# Autor: Michal Töpfer, AC Mladá Boleslav
#
# Skirpt pracuje přímo ve složce, ve které je spuštěn, a pro každý '*.heatresultxml' soubor vytvoří '*.txt' soubor,
# který se dá importovat do AK2.
# Pokud vše pracuje správně, jsou ID závodů nastavená podle ID ze závodní kanceláře.
#####################################################

import sys
from pathlib import Path
import xml.etree.ElementTree as etree
import csv


def try_parse_int(s):
    try:
        return int(s)
    except ValueError:
        return 999999


def process_competitor(competitor):
    c = competitor.attrib

    if "Lane" not in c and "Bib" not in c:
        raise Exception("V souboru není určena dráha ani Bib závodníka/závodníků.")

    result = [
        c.get("Rank", ""),
        c.get("Bib", ""),
        c.get("Lane", ""),
        c.get("Firstname", ""),
        c.get("Lastname", ""),
        c.get("Club", ""),
        "00"  # birth year - unused
    ]

    if "Runtime" in c:
        result.append(c["Runtime"])
    elif "State" in c:  # DQ, DNS, ...
        if c["State"] == "DSQ":
            result.append("disqualified")
        elif c["State"] == "DNS":
            result.append("did not start")
        elif c["State"] == "DNF":
            result.append("did not finish")
        else:
            result.append("")
    else:
        result.append("")

    #result.append("1067")  # something
    #result.append("591800")  # time in ms
    #result.append("1")  # camera picture
    #result.append("3")  # something
    #result.append("")  # something
    #result.append("")  # something
    #result.append("")  # something
    #
    #if "DifferenceToWinner" in c:  # DQ, DNS, ...
    #    if c["DifferenceToWinner"] == "Winner":
    #        result.append("winner")
    #    else:
    #        result.append("+" + c["DifferenceToWinner"])
    #
    #result.append("591800")  # time in ms

    #if "Rank" in c and c["Rank"] == "1":
    #    result.append("winner")
    #else:
    #    result.append("")  # something
    #
    #result.append("")  # something
    #result.append("")  # something
    #result.append("")  # something

    return result


def process_competitor_csv(competitor):
    c = competitor.attrib
    lane = c.get("Lane")
    if "Runtime" in c:
        return [lane, c["Runtime"]]
    if "State" in c:
        return [lane, c["State"]]
    else:
        raise Exception(f"Chybějíci výsledek pro dráhu {lane}.")


def process_heatresult(f):
    heat_result = etree.parse(f).getroot()
    results = heat_result[0]
    heat_number = heat_result.attrib["Nr"]
    filename = f.parent / (heat_number + ".txt")
    try:
        with open(filename, "w", newline="") as out_file:
            writer = csv.writer(out_file, delimiter="\t")
            for competitor in results:
                writer.writerow(process_competitor(competitor))

        filename = f.parent / (heat_number + ".csv")
        with open(filename, "w", newline="") as out_file:
            writer = csv.writer(out_file, delimiter=";")
            writer.writerow(["Track", "BestResult"])
            for competitor in results:
                writer.writerow(process_competitor_csv(competitor))
        print(f"#{heat_number} úspěšně zpracováno ({heat_result.attrib['Name']})")
        if "Wind" in heat_result.attrib:
            print(f"  vítr: {heat_result.attrib['Wind']} m/s")
    except Exception as e:
        print(f"CHYBA: #{heat_number} ({heat_result.attrib['Name']})")
        print(" ", e)


path = Path(".")
if len(sys.argv) == 2:
    path = Path("./" + sys.argv[1])

files = path.glob("*.heatresultxml")
files = sorted(files, key=lambda x: try_parse_int(x.stem))

for f in files:
    try:
        process_heatresult(f)
    except:
        print(f"CHYBA při zpracování souboru '{f}'. Soubor je pravděpodobně poškozen.")

print()
input("Stiskněte Enter pro ukončení programu...")