-
-
Notifications
You must be signed in to change notification settings - Fork 9
/
Copy pathengine.py
949 lines (784 loc) · 32 KB
/
engine.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
# Copyright (c) 2013 Shotgun Software Inc.
#
# CONFIDENTIAL AND PROPRIETARY
#
# This work is provided "AS IS" and subject to the Shotgun Pipeline Toolkit
# Source Code License included in this distribution package. See LICENSE.
# By accessing, using, copying or modifying this work you indicate your
# agreement to the Shotgun Pipeline Toolkit Source Code License. All rights
# not expressly granted therein are reserved by Shotgun Software Inc.
"""A Clarisse engine for Tank."""
import os
import sys
import time
import re
import inspect
import logging
from functools import wraps, partial
import tank
import traceback
from tank.log import LogManager
from tank.platform import Engine
from tank.platform.constants import SHOTGUN_ENGINE_NAME
import ix
__author__ = "Diego Garcia Huerta"
__contact__ = "https://www.linkedin.com/in/diegogh/"
# initialize our shotgun structure for the session
if not hasattr(ix, "shotgun"):
# use a dummy class to keep references to menus
ix.shotgun = lambda: None
ix.shotgun.menu_callbacks = {}
def show_error(msg):
print("Shotgun Error | Clarisse engine | %s " % msg)
ix.application.message_box(
msg,
"Shotgun Error | Clarisse engine",
ix.api.AppDialog.cancel(),
ix.api.AppDialog.STYLE_OK,
)
def show_warning(msg):
ix.application.message_box(
msg,
"Shotgun Warning | Clarisse engine",
ix.api.AppDialog.cancel(),
ix.api.AppDialog.STYLE_OK,
)
def show_info(msg):
ix.application.message_box(
msg,
"Shotgun Info | Clarisse engine",
ix.api.AppDialog.cancel(),
ix.api.AppDialog.STYLE_OK,
)
def display_error(msg):
t = time.asctime(time.localtime())
print("%s - Shotgun Error | Clarisse engine | %s " % (t, msg))
ix.application.log_error(
"%s - Shotgun Error | Clarisse engine | %s " % (t, msg)
)
def display_warning(msg):
t = time.asctime(time.localtime())
ix.application.log_warning(
"%s - Shotgun Warning | Clarisse engine | %s " % (t, msg)
)
def display_info(msg):
t = time.asctime(time.localtime())
ix.application.log_info(
"%s - Shotgun Info | Clarisse engine | %s " % (t, msg)
)
def display_debug(msg):
if os.environ.get("TK_DEBUG") == "1":
t = time.asctime(time.localtime())
ix.application.log_info(
"%s - Shotgun Debug | Clarisse engine | %s " % (t, msg)
)
# we use a trick with decorators to get some sort of event notification
# when the scene is saved/loaded, etc... we could use a timer similar to
# what tk-houdini uses but this other aproach is more generic
def wrapped(function, watcher, post_callback=None, pre_callback=None):
@wraps(function)
def wrapper(*args, **kwargs):
try:
if pre_callback is not None:
pre_callback(watcher)
result = function(*args, **kwargs)
if post_callback is not None:
post_callback(watcher)
return result
except:
raise
wrapper._original = function
return wrapper
SCENE_EVENT_NAMES = (
"new_project",
"clear_project",
"import_project",
"load_project",
"save_project",
"load_startup_scene",
)
SCENE_QUIT_EVENT_NAME = "quit"
class SceneEventWatcher(object):
"""
Encapsulates event handling for multiple scene events and routes them
into a single callback.
This uses monkey patching some of the functions in the clarisse application
Specifying run_once=True in the constructor causes all events to be
cleaned up after the first one has triggered
"""
def __init__(self, cb_fn, run_once=False):
"""
Constructor.
:param cb_fn: Callback to invoke everytime a scene event happens.
:param scene_events: List of scene events to watch for. Defaults to
new, open and save.
:param run_once: If True, the watcher will notify only on the first
event. Defaults to False.
"""
self.__cb_fn = cb_fn
self.__run_once = run_once
self.__wrapped_fns = {}
# register scene event callbacks:
self.start_watching()
def start_watching(self):
"""
Starts watching for scene events.
"""
# if currently watching then stop:
self.stop_watching()
# now add callbacks to watch for some scene events:
for event_name in SCENE_EVENT_NAMES:
try:
event_fn = getattr(ix.application, event_name)
event_fn = wrapped(
event_fn,
self,
post_callback=SceneEventWatcher.__scene_event_callback,
)
self.__wrapped_fns[event_name] = event_fn
setattr(ix.application, event_name, event_fn)
display_debug("Registered callback on %s " % event_name)
except Exception:
traceback.print_exc()
# report warning...
continue
# create a callback that will be run when Clarisse
# exits so we can do some clean-up:
event_fn = getattr(ix.application, SCENE_QUIT_EVENT_NAME)
event_fn = wrapped(
event_fn,
self,
pre_callback=SceneEventWatcher.__clarisse_exiting_callback,
)
self.__wrapped_fns[SCENE_QUIT_EVENT_NAME] = event_fn
setattr(ix.application, SCENE_QUIT_EVENT_NAME, event_fn)
def stop_watching(self):
"""
Stops watching the Clarisse scene.
"""
for event_name, event_fn in self.__wrapped_fns.iteritems():
setattr(ix.application, event_name, event_fn._original)
self.__wrapped_fns = {}
@staticmethod
def __scene_event_callback(watcher):
"""
Called on a scene event:
"""
if watcher.__run_once:
watcher.stop_watching()
watcher.__cb_fn()
@staticmethod
def __clarisse_exiting_callback(watcher):
"""
Called on Clarisse exit - should clean up any existing calbacks
"""
watcher.stop_watching()
###############################################################################
# methods to support the state when the engine cannot start up
# for example if a non-tank file is loaded in clarisse
def refresh_engine(engine_name, prev_context, menu_name):
"""
refresh the current engine
"""
current_engine = tank.platform.current_engine()
if not current_engine:
# If we don't have an engine for some reason then we don't have
# anything to do.
return
scene_name = ix.application.get_current_project_filename()
# This is a File->New call, so we just leave the engine in the current
# context and move on.
if scene_name == "":
if prev_context != tank.platform.current_engine().context:
current_engine.change_context(ctx)
return
# determine the tk instance and ctx to use:
tk = current_engine.sgtk
# loading a scene file
new_path = os.path.abspath(scene_name)
# this file could be in another project altogether, so create a new
# API instance.
try:
tk = tank.tank_from_path(new_path)
# and construct the new context for this path:
ctx = tk.context_from_path(new_path, prev_context)
except tank.TankError:
try:
ctx = current_engine.sgtk.context_from_entity_dictionary(
current_engine.context.project
)
except tank.TankError:
(exc_type, exc_value, exc_traceback) = sys.exc_info()
message = ""
message += "Shotgun Clarisse Engine cannot be started:.\n"
message += "Please contact you technical support team for more "
message += "information.\n\n"
message += "Exception: %s - %s\n" % (exc_type, exc_value)
message += "Traceback (most recent call last):\n"
message += "\n".join(traceback.format_tb(exc_traceback))
# build disabled menu
create_sgtk_disabled_menu(menu_name)
display_error(message)
return
# now remove the shotgun disabled menu if it exists.
remove_sgtk_disabled_menu(menu_name)
# shotgun menu may have been removed, so add it back in if its not already
# there.
current_engine.create_shotgun_menu()
if ctx != tank.platform.current_engine().context:
current_engine.change_context(ctx)
def on_scene_event_callback(engine_name, prev_context, menu_name):
"""
Callback that's run whenever a scene is saved or opened.
"""
try:
refresh_engine(engine_name, prev_context, menu_name)
except Exception:
(exc_type, exc_value, exc_traceback) = sys.exc_info()
message = ""
message += (
"Message: Shotgun encountered a problem changing the "
"Engine's context.\n"
)
message += "Please contact you technical support team for more "
message += "information.\n\n"
message += "Exception: %s - %s\n" % (exc_type, exc_value)
message += "Traceback (most recent call last):\n"
message += "\n".join(traceback.format_tb(exc_traceback))
show_error(message)
def sgtk_disabled_message():
"""
Explain why tank is disabled.
"""
msg = (
"Shotgun integration is disabled because it cannot recognize "
"the currently opened file. Try opening another file or restarting "
"Clarisse."
)
show_warning(msg)
def clear_sgtk_menu(menu_name):
if not ix.is_gui_application():
# don't create menu in not interactive mode
return
sg_menu = get_sgtk_root_menu(menu_name)
sg_menu.remove_all_commands()
ix.shotgun.menu_callbacks = {}
def get_sgtk_root_menu(menu_name):
menu = ix.application.get_main_menu()
sg_menu = menu.get_item(menu_name + ">")
if not sg_menu:
sg_menu = menu.add_command(menu_name + ">")
return sg_menu
def create_sgtk_disabled_menu(menu_name):
"""
Render a special "shotgun is disabled" menu
"""
if not ix.is_gui_application():
# don't create menu in not interactive mode
return
sg_menu = get_sgtk_root_menu(menu_name)
menu_item = menu_name + ">Sgtk is disabled."
ix.shotgun.menu_callbacks[menu_item] = sgtk_disabled_message
menu.add_command_as_script(
menu_name + ">Sgtk is disabled.",
"ix.shotgun.menu_callbacks[%s]" % menu_item,
)
def remove_sgtk_disabled_menu(menu_name):
"""
Clear the shotgun menu
"""
clear_sgtk_menu(menu_name)
###############################################################################
# The Tank Clarisse engine
class ClarisseEngine(Engine):
"""
Toolkit engine for Clarisse.
"""
def __get_platform_resource_path(self, filename):
"""
Returns the full path to the given platform resource file or folder.
Resources reside in the core/platform/qt folder.
:return: full path
"""
tank_platform_folder = os.path.abspath(inspect.getfile(tank.platform))
return os.path.join(tank_platform_folder, "qt", filename)
def __toggle_debug_logging(self):
"""
Toggles global debug logging on and off in the log manager.
This will affect all logging across all of toolkit.
"""
# flip debug logging
LogManager().global_debug = not LogManager().global_debug
def __open_log_folder(self):
"""
Opens the file system folder where log files are being stored.
"""
self.log_info("Log folder location: '%s'" % LogManager().log_folder)
if self.has_ui:
# only import QT if we have a UI
from sgtk.platform.qt import QtGui, QtCore
url = QtCore.QUrl.fromLocalFile(LogManager().log_folder)
status = QtGui.QDesktopServices.openUrl(url)
if not status:
self._engine.log_error("Failed to open folder!")
def __register_open_log_folder_command(self):
"""
# add a 'open log folder' command to the engine's context menu
# note: we make an exception for the shotgun engine which is a
# special case.
"""
if self.name != SHOTGUN_ENGINE_NAME:
self.register_command(
"Open Log Folder",
self.__open_log_folder,
{
"short_name": "open_log_folder",
"icon": self.__get_platform_resource_path("folder_256.png"),
"description": (
"Opens the folder where log files are being stored."
),
"type": "context_menu",
},
)
def __register_reload_command(self):
"""
Registers a "Reload and Restart" command with the engine if any
running apps are registered via a dev descriptor.
"""
from tank.platform import restart
self.register_command(
"Reload and Restart",
restart,
{
"short_name": "restart",
"icon": self.__get_platform_resource_path("reload_256.png"),
"type": "context_menu",
},
)
@property
def context_change_allowed(self):
"""
Whether the engine allows a context change without the need for a
restart.
"""
return True
@property
def host_info(self):
"""
:returns: A dictionary with information about the application hosting
this engine.
The returned dictionary is of the following form on success:
{
"name": "Clarisse",
"version": "2017 Update 4",
}
The returned dictionary is of following form on an error preventing
the version identification.
{
"name": "Clarisse",
"version: "unknown"
}
"""
host_info = {"name": "Clarisse", "version": "unknown"}
try:
clarisse_ver = ix.application.get_version_name()
host_info["version"] = clarisse_ver
except:
# Fallback to 'Clarisse' initialized above
pass
return host_info
###########################################################################
# init and destroy
def pre_app_init(self):
"""
Runs after the engine is set up but before any apps have been
initialized.
"""
# unicode characters returned by the shotgun api need to be converted
# to display correctly in all of the app windows
from tank.platform.qt import QtCore
# tell QT to interpret C strings as utf-8
utf8 = QtCore.QTextCodec.codecForName("utf-8")
QtCore.QTextCodec.setCodecForCStrings(utf8)
self.logger.debug("set utf-8 codec for widget text")
def init_engine(self):
"""
Initializes the Clarisse engine.
"""
self.logger.debug("%s: Initializing...", self)
# check that we are running an ok version of clarisse
current_os = sys.platform.lower()
if current_os not in ["darwin", "win32", "linux64"]:
raise tank.TankError(
"The current platform is not supported! Supported platforms "
"are Mac, Linux 64 and Windows 64."
)
clarisse_build_version = ix.application.get_version()
clarisse_ver = float(".".join(clarisse_build_version.split(".")[:2]))
if clarisse_ver < 3.6:
msg = "Shotgun integration is not compatible with Clarisse "
msg += "versions older than 3.6"
raise tank.TankError(msg)
if clarisse_ver > 4.0:
# show a warning that this version of Clarisse isn't yet fully
# tested with Shotgun:
msg = (
"The Shotgun Pipeline Toolkit has not yet been fully tested "
"with Clarisse %s.\n"
"You can continue to use Toolkit but you may experience bugs "
"or instability."
"\n\nUse at your own risk." % (clarisse_ver)
)
# determine if we should show the compatibility warning dialog:
show_warning_dlg = (
self.has_ui
and "SGTK_COMPATIBILITY_DIALOG_SHOWN" not in os.environ
)
if show_warning_dlg:
# make sure we only show it once per session:
os.environ["SGTK_COMPATIBILITY_DIALOG_SHOWN"] = "1"
# split off the major version number - accomodate complex
# version strings and decimals:
major_version_number_str = clarisse_build_version.split(".")[0]
if (
major_version_number_str
and major_version_number_str.isdigit()
):
# check against the compatibility_dialog_min_version:
if int(major_version_number_str) < self.get_setting(
"compatibility_dialog_min_version"
):
show_warning_dlg = False
if show_warning_dlg:
# Note, title is padded to try to ensure dialog isn't insanely
# narrow!
show_info(msg)
# always log the warning to the script editor:
self.logger.warning(msg)
# In the case of Clarisse on Windows, we have the possility of
# locking up if we allow the PySide shim to import
# QtWebEngineWidgets. We can stop that happening here by setting
# the environment variable.
if current_os.startswith("win"):
self.logger.debug(
"Clarisse on Windows can deadlock if QtWebEngineWidgets "
"is imported. "
"Setting SHOTGUN_SKIP_QTWEBENGINEWIDGETS_IMPORT=1..."
)
os.environ["SHOTGUN_SKIP_QTWEBENGINEWIDGETS_IMPORT"] = "1"
# add qt paths and dlls
self._init_pyside()
# default menu name is Shotgun but this can be overriden
# in the configuration to be Sgtk in case of conflicts
self._menu_name = "Shotgun"
if self.get_setting("use_sgtk_as_menu_name", False):
self._menu_name = "Sgtk"
if self.get_setting("automatic_context_switch", True):
# need to watch some scene events in case the engine needs
# rebuilding:
cb_fn = partial(
on_scene_event_callback,
engine_name=self.instance_name,
prev_context=self.context,
menu_name=self._menu_name,
)
self.__watcher = SceneEventWatcher(cb_fn, run_once=False)
self.logger.debug("Registered open and save callbacks.")
def create_shotgun_menu(self):
"""
Creates the main shotgun menu in clarisse.
Note that this only creates the menu, not the child actions
:return: bool
"""
# only create the shotgun menu if not in batch mode and menu doesn't
# already exist
if self.has_ui:
self._menu_handle = get_sgtk_root_menu(self._menu_name)
# create our menu handler
tk_clarisse = self.import_module("tk_clarisse")
self._menu_generator = tk_clarisse.MenuGenerator(
self, self._menu_handle
)
self._menu_generator.create_menu()
return True
return False
def _initialise_qapplication(self):
"""
Ensure the QApplication is initialized
"""
from sgtk.platform.qt import QtGui
qt_app = QtGui.QApplication.instance()
if qt_app is None:
self.log_debug("Initialising main QApplication...")
qt_app = QtGui.QApplication([])
qt_app.setWindowIcon(QtGui.QIcon(self.icon_256))
qt_app.setQuitOnLastWindowClosed(False)
# set up the dark style
self._initialize_dark_look_and_feel()
import pyqt_clarisse
pyqt_clarisse.exec_(qt_app)
def post_app_init(self):
"""
Called when all apps have initialized
"""
self._initialise_qapplication()
# for some readon this engine command get's lost so we add it back
self.__register_reload_command()
self.create_shotgun_menu()
# Run a series of app instance commands at startup.
self._run_app_instance_commands()
def post_context_change(self, old_context, new_context):
"""
Runs after a context change. The Clarisse event watching will be
stopped and new callbacks registered containing the new context
information.
:param old_context: The context being changed away from.
:param new_context: The new context being changed to.
"""
# restore the open log folder, it get's removed whenever the first time
# a context is changed
self.__register_open_log_folder_command()
self.__register_reload_command()
if self.get_setting("automatic_context_switch", True):
# We need to stop watching, and then replace with a new watcher
# that has a callback registered with the new context baked in.
# This will ensure that the context_from_path call that occurs
# after a File->Open receives an up-to-date "previous" context.
self.__watcher.stop_watching()
cb_fn = partial(
on_scene_event_callback,
engine_name=self.instance_name,
prev_context=self.context,
menu_name=self._menu_name,
)
self.__watcher = SceneEventWatcher(cb_fn, run_once=False)
self.logger.debug(
"Registered new open and save callbacks before"
" changing context."
)
# finally create the menu with the new context if needed
if old_context != new_context:
self.create_shotgun_menu()
def _run_app_instance_commands(self):
"""
Runs the series of app instance commands listed in the 'run_at_startup'
setting of the environment configuration yaml file.
"""
# Build a dictionary mapping app instance names to dictionaries of
# commands they registered with the engine.
app_instance_commands = {}
for (command_name, value) in self.commands.iteritems():
app_instance = value["properties"].get("app")
if app_instance:
# Add entry 'command name: command function' to the command
# dictionary of this app instance.
command_dict = app_instance_commands.setdefault(
app_instance.instance_name, {}
)
command_dict[command_name] = value["callback"]
# Run the series of app instance commands listed in the
# 'run_at_startup' setting.
for app_setting_dict in self.get_setting("run_at_startup", []):
app_instance_name = app_setting_dict["app_instance"]
# Menu name of the command to run or '' to run all commands of the
# given app instance.
setting_command_name = app_setting_dict["name"]
# Retrieve the command dictionary of the given app instance.
command_dict = app_instance_commands.get(app_instance_name)
if command_dict is None:
self.logger.warning(
(
"%s configuration setting 'run_at_startup'"
" requests app '%s' that is not installed."
),
self.name,
app_instance_name,
)
else:
if not setting_command_name:
# Run all commands of the given app instance.
# Run these commands once Clarisse will have completed its
# UI update and be idle in order to run them after the ones
# that restore the persisted Shotgun app panels.
for (
command_name,
command_function,
) in command_dict.iteritems():
self.logger.debug(
"%s startup running app '%s' command '%s'.",
self.name,
app_instance_name,
command_name,
)
clarisse.utils.executeDeferred(command_function)
else:
# Run the command whose name is listed in the
# 'run_at_startup' setting.
# Run this command once Clarisse will have completed its
# UI update and be idle in order to run it after the ones
# that restore the persisted Shotgun app panels.
command_function = command_dict.get(setting_command_name)
if command_function:
self.logger.debug(
"%s startup running app '%s' command '%s'.",
self.name,
app_instance_name,
setting_command_name,
)
clarisse.utils.executeDeferred(command_function)
else:
known_commands = ", ".join(
"'%s'" % name for name in command_dict
)
self.logger.warning(
(
"%s configuration setting 'run_at_startup' "
"requests app '%s' unknown command '%s'. "
"Known commands: %s"
),
self.name,
app_instance_name,
setting_command_name,
known_commands,
)
def destroy_engine(self):
"""
Stops watching scene events and tears down menu.
"""
self.logger.debug("%s: Destroying...", self)
if self.get_setting("automatic_context_switch", True):
# stop watching scene events
self.__watcher.stop_watching()
def _init_pyside(self):
"""
Handles the pyside init
"""
# first see if pyside2 is present
try:
from PySide2 import QtGui
except:
# fine, we don't expect PySide2 to be present just yet
self.logger.debug("PySide2 not detected - trying for PySide now...")
else:
# looks like pyside2 is already working! No need to do anything
self.logger.debug(
"PySide2 detected - the existing version will be used."
)
return
# then see if pyside is present
try:
from PySide import QtGui
except:
# must be that a PySide version is not installed,
self.logger.debug(
"PySide not detected - it will be added to the setup now..."
)
else:
# looks like pyside is already working! No need to do anything
self.logger.debug(
"PySide detected - the existing version will be used."
)
return
current_os = sys.platform.lower()
if current_os == "darwin":
desktop_path = os.environ.get("SHOTGUN_DESKTOP_INSTALL_PATH",
"/Applications/Shotgun.app")
sys.path.append(os.path.join(desktop_path, "Contents", "Resources",
"Python", "lib", "python2.7",
"site-packages"))
elif current_os == "win32":
desktop_path = os.environ.get("SHOTGUN_DESKTOP_INSTALL_PATH",
"C:/Program Files/Shotgun")
sys.path.append(os.path.join(desktop_path,
"Python", "Lib", "site-packages"))
elif current_os == "linux2":
desktop_path = os.environ.get("SHOTGUN_DESKTOP_INSTALL_PATH",
"/opt/Shotgun/Shotgun")
sys.path.append(os.path.join(desktop_path,
"Python", "Lib", "site-packages"))
else:
self.logger.error("Unknown platform - cannot initialize PySide!")
# now try to import it
try:
from PySide import QtGui
except Exception as exception:
traceback.print_exc()
self.logger.error(
"PySide could not be imported! Apps using pyside will not "
"operate correctly! Error reported: %s",
exception,
)
def _get_dialog_parent(self):
"""
Clarisse is not Qt Based so we do not have anything to return here.
"""
return None
@property
def has_ui(self):
"""
Detect and return if clarisse is running in batch mode
"""
if not ix.is_gui_application():
# batch mode or prompt mode
return False
else:
return True
###########################################################################
# logging
def _emit_log_message(self, handler, record):
"""
Called by the engine to log messages in Clarisse script editor.
All log messages from the toolkit logging namespace will be passed to
this method.
:param handler: Log handler that this message was dispatched from.
Its default format is "[levelname basename] message".
:type handler: :class:`~python.logging.LogHandler`
:param record: Standard python logging record.
:type record: :class:`~python.logging.LogRecord`
"""
# Give a standard format to the message:
# Shotgun <basename>: <message>
# where "basename" is the leaf part of the logging record name,
# for example "tk-multi-shotgunpanel" or "qt_importer".
if record.levelno < logging.INFO:
formatter = logging.Formatter(
"Debug: Shotgun %(basename)s: %(message)s"
)
else:
formatter = logging.Formatter("Shotgun %(basename)s: %(message)s")
msg = formatter.format(record)
# Select Clarisse display function to use according to the logging
# record level.
if record.levelno >= logging.ERROR:
fct = display_error
elif record.levelno >= logging.WARNING:
fct = display_warning
elif record.levelno >= logging.INFO:
fct = display_info
else:
fct = display_debug
# Display the message in Clarisse script editor in a thread safe manner
self.async_execute_in_main_thread(fct, msg)
###########################################################################
# scene and project management
def close_windows(self):
"""
Closes the various windows (dialogs, panels, etc.) opened by the engine
"""
# Make a copy of the list of Tank dialogs that have been created by the
# engine and are still opened since the original list will be updated
# when each dialog is closed.
opened_dialog_list = self.created_qt_dialogs[:]
# Loop through the list of opened Tank dialogs.
for dialog in opened_dialog_list:
dialog_window_title = dialog.windowTitle()
try:
# Close the dialog and let its close callback remove it from
# the original dialog list.
self.logger.debug("Closing dialog %s.", dialog_window_title)
dialog.close()
except Exception as exception:
traceback.print_exc()
self.logger.error(
"Cannot close dialog %s: %s", dialog_window_title, exception
)