tracker/rtracker.py at master · arifchauhan/tracker · 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
#!/usr/bin/env python3

import RPi.GPIO as GPIO
import time
import requests
import datetime
import base64
import LCD
import json

TRIG = 16
ECHO = 18
URL = 'http://192.168.0.101:8000/api/attendance/'


def setup():
    GPIO.setmode(GPIO.BOARD)
    GPIO.setup(TRIG, GPIO.OUT)
    GPIO.setup(ECHO, GPIO.IN)


def distance():
    GPIO.output(TRIG, 0)
    time.sleep(0.000002)
    GPIO.output(TRIG, 1)
    time.sleep(0.00001)
    GPIO.output(TRIG, 0)
    while GPIO.input(ECHO) == 0: a = 0
    start = time.time()
    while GPIO.input(ECHO) == 1: a = 1
    return (time.time() - start) * 340 / 2 * 100


def clear_LCD():
    LCD.lcd_string("---- HELLO -----", 0x80)
    LCD.lcd_string("________________", 0xC0)


def wait_LCD(retry=False):
    LCD.lcd_string('     Retrying...' if retry else '   Recognizing..', 0x80)
    LCD.lcd_string('  Please  Wait..', 0xC0)


def animate_inout(inout):
    if inout:
        for i in range(1, 8):
            start = '{}{}'.format(('>' * i), (' ' * (7 - i)))
            end = '{}{}'.format((' ' * (7 - i)), ('<' * i))
            LCD.lcd_string('{}IN{}'.format(start, end), 0xC0)
            time.sleep(0.2)
    else:
        for i in range(1, 7):
            end = '{}{}'.format(('>' * i), (' ' * (7 - i)))
            start = '{}{}'.format((' ' * (7 - i)), ('<' * i))
            LCD.lcd_string('{}OUT{}'.format(start, end), 0xC0)
            time.sleep(0.2)
        LCD.lcd_string('<<<<<<<OUT>>>>>>', 0xC0)


def loop():
    b = False
    clear_LCD()
    count = 3
    while True:
        time.sleep(1)
        dis = distance()
        if dis < 100:
            print('Distance: {} cm'.format(int(dis)))
            if b:
                print('Detected, recognizing...')
                continue
            wait_LCD()
            data = {'operation': 200}
            response = requests.post(URL, json=data)
            if response.status_code == 201:
                b = True
                data = json.loads(response.text)
                user = str(data['user'])
                print('{}: {}'.format(user, 'IN' if data['inout'] else 'OUT'))
                LCD.lcd_string(user, 0x80)
                animate_inout(data['inout'])
            else:
                wait_LCD(retry=True)
                print('Unknown user')
        else:
            count = count - 1
            if count == 0:
                clear_LCD()
                count = 5
                b = False


def destroy():
    print("Process killed")
    GPIO.cleanup()

if __name__ == "__main__":
    setup()
    try:
        loop()
    except KeyboardInterrupt:
        destroy()