7.7 DRILL — Ledd 4–5: utskrift/lagring og meny/`main` som binder kjeden

Slik ser fila ut:

Innholdet i lopper.txt:

bod;vare;pris;solgt
B1;sykkel;450;1
B2;bokkasse;120;3
B1;lampe;80;2
B3;kjokkenbord;900;1
B2;platesamling;250;1

Innlesingen fra kap. 7.2:

def les_salg(filnavn):
    f = open(filnavn, 'r')
    overskrift = f.readline().strip().split(';')
    k_pris = overskrift.index('pris')
    k_solgt = overskrift.index('solgt')
    tabell = [overskrift]
    for linje in f:
        felt = linje.strip().split(';')
        felt[k_pris] = int(felt[k_pris])
        felt[k_solgt] = int(felt[k_solgt])
        tabell.append(felt)
    f.close()
    return tabell

Analysen fra kap. 7.4 — omsetningen er pris ganger antall solgte:

def omsetning_per_bod(tabell):
    k_bod = tabell[0].index('bod')
    k_pris = tabell[0].index('pris')
    k_solgt = tabell[0].index('solgt')
    d = {}
    for rad in tabell[1:]:
        belop = rad[k_pris] * rad[k_solgt]
        d[rad[k_bod]] = d.get(rad[k_bod], 0) + belop
    return d

Utskriften, som kaller analysen:

def skriv_oversikt(tabell):
    d = omsetning_per_bod(tabell)
    print('Bod'.ljust(6) + 'Omsetning'.rjust(10))
    print('-' * 16)
    for bod in sorted(d):
        print(bod.ljust(6) + '{:10.2f}'.format(d[bod]))

Utskrift: ingen — blokken definerer bare funksjonen. Den skriver først når den blir kalt med en ferdig tabell.

Kjører du skriv_oversikt(les_salg('lopper.txt')) etter at de tre funksjonene over er definert, skriver programmet:

Utskrift:

Bod    Omsetning
----------------
B1        610.00
B2        610.00
B3        900.00

B1 har solgt sykkel for 450 og to lamper for 80, altså 610 kroner.

Slik ville sensor sett på det. Utskriftsdeloppgaven er typisk 10 %, og fire ting gir uttelling hver for seg: at funksjonen kaller analysen i stedet for å regne selv, overskriftslinja, fast kolonnebredde, og sorted(d) som gir en forutsigbar rekkefølge.

Flere korrekte løsninger. print(bod.ljust(6), '{:10.2f}'.format(d[bod])) med komma i stedet for + gir nesten det samme — bare med ett ekstra mellomrom mellom kolonnene, fordi print selv skiller argumentene med mellomrom. Begge gir full pott; du må bare være konsekvent, ellers står overskriften skjevt.

Kantfeilen: '{:10.2f}'.format(...) på en streng gir ValueError: Unknown format code 'f' for object of type 'str'. Er du usikker på om kolonnen er tall, konverter først — eller bruk str(x).rjust(10), som virker på alt.

def lagre_rapport(tabell, filnavn):
    d = omsetning_per_bod(tabell)
    f = open(filnavn, 'w')
    f.write('bod;omsetning\n')
    for bod in sorted(d):
        f.write(bod + ';' + str(d[bod]) + '\n')
    f.close()
    print('Rapport lagret i', filnavn)

Utskrift: ingen — blokken definerer bare funksjonen. Den skriver først når den blir kalt.

Kjører du lagre_rapport(les_salg('lopper.txt'), 'rapport.txt') etter at les_salg og omsetning_per_bod er definert, skriver programmet:

Utskrift:

Rapport lagret i rapport.txt

Og fila som ble laget:

Utskrift:

Rapport lagret i rapport.txt

bod;omsetning
B1;610
B2;610
B3;900

Legg merke til at rapporten har samme format som inndatafila: overskriftsrad, semikolon, én rad per linje. Det er ikke tilfeldig — da kan rapporten leses inn igjen med den samme innlesingsfunksjonen, og det er nettopp den slags gjenbruk oppgavetekstene noen ganger ber om i den siste deloppgaven.

Slik ville sensor sett på det: fire ting gir uttelling — filnavnet som parameter, '\n' på hver linje, close() etter løkka, og beskjeden til brukeren. Den siste glemmes ofte, og den er nevnt eksplisitt i oppgavetekstene: brukeren skal få vite at det gikk bra.

Kantfeilen: f.write(d[bod]) uten str(...) gir TypeError: write() argument must be str, not int. Denne krasjer, og er derfor snillere enn det glemte linjeskiftet.

import pickle

def lagre_binaer(d, filnavn):
    f = open(filnavn, 'wb')
    pickle.dump(d, f)
    f.close()

def hent_binaer(filnavn):
    f = open(filnavn, 'rb')
    d = pickle.load(f)
    f.close()
    return d

inn = {'B1': 610, 'B2': 610, 'B3': 900}
lagre_binaer(inn, 'omsetning.pkl')
ut = hent_binaer('omsetning.pkl')
print(ut)
print(ut == inn)
print(type(ut['B1']))

Utskrift:

{'B1': 610, 'B2': 610, 'B3': 900}
True
<class 'int'>

True er hele poenget. Ordboka kom tilbake med de samme nøklene, de samme verdiene og de samme typene — 610 er fortsatt et heltall, ikke strengen '610'.

Innholdet i omsetning.pkl vises ikke her, og skal ikke vises: fila er binær og ikke lesbar tekst.

Sammenlign med tekstfil-varianten:

inn = {'B1': 610, 'B2': 610}
f = open('omsetning.txt', 'w')
for bod in sorted(inn):
    f.write(bod + ';' + str(inn[bod]) + '\n')
f.close()

ut = {}
f = open('omsetning.txt', 'r')
for linje in f:
    felt = linje.strip().split(';')
    ut[felt[0]] = int(felt[1])
f.close()
print(ut == inn)

Utskrift:

True

Samme resultat — men veien tilbake krevde fem linjer med parsing, og du måtte huske int(). Det er nettopp den jobben pickle fjerner.

Slik ville sensor sett på det: deloppgaven er som regel liten, og de fire tingene som gir uttelling er import pickle, 'wb'/'rb', dump/load i riktig retning, og close(). Bytter du om 'w' og 'wb', krasjer det med TypeError: write() argument must be str, not bytes — så feilen er i det minste synlig.

Kantfeilen: pickle.load på en fil som ikke finnes gir FileNotFoundError, akkurat som vanlig lesing. Skal programmet takle at rapporten ikke er lagret ennå, hører den i try/except — se løkke 5.

Funksjonen selv inneholder input(), så blokken under definerer bare. Den kan ikke kjøres uten en bruker ved tastaturet:

def meny(tabell):
    while True:
        print()
        print('1 - vis omsetning per bod')
        print('2 - lagre rapport til fil')
        print('q - avslutt')
        valg = input('Valg: ')
        if valg == '1':
            skriv_oversikt(tabell)
        elif valg == '2':
            lagre_rapport(tabell, 'rapport.txt')
        elif valg == 'q':
            print('Ha det.')
            break
        else:
            print('Ugyldig valg, prøv igjen.')

Utskrift: ingen — blokken definerer bare funksjonen. Den venter på tastatur når den blir kalt.

Kjøringen under er ekte. Brukeren valgte 1, deretter 2, deretter en skrivefeil, og til slutt q. Pilene står ikke på skjermen — de er lagt til her for å vise hva brukeren tastet inn.

Kjøring:

1 - vis omsetning per bod
2 - lagre rapport til fil
q - avslutt
Valg: 1        <- skrevet av brukeren
Bod    Omsetning
----------------
B1        610.00
B2        610.00
B3        900.00

1 - vis omsetning per bod
2 - lagre rapport til fil
q - avslutt
Valg: 2        <- skrevet av brukeren
Rapport lagret i rapport.txt

1 - vis omsetning per bod
2 - lagre rapport til fil
q - avslutt
Valg: 5        <- skrevet av brukeren
Ugyldig valg, prøv igjen.

1 - vis omsetning per bod
2 - lagre rapport til fil
q - avslutt
Valg: q        <- skrevet av brukeren
Ha det.

Slik ville sensor sett på det. Fem ting gir uttelling hver for seg:

def les_salg(filnavn):
    f = open(filnavn, 'r')
    overskrift = f.readline().strip().split(';')
    k_pris = overskrift.index('pris')
    k_solgt = overskrift.index('solgt')
    tabell = [overskrift]
    for linje in f:
        felt = linje.strip().split(';')
        felt[k_pris] = int(felt[k_pris])
        felt[k_solgt] = int(felt[k_solgt])
        tabell.append(felt)
    f.close()
    return tabell

Deloppgave b) — analysen (ledd 3). Den tar tabellen som parameter og leser aldri fil:

def omsetning_per_bod(tabell):
    k_bod = tabell[0].index('bod')
    k_pris = tabell[0].index('pris')
    k_solgt = tabell[0].index('solgt')
    d = {}
    for rad in tabell[1:]:
        belop = rad[k_pris] * rad[k_solgt]
        d[rad[k_bod]] = d.get(rad[k_bod], 0) + belop
    return d

Deloppgave c) — utskriften (ledd 5). Den kaller b):

def skriv_oversikt(tabell):
    d = omsetning_per_bod(tabell)
    print('Bod'.ljust(6) + 'Omsetning'.rjust(10))
    print('-' * 16)
    for bod in sorted(d):
        print(bod.ljust(6) + '{:10.2f}'.format(d[bod]))

Utskrift: ingen — blokken definerer bare funksjonen. Den skriver når meny kaller den.

Deloppgave d) — lagringen (ledd 5). Den kaller også b):

def lagre_rapport(tabell, filnavn):
    d = omsetning_per_bod(tabell)
    f = open(filnavn, 'w')
    f.write('bod;omsetning\n')
    for bod in sorted(d):
        f.write(bod + ';' + str(d[bod]) + '\n')
    f.close()
    print('Rapport lagret i', filnavn)

Utskrift: ingen — blokken definerer bare funksjonen. Den skriver når meny kaller den.

Deloppgave e) — menyen (ledd 4). Hver gren er ett kall:

def meny(tabell):
    while True:
        print()
        print('1 - vis omsetning per bod')
        print('2 - lagre rapport til fil')
        print('q - avslutt')
        valg = input('Valg: ')
        if valg == '1':
            skriv_oversikt(tabell)
        elif valg == '2':
            lagre_rapport(tabell, 'rapport.txt')
        elif valg == 'q':
            print('Ha det.')
            break
        else:
            print('Ugyldig valg, prøv igjen.')

Utskrift: ingen — blokken definerer bare funksjonen. Den venter på tastatur når den blir kalt.

Deloppgave f) — main (limet). Fem linjer, og fila leses ett eneste sted:

def main(filnavn):
    try:
        tabell = les_salg(filnavn)
    except FileNotFoundError:
        print('Fant ikke fila', filnavn)
        return
    print('Leste', len(tabell) - 1, 'salgslinjer.')
    meny(tabell)

Utskrift: ingen — blokken definerer bare funksjonen. Den kjører når du kaller main(...).

Hele appen kjørt. Brukeren ser omsetningen, lagrer rapporten og avslutter:

Kjøring:

Leste 5 salgslinjer.

1 - vis omsetning per bod
2 - lagre rapport til fil
q - avslutt
Valg: 1        <- skrevet av brukeren
Bod    Omsetning
----------------
B1        610.00
B2        610.00
B3        900.00

1 - vis omsetning per bod
2 - lagre rapport til fil
q - avslutt
Valg: 2        <- skrevet av brukeren
Rapport lagret i rapport.txt

1 - vis omsetning per bod
2 - lagre rapport til fil
q - avslutt
Valg: q        <- skrevet av brukeren
Ha det.

Pilene står ikke på skjermen; de viser hva brukeren tastet.

Og hva som skjer når fila mangler:

def les_salg(filnavn):
    f = open(filnavn, 'r')
    f.close()
    return []

def main(filnavn):
    try:
        tabell = les_salg(filnavn)
    except FileNotFoundError:
        print('Fant ikke fila', filnavn)
        return
    print('Leste', len(tabell), 'linjer.')

main('finnes-ikke.txt')

Utskrift:

Fant ikke fila finnes-ikke.txt

En setning brukeren forstår, i stedet for fem linjer Traceback.

Slik ville sensor sett på det, deloppgave for deloppgave: