class ElementList(QtGui.QListWidget):
def __init__(self, parent):
super(ElementList, self).__init__()
self.parent = parent
self.project = Project()
self.elements = self.project.list_assets()
self.initUI()
def initUI(self):
#define gui elements
self.refreshList('Asset')
#Update the list based on the input element type
def refreshList(self, element):
if element == 'Asset':
self.elements = self.project.list_assets()
else:
self.elements = self.project.list_shots()
self.clear()
for e in self.elements:
self.addItem(e)
def setElement(self, element):
for idx in xrange(self.count()):
eItem = self.item(idx)
if str(eItem.text())==element:
self.setCurrentRow(idx)
break
class AssetListWindow(QtGui.QListWidget):
def __init__(self, parent):
super(AssetListWindow, self).__init__()
self.parent = parent
self.current_selection = None
self.project = Project()
self.initUI()
def initUI(self):
self.currentItemChanged.connect(self.set_current_item)
def set_current_item(self, index):
self.current_selection = str(index.text())
body = self.project.get_body(self.current_selection)
element = body.get_element(str(self.parent.departmentMenu.currentText()))
path = element.get_app_filepath()
self.parent.filePath = path
def refreshList(self, department):
if department in Department.FRONTEND:
asset_filter = None
if(self.typeFilter.currentIndex()):
asset_filter_str = str(self.typeFilter.currentText())
asset_filter = (Asset.TYPE, operator.eq, asset_filter_str)
self.elements = self.project.list_assets(asset_filter)
else:
self.elements = self.project.list_shots()
self.clear()
for e in self.elements:
self.addItem(e)
class ElementList(QtGui.QListWidget):
def __init__(self, parent):
super(ElementList, self).__init__()
self.parent = parent
self.project = Project()
self.elements = self.project.list_assets()
self.initUI()
def initUI(self):
#define gui elements
self.refreshList('Asset')
#Update the list based on the input element type
def refreshList(self, element):
if element == 'Asset':
self.elements = self.project.list_assets()
else:
self.elements = self.project.list_shots()
self.clear()
for e in self.elements:
self.addItem(e)
def setElement(self, element):
for idx in xrange(self.count()):
eItem = self.item(idx)
if str(eItem.text())==element:
self.setCurrentRow(idx)
break
class ElementList(QtWidgets.QListWidget):
def __init__(self, parent):
super(ElementList, self).__init__()
self.parent = parent
self.project = Project()
self.elements = self.project.list_assets()
self.initUI()
def initUI(self):
#define gui elements
self.refreshList('Asset')
#Update the list based on the input element type
def refreshList(self, element):
if element == 'Asset':
self.elements = self.project.list_assets()
elif element == 'Shot':
self.elements = self.project.list_shots()
elif element == 'Tool':
self.elements = self.project.list_tools()
elif element == 'CrowdCycle':
self.elements = self.project.list_crowd_cycles()
else:
self.elements = list()
message_gui.error(
'There was a problem loading in the elements from of ' +
str(element) + ' type.')
self.clear()
for e in self.elements:
self.addItem(e)
def setElement(self, element):
for idx in xrange(self.count()):
eItem = self.item(idx)
if str(eItem.text()) == element:
self.setCurrentRow(idx)
break
class CheckoutWindow(QtGui.QWidget):
finished = QtCore.pyqtSignal()
def __init__(self, parent, dept_list=Department.ALL):
super(CheckoutWindow, self).__init__()
self.parent = parent
self.project = Project()
self.environment = Environment()
self.initUI(dept_list)
def initUI(self, dept_list):
#define gui elements
self.resize(WINDOW_WIDTH, WINDOW_HEIGHT)
self.setWindowTitle('Checkout')
self.dept_tabs = QtGui.QTabWidget()
self.dept_list = dept_list
self.result = None
#create tabs
for dept in dept_list:
tab = DepartmentTab(self)
self.dept_tabs.addTab(tab, dept)
tab_layout = QtGui.QHBoxLayout()
element_list = QtGui.QListWidget()
commentBox = QtGui.QTextEdit()
commentBox.setReadOnly(True)
tab.commentBox = commentBox
if dept in Department.FRONTEND:
for asset in self.project.list_assets():
item = QtGui.QListWidgetItem(asset)
element_list.addItem(item)
element_list.currentItemChanged.connect(
self.set_current_item)
elif dept in Department.BACKEND:
for shot in self.project.list_shots():
item = QtGui.QListWidgetItem(shot)
element_list.addItem(item)
element_list.currentItemChanged.connect(
self.set_current_item)
tab_layout.addWidget(element_list)
tab_layout.addWidget(commentBox)
tab.setLayout(tab_layout)
#create buttons
self.checkout_button = QtGui.QPushButton('Checkout')
self.checkout_button.clicked.connect(self.checkout)
self.cancel_button = QtGui.QPushButton('Cancel')
self.cancel_button.clicked.connect(self.close)
#create button layout
button_layout = QtGui.QHBoxLayout()
button_layout.addWidget(self.checkout_button)
button_layout.addWidget(self.cancel_button)
self.img = QtGui.QLabel()
pixmap = QtGui.QPixmap(os.environ['BYU_TOOLS_DIR'] +
'/byugui/assets/images/taijitu.jpg')
scaled = pixmap.scaledToWidth(self.size().width())
self.img.setPixmap(scaled)
#create main layout
main_layout = QtGui.QVBoxLayout()
self.setLayout(main_layout)
main_layout.addWidget(self.img)
main_layout.setSpacing(5)
main_layout.setMargin(6)
main_layout.addWidget(self.dept_tabs)
main_layout.addLayout(button_layout)
self.show()
def set_current_item(self, index):
current_dept = self.dept_list[self.dept_tabs.currentIndex()]
if current_dept in Department.FRONTEND:
self.current_item = str(index.text())
elif current_dept in Department.BACKEND:
self.current_item = str(index.text())
asset_obj = self.project.get_body(self.current_item)
element_obj = asset_obj.get_element(current_dept)
last_publish = element_obj.get_last_publish()
last_publish_comment = None
if last_publish is not None:
last_publish_comment = "Last published {0} by {1} \n \"{2}\"".format(
last_publish[1], last_publish[0], last_publish[2])
else:
last_publish_comment = "No publishes for this element"
currentTab = self.dept_tabs.currentWidget()
currentTab.commentBox.setText(last_publish_comment)
def checkout(self):
"""
Checks out the currently selected item
:return:
"""
current_user = self.environment.get_current_username()
current_dept = self.dept_list[self.dept_tabs.currentIndex()]
asset_obj = self.project.get_body(self.current_item)
element_obj = asset_obj.get_element(current_dept)
element_path = element_obj.checkout(current_user)
if element_path != None:
self.result = element_path
self.close()
def closeEvent(self, event):
self.finished.emit()
event.accept()
class ElementBrowser(QtGui.QWidget):
ASSETS = "Assets"
SHOTS = "Shots"
BODY_DATA_COLUMN = 1
BODY_DESCRIPTION_COLUMN = 7
@staticmethod
def dark_palette():
palette = QtGui.QPalette()
base_color = QtGui.QColor(39,39,39)
alt_color = QtGui.QColor(30,30,30)
text_color = QtGui.QColor(192,192,192)
highlight_color = QtGui.QColor(57,86,115)
highlight_text_color = QtCore.Qt.white
disabled_alt_color = QtGui.QColor(49,49,49)
disabled_base_color = QtGui.QColor(40,40,40)
disabled_text_color = QtGui.QColor(100,100,100)
palette.setColor(QtGui.QPalette.Window, base_color)
palette.setColor(QtGui.QPalette.WindowText, text_color)
palette.setColor(QtGui.QPalette.Base, base_color)
palette.setColor(QtGui.QPalette.AlternateBase, alt_color)
palette.setColor(QtGui.QPalette.ToolTipBase, alt_color)
palette.setColor(QtGui.QPalette.ToolTipText, text_color)
palette.setColor(QtGui.QPalette.Button, base_color)
palette.setColor(QtGui.QPalette.ButtonText, text_color)
palette.setColor(QtGui.QPalette.Text, text_color)
palette.setColor(QtGui.QPalette.Highlight, highlight_color)
palette.setColor(QtGui.QPalette.HighlightedText, highlight_text_color)
palette.setColor(QtGui.QPalette.Disabled, QtGui.QPalette.Window, disabled_base_color)
palette.setColor(QtGui.QPalette.Disabled, QtGui.QPalette.WindowText, disabled_text_color)
palette.setColor(QtGui.QPalette.Disabled, QtGui.QPalette.Base, disabled_text_color)
palette.setColor(QtGui.QPalette.Disabled, QtGui.QPalette.AlternateBase, disabled_alt_color)
palette.setColor(QtGui.QPalette.Disabled, QtGui.QPalette.Button, disabled_base_color)
palette.setColor(QtGui.QPalette.Disabled, QtGui.QPalette.ButtonText, disabled_text_color)
palette.setColor(QtGui.QPalette.Disabled, QtGui.QPalette.Text, disabled_text_color)
return palette
def __init__(self):
QtGui.QWidget.__init__(self)
self.setWindowTitle("Element Browser")
self.setGeometry(0, 0, REF_WINDOW_WIDTH, REF_WINDOW_HEIGHT)
self.palette = self.dark_palette()
self.setPalette(self.palette)
# initialize project
self.project = Project()
self.user_list = self.project.list_users()
self.user_completer = QtGui.QCompleter(self.user_list)
#filters
self.filter_label = QtGui.QLabel("Filter by: ")
self.dept_filter_label = QtGui.QLabel("Department")
self.dept_filter_label.setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter)
self.dept_filter = QtGui.QComboBox()
self.dept_filter.addItem("all")
for each in Department.ALL:
self.dept_filter.addItem(each)
self.dept_list = Department.ALL
self.type_filter_label = QtGui.QLabel("Asset Type")
self.type_filter_label.setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter)
self.type_filter = QtGui.QComboBox()
self.type_filter.addItem("all")
for each in AssetType.ALL:
self.type_filter.addItem(each)
self.name_filter_label = QtGui.QLabel("Name")
self.name_filter_label.setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter)
self.name_filter = QtGui.QLineEdit()
# menu bar
self.menu_bar = QtGui.QMenuBar()
self.view_menu = QtGui.QMenu("View")
self.menu_bar.addMenu(self.view_menu)
self.expand_action = self.view_menu.addAction("Expand All")
self.user_list_action = self.view_menu.addAction("User Directory")
self.theme_action = self.view_menu.addAction("Default Theme")
self.theme_action.setCheckable(True)
# asset/shot menu
self.body_menu = QtGui.QComboBox()
self.body_menu.addItem(self.ASSETS)
self.body_menu.addItem(self.SHOTS)
self.current_body = self.ASSETS
self._set_bodies()
# new button
self.new_button = QtGui.QPushButton("New")
# refresh button
self.refresh_button = QtGui.QPushButton("Refresh")
# tree
self.tree = QtGui.QTreeWidget()
self.tree.setItemDelegate(TreeGridDelegate(self.tree))
self.columnCount = 8
self.tree.setColumnCount(self.columnCount)
tree_header = QtGui.QTreeWidgetItem(["name", "", "assigned", "status", "start", "end", "publish", "note"])
self.tree.setHeaderItem(tree_header)
self.init_tree = [None]*self.columnCount
self.init_tree[0] = self.init_name
self.init_tree[1] = self.init_dept
self.init_tree[2] = self.init_assigned_user
self.init_tree[3] = self.init_status
self.init_tree[4] = self.init_start_date
self.init_tree[5] = self.init_end_date
self.init_tree[6] = self.init_last_publish
self.init_tree[7] = self.init_note
self._build_tree()
self.update_tree = [None]*self.columnCount
self.update_tree[0] = self.update_name
self.update_tree[1] = self.update_dept
self.update_tree[2] = self.update_assigned_user
self.update_tree[3] = self.update_status
self.update_tree[4] = self.update_start_date
self.update_tree[5] = self.update_end_date
self.update_tree[6] = self.update_last_publish
self.update_tree[7] = self.update_note
# status bar
self.status_bar = QtGui.QStatusBar()
# connect events
self.expand_action.triggered.connect(self._expand_all)
self.user_list_action.triggered.connect(self._show_user_directory)
self.theme_action.triggered.connect(self._toggle_theme)
self.body_menu.currentIndexChanged.connect(self._body_changed)
self.new_button.clicked.connect(self._new_body)
self.refresh_button.clicked.connect(self._refresh)
self.tree.itemExpanded.connect(self._load_elements)
self.tree.itemChanged.connect(self._item_edited)
self.dept_filter.currentIndexChanged.connect(self._dept_filter_changed)
self.name_filter.editingFinished.connect(self._filter_by_name)
self.type_filter.currentIndexChanged.connect(self._refresh)
# layout
layout = QtGui.QVBoxLayout(self)
layout.setSpacing(5)
layout.setMargin(6)
options_layout = QtGui.QGridLayout()
options_layout.addWidget(self.body_menu, 0, 0)
options_layout.addWidget(self.new_button, 0, 1)
options_layout.addWidget(self.refresh_button, 0, 3)
options_layout.setColumnMinimumWidth(0, 100)
options_layout.setColumnMinimumWidth(1, 100)
options_layout.setColumnMinimumWidth(3, 100)
options_layout.setColumnStretch(2, 1)
filter_layout = QtGui.QGridLayout()
filter_layout.addWidget(self.filter_label, 0, 0)
filter_layout.addWidget(self.dept_filter_label, 0, 1)
filter_layout.addWidget(self.dept_filter, 0, 2)
filter_layout.addWidget(self.name_filter_label, 0, 3)
filter_layout.addWidget(self.name_filter, 0, 4)
filter_layout.addWidget(self.type_filter_label, 0, 5)
filter_layout.addWidget(self.type_filter, 0, 6)
filter_layout.setColumnMinimumWidth(0, 50)
filter_layout.setColumnMinimumWidth(1, 100)
filter_layout.setColumnMinimumWidth(2, 100)
filter_layout.setColumnMinimumWidth(3, 100)
filter_layout.setColumnMinimumWidth(4, 100)
filter_layout.setColumnMinimumWidth(5, 100)
filter_layout.setColumnMinimumWidth(6, 100)
cal = QtGui.QCalendarWidget()
filter_layout.setColumnStretch(7, 1)
layout.addWidget(self.menu_bar)
layout.addLayout(options_layout)
layout.addWidget(self.tree)
layout.addLayout(filter_layout)
# layout.addWidget(self.filter_label)
# layout.addWidget(self.type_filter)
# layout.addWidget(self.dept_filter)
layout.addWidget(self.status_bar)
self.setLayout(layout)
request_email.check_user_email(self)
def _build_tree(self):
self.tree.clear()
tree_state = self.tree.blockSignals(True)
for body in self.bodies:
if(str(self.name_filter.text()) in body):
tree_item = QtGui.QTreeWidgetItem([body])
self.tree.addTopLevelItem(tree_item)
tree_flags = tree_item.flags()
tree_item.setFlags(tree_flags | QtCore.Qt.ItemIsEditable)
# for col in xrange(self.columnCount):
# tree_item.setBackground(col, QtGui.QColor(30,30,30))
body_obj = self.project.get_body(body)
self._load_body(body_obj, tree_item)
tree_item.addChild(QtGui.QTreeWidgetItem()) # empty item
self.tree.blockSignals(tree_state)
def _load_body(self, body, item):
tree_state = self.tree.blockSignals(True)
item.setText(0, body.get_name())
namelabel = TreeLabel(body.get_name())
self.tree.setItemWidget(item, 0, namelabel)
if self.current_body==self.ASSETS:
body_type = body.get_type()
item.setText(self.BODY_DATA_COLUMN, body_type)
combobox = TreeComboBoxItem(item, self.BODY_DATA_COLUMN)
type_idx = 0
for idx, type in enumerate(AssetType.ALL):
combobox.addItem(type)
if type == body_type:
type_idx = idx
combobox.setCurrentIndex(type_idx)
self.tree.setItemWidget(item, self.BODY_DATA_COLUMN, combobox)
elif self.current_body==self.SHOTS:
item.setText(self.BODY_DATA_COLUMN, str(body.get_frame_range()))
else:
self.status_bar.showMessage("Error: unknown body type")
description_edit = TreeLineEdit(body.get_description(), item, self.BODY_DESCRIPTION_COLUMN)
self.tree.setItemWidget(item, self.BODY_DESCRIPTION_COLUMN, description_edit)
for col in xrange(self.BODY_DATA_COLUMN+1, self.columnCount-1): # disable remaining columns
emptylabel = TreeLabel()
self.tree.setItemWidget(item, col, emptylabel)
self.tree.blockSignals(tree_state)
def _load_elements(self, item):
tree_state = self.tree.blockSignals(True)
body = str(item.text(0))
body_obj = self.project.get_body(body)
elements = []
for dept in self.dept_list:
dept_elements = body_obj.list_elements(dept)
for dept_element in dept_elements:
elements.append((dept, dept_element))
item.takeChildren() # clear children
for dept, element in elements:
element_obj = body_obj.get_element(dept, element)
child_item = QtGui.QTreeWidgetItem()
item.addChild(child_item)
child_item.setFlags(child_item.flags() | QtCore.Qt.ItemIsEditable)
for col, init in enumerate(self.init_tree):
init(element_obj, child_item, col)
self.tree.blockSignals(tree_state)
def _expand_all(self):
# self.tree.expandAll()
count = self.tree.topLevelItemCount()
for i in xrange(count):
item = self.tree.topLevelItem(i)
self.tree.expandItem(item)
def _show_user_directory(self):
user_directory = UserListDialog(self)
user_directory.show()
def _toggle_theme(self):
checked = self.theme_action.isChecked()
if not checked:
self.palette = self.dark_palette()
else:
self.palette = QtGui.QPalette()
self.setPalette(self.palette)
def _set_bodies(self):
if self.current_body == self.ASSETS:
asset_filter = None
if(self.type_filter.currentIndex()):
asset_filter_str = str(self.type_filter.currentText())
asset_filter = (Asset.TYPE, operator.eq, asset_filter_str)
self.bodies = self.project.list_assets(asset_filter)
elif self.current_body == self.SHOTS:
self.bodies = self.project.list_shots()
else:
self.bodies = []
def _item_edited(self, item, column):
parent = item.parent()
if parent is not None:
body = str(parent.text(0))
body_obj = self.project.get_body(body)
element = str(item.text(0))
dept = str(item.text(1))
element_obj = body_obj.get_element(dept, element)
self.update_tree[column](element_obj, item, column)
# self.tree.resizeColumnToContents(column)
else:
body = str(item.text(0))
body_obj = self.project.get_body(body)
if column==self.BODY_DATA_COLUMN:
self._update_body_data(body_obj, item)
elif column==self.BODY_DESCRIPTION_COLUMN:
self._update_body_description(body_obj, item)
def _refresh(self): # TODO: maintain expanded rows on refresh
self._set_bodies()
self._build_tree()
self.status_bar.clearMessage()
def _body_changed(self, index):
self.current_body = str(self.body_menu.itemText(index))
if(self.body_menu.currentIndex()):
self.type_filter.setEnabled(False)
self.type_filter_label.setEnabled(False)
else:
self.type_filter.setEnabled(True)
self.type_filter_label.setEnabled(True)
self._refresh()
def _dept_filter_changed(self):
if(self.dept_filter.currentIndex()):
self.dept_list = [str(self.dept_filter.currentText())]
else:
self.dept_list = Department.ALL
self._refresh()
def _filter_by_name(self):
self._refresh()
# target = str(self.name_filter.text())
# for i in reversed(xrange(self.tree.topLevelItemCount())):
# if (target not in self.tree.topLevelItem(i).text(0)):
# self.tree.takeTopLevelItem(i)
def _new_body(self):
from byugui import new_asset_gui
self.new_body_dialog = new_asset_gui.CreateWindow(self)
if self.current_body == self.ASSETS:
self.new_body_dialog.setCurrentIndex(self.new_body_dialog.ASSET_INDEX)
elif self.current_body == self.SHOTS:
self.new_body_dialog.setCurrentIndex(self.new_body_dialog.SHOT_INDEX)
self.new_body_dialog.finished.connect(self._refresh)
def _update_body_data(self, body, item):
if self.current_body==self.ASSETS:
body.update_type(str(item.text(self.BODY_DATA_COLUMN)))
elif self.current_body==self.SHOTS:
body.update_frame_range(int(item.text(self.BODY_DATA_COLUMN)))
else:
self.status_bar.showMessage("Error: unknown body type")
def _update_body_description(self, body, item):
body.update_description(str(item.text(self.BODY_DESCRIPTION_COLUMN)))
def _valid_date(self, date):
try:
date_obj = datetime.datetime.strptime(date, "%Y-%m-%d").date()
return str(date_obj)
except ValueError:
self.status_bar.showMessage(date+" not a valid date, please use format: YYYY-MM-DD")
return None
def init_name(self, element, item, column):
item.setText(column, element.get_name())
namelabel = TreeLabel(element.get_name())
self.tree.setItemWidget(item, column, namelabel)
def init_dept(self, element, item, column):
item.setText(column, element.get_department())
deptlabel = TreeLabel(element.get_department())
self.tree.setItemWidget(item, column, deptlabel)
def init_assigned_user(self, element, item, column):
user = element.get_assigned_user()
item.setText(column, user)
lineedit = TreeLineEdit(user, item, column)
lineedit.setCompleter(self.user_completer)
self.tree.setItemWidget(item, column, lineedit)
def init_status(self, element, item, column):
item.setText(column, element.get_status())
combobox = TreeComboBoxItem(item, column)
element_type = element.get_status()
type_idx = 0
for idx, type in enumerate(Status.ALL):
combobox.addItem(type)
if type == element_type:
type_idx = idx
combobox.setCurrentIndex(type_idx)
self.tree.setItemWidget(item, column, combobox)
def init_start_date(self, element, item, column):
item.setText(column, element.get_start_date())
def init_end_date(self, element, item, column):
item.setText(column, element.get_end_date())
def init_last_publish(self, element, item, column):
publish = element.get_last_publish()
if publish is not None:
item.setText(column, publish[0]+", "+publish[1]+", "+publish[2])
else:
item.setText(column, "")
def init_note(self, element, item, column):
item.setText(column, element.get_last_note())
def update_name(self, element, item, column):
self.status_bar.showMessage("can't change name")
def update_dept(self, element, item, column):
self.status_bar.showMessage("can't change department")
def update_assigned_user(self, element, item, column):
user = str(item.text(column))
if user in self.user_list:
element.update_assigned_user(user)
self.status_bar.clearMessage()
else:
self.tree.itemWidget(item, column).setText(element.get_assigned_user())
self.status_bar.showMessage('"' + user + '" is not a valid username')
def update_status(self, element, item, column):
element.update_status(str(item.text(column)))
self.status_bar.clearMessage()
def update_start_date(self, element, item, column):
date_str = str(item.text(column))
valid_date_str = self._valid_date(date_str)
if valid_date_str:
element.update_start_date(valid_date_str)
self.status_bar.clearMessage()
else:
self.init_start_date(element, item, column)
def update_end_date(self, element, item, column):
date_str = str(item.text(column))
valid_date_str = self._valid_date(date_str)
if valid_date_str:
element.update_end_date(valid_date_str)
self.status_bar.clearMessage()
else:
self.init_end_date(element, item, column)
def update_last_publish(self, element, item, column):
self.status_bar.showMessage("can't modify publish data")
self.init_last_publish(element, item, column)
def update_note(self, element, item, column):
element.update_notes(str(item.text(column)))
self.status_bar.clearMessage()
class ElementList(QtWidgets.QTreeWidget):
def __init__(self, parent):
super(ElementList, self).__init__()
self.parent = parent
self.project = Project()
self.elements = self.project.list_assets()
self.initUI()
def initUI(self):
#define gui elements
self.refreshList('Asset')
def recurseTree(self, treeItem, array, asset):
#This is for setting bottom level text attributes
if len(array) == 0:
treeItem.setText(1, asset)
treeItem.setTextColor(0, "#3c83f9")
font = QtGui.QFont()
font.setPointSize(12)
font.setBold(False)
treeItem.setFont(0, font)
return
else: #This is for setting parent level text attributes and settin up the structure
item = QtWidgets.QTreeWidgetItem(array[0])
item.setText(0, array[0])
item.setText(1, "This is not a file")
item.setTextColor(0, "#d0d0d0")
font = QtGui.QFont()
font.setPointSize(11)
font.setBold(True)
item.setFont(0, font)
skip = False
# this is to check if the child already exists
for i in range(0, treeItem.childCount()):
if treeItem.child(i).text(0) == item.text(0):
item = treeItem.child(i)
skip = True
if skip == False: # Executes if the child doesnt already exist
treeItem.addChild(item)
newArray = array[1:]
self.recurseTree(item, newArray, asset)
return
#Update the list based on the input element type
def refreshList(self, element):
if element == 'Asset':
self.elements = self.project.list_assets()
elif element == 'Shot':
self.elements = self.project.list_shots()
elif element == 'Tool':
self.elements = self.project.list_tools()
elif element == 'CrowdCycle':
self.elements = self.project.list_crowd_cycles()
else:
self.elements = list()
message_gui.error(
'There was a problem loading in the elements from of ' +
str(element) + ' type.')
self.clear()
for e in self.elements:
asset_array = e.split("_")
firstelement = self.findItems(asset_array[0], 0, 0)
if not firstelement:
item = QtWidgets.QTreeWidgetItem(asset_array[0])
item.setText(0, asset_array[0])
item.setTextColor(0, "#d0d0d0")
font = QtGui.QFont()
font.setPointSize(11)
font.setBold(True)
item.setFont(0, font)
self.recurseTree(item, asset_array[1:], e)
self.insertTopLevelItem(0, item)
else:
self.recurseTree(firstelement[0], asset_array[1:], e)
#self.addItem(e)
def setElement(self, element):
print(element)
class CheckoutWindow(QtGui.QWidget):
finished = QtCore.pyqtSignal()
def __init__(self, parent, dept_list=Department.ALL):
super(CheckoutWindow, self).__init__()
self.parent = parent
self.project = Project()
self.environment = Environment()
self.initUI(dept_list)
def initUI(self, dept_list):
#define gui elements
self.resize(WINDOW_WIDTH,WINDOW_HEIGHT)
self.setWindowTitle('Checkout')
self.dept_tabs = QtGui.QTabWidget()
self.dept_list = dept_list
self.result = None
#create tabs
for dept in dept_list:
tab = DepartmentTab(self)
#tab = QtGui.QWidget()
# self.dept_tabs.insertTab(self.ASSET_INDEX, tab, dept)
self.dept_tabs.addTab(tab, dept)
tab_layout = QtGui.QVBoxLayout()
element_list = QtGui.QListWidget()
if dept in Department.FRONTEND:
for asset in self.project.list_assets():
item = QtGui.QListWidgetItem(asset)
element_list.addItem(item)
element_list.currentItemChanged.connect(self.set_current_item)
elif dept in Department.BACKEND:
for shot in self.project.list_shots():
item = QtGui.QListWidgetItem(shot)
element_list.addItem(item)
element_list.currentItemChanged.connect(self.set_current_item)
tab_layout.addWidget(element_list)
tab.setLayout(tab_layout)
#create buttons
self.checkout_button = QtGui.QPushButton('Checkout')
self.checkout_button.clicked.connect(self.checkout)
self.cancel_button = QtGui.QPushButton('Cancel')
self.cancel_button.clicked.connect(self.close)
#create button layout
button_layout = QtGui.QHBoxLayout()
#button_layout.setSpacing(2)
button_layout.addWidget(self.checkout_button)
button_layout.addWidget(self.cancel_button)
self.img = QtGui.QLabel()
pixmap = QtGui.QPixmap(os.environ['BYU_TOOLS_DIR'] + '/byugui/assets/images/taijitu.jpg')
scaled = pixmap.scaledToWidth(self.size().width())
self.img.setPixmap(scaled)
#create main layout
main_layout = QtGui.QVBoxLayout()
self.setLayout(main_layout)
main_layout.addWidget(self.img)
main_layout.setSpacing(5)
main_layout.setMargin(6)
main_layout.addWidget(self.dept_tabs)
main_layout.addLayout(button_layout)
self.show()
def set_current_item(self, index):
current_dept = self.dept_list[self.dept_tabs.currentIndex()]
if current_dept in Department.FRONTEND:
self.current_item = str(index.text())
elif current_dept in Department.BACKEND:
self.current_item = str(index.text())
def checkout(self):
"""
Checks out the currently selected item
:return:
"""
current_user = self.environment.get_current_username()
current_dept = self.dept_list[self.dept_tabs.currentIndex()]
asset_obj = self.project.get_body(self.current_item)
element_obj = asset_obj.get_element(current_dept)
element_path = element_obj.checkout(current_user)
if element_path != None:
# self.parent.close()
self.result = element_path
self.close()
def closeEvent(self, event):
self.finished.emit()
event.accept()
class SelectionWindow(QtWidgets.QDialog):
finished = QtCore.Signal()
def __init__(self, parent, dept_list=Department.ALL, showWindow=True):
super(SelectionWindow, self).__init__()
self.parent = parent
self.project = Project()
self.environment = Environment()
self.initUI(dept_list, showWindow)
def initUI(self, dept_list, showWindow):
#define gui elements
self.resize(WINDOW_WIDTH, WINDOW_HEIGHT)
self.setWindowTitle('Select Asset')
self.dept_tabs = QtWidgets.QTabWidget()
self.dept_list = dept_list
self.result = None
self.instructions = QtWidgets.QLabel(
'Select the asset you would like affect:')
#create Tabs
self.createTabs()
#create buttons
self.select_button = QtWidgets.QPushButton('Select')
self.select_button.clicked.connect(self.select)
self.cancel_button = QtWidgets.QPushButton('Cancel')
self.cancel_button.clicked.connect(self.close)
#create button layout
button_layout = QtWidgets.QHBoxLayout()
button_layout.addWidget(self.select_button)
button_layout.addWidget(self.cancel_button)
self.img = QtWidgets.QLabel()
image_path = os.path.join(os.environ['BYU_TOOLS_DIR'], 'byugui',
'assets', 'images', 'film-banner.jpg')
pixmap = QtGui.QPixmap(image_path)
scaled = pixmap.scaledToWidth(self.size().width())
self.img.setPixmap(scaled)
#create main layout
main_layout = QtWidgets.QVBoxLayout()
self.setLayout(main_layout)
main_layout.addWidget(self.img)
main_layout.setSpacing(5)
# main_layout.setMargin(6)
main_layout.addWidget(self.instructions)
main_layout.addWidget(self.dept_tabs)
main_layout.addLayout(button_layout)
if showWindow:
self.show()
def createTabs(self):
#remember the current index so that we can restore it when we create the tabs
currIndex = self.dept_tabs.currentIndex()
#clear out the old tabs
self.dept_tabs.clear()
#create tabs
for dept in self.dept_list:
tab = DepartmentTab(self)
self.dept_tabs.addTab(tab, dept)
tab_layout = QtWidgets.QHBoxLayout()
element_list = QtWidgets.QListWidget()
if dept in Department.ASSET_DEPTS:
for asset in self.project.list_assets():
item = QtWidgets.QListWidgetItem(asset)
element_list.addItem(item)
element_list.currentItemChanged.connect(
self.set_current_item)
elif dept in Department.SHOT_DEPTS:
for shot in self.project.list_shots():
item = QtWidgets.QListWidgetItem(shot)
element_list.addItem(item)
element_list.currentItemChanged.connect(
self.set_current_item)
tab_layout.addWidget(element_list)
tab.setLayout(tab_layout)
#restore the previous index
self.dept_tabs.setCurrentIndex(currIndex)
def set_current_item(self, index):
current_dept = self.dept_list[self.dept_tabs.currentIndex()]
if current_dept in Department.ASSET_DEPTS:
self.current_item = str(index.text())
elif current_dept in Department.SHOT_DEPTS:
self.current_item = str(index.text())
asset_obj = self.project.get_body(self.current_item)
element_obj = asset_obj.get_element(current_dept)
last_publish = element_obj.get_last_publish()
last_publish_comment = None
if last_publish is not None:
last_publish_comment = 'Last published {0} by {1} \n \"{2}\"'.format(
last_publish[1], last_publish[0], last_publish[2])
else:
last_publish_comment = 'No publishes for this element'
currentTab = self.dept_tabs.currentWidget()
def hasPreviousPublish(self, body, department):
asset_obj = self.project.get_body(body)
element_obj = asset_obj.get_element(department)
last_publish = element_obj.get_last_publish()
if last_publish is None:
return False
return True
def select(self):
'''
Selects the currently selected item
:return:
'''
current_user = self.environment.get_current_username()
current_dept = self.dept_list[self.dept_tabs.currentIndex()]
asset_obj = self.project.get_body(self.current_item)
element_obj = asset_obj.get_element(current_dept)
if element_obj != None:
self.result = element_obj
self.close()
def closeEvent(self, event):
self.finished.emit()
event.accept()
class ReferenceWindow(QtGui.QWidget):
finished = QtCore.pyqtSignal()
def __init__(self, parent, src, dept_list=Department.ALL):
super(ReferenceWindow, self).__init__()
self.project = Project()
self.parent = parent
self.src = src
self.filePaths = []
self.done = True
self.reference = False
self.initUI(dept_list)
def initUI(self, dept_list):
#define gui elements
self.setGeometry(300, 300, WINDOW_WIDTH, WINDOW_HEIGHT)
self.setWindowTitle('Taijitu Reference Manager')
self.departmentMenu = QtGui.QComboBox()
for i in dept_list:
self.departmentMenu.addItem(i)
self.departmentMenu.activated[str].connect(self.setElementType)
self.assetList = AssetListWindow(self)
for asset in self.project.list_assets():
item = QtGui.QListWidgetItem(asset)
self.assetList.addItem(item)
self.typeFilterLabel = QtGui.QLabel("Type Filter")
self.typeFilterLabel.setAlignment(QtCore.Qt.AlignRight
| QtCore.Qt.AlignVCenter)
self.typeFilter = QtGui.QComboBox()
self.typeFilter.addItem("all")
for i in AssetType.ALL:
self.typeFilter.addItem(i)
self.typeFilter.currentIndexChanged.connect(self.setElementType)
self.referenceButton = QtGui.QPushButton('Reference')
self.referenceButton.clicked.connect(self.createReference)
self.cancelButton = QtGui.QPushButton('Cancel')
self.cancelButton.clicked.connect(self.close)
#set gui layout
self.grid = QtGui.QGridLayout(self)
self.setLayout(self.grid)
self.grid.addWidget(self.departmentMenu, 0, 0)
self.grid.addWidget(self.assetList, 1, 0, 1, 0)
self.grid.addWidget(self.typeFilterLabel, 2, 0)
self.grid.addWidget(self.typeFilter, 2, 1)
self.grid.addWidget(self.referenceButton, 3, 0)
self.grid.addWidget(self.cancelButton, 3, 1)
self.show()
def setElementType(self, idx=0):
department = str(self.departmentMenu.currentText())
self.refreshList(department)
def createReference(self):
selected = []
del self.filePaths[:]
for item in self.assetList.selectedItems():
body = self.project.get_body(str(item.text()))
element = body.get_element(str(self.departmentMenu.currentText()))
path = element.get_app_filepath()
self.filePaths.append(path)
selected.append(str(item.text()))
checkout = self.project.get_checkout(os.path.dirname(self.src))
if checkout is not None:
body_name = checkout.get_body_name()
body = self.project.get_body(body_name)
for sel in selected:
body.add_reference(sel)
self.done = False
self.reference = True
self.close()
def refreshList(self, department):
if department in Department.FRONTEND:
asset_filter = None
if (self.typeFilter.currentIndex()):
asset_filter_str = str(self.typeFilter.currentText())
asset_filter = (Asset.TYPE, operator.eq, asset_filter_str)
self.elements = self.project.list_assets(asset_filter)
else:
self.elements = self.project.list_shots()
self.assetList.clear()
for e in self.elements:
self.assetList.addItem(e)
def closeEvent(self, event):
self.finished.emit()
event.accept()
class ElementBrowser(QtGui.QWidget):
ASSETS = "Assets"
SHOTS = "Shots"
BODY_DATA_COLUMN = 1
BODY_DESCRIPTION_COLUMN = 7
@staticmethod
def dark_palette():
palette = QtGui.QPalette()
base_color = QtGui.QColor(39,39,39)
alt_color = QtGui.QColor(30,30,30)
text_color = QtGui.QColor(192,192,192)
highlight_color = QtGui.QColor(57,86,115)
highlight_text_color = QtCore.Qt.white
disabled_alt_color = QtGui.QColor(49,49,49)
disabled_base_color = QtGui.QColor(40,40,40)
disabled_text_color = QtGui.QColor(100,100,100)
palette.setColor(QtGui.QPalette.Window, base_color)
palette.setColor(QtGui.QPalette.WindowText, text_color)
palette.setColor(QtGui.QPalette.Base, base_color)
palette.setColor(QtGui.QPalette.AlternateBase, alt_color)
palette.setColor(QtGui.QPalette.ToolTipBase, alt_color)
palette.setColor(QtGui.QPalette.ToolTipText, text_color)
palette.setColor(QtGui.QPalette.Button, base_color)
palette.setColor(QtGui.QPalette.ButtonText, text_color)
palette.setColor(QtGui.QPalette.Text, text_color)
palette.setColor(QtGui.QPalette.Highlight, highlight_color)
palette.setColor(QtGui.QPalette.HighlightedText, highlight_text_color)
palette.setColor(QtGui.QPalette.Disabled, QtGui.QPalette.Window, disabled_base_color)
palette.setColor(QtGui.QPalette.Disabled, QtGui.QPalette.WindowText, disabled_text_color)
palette.setColor(QtGui.QPalette.Disabled, QtGui.QPalette.Base, disabled_text_color)
palette.setColor(QtGui.QPalette.Disabled, QtGui.QPalette.AlternateBase, disabled_alt_color)
palette.setColor(QtGui.QPalette.Disabled, QtGui.QPalette.Button, disabled_base_color)
palette.setColor(QtGui.QPalette.Disabled, QtGui.QPalette.ButtonText, disabled_text_color)
palette.setColor(QtGui.QPalette.Disabled, QtGui.QPalette.Text, disabled_text_color)
return palette
def __init__(self):
QtGui.QWidget.__init__(self)
self.setWindowTitle("Element Browser")
self.setGeometry(0, 0, REF_WINDOW_WIDTH, REF_WINDOW_HEIGHT)
self.palette = self.dark_palette()
self.setPalette(self.palette)
# initialize project
self.project = Project()
self.user_list = self.project.list_users()
self.user_completer = QtGui.QCompleter(self.user_list)
#filters
self.filter_label = QtGui.QLabel("Filter by: ")
self.dept_filter_label = QtGui.QLabel("Department")
self.dept_filter_label.setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter)
self.dept_filter = QtGui.QComboBox()
self.dept_filter.addItem("all")
for each in Department.ALL:
self.dept_filter.addItem(each)
self.dept_list = Department.ALL
self.type_filter_label = QtGui.QLabel("Asset Type")
self.type_filter_label.setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter)
self.type_filter = QtGui.QComboBox()
self.type_filter.addItem("all")
for each in AssetType.ALL:
self.type_filter.addItem(each)
self.name_filter_label = QtGui.QLabel("Name")
self.name_filter_label.setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter)
self.name_filter = QtGui.QLineEdit()
# menu bar
self.menu_bar = QtGui.QMenuBar()
self.view_menu = QtGui.QMenu("View")
self.menu_bar.addMenu(self.view_menu)
self.expand_action = self.view_menu.addAction("Expand All")
self.user_list_action = self.view_menu.addAction("User Directory")
self.theme_action = self.view_menu.addAction("Default Theme")
self.theme_action.setCheckable(True)
# asset/shot menu
self.body_menu = QtGui.QComboBox()
self.body_menu.addItem(self.ASSETS)
self.body_menu.addItem(self.SHOTS)
self.current_body = self.ASSETS
self._set_bodies()
# new button
self.new_button = QtGui.QPushButton("New")
# refresh button
self.refresh_button = QtGui.QPushButton("Refresh")
# tree
self.tree = QtGui.QTreeWidget()
self.tree.setItemDelegate(TreeGridDelegate(self.tree))
self.columnCount = 8
self.tree.setColumnCount(self.columnCount)
tree_header = QtGui.QTreeWidgetItem(["name", "", "assigned", "status", "start", "end", "publish", "note"])
self.tree.setHeaderItem(tree_header)
tree_header_view = self.tree.header()
tree_header_view.resizeSection(4, 120)
tree_header_view.resizeSection(5, 120)
self.init_tree = [None]*self.columnCount
self.init_tree[0] = self.init_name
self.init_tree[1] = self.init_dept
self.init_tree[2] = self.init_assigned_user
self.init_tree[3] = self.init_status
self.init_tree[4] = self.init_start_date
self.init_tree[5] = self.init_end_date
self.init_tree[6] = self.init_last_publish
self.init_tree[7] = self.init_note
self._build_tree()
self.update_tree = [None]*self.columnCount
self.update_tree[0] = self.update_name
self.update_tree[1] = self.update_dept
self.update_tree[2] = self.update_assigned_user
self.update_tree[3] = self.update_status
self.update_tree[4] = self.update_start_date
self.update_tree[5] = self.update_end_date
self.update_tree[6] = self.update_last_publish
self.update_tree[7] = self.update_note
# status bar
self.status_bar = QtGui.QStatusBar()
# connect events
self.expand_action.triggered.connect(self._expand_all)
self.user_list_action.triggered.connect(self._show_user_directory)
self.theme_action.triggered.connect(self._toggle_theme)
self.body_menu.currentIndexChanged.connect(self._body_changed)
self.new_button.clicked.connect(self._new_body)
self.refresh_button.clicked.connect(self._refresh)
self.tree.itemExpanded.connect(self._load_elements)
self.tree.itemChanged.connect(self._item_edited)
self.dept_filter.currentIndexChanged.connect(self._dept_filter_changed)
self.name_filter.editingFinished.connect(self._filter_by_name)
self.type_filter.currentIndexChanged.connect(self._refresh)
# layout
layout = QtGui.QVBoxLayout(self)
layout.setSpacing(5)
layout.setMargin(6)
options_layout = QtGui.QGridLayout()
options_layout.addWidget(self.body_menu, 0, 0)
options_layout.addWidget(self.new_button, 0, 1)
options_layout.addWidget(self.refresh_button, 0, 3)
options_layout.setColumnMinimumWidth(0, 100)
options_layout.setColumnMinimumWidth(1, 100)
options_layout.setColumnMinimumWidth(3, 100)
options_layout.setColumnStretch(2, 1)
filter_layout = QtGui.QGridLayout()
filter_layout.addWidget(self.filter_label, 0, 0)
filter_layout.addWidget(self.dept_filter_label, 0, 1)
filter_layout.addWidget(self.dept_filter, 0, 2)
filter_layout.addWidget(self.name_filter_label, 0, 3)
filter_layout.addWidget(self.name_filter, 0, 4)
filter_layout.addWidget(self.type_filter_label, 0, 5)
filter_layout.addWidget(self.type_filter, 0, 6)
filter_layout.setColumnMinimumWidth(0, 50)
filter_layout.setColumnMinimumWidth(1, 100)
filter_layout.setColumnMinimumWidth(2, 100)
filter_layout.setColumnMinimumWidth(3, 100)
filter_layout.setColumnMinimumWidth(4, 100)
filter_layout.setColumnMinimumWidth(5, 100)
filter_layout.setColumnMinimumWidth(6, 100)
filter_layout.setColumnStretch(7, 1)
layout.addWidget(self.menu_bar)
layout.addLayout(options_layout)
layout.addWidget(self.tree)
layout.addLayout(filter_layout)
layout.addWidget(self.status_bar)
self.setLayout(layout)
request_email.check_user_email(self)
def _build_tree(self):
self.tree.clear()
tree_state = self.tree.blockSignals(True)
for body in self.bodies:
if(str(self.name_filter.text()) in body):
tree_item = QtGui.QTreeWidgetItem([body])
self.tree.addTopLevelItem(tree_item)
tree_flags = tree_item.flags()
tree_item.setFlags(tree_flags | QtCore.Qt.ItemIsEditable)
# for col in xrange(self.columnCount):
# tree_item.setBackground(col, QtGui.QColor(30,30,30))
body_obj = self.project.get_body(body)
self._load_body(body_obj, tree_item)
tree_item.addChild(QtGui.QTreeWidgetItem()) # empty item
self.tree.blockSignals(tree_state)
def _load_body(self, body, item):
tree_state = self.tree.blockSignals(True)
item.setText(0, body.get_name())
namelabel = TreeLabel(body.get_name())
self.tree.setItemWidget(item, 0, namelabel)
if self.current_body==self.ASSETS:
body_type = body.get_type()
item.setText(self.BODY_DATA_COLUMN, body_type)
combobox = TreeComboBoxItem(item, self.BODY_DATA_COLUMN)
type_idx = 0
for idx, type in enumerate(AssetType.ALL):
combobox.addItem(type)
if type == body_type:
type_idx = idx
combobox.setCurrentIndex(type_idx)
self.tree.setItemWidget(item, self.BODY_DATA_COLUMN, combobox)
elif self.current_body==self.SHOTS:
item.setText(self.BODY_DATA_COLUMN, str(body.get_frame_range()))
else:
self.status_bar.showMessage("Error: unknown body type")
description_edit = TreeLineEdit(body.get_description(), item, self.BODY_DESCRIPTION_COLUMN)
self.tree.setItemWidget(item, self.BODY_DESCRIPTION_COLUMN, description_edit)
for col in xrange(self.BODY_DATA_COLUMN+1, self.columnCount-1): # disable remaining columns
emptylabel = TreeLabel()
self.tree.setItemWidget(item, col, emptylabel)
self.tree.blockSignals(tree_state)
def _load_elements(self, item):
tree_state = self.tree.blockSignals(True)
body = str(item.text(0))
body_obj = self.project.get_body(body)
elements = []
for dept in self.dept_list:
dept_elements = body_obj.list_elements(dept)
for dept_element in dept_elements:
elements.append((dept, dept_element))
item.takeChildren() # clear children
for dept, element in elements:
element_obj = body_obj.get_element(dept, element)
child_item = QtGui.QTreeWidgetItem()
item.addChild(child_item)
child_item.setFlags(child_item.flags() | QtCore.Qt.ItemIsEditable)
for col, init in enumerate(self.init_tree):
init(element_obj, child_item, col)
self.tree.blockSignals(tree_state)
def _expand_all(self):
# self.tree.expandAll()
count = self.tree.topLevelItemCount()
for i in xrange(count):
item = self.tree.topLevelItem(i)
self.tree.expandItem(item)
def _show_user_directory(self):
user_directory = UserListDialog(self)
user_directory.show()
def _toggle_theme(self):
checked = self.theme_action.isChecked()
if not checked:
self.palette = self.dark_palette()
else:
self.palette = QtGui.QPalette()
self.setPalette(self.palette)
def _set_bodies(self):
if self.current_body == self.ASSETS:
asset_filter = None
if(self.type_filter.currentIndex()):
asset_filter_str = str(self.type_filter.currentText())
asset_filter = (Asset.TYPE, operator.eq, asset_filter_str)
self.bodies = self.project.list_assets(asset_filter)
elif self.current_body == self.SHOTS:
self.bodies = self.project.list_shots()
else:
self.bodies = []
def _item_edited(self, item, column):
parent = item.parent()
if parent is not None:
body = str(parent.text(0))
body_obj = self.project.get_body(body)
element = str(item.text(0))
dept = str(item.text(1))
element_obj = body_obj.get_element(dept, element)
self.update_tree[column](element_obj, item, column)
# self.tree.resizeColumnToContents(column)
else:
body = str(item.text(0))
body_obj = self.project.get_body(body)
if column==self.BODY_DATA_COLUMN:
self._update_body_data(body_obj, item)
elif column==self.BODY_DESCRIPTION_COLUMN:
self._update_body_description(body_obj, item)
def _refresh(self): # TODO: maintain expanded rows on refresh
self._set_bodies()
self._build_tree()
self.status_bar.clearMessage()
def _body_changed(self, index):
self.current_body = str(self.body_menu.itemText(index))
if(self.body_menu.currentIndex()):
self.type_filter.setEnabled(False)
self.type_filter_label.setEnabled(False)
else:
self.type_filter.setEnabled(True)
self.type_filter_label.setEnabled(True)
self._refresh()
def _dept_filter_changed(self):
if(self.dept_filter.currentIndex()):
self.dept_list = [str(self.dept_filter.currentText())]
else:
self.dept_list = Department.ALL
self._refresh()
def _filter_by_name(self):
self._refresh()
# target = str(self.name_filter.text())
# for i in reversed(xrange(self.tree.topLevelItemCount())):
# if (target not in self.tree.topLevelItem(i).text(0)):
# self.tree.takeTopLevelItem(i)
def _new_body(self):
from byugui import new_asset_gui
self.new_body_dialog = new_asset_gui.CreateWindow(self)
if self.current_body == self.ASSETS:
self.new_body_dialog.setCurrentIndex(self.new_body_dialog.ASSET_INDEX)
elif self.current_body == self.SHOTS:
self.new_body_dialog.setCurrentIndex(self.new_body_dialog.SHOT_INDEX)
self.new_body_dialog.finished.connect(self._refresh)
def _update_body_data(self, body, item):
if self.current_body==self.ASSETS:
body.update_type(str(item.text(self.BODY_DATA_COLUMN)))
elif self.current_body==self.SHOTS:
body.update_frame_range(int(item.text(self.BODY_DATA_COLUMN)))
else:
self.status_bar.showMessage("Error: unknown body type")
def _update_body_description(self, body, item):
body.update_description(str(item.text(self.BODY_DESCRIPTION_COLUMN)))
def _valid_date(self, date):
try:
date_obj = datetime.datetime.strptime(date, "%Y-%m-%d").date()
return str(date_obj)
except ValueError:
self.status_bar.showMessage(date+" not a valid date, please use format: YYYY-MM-DD")
return None
def init_name(self, element, item, column):
item.setText(column, element.get_name())
namelabel = TreeLabel(element.get_name())
self.tree.setItemWidget(item, column, namelabel)
def init_dept(self, element, item, column):
item.setText(column, element.get_department())
deptlabel = TreeLabel(element.get_department())
self.tree.setItemWidget(item, column, deptlabel)
def init_assigned_user(self, element, item, column):
user = element.get_assigned_user()
item.setText(column, user)
lineedit = TreeLineEdit(user, item, column)
lineedit.setCompleter(self.user_completer)
self.tree.setItemWidget(item, column, lineedit)
def init_status(self, element, item, column):
item.setText(column, element.get_status())
combobox = TreeComboBoxItem(item, column)
element_type = element.get_status()
type_idx = 0
for idx, type in enumerate(Status.ALL):
combobox.addItem(type)
if type == element_type:
type_idx = idx
combobox.setCurrentIndex(type_idx)
self.tree.setItemWidget(item, column, combobox)
def init_start_date(self, element, item, column):
start_date = element.get_start_date()
item.setText(column, " "+start_date)
start_dateedit = TreeDateEdit(start_date, item, column, self.tree)
self.tree.setItemWidget(item, column, start_dateedit)
def init_end_date(self, element, item, column):
end_date = element.get_end_date()
item.setText(column, " "+end_date)
end_dateedit = TreeDateEdit(end_date, item, column, self.tree)
self.tree.setItemWidget(item, column, end_dateedit)
def init_last_publish(self, element, item, column):
publish = element.get_last_publish()
if publish is not None:
item.setText(column, publish[0]+", "+publish[1]+", "+publish[2])
else:
item.setText(column, "")
def init_note(self, element, item, column):
item.setText(column, element.get_last_note())
def update_name(self, element, item, column):
self.status_bar.showMessage("can't change name")
def update_dept(self, element, item, column):
self.status_bar.showMessage("can't change department")
def update_assigned_user(self, element, item, column):
user = str(item.text(column))
if user in self.user_list:
element.update_assigned_user(user)
self.status_bar.clearMessage()
else:
self.tree.itemWidget(item, column).setText(element.get_assigned_user())
self.status_bar.showMessage('"' + user + '" is not a valid username')
def update_status(self, element, item, column):
element.update_status(str(item.text(column)))
self.status_bar.clearMessage()
def update_start_date(self, element, item, column):
date_str = str(item.text(column))
valid_date_str = self._valid_date(date_str)
if valid_date_str:
element.update_start_date(valid_date_str)
item.setText(column, " "+date_str)
self.status_bar.clearMessage()
else:
self.init_start_date(element, item, column)
def update_end_date(self, element, item, column):
date_str = str(item.text(column))
valid_date_str = self._valid_date(date_str)
if valid_date_str:
element.update_end_date(valid_date_str)
item.setText(column, " "+date_str)
self.status_bar.clearMessage()
else:
self.init_end_date(element, item, column)
def update_last_publish(self, element, item, column):
self.status_bar.showMessage("can't modify publish data")
self.init_last_publish(element, item, column)
def update_note(self, element, item, column):
element.update_notes(str(item.text(column)))
self.status_bar.clearMessage()
class CheckoutWindow(QtWidgets.QWidget):
finished = QtCore.Signal()
def __init__(self, parent, dept_list=Department.ALL):
super(CheckoutWindow, self).__init__()
self.parent = parent
self.project = Project()
self.environment = Environment()
self.initUI(dept_list)
def initUI(self, dept_list):
#define gui elements
self.resize(WINDOW_WIDTH,WINDOW_HEIGHT)
self.setWindowTitle('Checkout')
self.dept_tabs = QtWidgets.QTabWidget()
self.dept_list = dept_list
self.result = None
#create checkbox to show only published assets
self.show_published = QtWidgets.QCheckBox("Display only assets or shots with previous publishes")
self.show_published.setCheckState(QtCore.Qt.Unchecked)
self.show_published.stateChanged.connect(self.changeBodyCheckoutVisibility)
#create Tabs
self.createTabs()
#create buttons
self.checkout_button = QtWidgets.QPushButton('Checkout')
self.checkout_button.clicked.connect(self.checkout)
self.cancel_button = QtWidgets.QPushButton('Cancel')
self.cancel_button.clicked.connect(self.close)
#create button layout
button_layout = QtWidgets.QHBoxLayout()
button_layout.addWidget(self.checkout_button)
button_layout.addWidget(self.cancel_button)
self.img = QtWidgets.QLabel()
banner_path = os.path.join(os.environ['BYU_TOOLS_DIR'], 'byugui', 'assets', 'images', 'film-banner.jpg')
pixmap = QtGui.QPixmap(banner_path)
scaled = pixmap.scaledToWidth(self.size().width())
self.img.setPixmap(scaled)
#create main layout
main_layout = QtWidgets.QVBoxLayout()
self.setLayout(main_layout)
main_layout.addWidget(self.img)
main_layout.setSpacing(5)
# main_layout.setMargin(6)
main_layout.addWidget(self.dept_tabs)
main_layout.addWidget(self.show_published)
main_layout.addLayout(button_layout)
self.show()
def createTabs(self):
#remember the current index so that we can restore it when we create the tabs
currIndex = self.dept_tabs.currentIndex()
#clear out the old tabs
self.dept_tabs.clear()
#create tabs
for dept in self.dept_list:
tab = DepartmentTab(self)
self.dept_tabs.addTab(tab, dept)
tab_layout = QtWidgets.QHBoxLayout()
element_list = QtWidgets.QListWidget()
commentBox = QtWidgets.QTextEdit()
commentBox.setReadOnly(True)
tab.commentBox = commentBox
if dept in Department.ASSET_DEPTS:
for asset in self.project.list_assets():
if not self.show_published.isChecked() or self.hasPreviousPublish(asset, dept):
item = QtWidgets.QListWidgetItem(asset)
element_list.addItem(item)
element_list.currentItemChanged.connect(self.set_current_item)
elif dept in Department.SHOT_DEPTS:
for shot in self.project.list_shots():
if not self.show_published.isChecked() or self.hasPreviousPublish(shot, dept):
item = QtWidgets.QListWidgetItem(shot)
element_list.addItem(item)
element_list.currentItemChanged.connect(self.set_current_item)
elif dept in Department.CROWD_DEPTS:
for crowdCycle in self.project.list_crowd_cycles():
if not self.show_published.isChecked() or self.hasPreviousPublish(crowdCycle, dept):
item = QtWidgets.QListWidgetItem(crowdCycle)
element_list.addItem(item)
element_list.currentItemChanged.connect(self.set_current_item)
tab_layout.addWidget(element_list)
tab_layout.addWidget(commentBox)
tab.setLayout(tab_layout)
#restore the previous index
self.dept_tabs.setCurrentIndex(currIndex)
def hasPreviousPublish(self, body, department):
asset_obj = self.project.get_body(body)
element_obj = asset_obj.get_element(department)
last_publish = element_obj.get_last_publish()
if last_publish is None:
return False
return True
def changeBodyCheckoutVisibility(self):
#recreate tabs the with the new check option
self.createTabs()
def set_current_item(self, index):
current_dept = self.dept_list[self.dept_tabs.currentIndex()]
if current_dept in Department.ASSET_DEPTS:
self.current_item = str(index.text())
elif current_dept in Department.SHOT_DEPTS:
self.current_item = str(index.text())
elif current_dept in Department.CROWD_DEPTS:
self.current_item = str(index.text())
#TODO what the heck? Why do we have three identical results from three different conditions? What are we trying to accomplish here? Admitadly the last one I added just following the crowd.
asset_obj = self.project.get_body(self.current_item)
element_obj = asset_obj.get_element(current_dept)
last_publish = element_obj.get_last_publish()
last_publish_comment = None
if last_publish is not None:
last_publish_comment = "Last published {0} by {1} \n \"{2}\"".format(last_publish[1], last_publish[0], last_publish[2])
else:
last_publish_comment = "No publishes for this element"
currentTab = self.dept_tabs.currentWidget()
currentTab.commentBox.setText(last_publish_comment)
def checkout(self):
"""
Checks out the currently selected item
:return:
"""
current_user = self.environment.get_current_username()
current_dept = self.dept_list[self.dept_tabs.currentIndex()]
asset_obj = self.project.get_body(self.current_item)
element_obj = asset_obj.get_element(current_dept)
element_path = element_obj.checkout(current_user)
if element_path != None:
self.result = element_path
self.close()
def closeEvent(self, event):
self.finished.emit()
event.accept()
class AssembleWindow(QtWidgets.QWidget):
finished = QtCore.Signal()
def __init__(self, parent, dept_list=Department.ALL):
super(AssembleWindow, self).__init__()
self.parent = parent
self.project = Project()
self.environment = Environment()
self.initUI(dept_list)
def initUI(self, dept_list):
#define gui elements
self.resize(WINDOW_WIDTH,WINDOW_HEIGHT)
self.setWindowTitle('Assemble')
self.dept_tabs = QtWidgets.QTabWidget()
self.dept_list = dept_list
self.result = None
#create Tabs
self.createTabs()
#create buttons
self.assemble_button = QtWidgets.QPushButton('Assemble')
self.assemble_button.clicked.connect(self.assemble)
self.cancel_button = QtWidgets.QPushButton('Cancel')
self.cancel_button.clicked.connect(self.close)
#create button layout
button_layout = QtWidgets.QHBoxLayout()
button_layout.addWidget(self.assemble_button)
button_layout.addWidget(self.cancel_button)
self.img = QtWidgets.QLabel()
banner_path = os.path.join(os.environ['BYU_TOOLS_DIR'], 'byugui', 'assets', 'images', 'film-banner.jpg')
pixmap = QtGui.QPixmap(banner_path)
scaled = pixmap.scaledToWidth(self.size().width())
self.img.setPixmap(scaled)
#create main layout
main_layout = QtWidgets.QVBoxLayout()
self.setLayout(main_layout)
main_layout.addWidget(self.img)
main_layout.setSpacing(5)
# main_layout.setMargin(6)
main_layout.addWidget(self.dept_tabs)
main_layout.addLayout(button_layout)
self.show()
def createTabs(self):
#remember the current index so that we can restore it when we create the tabs
currIndex = self.dept_tabs.currentIndex()
#clear out the old tabs
self.dept_tabs.clear()
#create tabs
for dept in self.dept_list:
tab = DepartmentTab(self)
self.dept_tabs.addTab(tab, dept)
tab_layout = QtWidgets.QHBoxLayout()
element_list = QtWidgets.QListWidget()
if dept in Department.ASSET_DEPTS:
for asset in self.project.list_assets():
item = QtWidgets.QListWidgetItem(asset)
element_list.addItem(item)
element_list.currentItemChanged.connect(self.set_current_item)
elif dept in Department.SHOT_DEPTS:
for shot in self.project.list_shots():
item = QtWidgets.QListWidgetItem(shot)
element_list.addItem(item)
element_list.currentItemChanged.connect(self.set_current_item)
elif dept in Department.TOOL_DEPTS:
for tool in self.project.list_tools():
item = QtWidgets.QListWidgetItem(tool)
element_list.addItem(item)
element_list.currentItemChanged.connect(self.set_current_item)
tab_layout.addWidget(element_list)
tab.setLayout(tab_layout)
#restore the previous index
self.dept_tabs.setCurrentIndex(currIndex)
def set_current_item(self, index):
current_dept = self.dept_list[self.dept_tabs.currentIndex()]
if current_dept in Department.ASSET_DEPTS:
self.current_item = str(index.text())
elif current_dept in Department.SHOT_DEPTS:
self.current_item = str(index.text())
elif current_dept in Department.TOOL_DEPTS:
self.current_item = str(index.text())
asset_obj = self.project.get_body(self.current_item)
element_obj = asset_obj.get_element(current_dept)
last_publish = element_obj.get_last_publish()
last_publish_comment = None
if last_publish is not None:
last_publish_comment = "Last published {0} by {1} \n \"{2}\"".format(last_publish[1], last_publish[0], last_publish[2])
else:
last_publish_comment = "No publishes for this element"
currentTab = self.dept_tabs.currentWidget()
def hasPreviousPublish(self, body, department):
asset_obj = self.project.get_body(body)
element_obj = asset_obj.get_element(department)
last_publish = element_obj.get_last_publish()
if last_publish is None:
return False
return True
def assemble(self):
"""
Checks out the currently selected item
:return:
"""
current_user = self.environment.get_current_username()
current_dept = self.dept_list[self.dept_tabs.currentIndex()]
# Make sure that the asset hasn't been assemble yet
startAssembling = False
if self.hasPreviousPublish(self.current_item, current_dept):
startAssembling = message_gui.light_error("This asset has already been assembled for " + current_dept + ".\n Are you sure you want to assemble it again?")
else:
startAssembling = True
if startAssembling is True:
asset_obj = self.project.get_body(self.current_item)
element_obj = asset_obj.get_element(current_dept)
element_path = element_obj.checkout(current_user)
if element_path != None:
self.result = self.current_item
self.close()
else:
return
def closeEvent(self, event):
self.finished.emit()
event.accept()
class ReferenceWindow(QtGui.QWidget):
finished = QtCore.pyqtSignal()
def __init__(self, parent, src, dept_list=Department.ALL):
super(ReferenceWindow, self).__init__()
self.project = Project()
self.parent = parent
self.src = src
self.filePath = None
self.done = True
self.reference = False
self.initUI(dept_list)
def initUI(self, dept_list):
#define gui elements
self.setGeometry(300,300,WINDOW_WIDTH,WINDOW_HEIGHT)
self.setWindowTitle('Taijitu Reference Manager')
self.departmentMenu = QtGui.QComboBox()
for i in dept_list:
self.departmentMenu.addItem(i)
self.departmentMenu.activated[str].connect(self.setElementType)
self.assetList = AssetListWindow(self)
for asset in self.project.list_assets():
item = QtGui.QListWidgetItem(asset)
self.assetList.addItem(item)
self.typeFilterLabel = QtGui.QLabel("Type Filter")
self.typeFilterLabel.setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter)
self.typeFilter = QtGui.QComboBox()
self.typeFilter.addItem("all")
for i in AssetType.ALL:
self.typeFilter.addItem(i)
self.typeFilter.currentIndexChanged.connect(self.setElementType)
self.referenceButton = QtGui.QPushButton('Reference')
self.referenceButton.clicked.connect(self.createReference)
self.cancelButton = QtGui.QPushButton('Cancel')
self.cancelButton.clicked.connect(self.close)
#set gui layout
self.grid = QtGui.QGridLayout(self)
self.setLayout(self.grid)
self.grid.addWidget(self.departmentMenu, 0, 0)
self.grid.addWidget(self.assetList, 1, 0, 1, 0)
self.grid.addWidget(self.typeFilterLabel, 2, 0)
self.grid.addWidget(self.typeFilter, 2, 1)
self.grid.addWidget(self.referenceButton, 3, 0)
self.grid.addWidget(self.cancelButton, 3, 1)
self.show()
def setElementType(self, idx=0):
department = str(self.departmentMenu.currentText())
self.refreshList(department)
def createReference(self):
checkout = self.project.get_checkout(os.path.dirname(self.src))
if checkout is not None:
body_name = checkout.get_body_name()
body = self.project.get_body(body_name)
body.add_reference(self.assetList.current_selection)
self.done = False
self.reference = True
self.close()
def refreshList(self, department):
if department in Department.FRONTEND:
asset_filter = None
if(self.typeFilter.currentIndex()):
asset_filter_str = str(self.typeFilter.currentText())
asset_filter = (Asset.TYPE, operator.eq, asset_filter_str)
self.elements = self.project.list_assets(asset_filter)
else:
self.elements = self.project.list_shots()
self.assetList.clear()
for e in self.elements:
self.assetList.addItem(e)
def closeEvent(self, event):
self.finished.emit()
event.accept()
class CheckoutWindow(QtWidgets.QWidget):
finished = QtCore.Signal()
def __init__(self, parent, dept_list=Department.ALL):
super(CheckoutWindow, self).__init__()
self.parent = parent
self.project = Project()
self.environment = Environment()
self.initUI(dept_list)
def initUI(self, dept_list):
#define gui elements
self.resize(WINDOW_WIDTH, WINDOW_HEIGHT)
self.setWindowTitle('Checkout')
self.dept_tabs = QtWidgets.QTabWidget()
self.dept_list = dept_list
self.result = None
#create checkbox to show only published assets
self.show_published = QtWidgets.QCheckBox(
"Display only assets or shots with previous publishes")
self.show_published.setCheckState(QtCore.Qt.Unchecked)
self.show_published.stateChanged.connect(
self.changeBodyCheckoutVisibility)
#create Tabs
self.createTabs()
#create buttons
self.checkout_button = QtWidgets.QPushButton('Checkout')
self.checkout_button.clicked.connect(self.checkout)
self.cancel_button = QtWidgets.QPushButton('Cancel')
self.cancel_button.clicked.connect(self.close)
#create button layout
button_layout = QtWidgets.QHBoxLayout()
button_layout.addWidget(self.checkout_button)
button_layout.addWidget(self.cancel_button)
self.img = QtWidgets.QLabel()
banner_path = os.path.join(os.environ['BYU_TOOLS_DIR'], 'byugui',
'assets', 'images', 'film-banner.jpg')
pixmap = QtGui.QPixmap(banner_path)
scaled = pixmap.scaledToWidth(self.size().width())
self.img.setPixmap(scaled)
#create main layout
main_layout = QtWidgets.QVBoxLayout()
self.setLayout(main_layout)
main_layout.addWidget(self.img)
main_layout.setSpacing(5)
# main_layout.setMargin(6)
main_layout.addWidget(self.dept_tabs)
main_layout.addWidget(self.show_published)
main_layout.addLayout(button_layout)
self.show()
#Recursivly goes through the asset's file name
def recurseTree(self, treeItem, array, asset):
#This is for setting bottom level text attributes
if len(array) == 0:
treeItem.setText(1, asset)
treeItem.setTextColor(0, "#3c83f9")
font = QtGui.QFont()
font.setPointSize(12)
font.setBold(False)
treeItem.setFont(0, font)
return
else: #This is for setting parent level text attributes and settin up the structure
item = QtWidgets.QTreeWidgetItem(array[0])
item.setText(0, array[0])
item.setText(1, "This is not a file")
item.setTextColor(0, "#d0d0d0")
font = QtGui.QFont()
font.setPointSize(11)
font.setBold(True)
item.setFont(0, font)
skip = False
# this is to check if the child already exists
for i in range(0, treeItem.childCount()):
if treeItem.child(i).text(0) == item.text(0):
item = treeItem.child(i)
skip = True
if skip == False: # Executes if the child doesnt already exist
treeItem.addChild(item)
newArray = array[1:]
self.recurseTree(item, newArray, asset)
return
def createTabs(self):
#remember the current index so that we can restore it when we create the tabs
currIndex = self.dept_tabs.currentIndex()
#clear out the old tabs
self.dept_tabs.clear()
#create tabs
for dept in self.dept_list:
tab = DepartmentTab(self)
self.dept_tabs.addTab(tab, dept)
tab_layout = QtWidgets.QHBoxLayout()
element_list = QtWidgets.QTreeWidget()
element_list.setColumnCount(1)
commentBox = QtWidgets.QTextEdit()
commentBox.setReadOnly(False)
tab.commentBox = commentBox
if dept in Department.ASSET_DEPTS:
for asset in self.project.list_assets():
#print(asset)
if not self.show_published.isChecked(
) or self.hasPreviousPublish(asset, dept):
asset_array = asset.split("_")
firstelement = element_list.findItems(
asset_array[0], 0, 0)
if not firstelement:
item = QtWidgets.QTreeWidgetItem(asset_array[0])
item.setText(0, asset_array[0])
item.setTextColor(0, "#d0d0d0")
font = QtGui.QFont()
font.setPointSize(11)
font.setBold(True)
item.setFont(0, font)
self.recurseTree(item, asset_array[1:], asset)
element_list.insertTopLevelItem(0, item)
else:
self.recurseTree(firstelement[0], asset_array[1:],
asset)
element_list.currentItemChanged.connect(
self.set_current_item)
elif dept in Department.SHOT_DEPTS:
for shot in self.project.list_shots():
#print(shot)
if not self.show_published.isChecked(
) or self.hasPreviousPublish(shot, dept):
shot_array = shot.split("_")
firstelement = element_list.findItems(
shot_array[0], 0, 0)
if not firstelement:
item = QtWidgets.QTreeWidgetItem(shot_array[0])
item.setText(0, shot_array[0])
item.setTextColor(0, "#d0d0d0")
font = QtGui.QFont()
font.setPointSize(11)
font.setBold(True)
item.setFont(0, font)
self.recurseTree(item, shot_array[1:], shot)
element_list.insertTopLevelItem(0, item)
else:
self.recurseTree(firstelement[0], shot_array[1:],
shot)
element_list.currentItemChanged.connect(
self.set_current_item)
elif dept in Department.CROWD_DEPTS:
for crowdCycle in self.project.list_crowd_cycles():
if not self.show_published.isChecked(
) or self.hasPreviousPublish(crowdCycle, dept):
crowdCycle_array = crowdCycle.split("_")
firstelement = element_list.findItems(
crowdCycle_array[0], 0, 0)
if not firstelement:
item = QtWidgets.QTreeWidgetItem(
crowdCycle_array[0])
item.setText(0, crowdCycle_array[0])
item.setTextColor(0, "#d0d0d0")
font = QtGui.QFont()
font.setPointSize(11)
font.setBold(True)
item.setFont(0, font)
self.recurseTree(item, crowdCycle_array[1:],
crowdCycle)
element_list.insertTopLevelItem(0, item)
else:
self.recurseTree(firstelement[0],
crowdCycle_array[1:], crowdCycle)
element_list.currentItemChanged.connect(
self.set_current_item)
tab_layout.addWidget(element_list)
tab_layout.addWidget(commentBox)
tab.setLayout(tab_layout)
#restore the previous index
self.dept_tabs.setCurrentIndex(currIndex)
def hasPreviousPublish(self, body, department):
asset_obj = self.project.get_body(body)
element_obj = asset_obj.get_element(department)
last_publish = element_obj.get_last_publish()
if last_publish is None:
return False
return True
def changeBodyCheckoutVisibility(self):
#recreate tabs the with the new check option
self.createTabs()
def set_current_item(self, index):
current_dept = self.dept_list[self.dept_tabs.currentIndex()]
if current_dept in Department.ASSET_DEPTS:
self.current_item = str(index.text(1))
elif current_dept in Department.SHOT_DEPTS:
self.current_item = str(index.text(1))
elif current_dept in Department.CROWD_DEPTS:
self.current_item = str(index.text(1))
#TODO what the heck? Why do we have three identical results from three different conditions? What are we trying to accomplish here? Admitadly the last one I added just following the crowd.
asset_obj = self.project.get_body(self.current_item)
element_obj = asset_obj.get_element(current_dept)
last_publish = element_obj.get_last_publish()
last_publish_comment = None
if last_publish is not None:
last_publish_comment = "Last published {0} by {1} \n \"{2}\"".format(
last_publish[1], last_publish[0], last_publish[2])
else:
last_publish_comment = "No publishes for this element"
currentTab = self.dept_tabs.currentWidget()
currentTab.commentBox.setText(last_publish_comment)
def checkout(self):
"""
Checks out the currently selected item
:return:
"""
current_user = self.environment.get_current_username()
current_dept = self.dept_list[self.dept_tabs.currentIndex()]
asset_obj = self.project.get_body(self.current_item)
element_obj = asset_obj.get_element(current_dept, force_create=True)
element_path = element_obj.checkout(current_user)
if element_path != None:
self.result = element_path
self.close()
def closeEvent(self, event):
self.finished.emit()
event.accept()
