mirror of
https://github.com/ynput/ayon-core.git
synced 2026-01-02 00:44:52 +01:00
72 lines
1.7 KiB
Python
72 lines
1.7 KiB
Python
"""Host API required Work Files tool"""
|
|
import os
|
|
import shutil
|
|
|
|
from . import lib
|
|
from avalon import api
|
|
|
|
# used to lock saving until previous save is done.
|
|
save_disabled = False
|
|
|
|
|
|
def file_extensions():
|
|
return api.HOST_WORKFILE_EXTENSIONS["harmony"]
|
|
|
|
|
|
def has_unsaved_changes():
|
|
if lib.server:
|
|
return lib.server.send({"function": "scene.isDirty"})["result"]
|
|
|
|
return False
|
|
|
|
|
|
def save_file(filepath):
|
|
global save_disabled
|
|
if save_disabled:
|
|
return lib.server.send(
|
|
{
|
|
"function": "show_message",
|
|
"args": "Saving in progress, please wait until it finishes."
|
|
})["result"]
|
|
|
|
save_disabled = True
|
|
temp_path = lib.get_local_harmony_path(filepath)
|
|
|
|
if lib.server:
|
|
if os.path.exists(temp_path):
|
|
try:
|
|
shutil.rmtree(temp_path)
|
|
except Exception as e:
|
|
raise Exception(f"cannot delete {temp_path}") from e
|
|
|
|
lib.server.send(
|
|
{"function": "scene.saveAs", "args": [temp_path]}
|
|
)["result"]
|
|
|
|
lib.zip_and_move(temp_path, filepath)
|
|
|
|
lib.workfile_path = filepath
|
|
|
|
scene_path = os.path.join(
|
|
temp_path, os.path.basename(temp_path) + ".xstage"
|
|
)
|
|
lib.server.send(
|
|
{"function": "AvalonHarmony.addPathToWatcher", "args": scene_path}
|
|
)
|
|
else:
|
|
os.environ["HARMONY_NEW_WORKFILE_PATH"] = filepath.replace("\\", "/")
|
|
|
|
save_disabled = False
|
|
|
|
|
|
def open_file(filepath):
|
|
lib.launch_zip_file(filepath)
|
|
|
|
|
|
def current_file():
|
|
"""Returning None to make Workfiles app look at first file extension."""
|
|
return None
|
|
|
|
|
|
def work_root(session):
|
|
return os.path.normpath(session["AVALON_WORKDIR"]).replace("\\", "/")
|