gazu 0.10.14__tar.gz → 0.10.15__tar.gz

{gazu-0.10.14 → gazu-0.10.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gazu
-Version: 0.10.14
+Version: 0.10.15
 Summary: Gazu is a client for Zou, the API to store the data of your CG production.
 Home-page: https://gazu.cg-wire.com/
 Author: CG Wire

gazu-0.10.15/gazu/__version__.py

@@ -0,0 +1 @@
+__version__ = "0.10.15"

{gazu-0.10.14 → gazu-0.10.15}/gazu/casting.py

@@ -45,6 +45,25 @@ def update_asset_casting(project, asset, casting, client=default):
     return raw.put(path, casting, client=client)
 
 
+def update_episode_casting(project, episode, casting, client=default):
+    """
+    Change casting of given episode with given casting (list of asset ids displayed
+    into the episode).
+
+    Args:
+        episode (str / dict): The episode dict or the episode ID.
+        casting (dict): The casting description.
+        Ex: `casting = [{"asset_id": "asset-1", "nb_occurences": 3}]`
+
+    Returns:
+        dict: Related episode.
+    """
+    episode = normalize_model_parameter(episode)
+    project = normalize_model_parameter(project)
+    path = "data/projects/%s/entities/%s/casting" % (project["id"], episode["id"])
+    return raw.put(path, casting, client=client)
+
+
 def get_asset_type_casting(project, asset_type, client=default):
     """
     Return casting for given asset_type.
@@ -117,6 +136,23 @@ def get_asset_casting(asset, client=default):
     return raw.get(path, client=client)
 
 
+def get_episode_casting(episode, client=default):
+    """
+    Return casting for given episode.
+    `[{"episode_id": "episode-1", "nb_occurences": 3}]}`
+    Args:
+        episode (dict): The episode dict
+
+    Returns:
+        dict: Casting for given episode.
+    """
+    path = "/data/projects/%s/entities/%s/casting" % (
+        episode["project_id"],
+        episode["id"],
+    )
+    return raw.get(path, client=client)
+
+
 def get_asset_cast_in(asset, client=default):
     """
     Return entity list where given asset is casted.

{gazu-0.10.14 → gazu-0.10.15}/gazu/exception.py

@@ -75,3 +75,15 @@ class TaskMustBeADictException(Exception):
     """
     Error raised when a task should be a dict.
     """
+
+
+class FileDoesntExistException(Exception):
+    """
+    Error raised when a file should be existed when we submit a preview.
+    """
+
+
+class ProjectDoesntExistException(Exception):
+    """
+    Error raised when a project isn't available.
+    """

{gazu-0.10.14 → gazu-0.10.15}/gazu/playlist.py

@@ -151,3 +151,115 @@ def update_playlist(playlist, client=default):
     return raw.put(
         "data/playlists/%s" % playlist["id"], playlist, client=client
     )
+
+
+def get_entity_preview_files(entity, client=default):
+    """
+    Get all preview files grouped by task type for a given entity.
+
+    Args:
+        entity (str / dict): The entity to retrieve files from or its ID.
+
+    Returns:
+        dict: A dict where keys are task type IDs and value array of revisions.
+    """
+    entity = normalize_model_parameter(entity)
+    return raw.get(
+        "data/playlists/entities/%s/preview-files" % entity["id"],
+        client=client
+    )
+
+
+def add_entity_to_playlist(
+    playlist,
+    entity,
+    preview_file=None,
+    persist=True,
+    client=default
+):
+    """
+    Add an entity to the playlist, use the last uploaded preview as revision
+    to review.
+
+    Args:
+        playlist (dict): Playlist object to modify.
+        entity (str / dict): The entity to add or its ID.
+        preview_file (str / dict): Set it to force a give revision to review.
+        persist (bool): Set it to True to save the result to the API.
+
+    Returns:
+        dict: Updated playlist.
+    """
+    entity = normalize_model_parameter(entity)
+
+    if preview_file is None:
+        preview_files = get_entity_preview_files(entity)
+        for task_type_id in preview_files.keys():
+            task_type_files = preview_files[task_type_id]
+            first_file = task_type_files[0]
+            if preview_file is None or \
+               preview_file["created_at"] < first_file["created_at"]:
+                preview_file = first_file
+
+    preview_file = normalize_model_parameter(preview_file)
+    playlist["shots"].append({
+        "entity_id": entity["id"],
+        "preview_file_id": preview_file["id"]
+    })
+    if persist:
+        update_playlist(playlist, client=client)
+    return playlist
+
+
+def remove_entity_from_playlist(
+    playlist,
+    entity,
+    persist=True,
+    client=default
+):
+    """
+    Remove all occurences of a given entity from a playlist.
+
+    Args:
+        playlist (dict): Playlist object to modify
+        entity (str / dict): the entity to remove or its ID
+
+    Returns:
+        dict: Updated playlist.
+    """
+    entity = normalize_model_parameter(entity)
+    playlist["shots"] = [
+        entry
+        for entry in playlist["shots"]
+        if entry["entity_id"] != entity["id"]
+    ]
+    if persist:
+        update_playlist(playlist, client=client)
+    return playlist
+
+
+def update_entity_preview(
+    playlist,
+    entity,
+    preview_file,
+    persist=True,
+    client=default
+):
+    """
+    Remove all occurences of a given entity from a playlist.
+
+    Args:
+        playlist (dict): Playlist object to modify
+        entity (str / dict): the entity to add or its ID
+
+    Returns:
+        dict: Updated playlist.
+    """
+    entity = normalize_model_parameter(entity)
+    preview_file = normalize_model_parameter(preview_file)
+    for entry in playlist["shots"]:
+        if entry["entity_id"] == entity["id"]:
+            entry["preview_file_id"] = preview_file["id"]
+    if persist:
+        update_playlist(playlist, client=client)
+    return playlist

{gazu-0.10.14 → gazu-0.10.15}/gazu/sync.py

@@ -56,6 +56,7 @@ def import_entities(entities, client=default):
     """
     Import entities from another instance to target instance (keep id and audit
     dates).
+
     Args:
         entities (list): Entities to import.
 
@@ -69,6 +70,7 @@ def import_tasks(tasks, client=default):
     """
     Import tasks from another instance to target instance (keep id and audit
     dates).
+
     Args:
         tasks (list): Tasks to import.
 
@@ -82,6 +84,7 @@ def import_entity_links(links, client=default):
     """
     Import enitity links from another instance to target instance (keep id and
     audit dates).
+
     Args:
         links (list): Entity links to import.
 
@@ -292,7 +295,8 @@ def get_sync_person_id_map(source_client, target_client):
 def push_assets(project_source, project_target, client_source, client_target):
     """
     Copy assets from source to target and preserve audit fields (`id`,
-    `created_at`, and `updated_at`)
+    `created_at`, and `updated_at`).
+
     Args:
         project_source (dict): The project to get assets from
         project_target (dict): The project to push assets to
@@ -321,6 +325,7 @@ def push_episodes(
     """
     Copy episodes from source to target and preserve audit fields (`id`,
     `created_at`, and `updated_at`)
+
     Args:
         project_source (dict): The project to get episodes from
         project_target (dict): The project to push episodes to
@@ -344,6 +349,7 @@ def push_sequences(
     """
     Copy sequences from source to target and preserve audit fields (`id`,
     `created_at`, and `updated_at`)
+
     Args:
         project_source (dict): The project to get sequences from
         project_target (dict): The project to push sequences to
@@ -364,7 +370,8 @@ def push_sequences(
 def push_shots(project_source, project_target, client_source, client_target):
     """
     Copy shots from source to target and preserve audit fields (`id`,
-    `created_at`, and `updated_at`)
+    `created_at`, and `updated_at`).
+
     Args:
         project_source (dict): The project to get shots from
         project_target (dict): The project to push shots to
@@ -386,8 +393,9 @@ def push_entity_links(
     project_source, project_target, client_source, client_target
 ):
     """
-    Copy assets from source to target and preserve audit fields (`id`,
-    `created_at`, and `updated_at`)
+    Copy entity links (breakdown, concepts) from source to target and preserve
+    audit fields (`id`, `created_at`, and `updated_at`).
+
     Args:
         project_source (dict): The project to get assets from
         project_target (dict): The project to push assets to
@@ -408,7 +416,8 @@ def push_project_entities(
 ):
     """
     Copy assets, episodes, sequences, shots and entity links from source to
-    target and preserve audit fields (`id`, `created_at`, and `updated_at`)
+    target and preserve audit fields (`id`, `created_at`, and `updated_at`).
+
     Args:
         project_source (dict): The project to get assets from
         project_target (dict): The project to push assets to
@@ -445,6 +454,7 @@ def push_tasks(
     Copy tasks from source to target and preserve audit fields (`id`,
     `created_at`, and `updated_at`)
     Attachments and previews are created too.
+
     Args:
         project_source (dict): The project to get assets from
         project_target (dict): The project to push assets to
@@ -479,6 +489,7 @@ def push_tasks_comments(project_source, client_source, client_target):
     Create a new comment into target api for each comment in source project
     but preserve only `created_at` field.
     Attachments and previews are created too.
+
     Args:
         project_source (dict): The project to get assets from
         project_target (dict): The project to push assets to
@@ -508,6 +519,7 @@ def push_task_comments(
     Create a new comment into target api for each comment in source task
     but preserve only `created_at` field.
     Attachments and previews are created too.
+
     Args:
         project_source (dict): The project to get assets from
         project_target (dict): The project to push assets to
@@ -547,6 +559,7 @@ def push_task_comment(
     Create a new comment into target api for each comment in source task
     but preserve only `created_at` field.
     Attachments and previews are created too.
+
     Args:
         project_source (dict): The project to get assets from
         project_target (dict): The project to push assets to

{gazu-0.10.14 → gazu-0.10.15}/gazu/task.py

@@ -257,25 +257,31 @@ def all_tasks_for_task_status(project, task_type, task_status, client=default):
 
 
 @cache
-def all_tasks_for_task_type(project, task_type, client=default):
+def all_tasks_for_task_type(
+    project,
+    task_type,
+    episode=None,
+    client=default
+):
     """
     Args:
         project (str / dict): The project dict or the project ID.
         task_type (str / dict): The task type dict or ID.
+        episode_id (str / dict): The episode dict or ID.
 
     Returns:
         list: Tasks for given project and task type.
     """
     project = normalize_model_parameter(project)
     task_type = normalize_model_parameter(task_type)
-    return raw.fetch_all(
-        "tasks",
-        {
-            "project_id": project["id"],
-            "task_type_id": task_type["id"],
-        },
-        client=client,
-    )
+    params = {
+        "project_id": project["id"],
+        "task_type_id": task_type["id"],
+    }
+    if episode is not None:
+        episode = normalize_model_parameter(episode)
+        params["episode_id"] = episode["id"]
+    return raw.fetch_all("tasks", params, client=client)
 
 
 @cache
@@ -1260,17 +1266,31 @@ def get_task_url(task, client=default):
     )
 
 
-def all_tasks_for_project(project, client=default):
+def all_tasks_for_project(
+    project,
+    task_type=None,
+    episode=None,
+    client=default
+):
     """
     Args:
-        project (str / dict): The project
+        project (str / dict): The project (or its ID) to get tasks from.
+        task_type (str / dict): The task type (or its ID) to filter tasks.
+        episode (str / dict): The episode (or its ID) to filter tasks.
 
     Returns:
         dict: Tasks related to given project.
     """
     project = normalize_model_parameter(project)
     path = "/data/projects/%s/tasks" % project["id"]
-    return raw.get(path, client=client)
+    params = {}
+    if task_type is not None:
+        task_type = normalize_model_parameter(task_type)
+        params["task_type_id"] = task_type["id"]
+    if episode is not None:
+        episode = normalize_model_parameter(episode)
+        params["episode_id"] = episode["id"]
+    return raw.get(path, params=params, client=client)
 
 
 def update_comment(comment, client=default):

{gazu-0.10.14 → gazu-0.10.15}/gazu.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gazu
-Version: 0.10.14
+Version: 0.10.15
 Summary: Gazu is a client for Zou, the API to store the data of your CG production.
 Home-page: https://gazu.cg-wire.com/
 Author: CG Wire

{gazu-0.10.14 → gazu-0.10.15}/tests/test_casting.py

@@ -37,6 +37,19 @@ class CastingTestCase(unittest.TestCase):
             )
             self.assertEqual(casting[0]["asset_id"], fakeid("asset-1"))
 
+    def test_update_episode_casting(self):
+        casting = [{"asset_id": fakeid("asset-1"), "nb_occurences": 3}]
+        path = "data/projects/%s/entities/%s/casting" % (
+            fakeid("project-01"),
+            fakeid("episode-01"),
+        )
+        with requests_mock.mock() as mock:
+            mock.put(gazu.client.get_full_url(path), text=json.dumps(casting))
+            episode = {"id": fakeid("episode-01")}
+            project = {"id": fakeid("project-01")}
+            casting = gazu.casting.update_episode_casting(project, episode, casting)
+            self.assertEqual(casting[0]["asset_id"], fakeid("asset-1"))
+
     def test_get_asset_type_casting(self):
         casting = {
             fakeid("asset-2"): [
@@ -92,6 +105,21 @@ class CastingTestCase(unittest.TestCase):
             casting = gazu.casting.get_shot_casting(shot)
             self.assertEqual(casting[0]["asset_id"], fakeid("asset-1"))
 
+    def test_get_episode_casting(self):
+        casting = [{"asset_id": fakeid("asset-1"), "nb_occurences": 3}]
+        path = "data/projects/%s/entities/%s/casting" % (
+            fakeid("project-01"),
+            fakeid("episode-01"),
+        )
+        with requests_mock.mock() as mock:
+            mock.get(gazu.client.get_full_url(path), text=json.dumps(casting))
+            episode = {
+                "id": fakeid("episode-01"),
+                "project_id": fakeid("project-01"),
+            }
+            casting = gazu.casting.get_episode_casting(episode)
+            self.assertEqual(casting[0]["asset_id"], fakeid("asset-1"))
+
     def test_get_asset_casting(self):
         casting = [{"asset_id": fakeid("asset-1"), "nb_occurences": 3}]
         path = "data/projects/%s/entities/%s/casting" % (

{gazu-0.10.14 → gazu-0.10.15}/tests/test_playlist.py

@@ -166,3 +166,48 @@ class TaskTestCase(unittest.TestCase):
             playlist = {"id": fakeid("playlist-1"), "name": "name_changed"}
             playlist = gazu.playlist.update_playlist(playlist)
             self.assertEqual(playlist["id"], fakeid("playlist-1"))
+
+    def test_add_entity_to_playlist(self):
+        with requests_mock.mock() as mock:
+            mock.put(
+                gazu.client.get_full_url(
+                    "data/playlists/%s" % fakeid("playlist-1")
+                ),
+                text=json.dumps(
+                    {"id": fakeid("playlist-1"), "name": "name_changed"}
+                ),
+            )
+            mock.get(
+                gazu.client.get_full_url(
+                    "data/playlists/entities/%s/preview-files" % fakeid("shot-1")
+                ),
+                text=json.dumps(
+                    {fakeid("task-type-1"): [{"id": fakeid("preview-1")}]}
+                ),
+            )
+            playlist = {
+                "id": fakeid("playlist-1"),
+                "name": "name_changed",
+                "shots": []
+            }
+            shot = {
+                "id": fakeid("shot-1"),
+                "name": "SH01"
+            }
+            playlist = gazu.playlist.add_entity_to_playlist(playlist, shot)
+            self.assertEqual(playlist["id"], fakeid("playlist-1"))
+            self.assertEqual(playlist["shots"], [{
+                "entity_id": fakeid("shot-1"),
+                "preview_file_id": fakeid("preview-1")
+            }])
+            playlist = gazu.playlist.update_entity_preview(
+                playlist,
+                shot,
+                fakeid("preview-2")
+            )
+            self.assertEqual(playlist["shots"], [{
+                "entity_id": fakeid("shot-1"),
+                "preview_file_id": fakeid("preview-2")
+            }])
+            playlist = gazu.playlist.remove_entity_from_playlist(playlist, shot)
+            self.assertEqual(playlist["shots"], [])

gazu-0.10.14/gazu/__version__.py

@@ -1 +0,0 @@
-__version__ = "0.10.14"