def go(self):
project = Project()
asset_list = project.list_assets()
self.item_gui = sfl.SelectFromList(l=asset_list,
parent=maya_main_window(),
title="Select an asset to clone")
self.item_gui.submitted.connect(self.results)
def publish(self):
# this is the function that we will use to publish.
project = Project()
asset_list = project.list_assets()
self.item_gui = sfl.SelectFromList(
l=asset_list,
parent=maya_main_window(),
title="Select an asset to publish to")
self.item_gui.submitted.connect(self.asset_results)
def run(self):
# step 1: Select the body
# step 2: assemble the element
project = Project()
asset_list = project.list_assets()
non_shot_list = []
for item in asset_list:
asset = project.get_asset(item)
if not asset.get_type() == AssetType.SHOT:
non_shot_list.append(item)
self.asset_gui = sfl.SelectFromList(
l=non_shot_list,
parent=houdini_main_window(),
title="Select a prop, character, or set to assemble")
self.asset_gui.submitted.connect(self.asset_results)
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['MEDIA_PROJECT_DIR'], 'pipe', 'am', 'gui', '_resources', '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()
import checkout
from pipe.am.project import Project
from pipe.am.environment import Environment, Department
from pipe.am.element import Element
from pipe.am.body import Body, Asset, Shot, AssetType
import maya.standalone
maya.standalone.initialize(name='python')
import maya.cmds as cmds
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'
class ElementBrowser(QtWidgets.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):
QtWidgets.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 = QtWidgets.QCompleter(self.user_list)
#filters
self.filter_label = QtWidgets.QLabel("Filter by: ")
self.dept_filter_label = QtWidgets.QLabel("Department")
self.dept_filter_label.setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter)
self.dept_filter = QtWidgets.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 = QtWidgets.QLabel("Asset Type")
self.type_filter_label.setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter)
self.type_filter = QtWidgets.QComboBox()
self.type_filter.addItem("all")
for each in AssetType.ALL:
self.type_filter.addItem(each)
self.name_filter_label = QtWidgets.QLabel("Name")
self.name_filter_label.setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter)
self.name_filter = QtWidgets.QLineEdit()
# menu bar
self.menu_bar = QtWidgets.QMenuBar()
self.view_menu = QtWidgets.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 = QtWidgets.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 = QtWidgets.QPushButton("New")
# refresh button
self.refresh_button = QtWidgets.QPushButton("Refresh")
# tree
self.tree = QtWidgets.QTreeWidget()
self.tree.setItemDelegate(TreeGridDelegate(self.tree))
self.columnCount = 8
self.tree.setColumnCount(self.columnCount)
tree_header = QtWidgets.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 = QtWidgets.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 = QtWidgets.QVBoxLayout(self)
layout.setSpacing(5)
layout.setMargin(6)
options_layout = QtWidgets.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 = QtWidgets.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 = QtWidgets.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, QtWidgets.QColor(30,30,30))
body_obj = self.project.get_body(body)
self._load_body(body_obj, tree_item)
tree_item.addChild(QtWidgets.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 range(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 = QtWidgets.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_body_gui
self.new_body_dialog = new_body_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()