def prepare_import(self, project_id): self.episodes = {} self.sequences = {} self.descriptor_fields = self.get_descriptor_field_map( project_id, "Shot") project = projects_service.get_project(project_id) self.is_tv_show = projects_service.is_tv_show(project)
def prepare_import(self, project_id): self.asset_type_map = {} self.asset_map = {} self.episode_map = {} self.sequence_map = {} self.shot_map = {} asset_types = assets_service.get_asset_types() for asset_type in asset_types: self.asset_type_map[asset_type["id"]] = slugify(asset_type["name"]) assets = assets_service.get_assets({"project_id": project_id}) for asset in assets: key = self.get_asset_key(asset) self.asset_map[key] = asset["id"] project = projects_service.get_project(project_id) self.is_tv_show = projects_service.is_tv_show(project) if self.is_tv_show: episodes = shots_service.get_episodes({"project_id": project_id}) for episode in episodes: self.episode_map[episode["id"]] = slugify(episode["name"]) sequences = shots_service.get_sequences({"project_id": project_id}) for sequence in sequences: key = self.get_sequence_key(sequence) self.sequence_map[sequence["id"]] = key shots = shots_service.get_shots({"project_id": project_id}) for shot in shots: key = self.get_shot_key(shot) self.shot_map[key] = shot["id"]
def prepare_import(self, project_id): self.episodes = {} self.entity_types = {} self.descriptor_fields = self.get_descriptor_field_map( project_id, "Asset") project = projects_service.get_project(project_id) self.is_tv_show = projects_service.is_tv_show(project) if self.is_tv_show: episodes = shots_service.get_episodes_for_project(project_id) self.episodes = { episode["name"]: episode["id"] for episode in episodes } self.created_assets = [] self.task_types_in_project_for_assets = ( TaskType.query.join(ProjectTaskTypeLink).filter( ProjectTaskTypeLink.project_id == project_id).filter( TaskType.for_entity == "Asset")) self.task_statuses = { status["id"]: [status[n] for n in ("name", "short_name")] for status in tasks_service.get_task_statuses() } self.current_user_id = get_current_user()["id"] self.task_types_for_ready_for_map = { task_type.name: str(task_type.id) for task_type in TaskType.query.join(ProjectTaskTypeLink).filter( ProjectTaskTypeLink.project_id == project_id).filter( TaskType.for_entity == "Shot").all() }
def prepare_import(self, project_id): self.episodes = {} self.entity_types = {} self.descriptor_fields = self.get_descriptor_field_map( project_id, "Asset") project = projects_service.get_project(project_id) self.is_tv_show = projects_service.is_tv_show(project) if self.is_tv_show: episodes = shots_service.get_episodes_for_project(project_id) self.episodes = { episode["name"]: episode["id"] for episode in episodes }
def prepare_import(self, project_id): self.episodes = {} self.sequences = {} self.descriptor_fields = self.get_descriptor_field_map( project_id, "Shot" ) project = projects_service.get_project(project_id) self.is_tv_show = projects_service.is_tv_show(project) self.created_shots = [] self.task_types_in_project_for_shots = ( TaskType.query.join(ProjectTaskTypeLink) .filter(ProjectTaskTypeLink.project_id == project_id) .filter(TaskType.for_entity == "Shot") ) self.task_statuses = { status["id"]: [status[n] for n in ("name", "short_name")] for status in get_task_statuses() } self.current_user_id = get_current_user()["id"]
def test_is_tv_show(self): self.assertFalse(projects_service.is_tv_show(self.project.serialize())) self.project.update({"production_type": "tvshow"}) self.assertTrue(projects_service.is_tv_show(self.project.serialize()))