def test_get_or_create_episode(self):
self.assertEqual(len(shots_service.get_episodes()), 1)
episode_name = "NE01"
shots_service.get_or_create_episode(self.project.id, episode_name)
self.assertEqual(len(shots_service.get_episodes()), 2)
shots_service.get_or_create_episode(self.project.id, episode_name)
self.assertEqual(len(shots_service.get_episodes()), 2)
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"]
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 get(self):
"""
Retrieve all episode entries. Filters can be specified in the query
string.
"""
criterions = query.get_query_criterions_from_request(request)
user_service.check_project_access(criterions)
return shots_service.get_episodes(criterions)
def get(self):
"""
Retrieve all episode entries. Filters can be specified in the query
string.
"""
criterions = query.get_query_criterions_from_request(request)
if not permissions.has_manager_permissions():
user_service.check_criterions_has_task_related(criterions)
return shots_service.get_episodes(criterions)
def prepare_import(self):
self.projects = {}
self.sequences = {}
self.episodes = {}
self.shots = {}
self.entity_types = {}
self.entities = {}
self.persons = {}
self.departments = {}
self.task_types = {}
self.task_statuses = {}
for project in Project.query.all():
self.projects[project.name] = project
for person in Person.query.all():
self.persons[person.first_name + " " + person.last_name] = person
for episode in shots_service.get_episodes():
self.episodes[str(episode["project_id"]) +
episode["name"]] = episode
for sequence in shots_service.get_sequences():
self.sequences["%s%s%s" %
(sequence["project_id"], sequence["parent_id"],
sequence["name"])] = sequence
for shot in shots_service.get_shots():
self.shots["%s%s%s" % (shot["project_id"], shot["parent_id"],
shot["name"])] = shot
for entity_type in EntityType.query.all():
self.entity_types[entity_type.name] = entity_type
for entity in Entity.query.all():
self.entities["%s%s%s" % (entity.project_id, entity.entity_type_id,
entity.name)] = entity
def test_get_episodes(self):
episodes = shots_service.get_episodes()
self.assertEquals(len(episodes), 1)
self.assertEquals(episodes[0]["id"], str(self.episode.id))
