mirror of
https://github.com/ynput/ayon-core.git
synced 2025-12-25 13:24:54 +01:00
* General: Connect to AYON server (base) (#3924) * implemented 'get_workfile_info' in entities * removed 'prepare_asset_update_data' which is not used * disable settings and project manager if in v4 mode * prepared conversion helper functions for v4 entities * prepared conversion functions for hero versions * fix hero versions * implemented get_archived_representations * fix get latest versions * return prepared changes * handle archived representation * raise exception on failed json conversion * map archived to active properly * make sure default fields are added * fix conversion of hero version entity * fix conversion of archived representations * fix some conversions of representations and versions * changed active behavior in queries * fixed hero versions * implemented basic thumbnail caching * added raw variants of crud methods * implemented methods to get and create thumbnail * fix from flat dict * implemented some basic folder conversion for updates * fix thumbnail updates for version * implemented v4 thumbnail integrator * simplified data mapping * 'get_thumbnail' function also expect entity type and entity id for which is the thumbnail received * implemented 'get_thumbnail' for server * fix how thumbnail id is received from entity * removed unnecessary method 'get_thumbnail_id_from_source' * implemented thumbnail resolver for v4 * removed unnecessary print * move create and delete project directly to server api * disable local settings action too on v4 * OP-3521 - added method to check and download updated addons from v4 server * OP-3521 - added more descriptive error message for missing source * OP-3521 - added default implementation of addon downloader to import * OP-3521 - added check for dependency package zips WIP - server doesn't contain required endpoint. Testing only with mockup data for now. * OP-3521 - fixed parsing of DependencyItem Added Server Url type and ServerAddonDownloader - v4 server doesn't know its own DNS for static files so it is sending unique name and url must be created during runtime. * OP-3521 - fixed creation of targed directories * change nev keys to look for and don't set them automatically * fix task type conversion * implemented base of loading v4 addons in v3 * Refactored argument name in Downloaders * Updated parsing to DependencyItem according to current schema * Implemented downloading of package from server * Updated resolving of failures Uses Enum items. * Introduced passing of authorization token Better to inject it than to have it from env var. * Remove weird parsing of server_url Not necessary, endpoints have same prefix. * Fix doubling asset version name in addons folder Zip file should already contain `addonName_addonVersion` as first subfolder * Fix doubling asset version name in addons folder Zip file should already contain `addonName_addonVersion` as first subfolder * Made server_endpoint optional Argument should be better for testing, but for calling from separate methods it would be better to encapsulate it. Removed unwanted temporary productionPackage value * Use existing method to pull addon info from Server to load v4 version of addon * Raise exception when server doesn't have any production dependency package * added ability to specify v3 alias of addon name * expect v3_alias as uppered constant * Re-implemented method to get addon info Previous implementation wouldn't work in Python2 hosts. Will be refactored in the future. * fix '__getattr__' * added ayon api to pyproject.toml and lock file * use ayon api in common connection * added mapping for label * use ayon_api in client codebase * separated clearing cache of url and username * bump ayon api version * rename env 'OP4_TEST' to 'USE_AYON_SERVER' * Move and renamend get_addons_info to get_addons_info_as_dict in addon_distribution Should be moved to ayon_api later * Replaced requests calls with ayon_api * Replaced OP4_TEST_ENABLED with AYON_SERVER_ENABLED fixed endpoints * Hound * Hound * OP-3521 - fix wrong key in get_representation_parents parents overloads parents * OP-3521 - changes for v4 of SiteSync addon * OP-3521 - fix names * OP-3521 - remove storing project_name It should be safer to go thorug self.dbcon apparently * OP-3521 - remove unwanted "context["folder"]" can be only in dummy test data * OP-3521 - move site sync loaders to addon * Use only project instead of self.project * OP-3521 - added missed get_progress_for_repre * base of settings conversion script * simplified ayon functions in start.py * added loading of settings from ayon server * added a note about colors * fix global and local settings functions * AvalonMongoDB is not using mongo connection on ayon server enabled * 'get_dynamic_modules_dirs' is not checking system settings for paths in setting * log viewer is disabled when ayon server is enabled * basic logic of enabling/disabled addons * don't use mongo logging if ayon server is enabled * update ayon api * bump ayon api again * use ayon_api to get addons info in modules/base * update ayon api * moved helper functions to get addons and dependencies dir to common functions * Initialization of AddonInfo is not crashing on unkonwn sources * renamed 'DependencyDownloader' to 'AyonServerDownloader' * renamed function 'default_addon_downloader' to 'get_default_addon_downloader' * Added ability to convert 'WebAddonSource' to 'ServerResourceSorce' * missing dependency package on server won't cause crash * data sent to downloaders don't contain ayon specific headers * modified addon distribution to not duplicate 'ayon_api' functionality * fix doubled function defintioin * unzip client file to addon destination * formatting - unify quotes * disable usage of mongo connection if in ayon mode * renamed window.py to login_window.py * added webpublisher settings conversion * added maya conversion function * reuse variable * reuse variable (similar to previous commit) * fix ayon addons loading * fix typo 'AyonSettingsCahe' -> 'AyonSettingsCache' * fix enabled state changes * fix rr_path in royal render conversion * avoid mongo calls in AYON state * implemented custom AYON start script * fix formatting (after black) * ayon_start cleanup * 'get_addons_dir' and 'get_dependencies_dir' store value to environment variable * add docstrings to local dir functions * addon info has full name * fix modules enabled states * removed unused 'run_disk_mapping_commands' * removed ayon logic from 'start.py' * fix warning message * renamed 'openpype_common' to 'ayon_common' * removed unused import * don't import igniter * removed startup validations of third parties * change what's shown in version info * fix which keys are applied from ayon values * fix method name * get applications from attribs * Implemented UI basics to be able change user or logout * merged server.py and credentials.py * add more metadata to urls * implemented change token * implemented change user ui functionality * implemented change user ui * modify window to handle username and token value * pass username to add server * fix show UI cases * added loggin action to tray * update ayon api * added missing dependency * convert applications to config in a right way * initial implementation of 'nuke' settings conversion * removed few nuke comments * implemented hiero conversion * added imageio conversion * added run ayon tray script * fix few settings conversions * Renamed class of source classes as they are not just for addons * implemented objec to track source transfer progress * Implemented distribution item with multiple sources * Implemented ayon distribution wrapper to care about multiple things during distribution * added 'cleanup' method for downlaoders * download gets tranfer progress object * Change UploadState enum * added missing imports * use AyonDistribution in ayon_start.py * removed unused functions * removed implemented TODOs * fix import * fix key used for Web source * removed temp development fix * formatting fix * keep information if source require distribution * handle 'require_distribution' attribute in distribution process * added path attribute to server source * added option to pass addons infor to ayon distribution * fix tests * fix formatting * Fix typo * Fix typo * remove '_try_convert_to_server_source' * renamed attributes and methods to match their content * it is possible to pass dependency package info to AyonDistribution * fix called methods in tests * added public properties for error message and error detail * Added filename to WebSourceInfo Useful for GDrive sharable links where target file name is unknown/unparsable, it should be provided explicitly. * unify source conversion by adding 'convert_source' function * Fix error message Co-authored-by: Roy Nieterau <roy_nieterau@hotmail.com> * added docstring for 'transfer_progress' * don't create metadata file on read * added few docstrings * add default folder fields to folder/task queries * fix generators * add dependencies when runnign from code * add sys paths from distribution to pythonpath env * fix missing applications * added missing conversions for maya renderers * fix formatting * update ayon api * fix hashes in lock file * Use better exception Co-authored-by: Ondřej Samohel <33513211+antirotor@users.noreply.github.com> * Use Python 3 syntax Co-authored-by: Ondřej Samohel <33513211+antirotor@users.noreply.github.com> * apply some of sugested changes in ayon_start * added some docstrings and suggested modifications * copy create env from develop * fix rendersettings conversion * change code by suggestions * added missing args to docstring * added missing docstrings * separated downloader and download factory * fix ayon settings * added some basic file docstring to ayon_settings * join else conditions * fix project settings conversion * fix created at conversion * fix workfile info query * fix publisher UI * added utils function 'get_ayon_appdirs' * fix 'get_all_current_info' * fix server url assignment when url is set * updated ayon api * added utils functions to create local site id for ayon * added helper functions to create global connection * create global connection in ayon start to start use site id * use ayon site id in ayon mode * formatting cleanup * added header docstring * fixes after ayon_api update * load addons from ynput appdirs * fix function call * added docstring * update ayon pyton api * fix settings access * use ayon_api to get root overrides in Anatomy * bumbayon version to 0.1.13 * nuke: fixing settings keys from settings * fix burnins definitions * change v4 to AYON in thumbnail integrate * fix one more v4 information * Fixes after rebase * fix extract burnin conversion * additional fix of extract burnin * SiteSync:added missed loaders or v3 compatibility (#4587) * Added site sync loaders for v3 compatibility * Fix get_progress_for_repre * use 'files.name' instead of 'files.baseName' * update ayon api to 0.1.14 * add common to include files * change arguments for hero version creation * skip shotgrid settings conversion if different ayon addon is used * added ayon icons * fix labels of application variants * added option to show login window always on top * login window on invalid credentials is always on top * update ayon api * update ayon api * add entityType to project and folders * AYON: Editorial hierarchy creation (#4699) * disable extract hierarchy avalon when ayon mode is enabled * implemented extract hierarchy to AYON --------- Co-authored-by: Petr Kalis <petr.kalis@gmail.com> Co-authored-by: Roy Nieterau <roy_nieterau@hotmail.com> Co-authored-by: Ondřej Samohel <33513211+antirotor@users.noreply.github.com> Co-authored-by: Jakub Jezek <jakubjezek001@gmail.com> * replace 'legacy_io' with context functions in load plugins * added 'get_global_context' to pipeline init * use context getters instead of legacy_io in publish plugins * use data on context instead of 'legacy_io' in submit publish job * skip query of asset docs in collect nuke reads * use context functions on other places * 'list_looks' expects project name * remove 'get_context_title' * don't pass AvalonMongoDB to prelaunch hooks * change how context is calculated in hiero * implemented function 'get_fps_for_current_context' for maya * initialize '_image_dir' and '_image_prefixes' in init * legacy creator is using 'get_current_project_name' * fill docstrings * use context functions in workfile builders * hound fixes * 'create_workspace_mel' can expect project settings * swapped order of arguments * use information from instance/context data * Use self.project_name in workfiles tool Co-authored-by: Roy Nieterau <roy_nieterau@hotmail.com> * Remove outdated todo Co-authored-by: Roy Nieterau <roy_nieterau@hotmail.com> * don't query project document in nuke lib * Fix access to context data * Use right function to get project name Co-authored-by: Roy Nieterau <roy_nieterau@hotmail.com> * fix submit max deadline and swap order of arguments * added 'get_context_label' to nuke * fix import * fix typo 'curent_context' -> 'current_context' * fix project_setting variable * fix submit publish job environments * use task from context * Removed unused import --------- Co-authored-by: Petr Kalis <petr.kalis@gmail.com> Co-authored-by: Roy Nieterau <roy_nieterau@hotmail.com> Co-authored-by: Ondřej Samohel <33513211+antirotor@users.noreply.github.com> Co-authored-by: Jakub Jezek <jakubjezek001@gmail.com>
828 lines
28 KiB
Python
828 lines
28 KiB
Python
import os
|
|
import logging
|
|
import shutil
|
|
import copy
|
|
|
|
import qtpy
|
|
from qtpy import QtWidgets, QtCore
|
|
|
|
from openpype.host import IWorkfileHost
|
|
from openpype.client import get_asset_by_id
|
|
from openpype.pipeline.workfile.lock_workfile import (
|
|
is_workfile_locked,
|
|
is_workfile_lock_enabled,
|
|
is_workfile_locked_for_current_process
|
|
)
|
|
from openpype.tools.utils import PlaceholderLineEdit
|
|
from openpype.tools.utils.delegates import PrettyTimeDelegate
|
|
from openpype.lib import emit_event
|
|
from openpype.tools.workfiles.lock_dialog import WorkfileLockDialog
|
|
from openpype.pipeline import (
|
|
registered_host,
|
|
legacy_io,
|
|
Anatomy,
|
|
get_current_project_name,
|
|
)
|
|
from openpype.pipeline.context_tools import (
|
|
compute_session_changes,
|
|
change_current_context
|
|
)
|
|
from openpype.pipeline.workfile import (
|
|
get_workfile_template_key,
|
|
create_workdir_extra_folders,
|
|
)
|
|
|
|
from .model import (
|
|
WorkAreaFilesModel,
|
|
PublishFilesModel,
|
|
|
|
FILEPATH_ROLE,
|
|
DATE_MODIFIED_ROLE,
|
|
)
|
|
from .save_as_dialog import SaveAsDialog
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class FilesView(QtWidgets.QTreeView):
|
|
doubleClickedLeft = QtCore.Signal()
|
|
doubleClickedRight = QtCore.Signal()
|
|
|
|
def mouseDoubleClickEvent(self, event):
|
|
if event.button() == QtCore.Qt.LeftButton:
|
|
self.doubleClickedLeft.emit()
|
|
|
|
elif event.button() == QtCore.Qt.RightButton:
|
|
self.doubleClickedRight.emit()
|
|
|
|
return super(FilesView, self).mouseDoubleClickEvent(event)
|
|
|
|
|
|
class SelectContextOverlay(QtWidgets.QFrame):
|
|
def __init__(self, parent):
|
|
super(SelectContextOverlay, self).__init__(parent)
|
|
|
|
self.setObjectName("WorkfilesPublishedContextSelect")
|
|
label_widget = QtWidgets.QLabel(
|
|
"Please choose context on the left<br/><",
|
|
self
|
|
)
|
|
label_widget.setAlignment(QtCore.Qt.AlignCenter)
|
|
|
|
layout = QtWidgets.QHBoxLayout(self)
|
|
layout.addWidget(label_widget, 1, QtCore.Qt.AlignCenter)
|
|
|
|
label_widget.setAttribute(QtCore.Qt.WA_TranslucentBackground)
|
|
|
|
parent.installEventFilter(self)
|
|
|
|
def eventFilter(self, obj, event):
|
|
if event.type() == QtCore.QEvent.Resize:
|
|
self.resize(obj.size())
|
|
|
|
return super(SelectContextOverlay, self).eventFilter(obj, event)
|
|
|
|
|
|
class FilesWidget(QtWidgets.QWidget):
|
|
"""A widget displaying files that allows to save and open files."""
|
|
file_selected = QtCore.Signal(str)
|
|
file_opened = QtCore.Signal()
|
|
workfile_created = QtCore.Signal(str)
|
|
published_visible_changed = QtCore.Signal(bool)
|
|
|
|
def __init__(self, parent):
|
|
super(FilesWidget, self).__init__(parent)
|
|
|
|
# Setup
|
|
self._asset_id = None
|
|
self._asset_doc = None
|
|
self._task_name = None
|
|
self._task_type = None
|
|
|
|
# Pype's anatomy object for current project
|
|
project_name = get_current_project_name()
|
|
self.anatomy = Anatomy(project_name)
|
|
self.project_name = project_name
|
|
# Template key used to get work template from anatomy templates
|
|
self.template_key = "work"
|
|
|
|
# This is not root but workfile directory
|
|
self._workfiles_root = None
|
|
self._workdir_path = None
|
|
self.host = registered_host()
|
|
self.host_name = os.environ["AVALON_APP"]
|
|
|
|
# Whether to automatically select the latest modified
|
|
# file on a refresh of the files model.
|
|
self.auto_select_latest_modified = True
|
|
|
|
# Avoid crash in Blender and store the message box
|
|
# (setting parent doesn't work as it hides the message box)
|
|
self._messagebox = None
|
|
|
|
# Filtering input
|
|
filter_widget = QtWidgets.QWidget(self)
|
|
|
|
published_checkbox = QtWidgets.QCheckBox("Published", filter_widget)
|
|
|
|
filter_input = PlaceholderLineEdit(filter_widget)
|
|
filter_input.setPlaceholderText("Filter files..")
|
|
|
|
filter_layout = QtWidgets.QHBoxLayout(filter_widget)
|
|
filter_layout.setContentsMargins(0, 0, 0, 0)
|
|
filter_layout.addWidget(filter_input, 1)
|
|
filter_layout.addWidget(published_checkbox, 0)
|
|
|
|
# Create the Files models
|
|
extensions = set(self._get_host_extensions())
|
|
|
|
views_widget = QtWidgets.QWidget(self)
|
|
# --- Workarea view ---
|
|
workarea_files_model = WorkAreaFilesModel(extensions)
|
|
|
|
# Create proxy model for files to be able sort and filter
|
|
workarea_proxy_model = QtCore.QSortFilterProxyModel()
|
|
workarea_proxy_model.setSourceModel(workarea_files_model)
|
|
workarea_proxy_model.setDynamicSortFilter(True)
|
|
workarea_proxy_model.setSortCaseSensitivity(QtCore.Qt.CaseInsensitive)
|
|
|
|
# Set up the file list tree view
|
|
workarea_files_view = FilesView(views_widget)
|
|
workarea_files_view.setModel(workarea_proxy_model)
|
|
workarea_files_view.setSortingEnabled(True)
|
|
workarea_files_view.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
|
|
|
|
# Date modified delegate
|
|
workarea_time_delegate = PrettyTimeDelegate()
|
|
workarea_files_view.setItemDelegateForColumn(1, workarea_time_delegate)
|
|
# smaller indentation
|
|
workarea_files_view.setIndentation(3)
|
|
|
|
# Default to a wider first filename column it is what we mostly care
|
|
# about and the date modified is relatively small anyway.
|
|
workarea_files_view.setColumnWidth(0, 330)
|
|
|
|
# --- Publish files view ---
|
|
publish_files_model = PublishFilesModel(
|
|
extensions, legacy_io, self.anatomy
|
|
)
|
|
|
|
publish_proxy_model = QtCore.QSortFilterProxyModel()
|
|
publish_proxy_model.setSourceModel(publish_files_model)
|
|
publish_proxy_model.setDynamicSortFilter(True)
|
|
publish_proxy_model.setSortCaseSensitivity(QtCore.Qt.CaseInsensitive)
|
|
|
|
publish_files_view = FilesView(views_widget)
|
|
publish_files_view.setModel(publish_proxy_model)
|
|
|
|
publish_files_view.setSortingEnabled(True)
|
|
publish_files_view.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
|
|
|
|
# Date modified delegate
|
|
publish_time_delegate = PrettyTimeDelegate()
|
|
publish_files_view.setItemDelegateForColumn(1, publish_time_delegate)
|
|
# smaller indentation
|
|
publish_files_view.setIndentation(3)
|
|
|
|
# Default to a wider first filename column it is what we mostly care
|
|
# about and the date modified is relatively small anyway.
|
|
publish_files_view.setColumnWidth(0, 330)
|
|
|
|
publish_context_overlay = SelectContextOverlay(views_widget)
|
|
publish_context_overlay.setVisible(False)
|
|
|
|
views_layout = QtWidgets.QHBoxLayout(views_widget)
|
|
views_layout.setContentsMargins(0, 0, 0, 0)
|
|
views_layout.addWidget(workarea_files_view, 1)
|
|
views_layout.addWidget(publish_files_view, 1)
|
|
|
|
# Home Page
|
|
# Build buttons widget for files widget
|
|
btns_widget = QtWidgets.QWidget(self)
|
|
|
|
workarea_btns_widget = QtWidgets.QWidget(btns_widget)
|
|
btn_save = QtWidgets.QPushButton("Save As", workarea_btns_widget)
|
|
btn_browse = QtWidgets.QPushButton("Browse", workarea_btns_widget)
|
|
btn_open = QtWidgets.QPushButton("Open", workarea_btns_widget)
|
|
|
|
workarea_btns_layout = QtWidgets.QHBoxLayout(workarea_btns_widget)
|
|
workarea_btns_layout.setContentsMargins(0, 0, 0, 0)
|
|
workarea_btns_layout.addWidget(btn_open, 1)
|
|
workarea_btns_layout.addWidget(btn_browse, 1)
|
|
workarea_btns_layout.addWidget(btn_save, 1)
|
|
|
|
publish_btns_widget = QtWidgets.QWidget(btns_widget)
|
|
btn_save_as_published = QtWidgets.QPushButton(
|
|
"Copy && Open", publish_btns_widget
|
|
)
|
|
btn_change_context = QtWidgets.QPushButton(
|
|
"Choose different context", publish_btns_widget
|
|
)
|
|
btn_select_context_published = QtWidgets.QPushButton(
|
|
"Copy && Open", publish_btns_widget
|
|
)
|
|
btn_cancel_published = QtWidgets.QPushButton(
|
|
"Cancel", publish_btns_widget
|
|
)
|
|
|
|
publish_btns_layout = QtWidgets.QHBoxLayout(publish_btns_widget)
|
|
publish_btns_layout.setContentsMargins(0, 0, 0, 0)
|
|
publish_btns_layout.addWidget(btn_save_as_published, 1)
|
|
publish_btns_layout.addWidget(btn_change_context, 1)
|
|
publish_btns_layout.addWidget(btn_select_context_published, 1)
|
|
publish_btns_layout.addWidget(btn_cancel_published, 1)
|
|
|
|
btns_layout = QtWidgets.QHBoxLayout(btns_widget)
|
|
btns_layout.setContentsMargins(0, 0, 0, 0)
|
|
btns_layout.addWidget(workarea_btns_widget, 1)
|
|
btns_layout.addWidget(publish_btns_widget, 1)
|
|
|
|
# Build files widgets for home page
|
|
main_layout = QtWidgets.QVBoxLayout(self)
|
|
main_layout.setContentsMargins(0, 0, 0, 0)
|
|
main_layout.addWidget(filter_widget, 0)
|
|
main_layout.addWidget(views_widget, 1)
|
|
main_layout.addWidget(btns_widget, 0)
|
|
|
|
# Register signal callbacks
|
|
published_checkbox.stateChanged.connect(self._on_published_change)
|
|
filter_input.textChanged.connect(self._on_filter_text_change)
|
|
|
|
workarea_files_view.doubleClickedLeft.connect(
|
|
self._on_workarea_open_pressed
|
|
)
|
|
workarea_files_view.customContextMenuRequested.connect(
|
|
self._on_workarea_context_menu
|
|
)
|
|
workarea_files_view.selectionModel().selectionChanged.connect(
|
|
self.on_file_select
|
|
)
|
|
|
|
btn_open.pressed.connect(self._on_workarea_open_pressed)
|
|
btn_browse.pressed.connect(self.on_browse_pressed)
|
|
btn_save.pressed.connect(self._on_save_as_pressed)
|
|
btn_save_as_published.pressed.connect(
|
|
self._on_published_save_as_pressed
|
|
)
|
|
btn_change_context.pressed.connect(
|
|
self._on_publish_change_context_pressed
|
|
)
|
|
btn_select_context_published.pressed.connect(
|
|
self._on_publish_select_context_pressed
|
|
)
|
|
btn_cancel_published.pressed.connect(
|
|
self._on_publish_cancel_pressed
|
|
)
|
|
|
|
# Store attributes
|
|
self._published_checkbox = published_checkbox
|
|
self._filter_input = filter_input
|
|
|
|
self._workarea_time_delegate = workarea_time_delegate
|
|
self._workarea_files_view = workarea_files_view
|
|
self._workarea_files_model = workarea_files_model
|
|
self._workarea_proxy_model = workarea_proxy_model
|
|
|
|
self._publish_time_delegate = publish_time_delegate
|
|
self._publish_files_view = publish_files_view
|
|
self._publish_files_model = publish_files_model
|
|
self._publish_proxy_model = publish_proxy_model
|
|
|
|
self._publish_context_overlay = publish_context_overlay
|
|
|
|
self._workarea_btns_widget = workarea_btns_widget
|
|
self._publish_btns_widget = publish_btns_widget
|
|
self._btn_open = btn_open
|
|
self._btn_browse = btn_browse
|
|
self._btn_save = btn_save
|
|
|
|
self._btn_save_as_published = btn_save_as_published
|
|
self._btn_change_context = btn_change_context
|
|
self._btn_select_context_published = btn_select_context_published
|
|
self._btn_cancel_published = btn_cancel_published
|
|
|
|
# Create a proxy widget for files widget
|
|
self.setFocusProxy(btn_open)
|
|
|
|
# Hide publish files widgets
|
|
publish_files_view.setVisible(False)
|
|
publish_btns_widget.setVisible(False)
|
|
btn_select_context_published.setVisible(False)
|
|
btn_cancel_published.setVisible(False)
|
|
|
|
self._publish_context_select_mode = False
|
|
|
|
@property
|
|
def published_enabled(self):
|
|
return self._published_checkbox.isChecked()
|
|
|
|
def _on_published_change(self):
|
|
published_enabled = self.published_enabled
|
|
|
|
self._workarea_files_view.setVisible(not published_enabled)
|
|
self._workarea_btns_widget.setVisible(not published_enabled)
|
|
|
|
self._publish_files_view.setVisible(published_enabled)
|
|
self._publish_btns_widget.setVisible(published_enabled)
|
|
|
|
self._update_filtering()
|
|
self._update_asset_task()
|
|
|
|
self.published_visible_changed.emit(published_enabled)
|
|
|
|
self._select_last_modified_file()
|
|
|
|
def _on_filter_text_change(self):
|
|
self._update_filtering()
|
|
|
|
def _update_filtering(self):
|
|
text = self._filter_input.text()
|
|
if self.published_enabled:
|
|
self._publish_proxy_model.setFilterFixedString(text)
|
|
else:
|
|
self._workarea_proxy_model.setFilterFixedString(text)
|
|
|
|
def set_save_enabled(self, enabled):
|
|
self._btn_save.setEnabled(enabled)
|
|
if not enabled and self._published_checkbox.isChecked():
|
|
self._published_checkbox.setChecked(False)
|
|
self._published_checkbox.setVisible(enabled)
|
|
|
|
def set_asset_task(self, asset_id, task_name, task_type):
|
|
if asset_id != self._asset_id:
|
|
self._asset_doc = None
|
|
self._asset_id = asset_id
|
|
self._task_name = task_name
|
|
self._task_type = task_type
|
|
self._update_asset_task()
|
|
|
|
def _update_asset_task(self):
|
|
if self.published_enabled and not self._publish_context_select_mode:
|
|
self._publish_files_model.set_context(
|
|
self._asset_id, self._task_name
|
|
)
|
|
has_valid_items = self._publish_files_model.has_valid_items()
|
|
self._btn_save_as_published.setEnabled(has_valid_items)
|
|
self._btn_change_context.setEnabled(has_valid_items)
|
|
|
|
else:
|
|
# Define a custom session so we can query the work root
|
|
# for a "Work area" that is not our current Session.
|
|
# This way we can browse it even before we enter it.
|
|
if self._asset_id and self._task_name and self._task_type:
|
|
session = self._get_session()
|
|
self._workdir_path = session["AVALON_WORKDIR"]
|
|
self._workfiles_root = self.host.work_root(session)
|
|
self._workarea_files_model.set_root(self._workfiles_root)
|
|
|
|
else:
|
|
self._workarea_files_model.set_root(None)
|
|
|
|
# Disable/Enable buttons based on available files in model
|
|
has_valid_items = self._workarea_files_model.has_valid_items()
|
|
self._btn_browse.setEnabled(True)
|
|
self._btn_open.setEnabled(has_valid_items)
|
|
|
|
if self._publish_context_select_mode:
|
|
self._btn_select_context_published.setEnabled(
|
|
bool(self._asset_id) and bool(self._task_name)
|
|
)
|
|
return
|
|
|
|
# Manually trigger file selection
|
|
if not has_valid_items:
|
|
self.on_file_select()
|
|
|
|
def _get_asset_doc(self):
|
|
if self._asset_id is None:
|
|
return None
|
|
|
|
if self._asset_doc is None:
|
|
self._asset_doc = get_asset_by_id(
|
|
self.project_name, self._asset_id
|
|
)
|
|
|
|
return self._asset_doc
|
|
|
|
def _get_session(self):
|
|
"""Return a modified session for the current asset and task"""
|
|
|
|
session = legacy_io.Session.copy()
|
|
self.template_key = get_workfile_template_key(
|
|
self._task_type,
|
|
self.host_name,
|
|
project_name=self.project_name
|
|
)
|
|
changes = compute_session_changes(
|
|
session,
|
|
self._get_asset_doc(),
|
|
self._task_name,
|
|
template_key=self.template_key
|
|
)
|
|
session.update(changes)
|
|
|
|
return session
|
|
|
|
def _enter_session(self):
|
|
"""Enter the asset and task session currently selected"""
|
|
|
|
session = legacy_io.Session.copy()
|
|
changes = compute_session_changes(
|
|
session,
|
|
self._get_asset_doc(),
|
|
self._task_name,
|
|
template_key=self.template_key
|
|
)
|
|
if not changes:
|
|
# Return early if we're already in the right Session context
|
|
# to avoid any unwanted Task Changed callbacks to be triggered.
|
|
return
|
|
|
|
change_current_context(
|
|
self._get_asset_doc(),
|
|
self._task_name,
|
|
template_key=self.template_key
|
|
)
|
|
|
|
def _get_event_context_data(self):
|
|
asset_id = None
|
|
asset_name = None
|
|
asset_doc = self._get_asset_doc()
|
|
if asset_doc:
|
|
asset_id = asset_doc["_id"]
|
|
asset_name = asset_doc["name"]
|
|
return {
|
|
"project_name": self.project_name,
|
|
"asset_id": asset_id,
|
|
"asset_name": asset_name,
|
|
"task_name": self._task_name,
|
|
"host_name": self.host_name
|
|
}
|
|
|
|
def _is_workfile_locked(self, filepath):
|
|
if not is_workfile_lock_enabled(self.host_name, self.project_name):
|
|
return False
|
|
if not is_workfile_locked(filepath):
|
|
return False
|
|
return not is_workfile_locked_for_current_process(filepath)
|
|
|
|
def open_file(self, filepath):
|
|
host = self.host
|
|
if self._is_workfile_locked(filepath):
|
|
# add lockfile dialog
|
|
WorkfileLockDialog(filepath)
|
|
|
|
if isinstance(host, IWorkfileHost):
|
|
has_unsaved_changes = host.workfile_has_unsaved_changes()
|
|
else:
|
|
has_unsaved_changes = host.has_unsaved_changes()
|
|
|
|
if has_unsaved_changes:
|
|
result = self.save_changes_prompt()
|
|
if result is None:
|
|
# Cancel operation
|
|
return False
|
|
|
|
# Save first if has changes
|
|
if result:
|
|
if isinstance(host, IWorkfileHost):
|
|
current_file = host.get_current_workfile()
|
|
else:
|
|
current_file = host.current_file()
|
|
if not current_file:
|
|
# If the user requested to save the current scene
|
|
# we can't actually automatically do so if the current
|
|
# file has not been saved with a name yet. So we'll have
|
|
# to opt out.
|
|
log.error("Can't save scene with no filename. Please "
|
|
"first save your work file using 'Save As'.")
|
|
return
|
|
|
|
# Save current scene, continue to open file
|
|
if isinstance(host, IWorkfileHost):
|
|
host.save_workfile(current_file)
|
|
else:
|
|
host.save_file(current_file)
|
|
|
|
event_data_before = self._get_event_context_data()
|
|
event_data_before["filepath"] = filepath
|
|
event_data_after = copy.deepcopy(event_data_before)
|
|
emit_event(
|
|
"workfile.open.before",
|
|
event_data_before,
|
|
source="workfiles.tool"
|
|
)
|
|
self._enter_session()
|
|
if isinstance(host, IWorkfileHost):
|
|
host.open_workfile(filepath)
|
|
else:
|
|
host.open_file(filepath)
|
|
emit_event(
|
|
"workfile.open.after",
|
|
event_data_after,
|
|
source="workfiles.tool"
|
|
)
|
|
self.file_opened.emit()
|
|
|
|
def save_changes_prompt(self):
|
|
self._messagebox = messagebox = QtWidgets.QMessageBox(parent=self)
|
|
messagebox.setWindowFlags(
|
|
messagebox.windowFlags() | QtCore.Qt.FramelessWindowHint
|
|
)
|
|
messagebox.setIcon(QtWidgets.QMessageBox.Warning)
|
|
messagebox.setWindowTitle("Unsaved Changes!")
|
|
messagebox.setText(
|
|
"There are unsaved changes to the current file."
|
|
"\nDo you want to save the changes?"
|
|
)
|
|
messagebox.setStandardButtons(
|
|
QtWidgets.QMessageBox.Yes
|
|
| QtWidgets.QMessageBox.No
|
|
| QtWidgets.QMessageBox.Cancel
|
|
)
|
|
|
|
result = messagebox.exec_()
|
|
if result == QtWidgets.QMessageBox.Yes:
|
|
return True
|
|
if result == QtWidgets.QMessageBox.No:
|
|
return False
|
|
return None
|
|
|
|
def get_filename(self):
|
|
"""Show save dialog to define filename for save or duplicate
|
|
|
|
Returns:
|
|
str: The filename to create.
|
|
|
|
"""
|
|
session = self._get_session()
|
|
|
|
if self.published_enabled:
|
|
filepath = self._get_selected_filepath()
|
|
extensions = [os.path.splitext(filepath)[1]]
|
|
else:
|
|
extensions = self._get_host_extensions()
|
|
|
|
window = SaveAsDialog(
|
|
parent=self,
|
|
root=self._workfiles_root,
|
|
anatomy=self.anatomy,
|
|
template_key=self.template_key,
|
|
extensions=extensions,
|
|
session=session
|
|
)
|
|
window.exec_()
|
|
|
|
return window.get_result()
|
|
|
|
def on_duplicate_pressed(self):
|
|
work_file = self.get_filename()
|
|
if not work_file:
|
|
return
|
|
|
|
src = self._get_selected_filepath()
|
|
dst = os.path.join(self._workfiles_root, work_file)
|
|
shutil.copyfile(src, dst)
|
|
|
|
self.workfile_created.emit(dst)
|
|
|
|
self.refresh()
|
|
|
|
def _get_selected_filepath(self):
|
|
"""Return current filepath selected in view"""
|
|
if self.published_enabled:
|
|
source_view = self._publish_files_view
|
|
else:
|
|
source_view = self._workarea_files_view
|
|
selection = source_view.selectionModel()
|
|
index = selection.currentIndex()
|
|
if not index.isValid():
|
|
return
|
|
|
|
return index.data(FILEPATH_ROLE)
|
|
|
|
def _on_workarea_open_pressed(self):
|
|
path = self._get_selected_filepath()
|
|
if not path:
|
|
print("No file selected to open..")
|
|
return
|
|
|
|
self.open_file(path)
|
|
|
|
def _get_host_extensions(self):
|
|
if isinstance(self.host, IWorkfileHost):
|
|
return self.host.get_workfile_extensions()
|
|
return self.host.file_extensions()
|
|
|
|
def on_browse_pressed(self):
|
|
ext_filter = "Work File (*{0})".format(
|
|
" *".join(self._get_host_extensions())
|
|
)
|
|
dir_key = "directory"
|
|
if qtpy.API in ("pyside", "pyside2", "pyside6"):
|
|
dir_key = "dir"
|
|
|
|
workfile_root = self._workfiles_root
|
|
# Find existing directory of workfile root
|
|
# - Qt will use 'cwd' instead, if path does not exist, which may lead
|
|
# to igniter directory
|
|
while workfile_root:
|
|
if os.path.exists(workfile_root):
|
|
break
|
|
workfile_root = os.path.dirname(workfile_root)
|
|
|
|
kwargs = {
|
|
"caption": "Work Files",
|
|
"filter": ext_filter,
|
|
dir_key: workfile_root
|
|
}
|
|
|
|
work_file = QtWidgets.QFileDialog.getOpenFileName(**kwargs)[0]
|
|
if work_file:
|
|
self.open_file(work_file)
|
|
|
|
def _on_save_as_pressed(self):
|
|
self._save_as_with_dialog()
|
|
|
|
def _save_as_with_dialog(self):
|
|
work_filename = self.get_filename()
|
|
if not work_filename:
|
|
return None
|
|
|
|
src_path = self._get_selected_filepath()
|
|
|
|
# Trigger before save event
|
|
event_data_before = self._get_event_context_data()
|
|
event_data_before.update({
|
|
"filename": work_filename,
|
|
"workdir_path": self._workdir_path
|
|
})
|
|
emit_event(
|
|
"workfile.save.before",
|
|
event_data_before,
|
|
source="workfiles.tool"
|
|
)
|
|
|
|
# Make sure workfiles root is updated
|
|
# - this triggers 'workio.work_root(...)' which may change value of
|
|
# '_workfiles_root'
|
|
self.set_asset_task(
|
|
self._asset_id, self._task_name, self._task_type
|
|
)
|
|
|
|
# Create workfiles root folder
|
|
if not os.path.exists(self._workfiles_root):
|
|
log.debug("Initializing Work Directory: %s", self._workfiles_root)
|
|
os.makedirs(self._workfiles_root)
|
|
|
|
# Prepare full path to workfile and save it
|
|
filepath = os.path.join(
|
|
os.path.normpath(self._workfiles_root), work_filename
|
|
)
|
|
|
|
# Update session if context has changed
|
|
self._enter_session()
|
|
|
|
if not self.published_enabled:
|
|
if isinstance(self.host, IWorkfileHost):
|
|
self.host.save_workfile(filepath)
|
|
else:
|
|
self.host.save_file(filepath)
|
|
else:
|
|
shutil.copyfile(src_path, filepath)
|
|
if isinstance(self.host, IWorkfileHost):
|
|
self.host.open_workfile(filepath)
|
|
else:
|
|
self.host.open_file(filepath)
|
|
|
|
# Create extra folders
|
|
create_workdir_extra_folders(
|
|
self._workdir_path,
|
|
self.host_name,
|
|
self._task_type,
|
|
self._task_name,
|
|
self.project_name
|
|
)
|
|
event_data_after = self._get_event_context_data()
|
|
event_data_after.update({
|
|
"filename": work_filename,
|
|
"workdir_path": self._workdir_path
|
|
})
|
|
# Trigger after save events
|
|
emit_event(
|
|
"workfile.save.after",
|
|
event_data_after,
|
|
source="workfiles.tool"
|
|
)
|
|
|
|
self.workfile_created.emit(filepath)
|
|
# Refresh files model
|
|
if self.published_enabled:
|
|
self._published_checkbox.setChecked(False)
|
|
else:
|
|
self.refresh()
|
|
return filepath
|
|
|
|
def _on_published_save_as_pressed(self):
|
|
self._save_as_with_dialog()
|
|
|
|
def _set_publish_context_select_mode(self, enabled):
|
|
self._publish_context_select_mode = enabled
|
|
|
|
# Show buttons related to context selection
|
|
self._publish_context_overlay.setVisible(enabled)
|
|
self._btn_cancel_published.setVisible(enabled)
|
|
self._btn_select_context_published.setVisible(enabled)
|
|
# Change enabled state based on select context
|
|
self._btn_select_context_published.setEnabled(
|
|
bool(self._asset_id) and bool(self._task_name)
|
|
)
|
|
|
|
self._btn_save_as_published.setVisible(not enabled)
|
|
self._btn_change_context.setVisible(not enabled)
|
|
|
|
# Change views and disable workarea view if enabled
|
|
self._workarea_files_view.setEnabled(not enabled)
|
|
if self.published_enabled:
|
|
self._workarea_files_view.setVisible(enabled)
|
|
self._publish_files_view.setVisible(not enabled)
|
|
else:
|
|
self._workarea_files_view.setVisible(True)
|
|
self._publish_files_view.setVisible(False)
|
|
|
|
# Disable filter widgets
|
|
self._published_checkbox.setEnabled(not enabled)
|
|
self._filter_input.setEnabled(not enabled)
|
|
|
|
def _on_publish_change_context_pressed(self):
|
|
self._set_publish_context_select_mode(True)
|
|
|
|
def _on_publish_select_context_pressed(self):
|
|
result = self._save_as_with_dialog()
|
|
if result is not None:
|
|
self._set_publish_context_select_mode(False)
|
|
self._update_asset_task()
|
|
|
|
def _on_publish_cancel_pressed(self):
|
|
self._set_publish_context_select_mode(False)
|
|
self._update_asset_task()
|
|
|
|
def on_file_select(self):
|
|
self.file_selected.emit(self._get_selected_filepath())
|
|
|
|
def refresh(self):
|
|
"""Refresh listed files for current selection in the interface"""
|
|
if self.published_enabled:
|
|
self._publish_files_model.refresh()
|
|
else:
|
|
self._workarea_files_model.refresh()
|
|
|
|
if self.auto_select_latest_modified:
|
|
self._select_last_modified_file()
|
|
|
|
def _on_workarea_context_menu(self, point):
|
|
index = self._workarea_files_view.indexAt(point)
|
|
if not index.isValid():
|
|
return
|
|
|
|
if not index.flags() & QtCore.Qt.ItemIsEnabled:
|
|
return
|
|
|
|
menu = QtWidgets.QMenu(self)
|
|
|
|
# Duplicate
|
|
action = QtWidgets.QAction("Duplicate", menu)
|
|
tip = "Duplicate selected file."
|
|
action.setToolTip(tip)
|
|
action.setStatusTip(tip)
|
|
action.triggered.connect(self.on_duplicate_pressed)
|
|
menu.addAction(action)
|
|
|
|
# Show the context action menu
|
|
global_point = self._workarea_files_view.mapToGlobal(point)
|
|
action = menu.exec_(global_point)
|
|
if not action:
|
|
return
|
|
|
|
def _select_last_modified_file(self):
|
|
"""Utility function to select the file with latest date modified"""
|
|
if self.published_enabled:
|
|
source_view = self._publish_files_view
|
|
else:
|
|
source_view = self._workarea_files_view
|
|
model = source_view.model()
|
|
|
|
highest_index = None
|
|
highest = 0
|
|
for row in range(model.rowCount()):
|
|
index = model.index(row, 0, parent=QtCore.QModelIndex())
|
|
if not index.isValid():
|
|
continue
|
|
|
|
modified = index.data(DATE_MODIFIED_ROLE)
|
|
if modified is not None and modified > highest:
|
|
highest_index = index
|
|
highest = modified
|
|
|
|
if highest_index:
|
|
source_view.setCurrentIndex(highest_index)
|