def get_asset_type_by_name(asset_type_name): asset_type = EntityType.get_by(name=asset_type_name) if asset_type is None or not is_asset_type(asset_type): raise AssetTypeNotFoundException return asset_type.serialize(obj_type="AssetType")
def sanitize_import_data(self, data): from zou.app.models.preview_file import PreviewFile from zou.app.models.entity_type import EntityType entity_ids = [] model_type = data.get("type", "Shot") if "entities_out" in data: entity_ids = data["entities_out"] del data["entities_out"] if "asset_type_id" in data: data["entity_type_id"] = data["asset_type_id"] del data["asset_type_id"] del data["asset_type_name"] if "sequence_id" in data: data["parent_id"] = data["sequence_id"] del data["sequence_id"] if ( "preview_file_id" in data and data["preview_file_id"] is not None and len(data["preview_file_id"]) > 0 ): preview_file = PreviewFile.get(data["preview_file_id"]) if preview_file is None: del data["preview_file_id"] elif "preview_file_id" in data: del data["preview_file_id"] if "frame_in" in data: data["data"]["frame_in"] = data["frame_in"] del data["frame_in"] if "frame_out" in data: data["data"]["frame_out"] = data["frame_out"] del data["frame_out"] if "fps" in data: data["data"]["fps"] = data["fps"] del data["fps"] for field in [ "entities_in", "episode_id", "episode_name", "project_name", "sequence_name", "tasks", "type", ]: if field in data: del data[field] if model_type in ["Shot", "Sequence", "Episode"]: entity_type = EntityType.get_by(name=model_type) data["entity_type_id"] = entity_type.id return (data, entity_ids)
def get_entity_type_by_name(name): """ Return entity type maching *name*. If it doesn't exist, it creates it. """ entity_type = EntityType.get_by(name=name) if entity_type is None: entity_type = EntityType.create(name=name) return entity_type.serialize()
def get_or_create_asset_type(name): """ For a given name, get matching asset type. Create if it does not exist. """ asset_type = EntityType.get_by(name=name) if asset_type is None: asset_type = EntityType.create(name=name) return asset_type.serialize(obj_type="AssetType")
def get_or_create_asset_type(name): """ For a given name, get matching asset type. Create if it does not exist. """ asset_type = EntityType.get_by(name=name) if asset_type is None: asset_type = EntityType.create(name=name) clear_asset_type_cache() events.emit( "asset-type:new", {"name": asset_type.name, "id": asset_type.id} ) return asset_type.serialize(obj_type="AssetType")
def guess_asset(project, asset_type_name, asset_name): asset_type_id = None if len(asset_type_name) > 0: asset_type = EntityType.get_by(name=asset_type_name) if asset_type is not None: asset_type_id = asset_type.id if len(asset_name) > 0: asset = Entity.get_by(name=asset_name, entity_type_id=asset_type_id, project_id=project["id"]) else: raise WrongPathFormatException( "Asset name was not found in given path.") return asset
def save_asset_type(asset_type_name): asset_type = EntityType.get_by(name=asset_type_name) if asset_type is None: asset_type = EntityType.create(name=asset_type_name) return asset_type.serialize(obj_type="AssetType")
def get_or_create_type(name): asset_type = EntityType.get_by(name=name) if asset_type is None: asset_type = EntityType(name=name) asset_type.save() return asset_type.serialize(obj_type="AssetType")
def get_entity_type(name): entity_type = EntityType.get_by(name=name) if entity_type is None: entity_type = EntityType.create(name=name) return entity_type.serialize()
def get_or_create_type(name): asset_type = EntityType.get_by(name=name) if asset_type is None: asset_type = EntityType(name=name) asset_type.save() return asset_type
def save_asset_type(asset_type_name): asset_type = EntityType.get_by(name=asset_type_name) if asset_type is None: asset_type = EntityType(name=asset_type_name) asset_type.save() return asset_type