def init_project(self):
"""
When PROJECT selected in UI
"""
# Show and set up PROPERTIES widget
self.project_properties_ui.show()
self.asset_properties_ui.hide()
self.sequence_properties_ui.hide()
self.shot_properties_ui.hide()
# Setup data
model_index = self.listProjects.currentIndex() # .selectedIndexes()[0]
project_id = model_index.data(QtCore.Qt.UserRole + 1)
project = self.eve_data.get_project(project_id)
self.selected_project = project
self.eve_data.get_project_assets(project)
self.eve_data.get_project_sequences(project)
# Clear SHOTS in UI
self.listShots.setModel(models.ListModel([]))
# Fill Project Properties widget
project_root = build_project_root(project.name)
self.project_properties_ui.project_ui.linProjectLocation.setText(project_root)
self.project_properties_ui.project_ui.linProjectLocation.setEnabled(False)
self.project_properties_ui.project_ui.linProjectName.setText(project.name)
self.project_properties_ui.project_ui.linProjectName.setEnabled(False)
self.project_properties_ui.project_ui.linHoudini.setText(project.houdini_build)
self.project_properties_ui.project_ui.linProjectWidth.setText(str(project.width))
self.project_properties_ui.project_ui.linProjectHeight.setText(str(project.height))
self.project_properties_ui.project_ui.txtDescription.setText(project.description)
# FILL ASSET and SEQUENCE WIDGETS
self.model_assets = models.ListModel(self.eve_data.project_assets)
self.listAssets.setModel(self.model_assets)
self.model_sequences = models.ListModel(self.eve_data.project_sequences)
self.listSequences.setModel(self.model_sequences)
# Enable/disable UI buttons depending on project existence
if os.path.exists(project_root):
self.project_properties_ui.btnCreateProject.setText(self.btn_project_update)
self.project_properties_ui.btnLaunchHoudini.setEnabled(True)
self.project_properties_ui.btnLaunchNuke.setEnabled(True)
self.project_properties_ui.btnOpenFolder.setEnabled(True)
else:
self.project_properties_ui.btnCreateProject.setText(self.btn_project_create)
self.project_properties_ui.btnLaunchHoudini.setEnabled(False)
self.project_properties_ui.btnLaunchNuke.setEnabled(False)
self.project_properties_ui.btnOpenFolder.setEnabled(False)
def init_asset_manager(self):
# Get Eve data
self.eve_data = eve_data.EveData(self.SQL_FILE_PATH)
self.project = self.eve_data.get_project_by_name(self.project_name)
# Fill Asset Types
model_asset_types = models.ListModel(self.eve_data.asset_types)
self.comAssetType.setModel(model_asset_types)
# Fill Asset list in UI
self.eve_data.get_project_assets(self.project)
self.model_assets = models.ListModel(self.eve_data.project_assets)
self.comAssetName.setModel(self.model_assets)
def init_shot(self):
"""
When SHOT selected in UI
"""
# Show and set up PROPERTIES widget
self.project_properties_ui.hide()
self.asset_properties_ui.hide()
self.sequence_properties_ui.hide()
self.shot_properties_ui.show()
# Setup data
model_index = self.listShots.currentIndex()
shot_id = model_index.data(QtCore.Qt.UserRole + 1)
shot = self.eve_data.get_shot(shot_id)
self.selected_shot = shot
# FILL SHOT ASSETS WIDGET
self.eve_data.get_shot_assets(shot_id)
self.model_shot_assets = models.ListModel(self.eve_data.shot_assets)
self.shot_properties_ui.shot_ui.listAssets.setModel(self.model_shot_assets)
# Fill SHOT WIDGET
self.shot_properties_ui.shot_ui.linShotName.setEnabled(False)
self.shot_properties_ui.shot_ui.linProjectName.setText(self.selected_project.name)
self.shot_properties_ui.shot_ui.linSequenceName.setText(self.selected_sequence.name)
self.shot_properties_ui.shot_ui.linShotName.setText(self.selected_shot.name)
self.shot_properties_ui.shot_ui.linStartFrame.setText(str(shot.start_frame))
self.shot_properties_ui.shot_ui.linEndFrame.setText(str(shot.end_frame))
self.shot_properties_ui.shot_ui.linWidth.setText(str(shot.width))
self.shot_properties_ui.shot_ui.linHeight.setText(str(shot.height))
self.shot_properties_ui.shot_ui.txtDescription.setText(shot.description)
def init_sequence(self):
"""
When SEQUENCE selected in UI
"""
# Show and set up PROPERTIES widget
self.project_properties_ui.hide()
self.asset_properties_ui.hide()
self.sequence_properties_ui.show()
self.shot_properties_ui.hide()
# Setup data for sequence
model_index = self.listSequences.currentIndex()
sequence_id = model_index.data(QtCore.Qt.UserRole + 1)
sequence = self.eve_data.get_sequence(sequence_id)
self.selected_sequence = sequence
# and shot
self.eve_data.get_sequence_shots(sequence.id)
self.model_shots = models.ListModel(self.eve_data.sequence_shots)
self.listShots.setModel(self.model_shots)
# Fill SEQUENCE WIDGET
self.sequence_properties_ui.sequence_ui.linSequenceName.setEnabled(False)
self.sequence_properties_ui.sequence_ui.linProjectName.setText(self.selected_project.name)
self.sequence_properties_ui.sequence_ui.linSequenceName.setText(sequence.name)
self.sequence_properties_ui.sequence_ui.txtDescription.setText(sequence.description)
def init_asset(self):
"""
When ASSET selected in UI
"""
# Show and set up PROPERTIES widget
self.project_properties_ui.hide()
self.asset_properties_ui.show()
self.sequence_properties_ui.hide()
self.shot_properties_ui.hide()
# Setup data
model_index = self.listAssets.currentIndex()
asset_id = model_index.data(QtCore.Qt.UserRole + 1)
asset = self.eve_data.get_asset(asset_id)
self.selected_asset = asset
# Fill ASSET WIDGET
self.asset_properties_ui.asset_ui.linAssetName.setEnabled(False)
self.asset_properties_ui.asset_ui.linProjectName.setText(self.selected_project.name)
self.asset_properties_ui.asset_ui.linAssetName.setText(asset.name)
model_asset_types = models.ListModel(self.eve_data.asset_types)
self.asset_properties_ui.asset_ui.comAssetType.setModel(model_asset_types)
# Find AssetType string by database index
# !!! Probably wrong implementation of model data! and can be done via ListModel() !!!!!
asset_type_string = self.eve_data.get_asset_type_string(asset.type)
self.asset_properties_ui.asset_ui.comAssetType.setCurrentText(asset_type_string)
self.asset_properties_ui.asset_ui.txtDescription.setText(asset.description)
def showEvent(self, event):
"""
Executed when AddProject class is shown (AddProject.show())
"""
# Clean UI
self.asset_ui.linProjectName.setText(self.project.name)
self.asset_ui.linAssetName.clear()
self.asset_ui.txtDescription.clear()
# Add asset types to ui
self.model_asset_types = models.ListModel(self.asset_types)
self.asset_ui.comAssetType.setModel(self.model_asset_types)
def showEvent(self, event):
"""
Executed when AddProject class is shown (AddProject.show())
"""
# Clean UI
self.linShotName.setText(self.shot.name)
# self.asset_ui.linAssetName.clear()
# self.asset_ui.txtDescription.clear()
#
# Add assets to ui
self.model_assets = models.ListModel(self.project_assets)
self.listAssets.setModel(self.model_assets)
def init_shot_manager(self):
"""
Load data for Shot Manager
:return:
"""
self.eve_data = eve_data.EveData(self.SQL_FILE_PATH)
self.project = self.eve_data.get_project_by_name(self.project_name)
self.eve_data.get_project_sequences(self.project)
self.model_sequences = models.ListModel(
self.eve_data.project_sequences)
self.boxSequence.setModel(self.model_sequences)
self.init_shots()
def init_shots(self):
"""
When sequence selected in Shot Manager Ui
:return:
"""
model_index = self.boxSequence.model().index(
self.boxSequence.currentIndex(), 0)
sequence_id = model_index.data(QtCore.Qt.UserRole + 1)
self.eve_data.get_sequence_shots(sequence_id)
self.model_shots = models.ListModel(self.eve_data.sequence_shots)
self.selected_sequence = self.eve_data.get_sequence(sequence_id)
self.boxShot.setModel(self.model_shots)
self.init_shot()
def __init__(self):
super(ProjectManager, self).__init__()
# SETUP UI
self.setupUi(self)
self.project_properties_ui = ProjectProperties()
self.asset_properties_ui = AssetProperties()
self.sequence_properties_ui = SequenceProperties()
self.shot_properties_ui = ShotProperties()
self.layoutProperties.addWidget(self.project_properties_ui)
self.layoutProperties.addWidget(self.asset_properties_ui)
self.layoutProperties.addWidget(self.sequence_properties_ui)
self.layoutProperties.addWidget(self.shot_properties_ui)
self.btn_project_create = 'Create Project'
self.btn_project_update = 'Update Project'
# SETUP ENVIRONMENT
os.environ['EVE_ROOT'] = os.environ['EVE_ROOT'].replace('\\', '/')
self.eve_root = os.environ['EVE_ROOT'] # E:/Eve/Eve
# Load Eve DB
# Create database file if not exists (first time Project Manager launch)
self.SQL_FILE_PATH = settings.SQL_FILE_PATH.format(self.eve_root)
if not os.path.exists(self.SQL_FILE_PATH):
if not os.path.exists(os.path.dirname(self.SQL_FILE_PATH)):
os.makedirs(os.path.dirname(self.SQL_FILE_PATH))
self.create_database()
self.eve_data = eve_data.EveData(self.SQL_FILE_PATH)
self.model_projects = models.ListModel(self.eve_data.projects)
self.model_assets = None
self.model_sequences = None
self.model_shots = None
self.model_shot_assets = None
# Eve data
self.selected_project = None
self.selected_asset = None
self.selected_sequence = None
self.selected_shot = None
# Load ADD ENTITY classes
self.AP = AddProject(self)
self.AA = AddAsset(self)
self.AE = AddSequence(self)
self.AS = AddShot(self)
self.LA = LinkAssets(self)
# Fill UI with data from database
self.init_pm()
# Connect functions
# Menu
self.actionEveDocs.triggered.connect(self.docs)
# Project section
self.listProjects.clicked.connect(self.init_project)
self.btnAddProject.clicked.connect(self.AP.exec_)
self.btnDelProject.clicked.connect(self.del_project)
# Asset section
self.listAssets.clicked.connect(self.init_asset)
self.btnAddAsset.clicked.connect(self.run_add_asset)
self.btnDelAsset.clicked.connect(self.del_asset)
# Sequence section
self.listSequences.clicked.connect(self.init_sequence)
self.btnAddSequence.clicked.connect(self.run_add_sequence)
self.btnDelSequence.clicked.connect(self.del_sequence)
# Shot section
self.listShots.clicked.connect(self.init_shot)
self.btnAddShot.clicked.connect(self.run_add_shot)
self.btnDelShot.clicked.connect(self.del_shot)
# Project properties
self.project_properties_ui.btnCreateProject.clicked.connect(self.run_create_project)
self.project_properties_ui.btnLaunchHoudini.clicked.connect(self.launch_houdini)
# Asset properties
self.asset_properties_ui.btnUpdateAsset.clicked.connect(self.update_asset)
self.asset_properties_ui.btnCreateHoudiniFile.clicked.connect(self.create_asset_file)
# Sequence properties
self.sequence_properties_ui.btnUpdateSequence.clicked.connect(self.update_sequence)
# Shot properties
self.shot_properties_ui.btnUpdateShot.clicked.connect(self.update_shot)
self.shot_properties_ui.shot_ui.btnLinkAsset.clicked.connect(self.run_link_assets)
self.shot_properties_ui.shot_ui.btnUnlinkAsset.clicked.connect(self.run_unlink_assets)
