def json_audit():
project = Project()
environment = Environment()
asset_dir = project.get_assets_dir()
failed_str = ""
for i, asset in enumerate(project.list_assets(AssetType.PROP)):
try:
asset_obj = project.get_body(asset)
element_obj = asset_obj.get_element(Department.MODEL,
force_create=True)
element_path = element_obj.checkout(
environment.get_current_username())
checkout.non_gui_open(element_path, asset)
json_exporter_non_gui.exportProp(asset_obj)
except Exception, err:
error_str = ""
error_str += "\nError exporting JSON for {0}".format(asset)
error_str += "\n" + traceback.format_exc()
print error_str
failed_str += error_str
continue
def go():
global checkout_window
project = Project()
environment = Environment()
nodes = hou.selectedNodes()
if len(nodes) == 1:
#if selected node is digital asset
if nodes[0].type().definition() is not None:
asset = nodes[0]
asset_name = nodes[0].type().name() #get name of asset
index = asset_name.find("_main")
asset_name = asset_name[:index]
src = nodes[0].type().definition().libraryFilePath()
current_user = environment.get_current_username()
if asset_name in project.list_assets():
body = project.get_asset(asset_name)
if os.path.exists(src):
if body is not None:
if Element.DEFAULT_NAME in body.list_elements(Department.ASSEMBLY):
element = body.get_element(Department.ASSEMBLY, Element.DEFAULT_NAME)
element_path = element.checkout(current_user)
hou.hda.uninstallFile(src, change_oplibraries_file=False)
hou.hda.installFile(element_path)
asset.allowEditingOfContents()
hou.ui.displayMessage("Checkout Successful!", title='Success!')
elif len(nodes) > 1:
hou.ui.displayMessage("Only one node can be selected for checkout")
else:
checkout_window = CheckoutWindow(hou.ui.mainQtWindow(), [Department.LIGHTING, Department.FX])
checkout_window.finished.connect(checkout_shot)
def audit():
project = Project()
environment = Environment()
asset_dir = project.get_assets_dir()
for i, asset in enumerate(project.list_assets(AssetType.PROP)):
asset_obj = project.get_body(asset)
element_obj = asset_obj.get_element(Department.MODEL,
force_create=True)
element_path = element_obj.checkout(environment.get_current_username())
#cmds.file(rename='/tmp/lol'+str(i)+'.mb')
#cmds.file(save=True)
checkout.non_gui_open(element_path, asset)
print 'Done'
def non_gui_publish_hda(hda=None,comment='N/A'):
if hda is None:
print ('Error with asset')
project = Project()
environment = Environment()
user = environment.get_current_username()
hdaName = hda.type().name()
department=None
if str(hda) not in Department.ALL:
print 'v1 asset'
department=Department.ASSEMBLY
else:
department=str(hda)
asset_name = hdaName.replace("_" + department, "") if department not in [Department.ASSEMBLY, Department.HDA] else hdaName.replace("_main", "")
body = project.get_body(asset_name)
if body is None:
message_gui.error('No asset in pipe')
return
#TODO: publish tools
if body.is_tool():
print (asset_name+' is tool')
return
department=Department.HDA
hda_src = hda.type().definition().libraryFilePath()
print hda_src
element = body.get_element(department, Element.DEFAULT_NAME,force_create=True)
try:
hda.type().definition().updateFromNode(hda)
except hou.OperationFailed, e:
message_gui.error('There was a problem publishing the HDA to the pipeline.\n', details=str(e))
return
class ExportDialog(QtWidgets.QWidget):
finished = QtCore.Signal()
def __init__(self, parent, renderNodes):
super(ExportDialog, self).__init__()
self.parent = parent
self.project = Project()
self.environment = Environment()
self.renderNodes = renderNodes
self.initUI()
def initUI(self):
# Window layout
self.setFixedSize(360, 420)
self.setWindowTitle('Confirm Export')
main_layout = QtWidgets.QVBoxLayout()
self.setLayout(main_layout)
# Job Name Input Widget
self.jobName = QtWidgets.QLineEdit('JobName')
self.jobName.selectAll()
name_layout = QtWidgets.QHBoxLayout()
name_layout.addWidget(QtWidgets.QLabel('Job Name:'))
name_layout.addWidget(self.jobName)
main_layout.addLayout(name_layout)
# Number of Cores Input Widget
self.numCores = QtWidgets.QLineEdit('4')
coresLayout = QtWidgets.QHBoxLayout()
coresLayout.addWidget(QtWidgets.QLabel('Number of Cores:'))
coresLayout.addWidget(self.numCores)
main_layout.addLayout(coresLayout)
# Priority and Start Time
# Priority
self.priority = QtWidgets.QComboBox()
priority_opt = [
'Very Low', 'Low', 'Medium', 'High', 'Very High', 'Critical'
]
for opt in priority_opt:
self.priority.addItem(opt)
self.priority.setCurrentIndex(2)
# Begin time
self.delay = QtWidgets.QComboBox()
delay_opts = ['Immediate', 'Manual', 'Delayed']
for opt in delay_opts:
self.delay.addItem(opt)
self.delay.currentIndexChanged.connect(self.delaytime)
# Combo box options layout
opts_layout = QtWidgets.QHBoxLayout()
opts_layout.addWidget(QtWidgets.QLabel('Priority:'))
opts_layout.addWidget(self.priority)
opts_layout.addWidget(QtWidgets.QLabel('Begin:'))
opts_layout.addWidget(self.delay)
opts_layout.addStretch()
main_layout.addLayout(opts_layout)
# Time-frame selection for delayed jobs
self.delay_time = QtWidgets.QLineEdit('5')
self.delay_unit = QtWidgets.QComboBox()
delay_unit_opts = ['mins', 'hours', 'days']
for opt in delay_unit_opts:
self.delay_unit.addItem(opt)
self.delay_unit.setCurrentIndex(0)
# Delay time layout
self.delay_layout = QtWidgets.QHBoxLayout()
self.delay_layout.addWidget(QtWidgets.QLabel('Delay time:'))
self.delay_layout.addWidget(self.delay_time)
self.delay_layout.addWidget(self.delay_unit)
self.delay_layout.addStretch()
self.delaytime(0)
main_layout.addLayout(self.delay_layout)
# Mantra node selection list
self.select = QtWidgets.QListWidget()
self.select.setSelectionMode(
QtWidgets.QAbstractItemView.ExtendedSelection)
for node in self.renderNodes:
item = QtWidgets.QListWidgetItem(node.name())
self.select.addItem(item)
# must come after adding to widget
item.setSelected(True)
main_layout.addWidget(self.select)
# Buttons
self.export_btn = QtWidgets.QPushButton('Export')
self.export_btn.clicked.connect(self.export)
self.cancel_btn = QtWidgets.QPushButton('Cancel')
self.cancel_btn.clicked.connect(self.close)
btn_layout = QtWidgets.QHBoxLayout()
btn_layout.addWidget(self.cancel_btn)
btn_layout.addWidget(self.export_btn)
main_layout.addLayout(btn_layout)
# Show/Hide the delay-time options
def delaytime(self, index):
i = 0
items = self.delay_layout.count()
while i < items:
item = self.delay_layout.itemAt(i).widget()
if item:
item.setVisible(index == 2)
i = i + 1
# Export selected mantra nodes for rendering
def export(self):
self.close()
# Get user, project, and time info so we can make a temp folder
user = self.environment.get_current_username()
projectName = self.project.get_name().lower()
time_now = datetime.datetime.now()
#Make a temp folder for the rib files based on the user and the current time
ribDir = self.project.get_project_dir(
) + '/ribs/' + user + '_' + time_now.strftime('%m%d%y_%H%M%S')
print 'ribDir', ribDir, ' renderNodes size: ', len(self.renderNodes)
os.makedirs(ribDir)
# Sanitize job title
title = re.sub(r'[{}"\']', '',
str(self.jobName.text())).strip(' \t\n\r')
if len(title) == 0:
title = self.empty_text
numCores = int(
re.sub(r'[{}"\']', '', str(self.numCores.text())).strip(' \t\n\r'))
if numCores < 1:
numCores = 1
# This job we send to tractor
job = author.Job()
job.title = title
job.priority = self.priority.currentIndex()
path = '/opt/pixar/RenderManProServer-21.5/bin/'
job.envkey = [
'setenv PATH=' + path +
' RMANTREE=/opt/pixar/RenderManProServer-21.5'
]
job.service = 'PixarRender'
job.comment = 'Spooled by ' + user
# Loop through each frame of our nodes and create frame tasks and append it to the job script
for index, node in enumerate(self.renderNodes):
# Make sure this node was selected for export
print node.name()
if self.select.item(index).isSelected():
name = node.name()
validFrameRange = node.parm('trange').eval()
if validFrameRange == 0:
start = int(hou.frame())
end = int(hou.frame())
step = 1
else:
start = int(node.parm('f1').eval())
end = int(node.parm('f2').eval())
step = int(node.parm('f3').eval())
task = author.Task()
task.title = '%s [%d-%d]' % (name, start, end)
oldOutputMode = node.parm('rib_outputmode').eval()
try:
oldDiskFile = node.parm('soho_diskfile').expression()
useExpression = True
print 'We are getting rid of expressiion'
except:
oldDiskFile = node.parm('soho_diskfile').eval()
useExpression = False
print 'we didn\'t get rid of them'
# Activate rib output
node.parm('rib_outputmode').set(True)
node.parm('soho_diskfile').deleteAllKeyframes()
node.parm('soho_diskfile').set(ribDir +
('/%s_$F04.rib' % name))
print 'start rib making'
script = os.path.join(self.project.get_project_dir(),
'byu-pipeline-tools', 'houdini-tools',
'parallelRibs', 'taskDistribution.sh')
subprocess.call([
'sh', script,
str(start),
str(end),
str(node.path()),
str(saveHipRenderCopy()),
str(numCores)
])
print 'finish rib making'
# Loop through every frame in framerange
for frame in range(start, end + 1, step):
subtask = author.Task()
subtask.title = 'Frame %04d' % (frame)
ribFile = '%s/%s_%04d.rib' % (ribDir, name, frame)
print 'Here is the rib file ', ribFile
# Commands for Debugging
cmdPATH = author.Command()
cmdPATH.argv = ['echo', '${PATH}']
cmdRMANTREE = author.Command()
cmdRMANTREE.argv = ['echo', '${RMANTREE}']
printenv = author.Command()
printenv.argv = ['printenv']
# subtask.addCommand(cmdPATH)
# subtask.addCommand(cmdRMANTREE)
# subtask.addCommand(printenv)
# Real Commands
command = author.Command()
command.argv = ['prman', '-progress', ribFile]
command.service = 'PixarRender'
subtask.addCommand(command)
task.addChild(subtask)
job.addChild(task)
# Restore rib output
node.parm('soho_outputmode').set(oldOutputMode)
if useExpression:
node.parm('soho_diskfile').setExpression(oldDiskFile)
else:
node.parm('soho_diskfile').set(oldDiskFile)
command = author.Command()
command.argv = ['rm', '-rf', ribDir]
job.addCleanup(command)
# print 'This is the new job script \n', job.asTcl()
# Attempt to spool job, with the option to keep trying
choice = True
while choice:
try:
job.spool()
message_gui.info('Job sent to Tractor!')
break
except Exception as err:
choice = message_gui.yes_or_no(
'We ran into this problem while spooling the job:\nWould you like to try again?',
details=str(err),
title='Continue?')
#Cleanup ifd files, if they didn't want to retry
if not choice:
shutil.rmtree(ribDir)
