electronic-voting-machine/final.py at master · akumar1503/electronic-voting-machine · 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
import RPi.GPIO as GPIO
from grove_rgb_lcd import *
import time

state = 0
no_of_pos = 0
pos = {}
light = 21
start_button = 17
result_button = 27
button1 = 26
button2 = 19
GPIO.setmode(GPIO.BCM)
GPIO.setup(light,GPIO.OUT)
GPIO.setup(start_button,GPIO.IN,pull_up_down=GPIO.PUD_UP)
GPIO.setup(result_button,GPIO.IN,pull_up_down=GPIO.PUD_UP)
GPIO.setup(button1,GPIO.IN,pull_up_down=GPIO.PUD_UP)
GPIO.setup(button2,GPIO.IN,pull_up_down=GPIO.PUD_UP)

def showLight(st):
    if st == 0:
        GPIO.output(light,GPIO.LOW)
    elif st == 1:
        GPIO.output(light,GPIO.HIGH)

def setupMachine():
    global state

    no_of_pos = int(raw_input("Enter no. of positions:"))
    for i in range(no_of_pos):
        name = raw_input("Enter position name (max 12 char long):")
        can = int(raw_input("Enter no. of candidates:"))
        can_list = {}
        for j in range(can):
            can_name = raw_input("Enter name of candidate no. " + str(j + 1) + ":")
            can_list[can_name] = 0
        pos[name] = can_list
    state = 1
    print(state)
    print("Setup successful")

def startForOnePerson():
    print ("starting...")
    setText("Starting...")
    time.sleep (1)

    for pos_name, can_list in pos.iteritems():
        #print position name in first row
        line1 = ""
        line2 = ""

        line1 += pos_name
        #print candidates in second row
        for can in can_list.keys():
            line2 += can + " "
        setText(line1 + "\n" + line2)
        time.sleep(0.5)

        showLight(1)
        while True:
            input1 = GPIO.input(button1)
            input2 = GPIO.input(button2)

            if input1 == False:
                pos[pos_name][can_list.keys()[0]] += 1
                line1 += "  " + can_list.keys()[0]
                time.sleep(2)
                print("vote to 1")
                break
            elif input2 == False:
                pos[pos_name][can_list.keys()[1]] += 1
                line1 += "  " + can_list.keys()[1]
                time.sleep(2)
                print("vote to 2")
        setText(line1 + "\n" + line2)
        showLight(0)
        time.sleep(2)
    print ("vote recorded")

def printResult():
    print ("Showing result:")

    line1 = "Showing result"
    for pos_name, can_list in pos.iteritems():
        mx = -1
        line2 = ""
        for x , y in can_list.iteritems():
            if y > mx:
                mx=y
        for x , y in can_list.iteritems():
            if y == mx:
                line2 += x + " "
        setText(line1 + "\n" + line2)
        time.sleep(3)

def startMachine():
    global state
    while True:
        start = GPIO.input(start_button)
        res = GPIO.input(result_button)
        if start == False:
            time.sleep(2)
            if state == 0:
                setupMachine()
                print(state)
            elif state == 1:
                startForOnePerson()

        elif res == False:
            time.sleep(2)
            printResult()
            break


setRGB(0,0,255)
startMachine()