class Key(BaseBackend):
name = "Key"
def __init__(self):
BaseBackend.__init__(self, models.Key)
self.config = Config()
self.encryption = Encryption(self.config["blacklist.keystore.psk"])
def get(self, *args, **kwargs):
(result, keydata) = BaseBackend.get(self, name=kwargs["name"])
if result:
keydata.data = self.encryption.decrypt(keydata.data)
return (result, keydata)
def add(self, *args, **kwargs):
if self.has(name=kwargs["name"]):
return (False, "Key already exists")
data = self.encryption.encrypt(kwargs["data"])
key = models.Key(
name=kwargs["name"],
data=data,
)
key.save()
return self.get(name=kwargs["name"])
class Broker:
def __init__(self):
self.config = Config()
self.context = zmq.Context()
self.broker = self.context.socket(zmq.REQ)
self.broker.connect(self.config["blacklist.broker"])
self.encryption = Encryption(self.config["blacklist.api.psk"])
def request_dispatcher(self, action):
data = {
"username": self.config["username"],
"password": self.config["password"],
"random": random.random()*time.mktime(datetime.datetime.timetuple(datetime.datetime.now()))
}
data["action"] = action
self.broker.send(self.encryption.encrypt(data))
return self.encryption.decrypt(self.broker.recv())
def update_listings(self):
return self.request_dispatcher("update_listings")
def update_whitelistings(self):
return self.request_dispatcher("update_whitelistings")
def update_rules(self):
return self.request_dispatcher("update_rules")
class Broker:
def __init__(self):
self.config = Config()
self.context = zmq.Context()
self.broker = self.context.socket(zmq.REQ)
self.broker.connect(self.config["blacklist.broker"])
self.encryption = Encryption(self.config["blacklist.api.psk"])
def request_dispatcher(self, action):
data = {
"username":
self.config["username"],
"password":
self.config["password"],
"random":
random.random() *
time.mktime(datetime.datetime.timetuple(datetime.datetime.now()))
}
data["action"] = action
self.broker.send(self.encryption.encrypt(data))
return self.encryption.decrypt(self.broker.recv())
def update_listings(self):
return self.request_dispatcher("update_listings")
def update_whitelistings(self):
return self.request_dispatcher("update_whitelistings")
def update_rules(self):
return self.request_dispatcher("update_rules")
class Key(BaseBackend): name = "Key" def __init__(self): BaseBackend.__init__(self, models.Key) self.config = Config() self.encryption = Encryption(self.config["blacklist.keystore.psk"]) def get(self, *args, **kwargs): (result, keydata) = BaseBackend.get(self, name=kwargs["name"]) if result: keydata.data = self.encryption.decrypt(keydata.data) return (result, keydata) def add(self, *args, **kwargs): if self.has(name=kwargs["name"]): return (False, "Key already exists") data = self.encryption.encrypt(kwargs["data"]) key = models.Key( name=kwargs["name"], data=data, ) key.save() return self.get(name=kwargs["name"])
