def test_handle_image(self):
self.handler = InputHandler(db="test")
header = {'Authorization':"michau:pass2"}
with open("wizytowka/1.jpg", "rb") as file:
self.handler.handle_image(file.read(), header)
db = DataBase()
print(db.view_database("michau"))
def setUp(self):
try:
os.remove("Database/Data/database.db")
except FileNotFoundError:
print("file already removed!")
self.db = DataBase()
self.user1 = {
"name": "Rick Sanchez",
"password": "******",
"email": "*****@*****.**"
}
self.user2 = {
"name": "Talos Valcoran",
"password": "******",
"email": "*****@*****.**"
}
self.user3 = {
"name": "Uzas",
"password": "******",
"email": "*****@*****.**"
}
self.add_user = lambda user: self.db.add_user(user["name"], user[
"password"], user["email"])
self.recname1 = "Indrick Boreale"
self.recname2 = "Vandred Anrathi"
self.recname3 = "Cyrion"
self.rec1 = {
self.recname1: {
"email": "*****@*****.**",
"telefon": "13432425253",
"website": "www.cptBoreale.dr"
}
}
self.rec2 = {
self.recname2: {
"alias": "The Exalted",
"email": "*****@*****.**",
"adres": "The Covenant"
}
}
self.rec3 = {
self.recname3: {
"fax": "123221",
"affiliation": "slaanesh",
"email": "*****@*****.**"
}
}
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 populate_db(self):
db = DataBase(name=self.name)
db.add_record(json.loads(open("Database/Data/BC1.json").read()))
db.add_record(json.loads(open("Database/Data/BC2.json").read()))
def __init__(self):
self.db = DataBase()