def search_shot(self, source_model, search_text):
'''
search shot
:param source_model: source model
:param search_text: text
:return:
'''
sequence_item_temp_list = []
for sequence_index in range(source_model.rowCount()):
sequence_item = source_model.item(sequence_index, 0)
sequence_item_temp = QStandardItem(sequence_item.text())
sequence_item_temp.setEditable(False)
for shot_name_index in range(sequence_item.rowCount()):
shot_name_item = sequence_item.child(shot_name_index, 0)
shot_name = shot_name_item.text()
if search_text in shot_name:
shot_name_item_temp = shot_name_item.clone()
retrieving = QStandardItem("Retrieving Tasks...")
retrieving.setEditable(False)
shot_name_item_temp.appendRow(retrieving)
shot_name_item_temp.setEditable(False)
sequence_item_temp.appendRow(shot_name_item_temp)
if sequence_item_temp.hasChildren():
sequence_item_temp_list.append(sequence_item_temp)
source_model.clear()
for item in sequence_item_temp_list:
source_model.appendRow(item)
def search_asset(self, source_model, search_text):
'''
search asset
:param source_model: source model
:param search_text: text
:return:
'''
asset_type_item_temp_list = []
for asset_type_index in range(source_model.rowCount()):
asset_type_item = source_model.item(asset_type_index, 0)
asset_type_item_temp = QStandardItem(asset_type_item.text())
asset_type_item_temp.setEditable(False)
for asset_name_index in range(asset_type_item.rowCount()):
asset_name_item = asset_type_item.child(asset_name_index, 0)
asset_name = asset_name_item.text()
if search_text in asset_name:
asset_name_item_temp = asset_name_item.clone()
retrieving = QStandardItem("Retrieving Tasks...")
retrieving.setEditable(False)
asset_name_item_temp.appendRow(retrieving)
asset_name_item_temp.setEditable(False)
asset_type_item_temp.appendRow(asset_name_item_temp)
if asset_type_item_temp.hasChildren():
asset_type_item_temp_list.append(asset_type_item_temp)
source_model.clear()
for item in asset_type_item_temp_list:
source_model.appendRow(item)
def search_mytask(self, source_model, search_text):
'''
search mytask
:param source_model: source model
:param search_text: text
:return:
'''
mytask_type_item_temp_list = []
for mytask_type_index in range(source_model.rowCount()):
mytask_type_item = source_model.item(mytask_type_index, 0)
mytask_type_item_temp = QStandardItem(mytask_type_item.text())
mytask_type_item_temp.setEditable(False)
for mytask_name_index in range(mytask_type_item.rowCount()):
mytask_name_item = mytask_type_item.child(mytask_name_index, 0)
mytask_name = mytask_name_item.text()
if search_text in mytask_name:
mytask_name_item_temp = mytask_name_item.clone()
if mytask_name_item.hasChildren():
for mytask_name_index in range(
mytask_name_item.rowCount()):
mytask_step_item = mytask_name_item.child(
mytask_name_index, 0)
mytask_step_item_temp = mytask_step_item.clone()
if mytask_step_item.hasChildren():
for mytask_step_index in range(
mytask_step_item.rowCount()):
mytask_step_task_item = mytask_step_item.child(
mytask_step_index, 0)
mytask_step_task_item_temp = mytask_step_task_item.clone(
)
mytask_step_item_temp.appendRow(
mytask_step_task_item_temp)
mytask_name_item_temp.appendRow(
mytask_step_item_temp)
mytask_name_item_temp.setEditable(False)
mytask_type_item_temp.appendRow(mytask_name_item_temp)
if mytask_type_item_temp.hasChildren():
mytask_type_item_temp_list.append(mytask_type_item_temp)
source_model.clear()
for item in mytask_type_item_temp_list:
source_model.appendRow(item)
