232 lines
7.3 KiB
Python
232 lines
7.3 KiB
Python
from platform import system
|
|
import json
|
|
import os
|
|
import locale
|
|
from util import Singleton, curr_directory, log
|
|
import pyaudio
|
|
from toxencryptsave import LibToxEncryptSave
|
|
|
|
|
|
class Settings(Singleton, dict):
|
|
"""
|
|
Settings of current profile + global app settings
|
|
"""
|
|
|
|
def __init__(self, name):
|
|
self.path = ProfileHelper.get_path() + str(name) + '.json'
|
|
self.name = name
|
|
if os.path.isfile(self.path):
|
|
with open(self.path) as fl:
|
|
data = fl.read()
|
|
inst = LibToxEncryptSave.get_instance()
|
|
if inst.has_password():
|
|
data = inst.pass_decrypt(data)
|
|
try:
|
|
info = json.loads(data)
|
|
except Exception as ex:
|
|
info = Settings.get_default_settings()
|
|
log('Parsing settings error: ' + str(ex))
|
|
super(self.__class__, self).__init__(info)
|
|
self.upgrade()
|
|
else:
|
|
super(self.__class__, self).__init__(Settings.get_default_settings())
|
|
self.save()
|
|
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 unicode(auto['path']), unicode(auto['name'])
|
|
|
|
@staticmethod
|
|
def set_auto_profile(path, name):
|
|
p = Settings.get_default_path() + 'toxygen.json'
|
|
data = json.dumps({'path': unicode(path.decode(locale.getpreferredencoding())), 'name': unicode(name)})
|
|
with open(p, 'w') as fl:
|
|
fl.write(data)
|
|
|
|
@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': []
|
|
}
|
|
|
|
@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 = inst.pass_encrypt(text)
|
|
with open(self.path, 'w') 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(unicode(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(unicode(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):
|
|
path = path.decode(locale.getpreferredencoding())
|
|
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 is_active_profile(path, name):
|
|
path = path.decode(locale.getpreferredencoding()) + 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_path():
|
|
return ProfileHelper.get_instance().get_dir()
|