76 lines
2.6 KiB
Python
76 lines
2.6 KiB
Python
|
import plugin_super_class
|
||
|
|
import json
|
||
|
|
import settings
|
||
|
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
class AvatarEncryption(plugin_super_class.PluginSuperClass):
|
||
|
|
|
||
|
|
def __init__(self, *args):
|
||
|
|
super(AvatarEncryption, self).__init__('AvatarEncryption', 'ae', *args)
|
||
|
|
self._path = settings.ProfileHelper.get_path() + 'avatars/'
|
||
|
|
self._contacts = self._profile._contacts[:]
|
||
|
|
|
||
|
|
def close(self):
|
||
|
|
if not self._encrypt_save.has_password():
|
||
|
|
return
|
||
|
|
i, data = 1, {}
|
||
|
|
|
||
|
|
self.save_contact_avatar(data, self._profile, 0)
|
||
|
|
for friend in self._contacts:
|
||
|
|
self.save_contact_avatar(data, friend, i)
|
||
|
|
i += 1
|
||
|
|
self.save_settings(json.dumps(data))
|
||
|
|
|
||
|
|
def start(self):
|
||
|
|
if not self._encrypt_save.has_password():
|
||
|
|
return
|
||
|
|
data = json.loads(self.load_settings())
|
||
|
|
|
||
|
|
self.load_contact_avatar(data, self._profile)
|
||
|
|
for friend in self._contacts:
|
||
|
|
self.load_contact_avatar(data, friend)
|
||
|
|
self._profile.update()
|
||
|
|
|
||
|
|
def save_contact_avatar(self, data, contact, i):
|
||
|
|
tox_id = contact.tox_id[:64]
|
||
|
|
data[str(tox_id)] = str(i)
|
||
|
|
path = self._path + tox_id + '.png'
|
||
|
|
if os.path.isfile(path):
|
||
|
|
with open(path, 'rb') as fl:
|
||
|
|
avatar = fl.read()
|
||
|
|
encr_avatar = self._encrypt_save.pass_encrypt(avatar)
|
||
|
|
with open(self._path + self._settings.name + '_' + str(i) + '.png', 'wb') as fl:
|
||
|
|
fl.write(encr_avatar)
|
||
|
|
os.remove(path)
|
||
|
|
|
||
|
|
def load_contact_avatar(self, data, contact):
|
||
|
|
tox_id = str(contact.tox_id[:64])
|
||
|
|
if tox_id not in data:
|
||
|
|
return
|
||
|
|
path = self._path + self._settings.name + '_' + data[tox_id] + '.png'
|
||
|
|
if os.path.isfile(path):
|
||
|
|
with open(path, 'rb') as fl:
|
||
|
|
avatar = fl.read()
|
||
|
|
decr_avatar = self._encrypt_save.pass_decrypt(avatar)
|
||
|
|
with open(self._path + str(tox_id) + '.png', 'wb') as fl:
|
||
|
|
fl.write(decr_avatar)
|
||
|
|
os.remove(path)
|
||
|
|
contact.load_avatar()
|
||
|
|
|
||
|
|
def load_settings(self):
|
||
|
|
try:
|
||
|
|
with open(plugin_super_class.path_to_data(self._short_name) + self._settings.name + '.json', 'rb') as fl:
|
||
|
|
data = fl.read()
|
||
|
|
return str(self._encrypt_save.pass_decrypt(data), 'utf-8') if data else '{}'
|
||
|
|
except:
|
||
|
|
return '{}'
|
||
|
|
|
||
|
|
def save_settings(self, data):
|
||
|
|
try:
|
||
|
|
data = self._encrypt_save.pass_encrypt(bytes(data, 'utf-8'))
|
||
|
|
with open(plugin_super_class.path_to_data(self._short_name) + self._settings.name + '.json', 'wb') as fl:
|
||
|
|
fl.write(data)
|
||
|
|
except:
|
||
|
|
pass
|