objectville/SubwayLoader.py at main · michael-borck/objectville · 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
from Subway        import Subway
from SubwayPrinter import SubwayPrinter

class SubwayLoader(object):

    def __init__(self):
        self.subway = Subway()

    def get_subway(self):
        return self.subway

    def loadFromFile(self, name):
        data = file(name, "U")
        self.loadStations(data)
        line = data.readline().strip()
        while line != "":
            self.loadLine(data, line)
            line = data.readline().strip()
        data.close()

    def loadStations(self, data):
        station = data.readline().strip()
        while station != "":
            self.subway.add_station(station)
            station = data.readline().strip()

    def loadLine(self, data, line):
        station1 = data.readline().strip()
        station2 = data.readline().strip()
        while station2 != "":
            self.subway.add_connection(station1,station2,line)
            station1 = station2
            station2 = data.readline().strip()

if __name__ == '__main__':
    """Takes the place of the book's LoadTester class."""
    loader = SubwayLoader()
    loader.loadFromFile("ObjectvilleSubway.txt")
    subway = loader.get_subway()
    print "Testing stations...",
    if subway.has_station("DRY Drive") and \
       subway.has_station("Servlet Springs") and \
       subway.has_station("Boards 'R' Us"):
        print "PASSED!"
    else:
        print "FAILED!"
    print "Testing connections...",
    if subway.has_connection("DRY Drive","PMP Place","Wirfs-Brock Line")   and \
       subway.has_connection("GoF Gardens","JSP Junction","Jacobson Line") and \
       subway.has_connection("LSP Lane","Head First Labs","Booch Line"):
        print "PASSED!"
    else:
        print "FAILED!"
    print
    p = SubwayPrinter()
    print "-" * 70
    p.print_directions(subway.get_directions("AJAX Rapids", "JavaRanch"))
    print "-" * 70
    p.print_directions(subway.get_directions("XHTML Expressway","JSP Junction"))
    print "-" * 70
    p.print_directions(subway.get_directions("Mighty Gumball, Inc.", "Choc-O-Holic, Inc."))
    print "-" * 70