slugscan/slugscan.py at master · SLUGSoc/slugscan · 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
122
123
124
125
#!/usr/bin/python

import time
import serial
import argparse
from sql import SqlAccess, NoMemberException, NoEventException, NoUnregisteredCardException
from io_cli import CLI
from io_gui import GUI, UserPermissionException
from io_gpio import GPIOAccess
import ConfigParser

# Input/Output
io = GUI()
gpio = GPIOAccess()

# Parse commandline arguments
argParser = argparse.ArgumentParser(description='RFID card register system')
argParser.add_argument('event', nargs='?', type=str, help='Name of the current event, uses event in db/slugscan.cfg if left empty')

args = argParser.parse_args()
eventName = args.event

if (eventName is None):
	io.log("Using event name from config file...")
	cfg = ConfigParser.ConfigParser()
	cfg.readfp(open(r'db/slugscan.cfg'))
	eventName = cfg.get('Session', 'event').lower()
else:
	eventName = eventName[0].lower()


# RDM6300 Flags
RESCAN_DELAY = 1.3
FLAG_START = '\x02';
FLAG_STOP =  '\x03';
RDM_READ_LENGTH = 14;

PortRF = serial.Serial('/dev/serial0',9600)


# Init SQL
sql = SqlAccess(eventName, io)


def cleanName(name):
	new = str(name)
	new = new.lower().capitalize()
	return new

def createMember(cardNum):
	# Create member prompt, if enable new user flag is set
	# Get user input
	io.log("No member entry present, creating new member...")
	try:
		name = io.getName()
		sql.createMember(cardNum,cleanName(name['fst']),cleanName(name['lst']))

	except UserPermissionException:
		unregId = -1
		try:
			card = sql.getUnregCardByNum(cardNum)
			unregId = card['id']
			sql.updateUnregCard(unregId)
		except NoUnregisteredCardException as e:
			# Card not yet seen (not in unregistered table)
			io.log(str(e))
			try:
				sql.createUnregCard(cardNum)
				card = sql.getUnregCardByNum(cardNum)
				unregId = card['id']
			except:
				io.log("Setting card as unregistered failed.")
		io.output("Card not yet registered, inform a committee member. [ID: " + str(unregId) + "]")
		gpio.notRegistered()
		time.sleep(2)
		return

	except Exception as e:
		io.error(e)
		io.error("Creating member failed, please rescan card.")
		return

	print "Successfully created new member, please rescan card to sign in."

def processCard(cardNum):
	print "Processing Card: " + cardNum
	gpio.successfulScan()
	try:
		member = sql.getMemberForCard(cardNum)
		io.log(member)
		sql.updateRegisterMember(member)

	except NoMemberException as e:
		io.log(e)
		createMember(cardNum)

	time.sleep(RESCAN_DELAY)
	io.output("Please scan card...")

def readRDM6300():
	cId = ""
	PortRF.flushInput()
	PortRF.flushOutput()
	readByte = PortRF.read()
	if readByte == FLAG_START:
		for i in range(RDM_READ_LENGTH + 1):
			readByte = PortRF.read()
			if readByte == FLAG_STOP:
				# Card finished reading, process it
				readByte = None
				io.log("Read Card: " + cId)
				processCard(cId)
				return

			elif (i > RDM_READ_LENGTH):
				# Exceeded possible card length
				io.error("Invalid card read, please rescan the card.")
				return

			cId = cId + str(readByte)

# Main control loop
while True:
	readRDM6300()
	io.update()