import matplotlib.pyplot as plt

def gjennomsnitt(tall):
    sum = 0
    for i in tall:
        sum += i
    return sum / len(tall)

def les_inn_aksjekurser(filnavn):
    fil = open(filnavn)
    priser = []
    for linje in fil:
        priser.append(float(linje.strip()))
    return priser

def hent_gjennomsnitt_siste_n_dager(priser, antall_dager):
    gjennomsnitt_liste = []    
    for i in range(len(priser)):
        if i == 0:
            gjennomsnitt_liste.append(priser[i])
        elif i < antall_dager:
            gjennomsnitt_liste.append(gjennomsnitt(priser[0: i]))
        else:
            gjennomsnitt_liste.append(gjennomsnitt(priser[i - antall_dager: i]))
    return gjennomsnitt_liste

def kjop_og_selg(priser, gjennomsnittspriser):
    har_kjopt = False
    kjop_pris = 0
    tjent_totalt = 0
    for i in range(10, len(priser)):
        if not har_kjopt and priser[i - 1] < gjennomsnittspriser[i] and priser[i] >= gjennomsnittspriser[i]:
            har_kjopt = True
            kjop_pris = priser[i]
            print("Kj?per aksjer til {:.2f} kr.".format(kjop_pris))
        elif har_kjopt and priser[i - 1] >= gjennomsnittspriser[i] and priser[i] < priser[i - 1]:
            har_kjopt = False
            tjent_totalt += priser[i] - kjop_pris
            print("Selger aksjer til {:.2f} kr.".format(priser[i]))
            print("Tjent p? denne handelen: {:.2f} kr.".format(priser[i] - kjop_pris))
    print("Tjent totalt: {:.2f} kr".format(tjent_totalt))



priser = les_inn_aksjekurser("apple.txt")
gjennomsnittspriser = hent_gjennomsnitt_siste_n_dager(priser, 10)

# plt.plot(priser)
# plt.plot(gjennomsnittspriser)
# plt.show()

kjop_og_selg(priser, gjennomsnittspriser)