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()