RFIDsheild/RFIDshield.py at main · ConnorLayson/RFIDsheild · 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
#Import libraries
import board
import adafruit_st25dv16
from digitalio import DigitalInOut, Direction, Pull

#Write to RFID chips
class rfid:
    def __init__(self):
        self.i2c = board.I2C()
        self.eeprom = adafruit_st25dv16.EEPROM_I2C(self.i2c)

    def write(self,to_write,head=0x04):
        #TBH, I used code I found and slightly modified it to work in here.
        #Not entirely sure how it works. Feel free to take this if you need it
        str=to_write
        l=len(str)
        buf = bytearray ([0xe1, 0x40, 0x40, 0x05, 0x03, 0x00, 0xd1, 0x01, 0x00, 0x55])
        buf[5] = (l+5)
        buf[8] = (l+1)
        self.eeprom[0:len(buf)]=buf
        self.eeprom[len(buf)]=head
        k=len(buf)+1
        self.eeprom[k:k+l]=bytearray(str)
        self.eeprom[k+l]=0xfe

        for i in range(0, 5):
            j = i * 16
            hex_string = ":".join("%02x" % b for b in self.eeprom[j:j+15])
            print(j, "> ", hex_string, "> ", self.eeprom[j:j+15])

#Button functions
class button:
    def __init__(self,button):
        #Set button pin
        if button == 1:
            self.button = DigitalInOut(board.D9)
        if button == 0:
            self.button = DigitalInOut(board.D8)
        self.button.direction = Direction.INPUT
        self.button.pull = Pull.UP

    def get_press(self):
        #For some reason, button.value is False if the button is pressed
        #Not sure why but I swapped it here
        if not self.button.value:
            return True
        else:
            return False

#All the LED things
class led_array:
    def __init__(self):
        #Sets all the LEDs and appends to a list for ease-of-use
        self.list = []
        led1 = DigitalInOut(board.D0)
        led1.direction = Direction.OUTPUT
        self.list.append(led1)
        led2 = DigitalInOut(board.D1)
        led2.direction = Direction.OUTPUT
        self.list.append(led2)
        led3 = DigitalInOut(board.D2)
        led3.direction = Direction.OUTPUT
        self.list.append(led3)
        led4 = DigitalInOut(board.D3)
        led4.direction = Direction.OUTPUT
        self.list.append(led4)
        led5 = DigitalInOut(board.D4)
        led5.direction = Direction.OUTPUT
        self.list.append(led5)
        led6 = DigitalInOut(board.D5)
        led6.direction = Direction.OUTPUT
        self.list.append(led6)
        led7 = DigitalInOut(board.D6)
        led7.direction = Direction.OUTPUT
        self.list.append(led7)
        led8 = DigitalInOut(board.D7)
        led8.direction = Direction.OUTPUT
        self.list.append(led8)

    def fill(self):
        #Turns on all the LEDs
        for led in self.list:
            led.value = True

    def clear(self):
        #Turns off all the LEDs
        for led in self.list:
            led.value = False

    def activate(self,led):
        #Turns on LED specified
        if type(led) == int:
            self.list[led].value = True
        #Turns on LEDs specified
        if type(led) == list:
            for light in led:
                self.list[light].value = True

    def deactivate(self,led):
        #Turns off LED specified
        if type(led) == int:
            self.list[led].value = False
        #Turns off LEDs specified
        if type(led) == list:
            for light in led:
                self.list[light].value = False

    def toggle(self,led):
        #Toggles LED specified
        if type(led) == int:
            if self.list[led].value == True:
                self.list[led].value = False
            else:
                self.list[led].value = True
        #toggles LEDs specified
        if type(led) == list:
            for light in led:
                if self.list[light].value == True:
                    self.list[light].value = False
                else:
                    self.list[light].value = True