|
@@ -24,8 +24,43 @@ import sys
|
|
|
import time
|
|
|
import serial
|
|
|
import urllib
|
|
|
+import logging
|
|
|
+import argparse
|
|
|
import requests
|
|
|
import configparser
|
|
|
+from requests.packages.urllib3.exceptions import InsecureRequestWarning
|
|
|
+
|
|
|
+
|
|
|
+logger = logging.getLogger('qrcode_reader')
|
|
|
+logging.basicConfig(level=logging.INFO)
|
|
|
+logging.getLogger('requests').setLevel(logging.WARNING)
|
|
|
+logging.getLogger('urllib3').setLevel(logging.WARNING)
|
|
|
+requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
|
|
|
+
|
|
|
+
|
|
|
+def convert_obj(obj):
|
|
|
+ try:
|
|
|
+ return int(obj)
|
|
|
+ except:
|
|
|
+ pass
|
|
|
+ if isinstance(obj, str):
|
|
|
+ obj_l = obj.lower()
|
|
|
+ if obj_l in ['true', 'on', 'yes']:
|
|
|
+ return True
|
|
|
+ elif obj_l in ['false', 'off', 'no']:
|
|
|
+ return False
|
|
|
+ return obj
|
|
|
+
|
|
|
+
|
|
|
+def convert(seq):
|
|
|
+ if isinstance(seq, dict):
|
|
|
+ d = {}
|
|
|
+ for key, item in seq.items():
|
|
|
+ d[key] = convert(item)
|
|
|
+ return d
|
|
|
+ if isinstance(seq, (list, tuple)):
|
|
|
+ return [convert(x) for x in seq]
|
|
|
+ return convert_obj(seq)
|
|
|
|
|
|
|
|
|
class Connector():
|
|
@@ -56,37 +91,46 @@ class Connector():
|
|
|
req.raise_for_status()
|
|
|
req.connection.close()
|
|
|
except requests.exceptions.ConnectionError as ex:
|
|
|
- print('unable to connect to %s: %s' % (self.login_url, ex))
|
|
|
+ logger.error('unable to connect to %s: %s' % (self.login_url, ex))
|
|
|
sys.exit(1)
|
|
|
|
|
|
def checkin(self, code):
|
|
|
+ msg = 'scanning code %s: ' % code
|
|
|
limit_field = self.cfg['event'].getint('limit_field')
|
|
|
if limit_field:
|
|
|
code = code[:limit_field]
|
|
|
- checkin_url = self.checkin_url + '?' + urllib.parse.urlencode({cfg['event']['field']: code})
|
|
|
- params = dict(self.cfg['actions'])
|
|
|
- req = self.session.put(checkin_url, json=params)
|
|
|
+ params = {cfg['event']['field']: code, '_errorMessage': 'code: %s' % code}
|
|
|
+ checkin_url = self.checkin_url + '?' + urllib.parse.urlencode(params)
|
|
|
+ json = convert(dict(self.cfg['actions']))
|
|
|
+ req = self.session.put(checkin_url, json=json)
|
|
|
+ error = False
|
|
|
try:
|
|
|
req.raise_for_status()
|
|
|
+ msg += 'ok'
|
|
|
except requests.exceptions.HTTPError as ex:
|
|
|
- print('error: %s' % req.json().get('message'))
|
|
|
+ error = True
|
|
|
+ msg += 'error: %s' % req.json().get('message')
|
|
|
+ if not error:
|
|
|
+ logger.info(msg)
|
|
|
+ else:
|
|
|
+ logger.warning(msg)
|
|
|
req.connection.close()
|
|
|
|
|
|
|
|
|
def scan(port):
|
|
|
retry = 1
|
|
|
while True:
|
|
|
- print('waiting for connection on port %s...' % port)
|
|
|
+ logger.debug('waiting for connection on port %s...' % port)
|
|
|
try:
|
|
|
ser = serial.Serial(port=port, timeout=1)
|
|
|
break
|
|
|
except serial.serialutil.SerialException as ex:
|
|
|
- if retry >= 10:
|
|
|
- print('unable to connect: %s' % ex)
|
|
|
+ if retry >= 20:
|
|
|
+ logger.error('unable to connect: %s' % ex)
|
|
|
sys.exit(2)
|
|
|
time.sleep(1)
|
|
|
retry += 1
|
|
|
- print('connected to %s' % port)
|
|
|
+ logger.info('connected to %s' % port)
|
|
|
ser_io = io.TextIOWrapper(io.BufferedRWPair(ser, ser, 1), newline='\r', line_buffering=True)
|
|
|
while True:
|
|
|
line = ser_io.readline().strip()
|
|
@@ -96,12 +140,22 @@ def scan(port):
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
- cfg = configparser.ConfigParser()
|
|
|
- cfg.read('qrcode_reader.ini')
|
|
|
+ parser = argparse.ArgumentParser()
|
|
|
+ parser.add_argument('-c', '--code', help='specify a single code', action='store')
|
|
|
+ parser.add_argument('--config', help='user a different configuration file (default: qrcode_reader.ini)',
|
|
|
+ action='store', default='qrcode_reader.ini')
|
|
|
+ args = parser.parse_args()
|
|
|
+
|
|
|
+ cfg = configparser.ConfigParser(interpolation=configparser.ExtendedInterpolation())
|
|
|
+ cfg.read(args.config)
|
|
|
+ if cfg['qrcode_reader'].getboolean('debug'):
|
|
|
+ logging.basicConfig(level=logging.DEBUG)
|
|
|
connector = Connector(cfg)
|
|
|
- try:
|
|
|
- for code in scan(port=cfg['connection']['port']):
|
|
|
- print('received code %s' % code)
|
|
|
- connector.checkin(code)
|
|
|
- except KeyboardInterrupt:
|
|
|
- print('exiting...')
|
|
|
+ if args.code:
|
|
|
+ connector.checkin(args.code)
|
|
|
+ else:
|
|
|
+ try:
|
|
|
+ for code in scan(port=cfg['connection']['port']):
|
|
|
+ connector.checkin(code)
|
|
|
+ except KeyboardInterrupt:
|
|
|
+ logger.info('exiting...')
|