from platform import system import json import os import locale from util import Singleton, curr_directory, log import pyaudio from toxencryptsave import LibToxEncryptSave import smileys class Settings(dict, Singleton): """ Settings of current profile + global app settings """ def __init__(self, name): Singleton.__init__(self) self.path = ProfileHelper.get_path() + str(name) + '.json' self.name = name if os.path.isfile(self.path): with open(self.path, 'rb') as fl: data = fl.read() inst = LibToxEncryptSave.get_instance() try: if inst.is_data_encrypted(data): data = inst.pass_decrypt(data) info = json.loads(str(data, 'utf-8')) except Exception as ex: info = Settings.get_default_settings() log('Parsing settings error: ' + str(ex)) super(Settings, self).__init__(info) self.upgrade() else: super(Settings, self).__init__(Settings.get_default_settings()) self.save() smileys.SmileyLoader(self) p = pyaudio.PyAudio() self.audio = {'input': p.get_default_input_device_info()['index'], 'output': p.get_default_output_device_info()['index']} @staticmethod def get_auto_profile(): path = Settings.get_default_path() + 'toxygen.json' if os.path.isfile(path): with open(path) as fl: data = fl.read() auto = json.loads(data) if 'path' in auto and 'name' in auto: return str(auto['path']), str(auto['name']) return '', '' @staticmethod def set_auto_profile(path, name): p = Settings.get_default_path() + 'toxygen.json' with open(p) as fl: data = fl.read() data = json.loads(data) data['path'] = str(path) data['name'] = str(name) with open(p, 'w') as fl: fl.write(json.dumps(data)) @staticmethod def reset_auto_profile(): p = Settings.get_default_path() + 'toxygen.json' with open(p) as fl: data = fl.read() data = json.loads(data) if 'path' in data: del data['path'] del data['name'] with open(p, 'w') as fl: fl.write(json.dumps(data)) @staticmethod def is_active_profile(path, name): path = path + name + '.tox' settings = Settings.get_default_path() + 'toxygen.json' if os.path.isfile(settings): with open(settings) as fl: data = fl.read() data = json.loads(data) if 'active_profile' in data: return path in data['active_profile'] return False @staticmethod def get_default_settings(): """ Default profile settings """ return { 'theme': 'default', 'ipv6_enabled': True, 'udp_enabled': True, 'proxy_type': 0, 'proxy_host': '0', 'proxy_port': 0, 'start_port': 0, 'end_port': 0, 'tcp_port': 0, 'notifications': True, 'sound_notifications': False, 'language': 'English', 'save_history': False, 'allow_inline': True, 'allow_auto_accept': True, 'auto_accept_path': None, 'show_online_friends': False, 'auto_accept_from_friends': [], 'friends_aliases': [], 'typing_notifications': False, 'calls_sound': True, 'blocked': [], 'plugins': [], 'notes': {}, 'smileys': True, 'smiley_pack': 'default', 'mirror_mode': False, 'width': 920, 'height': 500, 'x': 400, 'y': 400, 'message_font_size': 14, 'unread_color': 'red', 'save_unsent_only': False, 'compact_mode': False } @staticmethod def supported_languages(): return { 'English': 'en_EN', 'Russian': 'ru_RU', 'French': 'fr_FR' } def upgrade(self): default = Settings.get_default_settings() for key in default: if key not in self: print(key) self[key] = default[key] self.save() def save(self): text = json.dumps(self) inst = LibToxEncryptSave.get_instance() if inst.has_password(): text = bytes(inst.pass_encrypt(bytes(text, 'utf-8'))) else: text = bytes(text, 'utf-8') with open(self.path, 'wb') as fl: fl.write(text) def close(self): path = Settings.get_default_path() + 'toxygen.json' if os.path.isfile(path): with open(path) as fl: data = fl.read() app_settings = json.loads(data) try: app_settings['active_profile'].remove(str(ProfileHelper.get_path() + self.name + '.tox')) except: pass data = json.dumps(app_settings) with open(path, 'w') as fl: fl.write(data) def set_active_profile(self): """ Mark current profile as active """ path = Settings.get_default_path() + 'toxygen.json' if os.path.isfile(path): with open(path) as fl: data = fl.read() app_settings = json.loads(data) else: app_settings = {} if 'active_profile' not in app_settings: app_settings['active_profile'] = [] profile_path = ProfileHelper.get_path() app_settings['active_profile'].append(str(profile_path + str(self.name) + '.tox')) data = json.dumps(app_settings) with open(path, 'w') as fl: fl.write(data) def export(self, path): text = json.dumps(self) with open(path + str(self.name) + '.json', 'w') as fl: fl.write(text) @staticmethod def get_default_path(): if system() == 'Linux': return os.getenv('HOME') + '/.config/tox/' elif system() == 'Windows': return os.getenv('APPDATA') + '/Tox/' class ProfileHelper(Singleton): """ Class with methods for search, load and save profiles """ def __init__(self, path, name): Singleton.__init__(self) self._path = path + name + '.tox' self._directory = path # create /avatars if not exists: directory = path + 'avatars' if not os.path.exists(directory): os.makedirs(directory) def open_profile(self): with open(self._path, 'rb') as fl: data = fl.read() if data: return data else: raise IOError('Save file has zero size!') def get_dir(self): return self._directory def save_profile(self, data): inst = LibToxEncryptSave.get_instance() if inst.has_password(): data = inst.pass_encrypt(data) with open(self._path, 'wb') as fl: fl.write(data) print('Profile saved successfully') def export_profile(self, new_path): new_path += os.path.basename(self._path) with open(self._path, 'rb') as fin: data = fin.read() with open(new_path, 'wb') as fout: fout.write(data) print('Profile exported successfully') @staticmethod def find_profiles(): """ Find available tox profiles """ path = Settings.get_default_path() result = [] # check default path if not os.path.exists(path): os.makedirs(path) for fl in os.listdir(path): if fl.endswith('.tox'): name = fl[:-4] result.append((path, name)) path = curr_directory() # check current directory for fl in os.listdir(path): if fl.endswith('.tox'): name = fl[:-4] result.append((path + '/', name)) return result @staticmethod def get_path(): return ProfileHelper.get_instance().get_dir()