def modify_site(site_path):
logger.info(site_path)
site_path = site_path.replace('\\', '/')
if os.path.isdir(site_path):
site_path = os.path.join(site_path, 'wizard_site')
if not os.path.isdir(site_path):
os.makedirs(site_path)
data_path = os.path.join(site_path, 'Data')
logger.info(data_path)
if not os.path.isdir(data_path):
os.makedirs(data_path)
avatars_path = os.path.join(data_path, 'avatars')
logger.info(avatars_path)
if not os.path.isdir(avatars_path):
os.makedirs(avatars_path)
os.environ[defaults._wizard_site_] = site_path
site_file = os.path.join(defaults._user_path_, 'site.wd')
utils.database().write(0, site_file, site_path)
subprocess.Popen('setx {} {}'.format(defaults._wizard_site_, site_path))
else:
logger.warning("{} doesn't exists".format(site_path))
def write_wall_file(self, settings, new=0):
try:
util.database().write(2, self.wall, settings)
if not new:
logger.debug('{} file updated'.format(defaults._wall_))
except:
logger.error(sys.exc_info()[1])
def write_file(self, settings, new=0):
try:
util.database().write(2, self.file, settings)
if not new:
logger.debug('{} file updated'.format(self.user + '.wd'))
except:
logger.error(sys.exc_info()[1])
def save(self, file, prefs=1):
data = self.ui.file_viewer_plainTextEdit.text().decode('utf-8')
if prefs:
utils.database().write(0, file, data, 1)
else:
with open(file, 'w') as f:
f.write(data)
self.file_modified(modified=0)
def write_chat_file(self, new=0):
try:
util.database().write(2, self.chat, self.history)
self.dump_count = 0
if not new:
logger.debug('{} file updated'.format(defaults._chat_))
except:
logger.error(sys.exc_info()[1])
def read_project():
project_tree = get_project_tree()
project_folder = get_project_folder()
if util.database().isfile(2, project_tree):
project = util.database().read(2, project_tree)
return project
elif not util.database().isfile(2, project_tree) and \
os.path.isdir(project_folder):
check_tree_file()
logger.debug(
'asset.wd file does not exists at {}, building it...'.format(
project_folder))
return {}
def __init__(self, asset, in_out=None):
self.database = util.database()
self.asset = asset
self.in_out = in_out
self.path = folder(self.asset).name
self.file = os.path.join(self.path, defaults._name_prefs_)
self.settings = self.read_settings()
def check_tree_file():
project_tree = get_project_tree()
if project_tree and not util.database().isfile(2, project_tree):
# Build base assets dictionnary
project = dict()
project[defaults._assets_] = dict()
project[defaults._sequences_] = dict()
project[defaults._library_] = dict()
project[defaults._editing_] = dict()
project[defaults._assets_][defaults._characters_] = dict()
project[defaults._assets_][defaults._props_] = dict()
project[defaults._assets_][defaults._sets_] = dict()
project[defaults._assets_][defaults._set_dress_] = dict()
project[defaults._assets_][defaults._vehicles_] = dict()
for lib in defaults._lib_categories_list_:
project[defaults._library_][lib] = dict()
for edit in defaults._editing_categories_list_:
project[defaults._editing_][edit] = dict()
# Write the file to the project path
write_project(project)
logger.debug('asset.wd file created at {}'.format(
get_project_folder()))
return project
else:
return 1
def __init__(self, asset):
self.database = util.database()
self.asset = asset
self.path = folder(self.asset).software
self.file = os.path.join(self.path, defaults._software_prefs_)
self.read_settings()
self.get_version()
def open(self):
try:
if not os.path.isfile(self.tickets_file):
self.create_tickets_file()
else:
self.settings = utils.database().read(0, self.tickets_file)
except:
logger.critical(str(traceback.format_exc()))
def __init__(self, asset, version=None):
self.database = util.database()
self.asset = asset
self.path = folder(self.asset).playblast()
self.file = os.path.join(self.path, defaults._playblast_prefs_)
self.read_settings()
self.version = self.get_version()
def read_chat_file(self):
try:
if self.is_archives():
self.history = util.database().read(2, self.chat)
else:
logger.warning("Chat file doesn't exist...")
except:
logger.error(sys.exc_info()[1])
def __init__(self, asset):
self.database = util.database()
self.asset = asset
self.path = folder(self.asset).export()
self.file = os.path.join(self.path, defaults._export_prefs_)
self.read_settings()
self.get_version()
self.version_path = folder(self.asset).export(
self.asset.export_version)
def open_pref_file(self):
try:
if self.is_prefs():
self.settings = util.database().read(1, defaults._user_)
else:
logger.warning(
"Prefs file doesn't exist : {}, creating it...".format(
defaults._user_))
except:
logger.error(sys.exc_info()[1])
def open_wall_file(self):
try:
if self.is_wall():
settings = util.database().read(2, self.wall)
return settings
else:
logger.warning("Wall file doesn't exist...")
except:
logger.error(sys.exc_info()[1])
def __init__(self):
self.database = utils.database()
self.init_shared_folder()
self.users = prefs.project_users
if self.is_file():
self.read()
else:
self.init_settings()
def open_file(self):
try:
if self.is_file():
settings = util.database().read(2, self.file)
return settings
else:
logger.warning("{} file doesn't exist...".format(self.user +
'.wd'))
except:
logger.error(sys.exc_info()[1])
def __init__(self, user_name=None, current_project=None):
self.database = util.database()
if not self.is_prefs():
# Create pref path if doesn't exists
if not os.path.exists(defaults._user_path_):
os.makedirs(defaults._user_path_)
# Create settings dictionnary
self.settings = dict()
self.settings[defaults._user_name_key_] = user_name
self.settings[defaults._asset_context_] = None
self.settings[defaults._popup_prefs_key_] = dict()
self.settings[defaults._popup_prefs_key_][
defaults._popup_enable_key_] = 1
self.settings[defaults._popup_prefs_key_][
defaults._popup_sound_key_] = 1
self.settings[defaults._popup_prefs_key_][
defaults._popup_sound_file_key_] = defaults._woo_key_
self.settings[defaults._popup_prefs_key_][
defaults._popup_duration_key_] = 3
self.settings[defaults._popup_prefs_key_][
defaults._popup_creation_key_] = 1
self.settings[defaults._popup_prefs_key_][
defaults._popup_publish_key_] = 1
self.settings[defaults._popup_prefs_key_][
defaults._popup_playblast_key_] = 1
self.settings[defaults._popup_prefs_key_][
defaults._popup_save_key_] = 1
self.settings[defaults._popup_prefs_key_][
defaults._popup_message_key_] = 1
self.settings[defaults._popup_prefs_key_][
defaults._popup_position_key_] = defaults._bt_r_key_
self.settings[defaults._current_project_key_] = current_project
self.settings[
defaults._user_theme_key_] = defaults._dark_theme_key_
self.settings[
defaults._user_chat_theme_key_] = defaults._purple_color_
self.settings[defaults._user_screen_index_key_] = 0
self.settings[defaults._windows_quit_key_] = 0
self.settings[defaults._show_updates_] = 1
self.settings[defaults._last_updates_] = "0.9.1"
self.settings[defaults._script_cache_] = ""
self.settings[defaults._show_new_version_] = 1
self.settings[defaults._show_error_handler_] = 0
self.settings[defaults._local_project_path_] = ''
self.settings[defaults._shutter_] = 0
self.settings[defaults._server_ip_] = None
self.settings[defaults._tab_context_] = 0
self.settings[defaults._chat_seen_] = dict()
# Write the .manager file as YAML with setting dict
logger.debug('user.wd file created')
self.write_pref_file(new=1)
else:
self.open_pref_file()
def is_site(): site_file = os.path.join(defaults._user_path_, 'site.wd') if os.path.isfile(site_file): site = utils.database().read(0, site_file) if site == 'null': return 0 else: if os.path.isdir(site): return 1 else: return 0 else: return 0
def __init__(self, user_name=None):
self.database = util.database()
if not user_name:
self.user_name = user().get_user()
else:
self.user_name = user_name
if self.user_name:
self.stats_file = os.path.join(os.environ[defaults._stats_var_],
(self.user_name + '.wd'))
if not self.is_file():
self.create_user()
else:
self.read()
def __init__(
self,
software=None,
):
self.database = util.database()
self.site = site()
self.user = user()
# Check if the given software is valid
if not software or software not in defaults._softwares_list_:
logger.warning('Please give a default software...')
else:
self.software = software
# Read the software prefs
self.settings = self.read_prefs()
def __init__(self):
self.database = util.database()
if not self.is_prefs():
# Create a basic settings dict
self.settings = dict()
self.settings[defaults._users_list_key_] = dict()
self.settings[defaults._projects_list_key_] = dict()
# Create the preferences file
self.write_pref_file(new=1)
self.create_user(add_user_name='admin',
promotion=defaults._staff_,
email='*****@*****.**',
password='******',
admin=1,
full_name='Administrator')
logger.debug('site.wd file created')
else:
self.open_pref_file()
def read_file(self, filepath):
extension = filepath.split('.')[-1]
if extension == 'wd':
self.ui.file_viewer_plainTextEdit.set_lexer(python=0, yaml=1)
self.ui.file_viewer_plainTextEdit.setText(utils.database().read(
0, filepath, 1).decode('utf-8'))
self.file = filepath
self.update_file()
else:
with open(filepath, 'r') as f:
data = f.read()
if os.path.splitext(filepath)[-1] == '.py':
self.ui.file_viewer_plainTextEdit.set_lexer(python=1, yaml=0)
else:
self.ui.file_viewer_plainTextEdit.set_lexer(python=0, yaml=0)
self.ui.file_viewer_plainTextEdit.setText(data)
self.file = filepath
self.update_file()
def __init__(self, project_path, user):
self.database = util.database()
self.user = user
self.project_path = project_path
if project_path:
self.file = os.path.join(project_path, self.user + '.wd')
if not self.is_file():
# Create settings dictionnary
settings = dict()
settings[defaults._events_] = []
settings[defaults._locks_] = []
# Write the .manager file as YAML with setting dict
logger.info('{} file created'.format(self.user + '.wd'))
self.write_file(settings, new=1)
else:
self.file = None
def write_pref_file(self, new=0):
try:
util.database().write(1, defaults._user_, self.settings)
logger.debug('user.wd file updated')
except:
logger.error(sys.exc_info()[1])
def __init__(self, asset):
self.database = util.database()
self.asset = asset
self.path = folder(self.asset).category
self.file = os.path.join(self.path, defaults._category_prefs_)
self.settings = self.read_settings()
def is_wall(self):
if self.wall:
return util.database().isfile(2, self.wall)
else:
return None
import sys
import os
import fix_qt
# external modules
from PyQt5 import QtWidgets, QtGui, QtCore
from PyQt5.QtCore import QThread, pyqtSignal
from PyQt5.QtWidgets import QApplication
from PyQt5.QtCore import QTimer
# wizard modules
import wizard.vars.defaults as defaults
from wizard.tools import utility as utils
# building local environment
main_site_path = utils.database().read(
0, os.path.join(defaults._user_path_, 'site.wd'))
site_path = os.path.join(main_site_path, defaults._site_)
stats_path = os.path.join(main_site_path, defaults._stats_)
os.environ[defaults._wizard_site_] = main_site_path
os.environ[defaults._site_var_] = site_path
os.environ[defaults._abs_site_path_] = os.path.abspath('')
os.environ[defaults._stats_var_] = stats_path
os.environ[defaults._project_db_env_] = ''
os.environ[defaults._current_assets_list_] = ''
# wizard modules
from wizard.prefs import project as project_prefs
import gui.build as build
from wizard.tools import log
from wizard.prefs.main import prefs
def is_prefs(self):
return util.database().isfile(1, defaults._user_)
def __init__(self):
self.database = util.database()
self.jokes_file = os.path.join(os.environ[defaults._wizard_site_],
defaults._jokes_)
self.read()
