jernbane/US-H.py at main · sleipner01/jernbane · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
import sqlite3
import datetime

con = sqlite3.connect('jernbanen.db')

"""
    Henter alle billettene for en kunde

    @param kundeNummer: kundenummeret til kunden
"""
def getTickets(kundeNummer):
    cur = con.cursor()
    res = cur.execute("""
                SELECT kundenummer, fornavn, etternavn, tidspunkt as "Kjøpetidspunkt", B.rutenummer,
                avgangsDato, rekkefolge as "Vogn-rekkefolge", plassNr, vogntype, j.navn, j2.navn, S1.avgangsTid, S2.ankomstTid

                FROM Kunde K
                JOIN Kundeordre KO ON K.kundenummer = KO.kundeId
                JOIN Billett B ON B.ordrenummer = KO.ordrenummer
                JOIN Jernbanestasjon j ON B.startStasjonId = j.stasjonId
                JOIN Jernbanestasjon j2  ON B.endeStasjonId = j2.stasjonId
                JOIN Stoppested s1 ON B.startStasjonId = s1.stasjonId AND s1.rutenummer = B.rutenummer
                JOIN Stoppested s2 ON B.endeStasjonId = s2.stasjonId AND s2.rutenummer = B.rutenummer

                JOIN (  Togrute t2
                    JOIN VognOppsett ON VognOppsett.vognOppsettId = t2.vognOppsettId
                    JOIN VognerPaaVognOppsett ON VognerPaaVognOppsett.vognOppsettId = VognOppsett.vognOppsettId
                    JOIN Vogn ON VognerPaaVognOppsett.vognId = Vogn.vognId) ON t2.rutenummer = b.rutenummer AND Vogn.vognId = B.vognId

                WHERE K.kundenummer = ? AND avgangsDato > ?
                ORDER BY avgangsDato, S1.avgangsTid ASC

                """, (str(kundeNummer) , "2023-04-01 12:00:00"))

                # Datoen er satt til 1. april 2023 for å sikre at data ikke blir filtrert bort
                # siden databasen ikke har noe data etter 9.april 2023
                # Burde ideelt sett blitt brukt str(Date.now())

    return res.fetchall()

"""
    Konverterer data fra databasen til en dict

    @param data: data fra databasen
"""
def dataToDict(data):
    dict = []
    for row in data:
        tmp = {}
        tmp["kundenummer"] = row[0]
        tmp["fornavn"] = row[1]
        tmp["etternavn"] = row[2]
        tmp["kjøpetidspunkt"] = row[3]
        tmp["rutenummer"] = row[4]
        tmp["avgangsDato"] = row[5]
        tmp["Vogn-rekkefølge"] = row[6]
        tmp["plassNr"] = row[7]
        tmp["vogntype"] = row[8]
        tmp["startStasjon"] = row[9]
        tmp["sluttStasjon"] = row[10]
        tmp["avreise"] = row[11]
        tmp["ankomst"] = row[12]

        dict.append(tmp)

    return dict

"""
    Printer ut alle kommende billetter for en kunde

    @param dict: dict med data fra databasen
"""
def printData(dict):
    if len(dict) == 0:
        print("Ingen reiser funnet for kunde\n")
        return

    print("\nKommende reiser for kunde: ", dict[0]["kundenummer"])
    for ticket in dict:
        print("""
            \rDato: %s
            \r---------------------------------
            \rFra: %9s     Til: %9s
            \rAvgang: %5s      Ankomst: %5s
            \rRutenummer: %s
            \r---------------------------------""" %(ticket["avgangsDato"], ticket["startStasjon"], ticket["sluttStasjon"], ticket["avreise"], ticket["ankomst"], ticket["rutenummer"]))

        if(ticket["vogntype"] == 1):
            coupe = ticket["plassNr"] // 2
            print("""\rSovevogn:
            \rVogn-rekkefølge: %3s  Kupe: %2s
            \rSeng-nr: %1s
            \r---------------------------------
            """ %(ticket["Vogn-rekkefølge"], coupe, ticket["plassNr"]))

        if(ticket["vogntype"] == 0):
            print("""\rSittevogn:
            \rVogn-rekkefølge: %3s Plass-nr: %2s
            \r---------------------------------
            """ %(ticket["Vogn-rekkefølge"], ticket["plassNr"]))


"""
    Henter ut alle brukere i databasen
"""
def getUsers():
    cur = con.cursor()
    res = cur.execute("""
        SELECT kundenummer, fornavn, etternavn FROM Kunde
    """)
    res = res.fetchall()

    users = []
    for user in res:
        tmp = {}
        tmp["kundenummer"] = user[0]
        tmp["fornavn"] = user[1]
        tmp["etternavn"] = user[2]
        users.append(tmp)

    print("Registrerte kunder:\n")
    print("Kundenummer\t| Fornavn \t| Etternavn")
    print("--------------------------------------------------")
    for user in users:
        print("%s \t\t| %s\t\t| %s" %(user["kundenummer"], user["fornavn"], user["etternavn"]))
    print("--------------------------------------------------")
    print(" ")
    print("Skriv q for å avslutte programmet\n")


def main():
    getUsers()
    res = ""
    while res != "q":
        res = input("Skriv inn ditt kundenummer for å se dine fremtidige reiser: ")
        if res == "q":
            break
        try:
            res = int(res)
        except:
            print("Ugyldig input, skriv inn et gyldig tall\n")
            continue
        else:
            data = getTickets(res)
            dict = dataToDict(data)
            printData(dict)
            break
main()
con.close()