class InputHandler:
cookies = {}
def __init__(self, db="user"):
self.DE = DataExtractor()
self.db = DataBase()
self.handlers = {
# "/register": lambda data, headers: self.add_user(data),
# "/download": lambda headers: self.send_databse(headers),
"/add_user":
lambda data, headers: self.add_user(data),
"/remove_user":
lambda data, headers: self.remove_user(data, headers),
"/login":
lambda headers: self.login(headers),
"/get_data":
lambda headers: self.get_data(
headers), #todo maybe separate get and post callbacks
"/handle_image":
lambda data, headers: self.handle_image(data, headers),
"/change_password":
lambda data, headers: self.change_password(data, headers),
"/change_data":
lambda data, headers: self.change_data(data, headers),
"/remove_record":
lambda data, headers: self.remove_record(data, headers),
}
def add_user(self, inp):
data = self.decode_json(inp)
print(data)
return 200 if self.db.add_user(data['name'], data['password'],
data["email"]) else 234
def change_password(self, data, headers):
if not self.verify_user(headers):
return 234
data = self.decode_json(data)
name = self.parse_header(headers)[0]
return 200 if self.db.update_user(name, data['password']) else 234
def remove_user(self, data, headers):
data = self.decode_json(data)
print(data)
if not self.verify_user(headers):
return 234
self.db.remove_user(data['name'])
return 200
def login(self, headers):
if self.verify_user(headers):
name = self.parse_header(headers)[0]
InputHandler.cookies[name] = ''.join(
random.choice(string.digits + string.ascii_letters)
for _ in range(20)) #todo remove cookies
return InputHandler.cookies[name]
else:
return None
def get_data(
self, #GET
headers):
if not self.verify_user(headers):
return 234
name = self.parse_header(headers)[0]
return self.db.view_database(name)
def remove_record(self, data, headers):
if not self.verify_user(headers):
return 234
record = data.decode('utf-8')
name = self.parse_header(headers)[0]
self.db.remove_record(name, record)
return 200
def change_data(self, data, headers):
if not self.verify_user(headers):
return 234
data = self.decode_json(data)
name = self.parse_header(headers)[0]
return 200 if self.db.update_record(name, data) else 234
def add_image(self, headers):
if not self.verify_user(headers):
return 234
name = self.parse_header(headers)[0]
return self.db.view_database(name)
def verify_user(self, headers):
name, password = self.parse_header(headers)
print(name, password)
if InputHandler.cookies.get(name) == password:
return True
return True if self.db.verify_user(name, password) else False
def handle_image(self, data, headers):
if not self.verify_user(headers):
return 234
user = self.parse_header(headers)[0]
OCR = self.DE.extract(data)
self.db.add_record(user, OCR)
return 200
def decode_json(self, data):
ret = data.decode('utf-8')
return ast.literal_eval(ret)
def get_handlers(self):
return self.handlers
def parse_header(self, header):
authorization = header.get("Authorization")
creds = authorization.split(':')
password = creds[1]
name = creds[0]
return name, password
#todo remove prints
#{
# "name":"michau",
# "email":"*****@*****.**",
# "password":"******"
#}
class sqlite_test:
def __init__(self):
self.db = DataBase()
def connect(self):
self.connection = sqlite3.connect(DB_FILENAME)
def do_stuff(self):
self.connection.execute("""
CREATE TABLE IF NOT EXISTS user (
username TEXT,
password TEXT,
email TEXT,
data JSON
)
""")
self.connection.commit()
some_data = {"sample": "raz",
"cos": "cosjescze"}
jobject = json.dumps(some_data)
self.connection.executemany("""
INSERT INTO user (username, password, email, data)
VALUES (?, ?, ?, ?)
""", (('wonsz_rzeczny', 'jest_niebezpieczny', "*****@*****.**", jobject),
('Radovid', '20pointsShani', "*****@*****.**", jobject)))
self.connection.commit()
for row in self.connection.execute("SELECT * FROM user"):
print(row)
def test(self):
self.test_add_user()
self.test_remove_user()
self.test_ver()
self.test_remove_record()
self.test_add_record()
self.test_update_record()
# name = "Radovid"
# data = {"smth":"2",
# "moar":"yas"}
# db.add_record(name, data)
# res = db.view_database("Bob Girlyman")
# print(res.fetchone())
def test_ver(self):
name = "Indrick Boreale"
password = "******"
self.db.add_user(name, password)
res = self.db.verify_user("Indrick Boreale", "SpessMehrens")
if res is True:
print("!!! 1ver test passed")
else:
print("!!! 1ver test failed")
self.db.remove_user("Indrick Boreale")
res = self.db.verify_user("Indrick Boreale", "SpessMehrens")
if res is False:
print("!!! 2ver test passed")
else:
print("!!! 2ver test failed")
def test_add_user(self):
self.db.remove_user("Indrick Boreale")
name = "Indrick Boreale"
password = "******"
self.db.add_user(name, password)
res = self.db.verify_user("Indrick Boreale", "SpessMehrens")
print("res val is " + str(res))
if res is True:
print("!!!add user test passed")
else:
print("!!!add user test failed")
def test_update(self):
self.db.update_user("test", "silver")
def test_remove_user(self):
self.db.remove_user("Indrick Boreale")
self.db.remove_user("Indrick Boreale")
res = self.db.verify_user("Indrick Boreale", "SpessMehrens")
if res is False:
print("!!!remove test passed")
else:
print("!!!remove test failes=d")
def test_add_record(self):
name = "Bob Gyrlyman"
dataname = "Marneus"
rec = {dataname: {"password": "******", "email": "*****@*****.**"}}
js = json.dumps(rec)
self.db.add_record(data=js, name=name)
self.db.add_record(data=js, name=name)
db = self.db.view_database(name)
if db[dataname] == rec[dataname]:
print("!!!1 add record test passed")
else:
print("!!!1 add record test failed")
self.db.remove_record(name, dataname)
self.db.add_record(data=js, name=name)
db = self.db.view_database(name)
if db[dataname] == rec[dataname]:
print("!!!2 add record test passed")
else:
print("!!!2 add record test failed")
def test_update_record(self):
name = "Bob Gyrlyman"
dataname = "Marneus"
rec = {dataname: {"password": "******", "email": "*****@*****.**"}}
js = json.dumps(rec)
self.db.add_record(data=js, name=name)
up = {dataname: {"password": "******", "email": "*****@*****.**"}}
self.db.update_record(name, up)
res = self.db.view_database(name)
if res[dataname] == up[dataname]:
print("!!! update record test passed")
else:
print("!!! update record test failed")
def test_remove_record(self):
name = "Bob Gyrlyman"
dataname = "Marneus"
rec = {dataname: {"password": "******", "email": "*****@*****.**"}}
js = json.dumps(rec)
self.db.add_record(data=js, name=name)
print("remove")
self.db.remove_record(name, dataname)
self.db.remove_record(name, dataname)
db = self.db.view_database("Bob Gyrlyman")
try:
db[dataname]
except KeyError:
print("!!! remove record test pass")
return
print("!!! remove record test failed")