Importowanie własnego modułu - Python

Question

Cześć drodzy pasjonaci. Mam pytanie odnoście importowania własnego modułu do innego pliku .py

Napisałem 2 programy. Pierwszy to quiz zaciągający się z pliku txt, który zapisuje do innego pliku (bestscore.txt) nickname i wynik punktowy gracza.

Drugi program zaciąga dane z pliku bestscore.txt i formuje tabele z 3-ma najlepszymi wynikami. 

I teraz ja bym chciał te dwa programy ze sobą połączyć nie kopiując i przyklejając kodu. W sensie po zakończeniu quizu dla gracza - wyświetlić tabele najlepszych wyników.

Pytanie:

Czy da się to zrobić jeżeli program dwa - nie jest napisany w formie funkcji tylko po prostu tak:

if __name__ == "__main__":
    import operator
    topscore = []
    with open("best_score.txt","r") as the_file:
        for line in the_file:
            words = line.split()
            topscore.append(words)
        topscore.sort(key=operator.itemgetter(1))

    place1 = topscore[-1]
    place2 = topscore[-2]
    place3 = topscore[-3]
    print("\n\tTABELA NALEPSZYCH WYNIKOW")
    print("\t" + "-"*28)
    print("\t| nick: {:10s}| {:3s}pkt.|".format(place1[0], place1[1]))
    print("\t" + "-"*28)
    print("\t| nick: {:10s}| {:3s}pkt.|".format(place2[0], place2[1]))
    print("\t" + "-"*28)
    print("\t| nick: {:10s}| {:3s}pkt.|".format(place3[0], place3[1]))
    print("\t" + "-"*28)

Jeszcze format fliku bestscore.txt:

Marysia 5
Seba 5
Zbych 4
Spirytus 7
Roman 2

PS. Wiem, że gdyby ten program był funkcją to mógłbym użyć : import file i potem file.funkcja(). Czy powyższy program może się wykonać w głównym programie nie będąc napisanym w formie funkcji?

Answer 1

Wystarczy że zdefiniujesz funkcję np. main() i ją wywołasz w sekcji if.... dodatkowo będziesz ją mógł wywołać z modułu:

def main():
    import operator
    topscore = []
    with open("best_score.txt","r") as the_file:
        for line in the_file:
            words = line.split()
            topscore.append(words)
        topscore.sort(key=operator.itemgetter(1))
 
    place1 = topscore[-1]
    place2 = topscore[-2]
    place3 = topscore[-3]
    print("\n\tTABELA NALEPSZYCH WYNIKOW")
    print("\t" + "-"*28)
    print("\t| nick: {:10s}| {:3s}pkt.|".format(place1[0], place1[1]))
    print("\t" + "-"*28)
    print("\t| nick: {:10s}| {:3s}pkt.|".format(place2[0], place2[1]))
    print("\t" + "-"*28)
    print("\t| nick: {:10s}| {:3s}pkt.|".format(place3[0], place3[1]))
    print("\t" + "-"*28)

if __name__ == "__main__":
    main()

 

Comments

@mokrowski - Jak zawsze niezawodny ;)

Wszystko działa jak powinno. Wielkie dzięki i pozdrawiam.