sorozatok - 2020. okt - emelt

from datetime import datetime
class epizod:
    def __init__(self, dt='', cim='', resz='', hossz='', latta=''):
        self.dt=dt
        self.cim=cim
        self.resz=resz
        self.hossz=hossz
        self.latta=latta

sorozatok=[]
db=0
latta=0
latta_perc=0

with open("lista.txt") as f:
    for line in f:
        epi = epizod()
        epi.dt=line.rstrip("\n")
        if epi.dt!="NI":
            db=db+1

        epi.cim=next(f).rstrip("\n")
        epi.resz=next(f).rstrip("\n")
        epi.hossz=next(f).rstrip("\n")
        epi.latta=next(f).rstrip("\n")
        if epi.latta=="1":
            latta=latta+1
            latta_perc=latta_perc+int(epi.hossz)



        sorozatok.append(epi)


print("1. feladat")

print("2. feladat")

print("A listában "+str(db)+" db vetítési dátummal rendelkező epizód van.")

print("3. feladat")

osszes = len(sorozatok)
szazalek = round(latta / osszes *100, 2)

print("A listában lévő epizódok "+str(szazalek)+"%-át látta.")

print("4. feladat")
nap = latta_perc // (24*60)
ora = (latta_perc - nap*24*60) // 60
perc = (latta_perc - nap*24*60) % 60

print("Sorozatnézéssel "+str(nap)+" napot "+str(ora)+" órát és "+str(perc)+" percet töltött.")

print("5. feladat")
dt = input("Adjon meg egy dátumot! Dátum= ")
dt=datetime.strptime(dt,"%Y.%m.%d")

for i in sorozatok:
    if i.latta=="0" and i.dt!="NI":
        a=datetime.strptime(i.dt,"%Y.%m.%d")
        if a <= dt:
            print(i.resz,'\t',i.cim)

print("6. feladat")
def hetnapja(ev, ho, nap):
    napok = ["v", "h", "k", "sze", "cs", "p", "szo"]
    honapok = [0, 3, 2, 5, 0, 3, 5, 1, 4, 6, 2, 4]

    if ho < 3:
        ev = ev-1
    return napok[(ev+ev // 4 - ev // 100 + ev // 400 + honapok[ho-1] + nap) % 7]

print("7. feladat")
hnap = input("Adja meg a hét egy napját (például cs)! Nap= ")

sorozat_cimek = set()
for i in sorozatok:
    if i.dt!="NI":
        a=datetime.strptime(i.dt,"%Y.%m.%d")
        if hnap==hetnapja(a.year,a.month,a.day):
            sorozat_cimek.add(i.cim)

if len(sorozat_cimek)==0:
    print("Az adott napon nem kerül adásba sorozat.")
else:
    for i in sorozat_cimek:
        print(i)

print("8. feladat")


sori_cimek = set()
file = open("summa.txt","w")

for i in sorozatok:
    sori_cimek.add(i.cim)

for i in sori_cimek:
    time = 0
    db = 0
    for j in sorozatok:
        if i == j.cim:
            time=time+int(j.hossz)
            db=db+1
    #print(i,time,db)
    szoveg = i+" "+str(time)+" "+str(db)+"\n"
    file.write(szoveg)

file.close()