Skip to content
This repository was archived by the owner on Sep 20, 2024. It is now read-only.

Adds button to open snapshot dialog instead. #8

Open
tonybarbieri wants to merge 2 commits into
base: master
Choose a base branch
from
Open

Adds button to open snapshot dialog instead. #8

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
11f2786
Adds button to save as dialog that opens snapshot dialog instead.
Jun 29, 2013
fc19e96
Make sure to return the snapshot dialog result.
Jun 30, 2013
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
101 changes: 57 additions & 44 deletions python/tk_multi_workfiles/save_as.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
from itertools import chain

import tank
from tank.platform.qt import QtCore, QtGui
from tank.platform.qt import QtCore, QtGui
from tank import TankError

from pprint import pprint
Expand All @@ -15,31 +15,31 @@

class SaveAs(object):
"""

"""

@staticmethod
def show_save_as_dlg(app):
"""

"""
handler = SaveAs(app)
handler._show_save_as_dlg()

def __init__(self, app):
"""
Construction
"""
self._app = app

self._work_template = self._app.get_template("template_work")
self._publish_template = self._app.get_template("template_publish")

def _show_save_as_dlg(self):
"""
Show the save as dialog
"""

# get the current file path:
try:
current_path = self._get_current_file_path()
Expand All @@ -49,7 +49,7 @@ def _show_save_as_dlg(self):
"Unable to continue!" % e)
QtGui.QMessageBox.critical(None, "Save As Error!", msg)
return

# determine if this is a publish path or not:
is_publish = self._publish_template.validate(current_path)
fields = {}
Expand All @@ -70,7 +70,7 @@ def _show_save_as_dlg(self):
else:
name = default_name
fields = self._app.context.as_template_fields(self._work_template)

try:
# make sure the work file name doesn't already exist:
# note, this could potentially be slow so for now lets
Expand All @@ -80,71 +80,84 @@ def _show_save_as_dlg(self):
test_name = name
if counter > 0:
test_name = "%s%d" % (name, counter)

test_fields = fields.copy()
test_fields["name"] = test_name

existing_files = self._app.tank.paths_from_template(self._work_template, test_fields, ["version"])
if not existing_files:
name = test_name
break

except TankError, e:
# this shouldn't be fatal so just log a debug message:
self._app.log_debug("Warning - failed to find a default name for Tank Save-As: %s" % e)


worker_cb = lambda details, wp=current_path, ip=is_publish: self.generate_new_work_file_path(wp, ip, details.get("name"), details.get("reset_version"))
with AsyncWorker(worker_cb) as preview_updater:
while True:
# show modal dialog:
from .save_as_form import SaveAsForm
(res, form) = self._app.engine.show_modal(title, self._app, SaveAsForm, preview_updater, is_publish, name)

if res == QtGui.QDialog.Accepted:

can_snapshot = False
snapshot_app = self._app.engine.apps.get("tk-multi-snapshot")
if snapshot_app:
try:
# This might fail if the snapshot tool doesn't have
# this feature.
can_snapshot = snapshot_app.can_snapshot()
except:
can_snapshot = False

(res, form) = self._app.engine.show_modal(title, self._app, SaveAsForm, preview_updater, is_publish, name, can_snapshot)

if res == SaveAsForm.SHOW_SNAPSHOT_RETURN_CODE:
return snapshot_app.show_snapshot_dlg()
elif res == QtGui.QDialog.Accepted:
# get details from UI:
name = form.name
reset_version = form.reset_version

details = self.generate_new_work_file_path(current_path, is_publish, name, reset_version)
new_path = details.get("path")
msg = details.get("message")

if not new_path:
# something went wrong!
QtGui.QMessageBox.information(None, "Unable to Save", "Unable to Save!\n\n%s" % msg)
continue

# ok, so do save-as:
try:
self.save_as(new_path)
except Exception, e:
self._app.log_exception("Something went wrong while saving!")

break
else:
break

def save_as(self, new_path):
"""
Do actual save-as of the current scene as the new
path - assumes all validity checking has already
been done
"""

# always try to create folders:
ctx_entity = self._app.context.task if self._app.context.task else self._app.context.entity
self._app.tank.create_filesystem_structure(ctx_entity.get("type"), ctx_entity.get("id"))

# and save the current file as the new path:
self._save_current_file_as(new_path)

def _save_current_file_as(self, path):
"""
Use hook to get the current work/scene file path
"""
self._app.execute_hook("hook_scene_operation", operation="save_as", file_path=path, context = self._app.context)

def generate_new_work_file_path(self, current_path, current_is_publish, new_name, reset_version):
"""
Generate a new work file path from the current path taking into
Expand All @@ -158,7 +171,7 @@ def generate_new_work_file_path(self, current_path, current_is_publish, new_name
if not new_name:
msg = "You must enter a name!"
return {"message":msg}

if not self._work_template.keys["name"].validate(new_name):
msg = "Your filename contains illegal characters!"
return {"message":msg}
Expand All @@ -168,7 +181,7 @@ def generate_new_work_file_path(self, current_path, current_is_publish, new_name

# start with fields from context:
fields = self._app.context.as_template_fields(self._work_template)

# add in any additional fields from current path:
base_template = self._publish_template if current_is_publish else self._work_template
if base_template.validate(current_path):
Expand All @@ -177,7 +190,7 @@ def generate_new_work_file_path(self, current_path, current_is_publish, new_name
else:
# just make sure there is a version
fields["version"] = 1

current_version = fields.get("version")
current_name = fields.get("name")

Expand All @@ -190,52 +203,52 @@ def generate_new_work_file_path(self, current_path, current_is_publish, new_name
max_work_version = versioning.get_max_workfile_version(fields)
max_publish_version = versioning.get_max_publish_version(new_name)
max_version = max(max_work_version, max_publish_version)
# now depending on what the source was

# now depending on what the source was
# and if the name has been changed:
new_version = None
if current_is_publish and new_name == current_name:
# we're ok to just copy publish across and version up
can_reset_version = False
new_version = max_version + 1

if new_version != current_version+1:
#(AD) - do we need a warning here?
pass

msg = None
else:
if max_version:
# already have a publish and/or work file
can_reset_version = False
new_version = max_version + 1

if max_version == max_work_version:
msg = "A work file with this name already exists. If you proceed, your file will use the next available version number."
else:
msg = "A publish file with this name already exists. If you proceed, your file will use the next available version number."

else:
# don't have an existing version
can_reset_version = True
msg = ""
if reset_version:
new_version = 1

# now create new path
if new_version:
fields["version"] = new_version
new_work_path = self._work_template.apply_fields(fields)

return {"path":new_work_path, "message":msg, "can_reset_version":can_reset_version}


def _get_current_file_path(self):
"""
Use hook to get the current work/scene file path
"""
return self._app.execute_hook("hook_scene_operation", operation="current_path", file_path="", context = self._app.context)




Loading