FabKey/FabKey.py at master · FabLabAvignon/FabKey · 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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
#!/usr/bin/python
import os, sys
import serial
import MySQLdb
import ConfigParser
from datetime import datetime

if not os.geteuid() == 0: # Check if is started as root
    sys.exit('Must be run as root.')

class KeyManager:
    def __init__(self, host, uname, passwd, name):
        # Open database connection
        try:
            self.dConn = MySQLdb.connect(
                host=host,
                user=uname,
                passwd=passwd,
                db=name)
        except Exception as e:
            sys.exit("Error: MySQL Server connect failed : " + str(e))

    def isValid(self, keyValue):
        # Init query
        dQuery = self.dConn.cursor()

        # Execute query
        if(dQuery.execute("SELECT keyType, keyOwner FROM keyList WHERE keyValue = %s", [keyValue]) != 0):
            sAnsw = dQuery.fetchone()

            if (not self.isExpired(keyValue)):
                keyType = sAnsw[0].split(":", 1)[0]

                # Get key type, !: Admin, 1: One time use, @: Hour check
                if (keyType == '!'):
                    return True

                if (keyType == '1'):
                    dQuery.execute("DELETE FROM keyList WHERE keyValue = %s", [keyValue])
                    self.dConn.commit()
                    return True

                if (keyType == '@'):
                    try:
                        keyParam = sAnsw[0].split(":", 1)[1]
                    except Exception:
                        print("Warning: Key '" + keyValue + "' has no keyType parameter, expected parameter.")
                        return False

                    nowTime = datetime.now().time()

                    try:
                        startTime = datetime.strptime(
                            keyParam.split("|", 1)[0], "%H-%M").time()
                        endTime = datetime.strptime(
                            keyParam.split("|", 1)[1], "%H-%M").time()
                    except Exception as e:
                        print("Warning: Key '" + keyValue + "' has configuration problem : " + str(e))
                        return False

                    if (startTime < nowTime and endTime > nowTime): # If in time range
                        return True
                    else:
                        return False
            else:
                return False
        else:
            print("Warning: No key found for '" + keyValue + "'.")
            return True

    def isExpired(self, keyValue):
        # Init query
        dQuery = self.dConn.cursor()

        # Execute query
        if(dQuery.execute("SELECT keyExpiry FROM keyList WHERE keyValue = %s", [keyValue]) != 0):
            keyExpiry = dQuery.fetchone()[0]
            if (keyExpiry != "*"): # If key is expirable
                nowDate = datetime.now().date()
                try:
                    expiryDate = datetime.strptime(
                        keyExpiry, "%d-%m-%y").date()
                except Exception:
                    print("Error: Key expiry date for '"
                        + keyValue + "' is invalid, please use * for unexpirable or date like this: DD-MM-YY.")
                    return True

                if (expiryDate > nowDate): # If not expired
                    return False
                else: # If expired
                    dQuery.execute("DELETE FROM keyList WHERE keyValue = %s", [keyValue])
                    self.dConn.commit()
                    print("Warning: Expired key '" + keyValue + "', removed from database.")
                    return True
            else: # If key is unexpirable
                return False
        else:
            print("Warning: No key found for '" + keyValue + "'.")
            return True

class SMSHandler:
    def __init__(self, serialPort):
        # Preconfigure serial
        self.serIO = serial.Serial()
        self.serIO.baudrate = 9600
        self.serIO.port = serialPort

        # Connect to serial port
        try:
            self.serIO.open()
        except Exception as e:
            sys.exit("Error: Failed to open serial port '" + serialPort + "' : " + str(e))

        # Flush in and out
        self.serIO.flushInput()
        self.serIO.flushOutput()


	sys.exit(self.sendSerial("pwd"))
        #while self.serIO.open():


    def getLastestSMS(self):
        self.lol = "mdr"

    def sendSerial(self, command2send):
        self.serIO.write(command2send)

	nothingAnswered = True
	while nothingAnswered:
            if self.serIO.inWaiting() > 0:
                nothingAnwered = False

        return self.serIO.readline()

# Loading config file
Config = ConfigParser.ConfigParser()
Config.read('config.cfg')

if __name__ == '__main__':
    keyMgr = KeyManager( # Init the keyManager
        Config.get("db", "host"),
        Config.get("db", "uname"),
        Config.get("db", "passwd"),
        Config.get("db", "name"))

    smsHandler = SMSHandler(Config.get("ser", "sPort")) # Start the smsHandler to handle incoming sms