Traffic-Light/script.py at main · WisconsinAutonomous/Traffic-Light · 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
import RPi.GPIO as GPIO
import time

#Static green, red or yellow

#Everything off

#normal loop

#Set up

GPIO.setmode(GPIO.BCM)


#configure pins

green = 15
yellow = 13
red = 11

#Configure pins as outputs
GPIO.setup(green, GPIO.OUT)
GPIO.setup(yellow, GPIO.OUT)
GPIO.setup(red, GPIO.OUT)

GPIO.output(green, GPIO.LOW)
GPIO.output(yellow, GPIO.LOW)
GPIO.output(red, GPIO.LOW)

#normal loop

def normal_loop():
    while True:
        GPIO.output(green, GPIO.HIGH)
        time.sleep(10)
        GPIO.output(green, GPIO.LOW)
        time.sleep(.5)

        GPIO.output(yellow, GPIO.HIGH)
        time.sleep(10)
        GPIO.output(yellow, GPIO.LOW)
        time.sleep(.5)

        GPIO.output(red, GPIO.HIGH)
        time.sleep(10)
        GPIO.output(red, GPIO.LOW)
        time.sleep(.5)
#Static green, red or yellow

#static green
def static_green():
    GPIO.output(yellow, GPIO.LOW)
    GPIO.output(red, GPIO.LOW)

    try:
        GPIO.output(green, GPIO.HIGH)
    except KeyboardInterrupt:
        print("Done with green")
#static red
def static_red():
    GPIO.output(yellow, GPIO.LOW)
    GPIO.output(green, GPIO.LOW)

    try:
        GPIO.output(red, GPIO.HIGH)
    except KeyboardInterrupt:
        print("Done with red")
#static yellow
def static_yellow():
    GPIO.output(green, GPIO.LOW)
    GPIO.output(red, GPIO.LOW)

    try:
        GPIO.output(yellow, GPIO.HIGH)
    except KeyboardInterrupt:
        print("Done with green")

#everything off
def every_thing_off():
    GPIO.output(green, GPIO.LOW)
    GPIO.output(red, GPIO.LOW)
    GPIO.output(yellow, GPIO.LOW)

def main():
    try :
        while True:
            command = input("List of Commands: /n 1)static green /n 2)static yellow /n 3) static red /n 4) normal loop /n 5) everything off /n 6) exit")

            if command == "static green":
                try:
                    static_green()
                except KeyboardInterrupt:
                    every_thing_off()
            elif command == "static yellow":
                try:
                    static_yellow()
                except KeyboardInterrupt:
                    every_thing_off()
            elif command == "static red":
                try:
                    static_red()
                except KeyboardInterrupt:
                    every_thing_off()
            elif command == "normal loop":
                try:
                    normal_loop()
                except KeyboardInterrupt:
                    every_thing_off()
            elif command == "everything off":
                try:
                    every_thing_off()
                except KeyboardInterrupt:
                    every_thing_off()
            elif command == "exit":
                every_thing_off()
                break

    finally:
        GPIO.cleanup()