def get_folder_from_episode(entity):
if shots_service.is_shot(entity) or shots_service.is_scene(entity):
sequence = shots_service.get_sequence_from_shot(entity)
elif shots_service.is_sequence(entity):
sequence = entity
try:
episode = shots_service.get_episode_from_sequence(sequence)
episode_name = episode["name"]
except:
episode_name = "e001"
return episode_name
def get_folder_from_sequence(entity):
if shots_service.is_shot(entity) or shots_service.is_scene(entity):
sequence = shots_service.get_sequence_from_shot(entity)
sequence_name = sequence["name"]
elif shots_service.is_sequence(entity):
sequence_name = entity["name"]
else:
sequence_name = ""
if "Seq" in sequence_name:
sequence_number = sequence.name[3:]
sequence_name = "S%s" % sequence_number.zfill(3)
return sequence_name
def get_file_name_template(tree, mode, entity):
try:
if entity["type"] == "AssetInstance":
return tree[mode]["file_name"]["instance"]
elif shots_service.is_shot(entity):
return tree[mode]["file_name"]["shot"]
elif shots_service.is_sequence(entity):
return tree[mode]["file_name"]["sequence"]
elif shots_service.is_scene(entity):
return tree[mode]["file_name"]["scene"]
else:
return tree[mode]["file_name"]["asset"]
except KeyError:
raise MalformedFileTreeException
def get_folder_path_template(tree, mode, entity):
try:
if entity["type"] == "AssetInstance":
if entity.get("target_asset_id", None) is not None:
return tree[mode]["folder_path"]["instance_asset"]
else:
return tree[mode]["folder_path"]["instance"]
elif shots_service.is_shot(entity):
return tree[mode]["folder_path"]["shot"]
elif shots_service.is_sequence(entity):
return tree[mode]["folder_path"]["sequence"]
elif shots_service.is_scene(entity):
return tree[mode]["folder_path"]["scene"]
else:
return tree[mode]["folder_path"]["asset"]
except KeyError:
raise MalformedFileTreeException
def test_is_scene(self):
self.assertTrue(shots_service.is_scene(self.scene.serialize()))
self.assertFalse(shots_service.is_scene(self.asset.serialize()))
