class SyncManager(object):
"""
Synchronization manager. This initializes and stores settings and handles the Sync object.
"""
def __init__(self):
self.server_address = ""
self.username = ""
self.password = ""
self.certificate = ""
self.certificate_file = None
self.sync = None
def __del__(self):
if self.certificate_file:
self.certificate_file.close()
def get_binary_sync_settings(self):
"""
returns packed sync settings
:return: binary settings
:rtype: bytes
"""
if self.sync:
return Packer.compress(json.dumps({
"server-address": self.server_address,
"username": self.username,
"password": self.password,
"certificate": self.certificate
}).encode('utf-8'))
else:
return b''
def load_binary_sync_settings(self, data):
"""
loads sync settings
:param bytes data: packed json data of sync settings
"""
settings_dict = json.loads(str(Packer.decompress(data), encoding='utf-8'))
if "server-address" in settings_dict and \
"username" in settings_dict and \
"password" in settings_dict and \
"certificate" in settings_dict:
self.server_address = settings_dict["server-address"]
self.username = settings_dict["username"]
self.password = settings_dict["password"]
self.certificate = settings_dict["certificate"]
if self.certificate_file:
self.certificate_file.close()
self.certificate_file = NamedTemporaryFile()
self.certificate_file.write(self.certificate.encode('utf-8'))
self.certificate_file.seek(0)
self.create_sync()
def ask_for_sync_settings(self):
"""
Ask the user for sync settings: Asks for server-URL, username and password.
"""
print("Bitte geben Sie die Einstellungen für Ihren Synchronisations-Server an...")
self.server_address = input("URL: ")
self.username = input("Benutzername: ")
self.password = input("Passwort: ")
line = input("Zertifikat im .pem-Format (beenden mit einer Leerzeile): ")
while len(line) > 0:
self.certificate += line + "\n"
line = input("")
self.certificate += line
if self.certificate_file:
self.certificate_file.close()
self.certificate_file = NamedTemporaryFile()
self.certificate_file.write(self.certificate.encode('utf-8'))
self.certificate_file.seek(0)
self.create_sync()
def set_server_address(self, url):
"""
Sets the url without ajax folder and php file names but with https://
:param url: the url
:type url: str
"""
self.server_address = url
def set_username(self, username):
"""
Sets the username.
:param username: the username
:type username: str
"""
self.username = username
def set_password(self, password):
"""
Sets the password.
:param password: the password
:type password: str
"""
self.password = password
def set_certificate(self, certificate):
"""
Sets the certificate from a string in PEM format.
:param certificate: certificate in PEM format
:type certificate: str
"""
self.certificate = certificate
if self.certificate_file:
self.certificate_file.close()
self.certificate_file = NamedTemporaryFile()
self.certificate_file.write(self.certificate.encode('utf-8'))
self.certificate_file.seek(0)
def create_sync(self):
"""
creates a sync object.
"""
self.sync = Sync(self.server_address, self.username, self.password, self.certificate_file.name)
def has_settings(self):
"""
Returns true if pull or push are possible
:return: Are there settings?
:rtype: bool
"""
return bool(self.sync)
def pull(self):
"""
pulls data from the sync server. Returns an empty string if no connection is possible.
:return: pulled base64 data
:rtype: str
"""
if self.sync:
return self.sync.pull()
else:
return False, ''
def push(self, data):
"""
pushes data to the sync server. If the push fails an error message is displayed.
:param str data: base64 data
"""
if self.sync:
if not self.sync.push(data):
print("Synchronisation fehlgeschlagen.")
else:
print("Sie haben keine gültigen Einstellungen für den sync server.")
def test_pull(self):
sync = Sync("https://ersatzworld.net/ctpwdgen-server/", 'inter', 'op',
'file.pem')
status, blob = sync.pull()
self.assertTrue(status)
self.assertEqual(str(b64encode(b'Test'), encoding='utf-8'), blob)
def test_pull(self):
sync = Sync("https://ersatzworld.net/ctpwdgen-server/", 'inter', 'op', 'file.pem')
status, blob = sync.pull()
self.assertTrue(status)
self.assertEqual(str(b64encode(b'Test'), encoding='utf-8'), blob)
def test_pull_empty_request(self):
sync = Sync("https://ersatzworld.net/ctpwdgen-server/", 'inter', 'op',
'file.pem')
status, blob = sync.pull()
self.assertTrue(status)
self.assertEqual('', blob)
def test_pull_empty_request(self):
sync = Sync("https://ersatzworld.net/ctpwdgen-server/", 'inter', 'op', 'file.pem')
status, blob = sync.pull()
self.assertTrue(status)
self.assertEqual('', blob)
