Biometric_Lock_using_Raspberry_PI/enroll.py at master · sagarpant/Biometric_Lock_using_Raspberry_PI · 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 time, sys, baseMethods,lcddriver


class Enroll:
    baseMethod = baseMethods.BaseMethods()
    global display
    display = lcddriver.lcd()

    def __init__(self):
        pass

    def main(self, mode):
        # mode = 1 for normal user
        # mode = 2 for authorizer

        self.body2()

        while True:
            resultRegModel = self.baseMethod.regModel()
            if resultRegModel == 10:
                display.lcd_clear()
                display.lcd_display_string("Fingerprints are", 1)
                display.lcd_display_string("not the same", 2)
                time.sleep(2)
                display.lcd_clear()
                self.body2()
            elif resultRegModel == 1:
                display.lcd_clear()
                display.lcd_display_string('Internal Error', 1)
                time.sleep(2)
                display.lcd_clear()
                sys.exit(0)
            else:
                break


        vtnOutput = self.baseMethod.valid_temp_num()
        if vtnOutput[0] == 0:  # if the function was executed successfuly
            pg = 0  # vtnOutput[1]
            id = vtnOutput[2]
            if id == 255:  # one extra condition of pg==0 removed
                display.lcd_clear()
                display.lcd_display_string("Memory full",1)
                time.sleep(2)
                display.lcd_clear()
                system.exit(0)
        else:
            display.lcd_clear()
            display.lcd_display_string("Internal Error!",1)
            time.sleep(2)
            display.lcd_clear()
            sys.exit(0)

        if self.baseMethod.store(pg, id):
            display.lcd_clear()
            display.lcd_display_string('Store Error!',1)
            time.sleep(2)
            display.lcd_clear()
            sys.exit(0)
        # Proceed from here as no errors found.

        if mode == 1:
            display.lcd_clear()
            display.lcd_display_string("Registration",1)
            display.lcd_display_string("Success",2)
            time.sleep(2)
            display.lcd_clear()
        elif mode == 2:
            try:
                fp = open('records.txt', 'a')
                fp.write(str(pg) + "," + str(id) + "\n")
            finally:
                fp.close()
            display.lcd_clear()
            display.lcd_display_string("Authorization",1)
            display.lcd_display_string("Success",2)
            time.sleep(2)
            display.lcd_clear()
        else:
            display.lcd_clear()
            display.lcd_display_string("Internal Error",1)
            time.sleep(2)
            display.lcd_clear()
    def body(self, comment1, comment2):
        display.lcd_clear()
        display.lcd_display_string(comment1,1)
        display.lcd_display_string(comment2,2)
        sys.stdout.flush()

        time.sleep(1)
        timeStamp = time.time()

        while self.baseMethod.genImg():
            if (time.time() - timeStamp) <= 120:
                time.sleep(0.1)
                sys.stdout.flush()
            else:
                break

        display.lcd_clear()
        display.lcd_display_string('Processing',1)
        sys.stdout.flush()

    def body2(self):
        self.body("Put New Finger", "Again 1")

        if self.baseMethod.img2Tz(1):
            display.lcd_clear()
            display.lcd_display_string('Conversion Error',1)
            time.sleep(2)
            display.lcd_clear()
            self.body("Put New Finger", "Again 1")

        self.body("Put New Finger", "Again 2")

        if self.baseMethod.img2Tz(2):
            display.lcd_clear()
            display.lcd_display_string('Conversion Error',1)
            time.sleep(2)
            display.lcd_clear()
            self.body("Put New Finger", "Again 2")