from dataclasses import dataclass

with open("movies.txt", "r") as file:
    contents = file.read()

lines = contents.split("\n")

@dataclass
class Movie:
    title: str
    year: int
    duration: int
    has_award: bool

movies : list[Movie] = []
for line in lines[2:]:
    parts = line.split(";")
    title = parts[2]
    year = int(parts[0])
    if parts[1] == "":
        duration = 0
    else:
        duration = int(parts[1])
    has_award = parts[8] == "Yes" 
    movies.append(Movie(title, year, duration, has_award))

with open("resultats.txt", "w") as file:
    duration_totale = 0
    for movie in movies:
        if movie.has_award:
            duration_totale += movie.duration
            file.write(f"{movie.title};{movie.year};{movie.duration};{movie.has_award}\n")
    file.write(f"Total duration: {duration_totale//60}h\n")