csf-labeler/scanner.py at main · cobuildlab/csf-labeler · 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
import os.path
from evdev import InputDevice, categorize, ecodes
from config import BARCODE_CONFIG
from threading import Thread
import time

from utils import custom_upper

KEY_UP = 0
KEY_DOWN = 1
LEFT_SHIFT = 42
KEY_ENTER = 28

# usb barcode scanner will match characters in this array based off keycode to verify correct
# string output due to different encoding
KEY_MAPPING = {'KEY_EQUAL': '+', 'KEY_SLASH': '/', 'KEY_SPACE': ' ', 'KEY_DOT': '.', 'KEY_MINUS': '-', 'KEY_Q': 'q',
               'KEY_W': 'w', 'KEY_E': 'e', 'KEY_R': 'r',
               'KEY_T': 't', 'KEY_Y': 'y',
               'KEY_U': 'u', 'KEY_I': 'i', 'KEY_O': 'o', 'KEY_P': 'p', 'KEY_A': 'a', 'KEY_S': 's', 'KEY_D': 'd',
               'KEY_F': 'f', 'KEY_G': 'g', 'KEY_H': 'h',
               'KEY_J': 'j', 'KEY_K': 'k', 'KEY_L': 'l', 'KEY_Z': 'z', 'KEY_X': 'x', 'KEY_C': 'c', 'KEY_V': 'v',
               'KEY_B': 'b', 'KEY_N': 'n', 'KEY_M': 'm',
               'KEY_1': '1', 'KEY_2': '2', 'KEY_3': '3', 'KEY_4': '4', 'KEY_5': '5', 'KEY_6': '6', 'KEY_7': '7',
               'KEY_8': '8', 'KEY_9': '9', 'KEY_0': '0'}


def get_scanner_device():
    for barcode_config in BARCODE_CONFIG:
        scanner_path = barcode_config["src"]
        path_exists = os.path.exists(scanner_path)
        if path_exists is False:
            continue
        try:
            code_scanner = InputDevice(scanner_path)
        except FileNotFoundError:
            continue
        if code_scanner:
            return code_scanner
    return None


class CodeScanner(Thread):
    def __init__(self):
        Thread.__init__(self, name="CodeScanner")
        self.code_scanner = None
        self.scanned_code = ""

    def reset(self):
        self.scanned_code = ""

    def run(self):
        # we check forever connections and disconnections from the Scanner
        while True:
            self.code_scanner = get_scanner_device()
            if self.code_scanner is None:
                print("CodeScanner:run:No scanner I'll try again in 5 seconds:")
                time.sleep(5)
                continue

            print("CodeScanner:run:code_scanner:", self.code_scanner)
            try:
                to_upper_case = False
                input_code = ""
                for event in self.code_scanner.read_loop():
                    if event.type != ecodes.EV_KEY:
                        continue

                    data = categorize(event)
                    if data.scancode == LEFT_SHIFT:
                        to_upper_case = True
                        continue

                    if data.keystate != KEY_UP:
                        continue

                    if data.scancode == KEY_ENTER:
                        self.scanned_code = input_code
                        input_code = ""
                        continue

                    if data.keycode in KEY_MAPPING:
                        if to_upper_case:
                            input_code += custom_upper(KEY_MAPPING[data.keycode])
                            to_upper_case = False
                        else:
                            input_code += KEY_MAPPING[data.keycode]
            except OSError as e:
                print("CodeScanner:run:Scanner offline:error:", e)
                self.code_scanner = None