button.py

#Poll Raspberry Pi Buttons and control LEDS
#By Tyler Spadgenske

import RPi.GPIO as gpio
import time

class Poll():
    def __init__(self):
        self.DEBUG = True

        #Setup pins and board
        gpio.setmode(gpio.BCM)

        #Pin 17 is Team 1
        gpio.setup(17, gpio.IN, pull_up_down=gpio.PUD_DOWN)
        #Pin 22 is Team 2
        gpio.setup(22, gpio.IN, pull_up_down=gpio.PUD_DOWN)
        #Pin 4 is Team 3
        gpio.setup(4, gpio.IN, pull_up_down=gpio.PUD_DOWN)
        #Pin 27 is Team 4
        gpio.setup(27, gpio.IN, pull_up_down=gpio.PUD_DOWN)
        #Pin 5 is Team 5
        gpio.setup(5, gpio.IN, pull_up_down=gpio.PUD_DOWN)
        #GPIO Needs to be setup between 3.3v and pins above

        #Put pins in variables
        self.center_button = 22
        self.right_button = 4
        self.left_button = 17
        self.fourth_button = 27
        self.fifth_button = 5
        
        self.first = ''

        self.center_press = 0
        self.right_press = 0
        self.left_press = 0
        self.fourth_press = 0
        self.fifth_press = 0

    def reset(self):
        self.first = ''

        self.center_press = 0
        self.right_press = 0
        self.left_press = 0
        self.fourth_press = 0
        self.fifth_press = 0
        
    def poll(self): #Returns first button pressed
        #Variables for press count (one for each button)
        self.center_press = 0
        self.right_press = 0
        self.left_press = 0
        self.fourth_press = 0
        self.fifth_press = 0

        #Check for button presses
        while True:
            self.first = self.check()
            if self.first != '':
                break
        time.sleep(.5)
        return self.first

    def check(self):
        center_input = gpio.input(self.center_button)
        left_input = gpio.input(self.left_button)
        right_input = gpio.input(self.right_button)
        fourth_input = gpio.input(self.fourth_button)
        fifth_input = gpio.input(self.fifth_button)

        #Center buttons stuff
        if center_input == gpio.HIGH:
            if self.center_press > 0:
                if self.DEBUG:
                    print('ADMIN: Console button 2 has been pressed')
                self.first = 1
            
            else:
                if self.DEBUG:
                    pass
            self.center_press += 1

        #Right buttons stuff
        if right_input == gpio.HIGH:
            if self.right_press > 0:
                if self.DEBUG:
                    print('ADMIN: Console button 3 has been pressed')
                self.first = 2
            
            else:
                if self.DEBUG:
                    pass
            self.right_press += 1

        #Left button stuff
        if left_input == gpio.HIGH:
            if self.left_press > 0:
                if self.DEBUG:
                    print('ADMIN: Console button 1 has been pressed')
                self.first = 0
            
            else:
                if self.DEBUG:
                    pass
            self.left_press += 1

        #Fourth button stuff
        if fourth_input == gpio.HIGH:
            if self.fourth_press > 0:
                if self.DEBUG:
                    print('ADMIN: Console button 4 has been pressed')
                self.first = 3
            
            else:
                if self.DEBUG:
                    pass
            self.fourth_press += 1

        #Fifth button stuff
        if fifth_input == gpio.HIGH:
            if self.fifth_press > 0:
                if self.DEBUG:
                    print('ADMIN: Console button 5 has been pressed')
                self.first = 4
            
            else:
                if self.DEBUG:
                    pass
            self.fifth_press += 1

        return self.first

if __name__ == '__main__':
    test = Poll()
    while True:
        test.first = ''
        winner = test.poll()
        print(winner)