gazu 0.10.29__tar.gz → 0.10.31__tar.gz

{gazu-0.10.29 → gazu-0.10.31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: gazu
-Version: 0.10.29
+Version: 0.10.31
 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.31/gazu/__version__.py

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

{gazu-0.10.29 → gazu-0.10.31}/gazu/files.py

@@ -1281,7 +1281,7 @@ def upload_project_avatar(project, file_path, client=default):
     Upload given file as project avatar.
 
     Args:
-        project (str / dict): The project dict or the project ID.
+        project (str / dict): The project dict or ID.
         file_path (str): Path of the file to upload as avatar.
     """
     path = (
@@ -1291,6 +1291,36 @@ def upload_project_avatar(project, file_path, client=default):
     return raw.upload(path, file_path, client=client)
 
 
+def download_organisation_avatar(organisation, file_path, client=default):
+    """
+    Download given organisation's avatar and save it at given location.
+    Args:
+        organisation (str / dict): The organisation dict or ID.
+        file_path (str): Location on hard drive where to save the file.
+    """
+    organisation = normalize_model_parameter(organisation)
+    return raw.download(
+        "pictures/thumbnails/organisations/%s.png" % (organisation["id"]),
+        file_path,
+        client=client,
+    )
+
+
+def upload_organisation_avatar(organisation, file_path, client=default):
+    """
+    Upload given file as organisation avatar.
+
+    Args:
+        organisation (str / dict): The organisation dict or ID.
+        file_path (str): Path of the file to upload as avatar.
+    """
+    path = (
+        "/pictures/thumbnails/organisations/%s"
+        % normalize_model_parameter(organisation)["id"]
+    )
+    return raw.upload(path, file_path, client=client)
+
+
 def update_preview(preview_file, data, client=default):
     """
     Update the data of given preview file.

{gazu-0.10.29 → gazu-0.10.31}/gazu/shot.py

@@ -68,7 +68,7 @@ def all_shots_for_sequence(sequence, client=default):
 def all_sequences_for_project(project, client=default):
     """
     Args:
-        sequence (str / dict): The sequence dict or the sequence ID.
+        sequence (str / dict): The project dict or the project ID.
 
     Returns:
         list: Sequences from database for given project.
@@ -83,7 +83,7 @@ def all_sequences_for_project(project, client=default):
 def all_sequences_for_episode(episode, client=default):
     """
     Args:
-        sequence (str / dict): The sequence dict or the sequence ID.
+        sequence (str / dict): The episode dict or the episode ID.
 
     Returns:
         list: Sequences which are children of given episode.

{gazu-0.10.29 → gazu-0.10.31}/gazu.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: gazu
-Version: 0.10.29
+Version: 0.10.31
 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.29 → gazu-0.10.31}/tests/test_files.py

@@ -1230,3 +1230,41 @@ class FilesTestCase(unittest.TestCase):
                     ),
                     {"id": fakeid("project-1")},
                 )
+
+    def test_upload_organisation_avatar(self):
+        with open("./tests/fixtures/v1.png", "rb") as test_file:
+            with requests_mock.Mocker() as mock:
+                mock_route(
+                    mock,
+                    "POST",
+                    "pictures/thumbnails/organisations/%s"
+                    % fakeid("organisation-1"),
+                    text={"id": fakeid("organisation-1")},
+                )
+
+                add_verify_file_callback(mock, {"file": test_file.read()})
+
+                self.assertEqual(
+                    gazu.files.upload_organisation_avatar(
+                        fakeid("organisation-1"),
+                        "./tests/fixtures/v1.png",
+                    ),
+                    {"id": fakeid("organisation-1")},
+                )
+
+    def test_download_organisation_avatar(self):
+        with open("./tests/fixtures/v1.png", "rb") as thumbnail_file:
+            with requests_mock.mock() as mock:
+                path = "pictures/thumbnails/organisations/{}.png".format(
+                    fakeid("organisation-1")
+                )
+                mock.get(gazu.client.get_full_url(path), body=thumbnail_file)
+                gazu.files.download_organisation_avatar(
+                    fakeid("organisation-1"), "./test.png"
+                )
+                self.assertTrue(os.path.exists("./test.png"))
+                self.assertEqual(
+                    os.path.getsize("./test.png"),
+                    os.path.getsize("./tests/fixtures/v1.png"),
+                )
+                os.remove("./test.png")

gazu-0.10.29/gazu/__version__.py

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