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