def get_projects(company):
"""
returns projects for "company"
:param company:
:return:
"""
d = {'root': ROOT, 'company': company, 'project': '*', 'context': 'source'}
path_object = PathObject(d)
return path_object.glob_project_element('project')
def get_companies():
"""
returns all companies the system is aware of.
:return:
"""
d = {'root': ROOT, 'company': '*'}
path_object = PathObject(d)
companies = path_object.glob_project_element('company')
if '_config' in companies:
companies.remove('_config')
return companies
class AssetCreator(LJDialog):
def __init__(self, parent=None, path_dict=None, task_mode=False):
LJDialog.__init__(self, parent)
self.task_mode = task_mode
self.resize(300, 60)
self.red_palette, self.green_palette, self.black_palette = define_palettes(
)
if not path_dict:
return
self.path_object = PathObject(path_dict)
self.scope = self.path_object.scope
if task_mode:
self.setWindowTitle('Create %s' % 'Task(s)')
else:
self.setWindowTitle('Create %s' % self.scope.title())
self.asset = None
self.asset_message_string = ''
self.asset_list = []
self.full_root = None
self.regex = ''
self.valid_categories_string = ''
self.seq = None
self.valid_categories = []
self.get_valid_categories()
# Environment Stuff
self.root = CONFIG['paths']['root']
self.project_management = CONFIG['account_info']['project_management']
self.schema = CONFIG['project_management'][
self.project_management]['api']['default_schema']
if self.scope == 'assets':
self.asset_string_example = CONFIG['rules']['path_variables'][
'asset']['example']
elif self.scope == 'shots':
self.asset_string_example = CONFIG['rules']['path_variables'][
'shotname']['example']
schema = CONFIG['project_management'][
self.project_management]['tasks'][self.schema]
self.proj_man_tasks = schema['long_to_short'][self.scope.lower()]
self.proj_man_tasks_short_to_long = schema['short_to_long'][
self.scope.lower()]
self.v_layout = QtWidgets.QVBoxLayout(self)
self.scope_row = QtWidgets.QHBoxLayout()
self.asset_row = QtWidgets.QHBoxLayout(self)
self.tasks = []
self.task_row = QtWidgets.QHBoxLayout(self)
self.task_combo = AdvComboBox()
self.clear_selection_button = QtWidgets.QPushButton('Clear Selection')
self.defaults_radio = QtWidgets.QRadioButton('Select Defaults')
self.none_radio = QtWidgets.QRadioButton('Select None')
self.radio_layout = QtWidgets.QHBoxLayout()
self.radio_layout.addItem(
QtWidgets.QSpacerItem(0, 0, QtWidgets.QSizePolicy.Expanding,
QtWidgets.QSizePolicy.Minimum))
self.radio_layout.addWidget(self.defaults_radio)
self.radio_layout.addWidget(self.none_radio)
self.defaults_radio.setChecked(True)
self.defaults_radio.hide()
self.none_radio.hide()
if self.task_mode:
self.create_button = QtWidgets.QPushButton('Create %s' % 'Task(s)')
else:
self.create_button = QtWidgets.QPushButton('Create %s' %
self.scope.title())
self.create_button.setEnabled(False)
# asset & shot stuff
self.asset_widget = AssetWidget(self,
scope=self.scope,
title=str(self.path_object.project))
self.asset_widget.name_row.combo.setEnabled(False)
self.asset_row.addWidget(self.asset_widget)
# task stuff
self.task_layout = QtWidgets.QVBoxLayout(self)
for each in self.proj_man_tasks:
checkbox = QtWidgets.QCheckBox('%s (%s)' %
(each, self.proj_man_tasks[each]))
checkbox.stateChanged.connect(self.on_checkbox_clicked)
self.task_layout.addWidget(checkbox)
self.v_layout.addLayout(self.asset_row)
self.v_layout.addLayout(self.radio_layout)
self.v_layout.addLayout(self.task_layout)
self.v_layout.addWidget(self.create_button)
self.v_layout.addItem(
QtWidgets.QSpacerItem(0, 0, QtWidgets.QSizePolicy.Minimum,
QtWidgets.QSizePolicy.Expanding))
self.asset_widget.message.hide()
# self.asset_widget.message.setPalette(self.red_palette)
self.load_categories()
self.asset_widget.category_row.combo.currentIndexChanged.connect(
self.on_category_selected)
self.asset_widget.name_row.combo.editTextChanged.connect(
self.process_asset_string)
self.create_button.clicked.connect(self.on_create_clicked)
if self.scope == 'shots':
self.asset_widget.category_row.combo.editTextChanged.connect(
self.on_seq_text_changed)
self.hide_layout_items(self.task_layout)
if self.task_mode:
self.on_set_vars()
def on_set_vars(self):
if self.path_object.seq:
if self.path_object.seq != '*':
index = self.asset_widget.category_row.combo.findText(
self.path_object.seq)
if index != -1:
self.asset_widget.category_row.combo.setCurrentIndex(index)
else:
self.asset_widget.category_row.combo.addItem(
self.path_object.seq)
index = self.asset_widget.category_row.combo.findText(
self.path_object.seq)
self.asset_widget.category_row.combo.setCurrentIndex(index)
if self.path_object.shot:
if self.path_object.shot != '*':
index = self.asset_widget.name_row.combo.findText(
self.path_object.shot)
if index != -1:
self.asset_widget.name_row.combo.setCurrentIndex(index)
else:
self.asset_widget.name_row.combo.addItem(
self.path_object.shot)
index = self.asset_widget.name_row.combo.findText(
self.path_object.shot)
self.asset_widget.name_row.combo.setCurrentIndex(index)
def on_checkbox_clicked(self):
self.find_all_checked_boxes()
def find_all_checked_boxes(self):
self.tasks = []
for i in range(self.task_layout.count()):
child = self.task_layout.itemAt(i)
if child.widget().checkState() == QtCore.Qt.Checked:
self.tasks.append(child.widget().text().split('(')[-1].replace(
')', ''))
@staticmethod
def hide_layout_items(layout):
for i in range(layout.count()):
child = layout.itemAt(i).widget()
child.hide()
@staticmethod
def show_layout_items(layout):
for i in range(layout.count()):
child = layout.itemAt(i).widget()
child.show()
def on_category_selected(self):
self.path_object.set_attr(
seq=self.asset_widget.category_row.combo.currentText())
self.asset_widget.name_row.combo.setEnabled(True)
self.load_assets()
def on_seq_text_changed(self):
self.seq = self.asset_widget.category_row.combo.currentText()
seq_rules = CONFIG['rules']['path_variables']['seq']['regex']
example = CONFIG['rules']['path_variables']['seq']['example']
self.regex = re.compile(r'%s' % seq_rules)
if re.match(self.regex, self.seq):
self.asset_widget.name_row.combo.setEnabled(True)
self.asset_widget.message.hide()
else:
self.asset_message_string = '%s is not a valid sequence name\n%s' % (
self.seq, example)
self.asset_widget.message.setText(self.asset_message_string)
# self.asset_widget.message.setPalette(self.red_palette)
self.asset_widget.message.show()
def load_categories(self):
if self.path_object.scope == 'assets':
categories = CONFIG['asset_categories']
cats = ['']
for c in categories:
if CONFIG['account_info']['project_management'] == 'shotgun':
cats.append(c)
else:
cats.append(categories[c])
self.asset_widget.category_row.combo.addItems(cats)
elif self.path_object.scope == 'shots':
seqs = self.path_object.glob_project_element('seq')
self.asset_widget.category_row.combo.addItems(seqs)
self.on_seq_text_changed()
def process_asset_string(self):
asset_rules = ''
self.seq = self.asset_widget.category_row.combo.currentText()
asset_string = self.asset_widget.name_row.combo.currentText()
if self.scope == 'assets':
asset_rules = r'^([a-zA-Z]{3,},\s*)*([a-zA-Z]{3,}$)|^([a-zA-Z]{3,},\s*)*([a-zA-Z]{3,}$)'
# example = CONFIG['rules']['path_variables']['shot']['regex']
if self.scope == 'shots':
asset_rules = CONFIG['rules']['path_variables']['shot']['regex']
self.regex = re.compile(r'%s' % asset_rules)
if re.match(self.regex, asset_string):
self.asset_message_string = '%s is a valid %s name' % (
asset_string, self.scope)
# self.asset_widget.message.setPalette(self.black_palette)
self.asset_widget.message.setText(self.asset_message_string)
self.asset_widget.message.hide()
self.show_layout_items(self.task_layout)
self.create_button.setEnabled(True)
if asset_string in self.asset_list:
self.asset_message_string = '%s already Exists!' % asset_string
self.asset_widget.message.setText(self.asset_message_string)
# self.asset_widget.message.setPalette(self.red_palette)
self.asset_widget.message.show()
self.hide_layout_items(self.task_layout)
self.create_button.setEnabled(False)
else:
self.asset_message_string = '%s is not a valid %s name\n%s' % (
asset_string, self.scope, self.asset_string_example)
self.asset_widget.message.setText(self.asset_message_string)
# self.asset_widget.message.setPalette(self.red_palette)
if asset_string == '':
self.asset_widget.message.hide()
else:
self.asset_widget.message.show()
self.hide_layout_items(self.task_layout)
self.create_button.setEnabled(False)
def get_valid_categories(self):
categories = CONFIG['asset_categories']
for each in categories:
self.valid_categories.append(each)
self.valid_categories_string = '%s, %s' % (
self.valid_categories_string, each)
@staticmethod
def remove_spaces_on_list(list_):
new_list = []
for each in list_:
new_list.append(each.replace(' ', ''))
return new_list
def load_assets(self):
self.asset_widget.name_row.combo.clear()
glob_path = self.path_object.path_root
glob_path = re.sub('/+$', '', glob_path)
list_ = glob.glob(glob_path)
assets = ['']
for each in list_:
assets.append(os.path.split(each)[-1])
self.asset_list = assets
self.asset_widget.name_row.combo.addItems(assets)
def load_tasks(self):
task_list = self.proj_man_tasks
tasks = ['']
for each in task_list:
tasks.append(each)
@staticmethod
def is_valid_asset():
pass
@staticmethod
def is_valid_shot():
pass
def on_create_clicked(self):
for each in self.tasks:
if self.scope == 'assets':
self.path_object.set_attr(
asset=self.asset_widget.name_row.combo.currentText())
self.path_object.set_attr(type=self.seq)
elif self.scope == 'shots':
self.path_object.set_attr(
shot=self.asset_widget.name_row.combo.currentText())
self.path_object.set_attr(seq=self.seq)
self.path_object.set_attr(task=each)
CreateProductionData(self.path_object.data,
project_management=self.project_management)
self.accept()
self.close()
