From fbe1a339e179354d0d02d382fe11cd06551f8b9c Mon Sep 17 00:00:00 2001 From: Jack Date: Wed, 2 Oct 2024 21:57:15 -0400 Subject: [PATCH] Idk what I did, forgot to commit --- tinyblast.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/tinyblast.py b/tinyblast.py index bf291d2..d7e56ab 100644 --- a/tinyblast.py +++ b/tinyblast.py @@ -2,6 +2,7 @@ import maya.OpenMaya as om import maya.OpenMayaMPx as ompx import maya.OpenMayaUI as omui import maya.cmds as cmds +import maya.utils as utils import os import subprocess import sys @@ -38,10 +39,11 @@ def custom_playblast(*args, **kwargs): input_file = result # The file output by playblast #output_directory = os.path.dirname(result) # Get the directory path output_directory = os.path.dirname(cmds.file(query=True, sceneName=True)) + scene_name = os.path.basename(cmds.file(query=True, sceneName=True)) input_filename = os.path.basename(result) # Get the filename with extension # Change the extension to .mp4 - output_filename = os.path.splitext(input_filename)[0] + ".mp4" + output_filename = os.path.splitext(scene_name)[0] + ".mp4" # Define the full path for the converted output file output_file = os.path.join(output_directory, output_filename) @@ -148,6 +150,9 @@ def get_playblast_options_window(): return wrapInstance(int(window_ptr), QtWidgets.QWidget) return None +def open_script_job(): + cmds.scriptJob(event=["SceneOpened", setup_script_job]) + def setup_script_job(): global playblast_job_id # Kill any previously running scriptJob @@ -187,7 +192,7 @@ def initializePlugin(mobject): mplugin = ompx.MFnPlugin(mobject, "Jack Christensen", "1.1.0", "Any") mplugin.registerCommand("tinyblast", tinyblast_cmd) om.MGlobal.displayInfo("Tinyblast plugin loaded.") - setup_script_job() + open_script_job() except Exception as e: om.MGlobal.displayError(f"Failed to initialize plugin: {str(e)}") raise