gazu 0.10.20__tar.gz → 0.10.22__tar.gz

{gazu-0.10.20 → gazu-0.10.22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gazu
-Version: 0.10.20
+Version: 0.10.22
 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.20 → gazu-0.10.22}/gazu/__init__.py

@@ -87,22 +87,8 @@ def log_out(client=raw.default_client):
     return tokens
 
 
-def refresh_token(client=raw.default_client):
-    headers = {"User-Agent": "CGWire Gazu %s" % __version__}
-    if "refresh_token" in client.tokens:
-        headers["Authorization"] = "Bearer %s" % client.tokens["refresh_token"]
-
-    response = client.session.get(
-        raw.get_full_url("auth/refresh-token", client=client),
-        headers=headers,
-    )
-    raw.check_status(response, "auth/refresh-token")
-
-    tokens = response.json()
-
-    client.tokens["access_token"] = tokens["access_token"]
-
-    return tokens
+def refresh_access_token(client=raw.default_client):
+    return client.refresh_access_token()
 
 
 def get_event_host(client=raw.default_client):

gazu-0.10.22/gazu/__version__.py

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

{gazu-0.10.20 → gazu-0.10.22}/gazu/client.py

@@ -36,32 +36,107 @@ class KitsuClient(object):
         host,
         ssl_verify=True,
         cert=None,
-        automatic_refresh_token=False,
+        use_refresh_token=True,
         callback_not_authenticated=None,
+        tokens={"access_token": None, "refresh_token": None},
+        access_token=None,
+        refresh_token=None,
     ):
-        self.tokens = {"access_token": "", "refresh_token": ""}
+        self.tokens = tokens
+        if access_token:
+            self.access_token = access_token
+        if refresh_token:
+            self.refresh_token = refresh_token
+        self.use_refresh_token = use_refresh_token
+        self.callback_not_authenticated = callback_not_authenticated
+
         self.session = requests.Session()
         self.session.verify = ssl_verify
         self.session.cert = cert
         self.host = host
         self.event_host = host
-        self.automatic_refresh_token = automatic_refresh_token
-        self.callback_not_authenticated = callback_not_authenticated
+
+    @property
+    def access_token(self):
+        return self.tokens.get("access_token", None)
+
+    @access_token.setter
+    def access_token(self, token):
+        self.tokens["access_token"] = token
+
+    @property
+    def refresh_token(self):
+        return self.tokens.get("refresh_token", None)
+
+    @refresh_token.setter
+    def refresh_token(self, token):
+        self.tokens["refresh_token"] = token
+
+    def refresh_access_token(self):
+        """
+        Refresh access tokens for this client.
+
+        Returns:
+            dict: The new access token.
+        """
+        response = self.session.get(
+            get_full_url("auth/refresh-token", client=self),
+            headers={
+                "User-Agent": "CGWire Gazu " + __version__,
+                "Authorization": "Bearer " + self.refresh_token,
+            },
+        )
+        check_status(response, "auth/refresh-token")
+        tokens = response.json()
+
+        self.access_token = tokens["access_token"]
+        self.refresh_token = None
+
+        return tokens
+
+    def make_auth_header(self):
+        """
+        Make headers required to authenticate.
+
+        Returns:
+            dict: Headers required to authenticate.
+        """
+        headers = {"User-Agent": "CGWire Gazu " + __version__}
+
+        if self.access_token:
+            headers["Authorization"] = "Bearer " + self.access_token
+
+        return headers
 
 
 def create_client(
     host,
     ssl_verify=True,
     cert=None,
-    automatic_refresh_token=False,
+    use_refresh_token=False,
     callback_not_authenticated=None,
+    **kwargs
 ):
+    """
+    Create a client with given parameters.
+
+    Args:
+        host (str): The host to use for requests.
+        ssl_verify (bool): Whether to verify SSL certificates.
+        cert (str): Path to a client certificate.
+        use_refresh_token (bool): Whether to automatically refresh tokens.
+        callback_not_authenticated (function): Function to call when not authenticated.
+
+    Returns:
+        KitsuClient: The created client.
+    """
     return KitsuClient(
         host,
         ssl_verify,
         cert=cert,
-        automatic_refresh_token=automatic_refresh_token,
+        use_refresh_token=use_refresh_token,
         callback_not_authenticated=callback_not_authenticated,
+        **kwargs
     )
 
 
@@ -81,8 +156,13 @@ except Exception:
 
 def host_is_up(client=default_client):
     """
+    Check if the host is up.
+
+    Args:
+        client (KitsuClient): The client to use for the request.
+
     Returns:
-        True if the host is up.
+        bool: True if the host is up.
     """
     try:
         response = client.session.head(client.host)
@@ -94,8 +174,12 @@ def host_is_up(client=default_client):
 def host_is_valid(client=default_client):
     """
     Check if the host is valid by simulating a fake login.
+
+    Args:
+        client (KitsuClient): The client to use for the request.
+
     Returns:
-        True if the host is valid.
+        bool: True if the host is valid.
     """
     if not host_is_up(client):
         return False
@@ -107,14 +191,23 @@ def host_is_valid(client=default_client):
 
 def get_host(client=default_client):
     """
+    Get client.host.
+
+    Args:
+        client (KitsuClient): The client to use for the request.
+
     Returns:
-        Host on which requests are sent.
+        str: The host of the client.
     """
     return client.host
 
 
 def get_api_url_from_host(client=default_client):
     """
+    Get the API url from the host.
+
+    Args:
+        client (KitsuClient): The client to use for the request.
     Returns:
         Zou url, retrieved from host.
     """
@@ -123,8 +216,14 @@ def get_api_url_from_host(client=default_client):
 
 def set_host(new_host, client=default_client):
     """
+    Set the host for the client.
+
+    Args:
+        new_host (str): The new host to set.
+        client (KitsuClient): The client to use for the request.
+
     Returns:
-        Set currently configured host on which requests are sent.
+        str: The new host.
     """
     client.host = new_host
     return client.host
@@ -132,16 +231,27 @@ def set_host(new_host, client=default_client):
 
 def get_event_host(client=default_client):
     """
+    Get the host on which listening for events.
+
+    Args:
+        client (KitsuClient): The client to use for the request.
+
     Returns:
-        Host on which listening for events.
+        str: The event host.
     """
     return client.event_host or client.host
 
 
 def set_event_host(new_host, client=default_client):
     """
+    Set the host on which listening for events.
+
+    Args:
+        new_host (str): The new host to set.
+        client (KitsuClient): The client to use for the request.
+
     Returns:
-        Set currently configured host on which listening for events.
+        str: The new event host.
     """
     client.event_host = new_host
     return client.event_host
@@ -153,6 +263,10 @@ def set_tokens(new_tokens, client=default_client):
 
     Args:
         new_tokens (dict): Tokens to use for authentication.
+        client (KitsuClient): The client to use for the request.
+
+    Returns:
+        dict: The stored tokens.
     """
     client.tokens = new_tokens
     return client.tokens
@@ -160,13 +274,15 @@ def set_tokens(new_tokens, client=default_client):
 
 def make_auth_header(client=default_client):
     """
+    Make headers required to authenticate.
+
+    Args:
+        client (KitsuClient): The client to use for the request.
+
     Returns:
-        Headers required to authenticate.
+        dict: Headers required to authenticate.
     """
-    headers = {"User-Agent": "CGWire Gazu %s" % __version__}
-    if "access_token" in client.tokens:
-        headers["Authorization"] = "Bearer %s" % client.tokens["access_token"]
-    return headers
+    return client.make_auth_header()
 
 
 def url_path_join(*items):
@@ -176,12 +292,17 @@ def url_path_join(*items):
 
     Args:
         items (list): Path elements
+
+    Returns:
+        str: The joined path.
     """
     return "/".join([item.lstrip("/").rstrip("/") for item in items])
 
 
 def get_full_url(path, client=default_client):
     """
+    Join host url with given path.
+
     Args:
         path (str): The path to integrate to host url.
 
@@ -219,6 +340,12 @@ def get(path, json_response=True, params=None, client=default_client):
     """
     Run a get request toward given path for configured host.
 
+    Args:
+        path (str): The path to query.
+        json_response (bool): Whether to return a json response.
+        params (dict): The parameters to pass to the request.
+        client (KitsuClient): The client to use for the request.
+
     Returns:
         The request result.
     """
@@ -243,6 +370,11 @@ def post(path, data, client=default_client):
     """
     Run a post request toward given path for configured host.
 
+    Args:
+        path (str): The path to query.
+        data (dict): The data to post.
+        client (KitsuClient): The client to use for the request.
+
     Returns:
         The request result.
     """
@@ -270,6 +402,11 @@ def put(path, data, client=default_client):
     """
     Run a put request toward given path for configured host.
 
+    Args:
+        path (str): The path to query.
+        data (dict): The data to put.
+        client (KitsuClient): The client to use for the request.
+
     Returns:
         The request result.
     """
@@ -291,6 +428,11 @@ def delete(path, params=None, client=default_client):
     """
     Run a delete request toward given path for configured host.
 
+    Args:
+        path (str): The path to query.
+        params (dict): The parameters to pass to the request.
+        client (KitsuClient): The client to use for the request.
+
     Returns:
         The request result.
     """
@@ -320,15 +462,16 @@ def get_message_from_response(
         default_message: str - An optional default value to revert to if no message is detected.
 
     Returns:
-        The message of a given response, or a default message - if any.
+        str: The message to display to the user.
     """
     message = default_message
     message_json = response.json()
 
-    for key in ["error", "message"]:
-        if message_json.get(key):
-            message = message_json[key]
-            break
+    if hasattr(message_json, "get"):
+        for key in ["error", "message"]:
+            if message_json.get(key):
+                message = message_json[key]
+                break
 
     return message
 
@@ -340,6 +483,8 @@ def check_status(request, path, client=None):
 
     Args:
         request (Request): The request to validate.
+        path (str): The path of the request.
+        client (KitsuClient): The client to use for the request.
 
     Returns:
         int: Status code
@@ -369,19 +514,18 @@ def check_status(request, path, client=None):
         )
     elif status_code in [401, 422]:
         try:
-            if client is not None and client.automatic_refresh_token:
-                from . import refresh_token
-
-                refresh_token(client=client)
-
+            if (
+                client
+                and client.refresh_token
+                and client.use_refresh_token
+                and request.json()["message"] == "Signature has expired"
+            ):
+                client.refresh_access_token()
                 return status_code, True
             else:
                 raise NotAuthenticatedException(path)
         except NotAuthenticatedException:
-            if (
-                client is not None
-                and client.callback_not_authenticated is not None
-            ):
+            if client and client.callback_not_authenticated:
                 retry = client.callback_not_authenticated(client, path)
                 if retry:
                     return status_code, True
@@ -410,6 +554,8 @@ def fetch_all(
     """
     Args:
         path (str): The path for which we want to retrieve all entries.
+        params (dict): The parameters to pass to the request.
+        client (KitsuClient): The client to use for the request.
         paginated (bool): Will query entries page by page.
         limit (int): Limit the number of entries per page.
 
@@ -453,6 +599,8 @@ def fetch_first(path, params=None, client=default_client):
     """
     Args:
         path (str): The path for which we want to retrieve the first entry.
+        params (dict): The parameters to pass to the request.
+        client (KitsuClient): The client to use for the request.
 
     Returns:
         dict: The first entry for which a model is required.
@@ -464,7 +612,7 @@ def fetch_first(path, params=None, client=default_client):
         return None
 
 
-def fetch_one(model_name, id, client=default_client):
+def fetch_one(model_name, id, params=None, client=default_client):
     """
     Function dedicated at targeting routes that returns a single model
     instance.
@@ -472,11 +620,15 @@ def fetch_one(model_name, id, client=default_client):
     Args:
         model_name (str): Model type name.
         id (str): Model instance ID.
+        params (dict): The parameters to pass to the request.
+        client (KitsuClient): The client to use for the request.
 
     Returns:
         dict: The model instance matching id and model name.
     """
-    return get(url_path_join("data", model_name, id), client=client)
+    return get(
+        url_path_join("data", model_name, id), params=params, client=client
+    )
 
 
 def create(model_name, data, client=default_client):
@@ -484,8 +636,9 @@ def create(model_name, data, client=default_client):
     Create an entry for given model and data.
 
     Args:
-        model (str): The model type involved
-        data (str): The data to use for creation
+        model_name (str): The model type involved.
+        data (str): The data to use for creation.
+        client (KitsuClient): The client to use for the request.
 
     Returns:
         dict: Created entry
@@ -498,9 +651,10 @@ def update(model_name, model_id, data, client=default_client):
     Update an entry for given model, id and data.
 
     Args:
-        model (str): The model type involved
-        id (str): The target model id
-        data (dict): The data to update
+        model_name (str): The model type involved.
+        model_id (str): The target model id.
+        data (dict): The data to update.
+        client (KitsuClient): The client to use for the request.
 
     Returns:
         dict: Updated entry
@@ -510,19 +664,31 @@ def update(model_name, model_id, data, client=default_client):
     )
 
 
-def upload(path, file_path, data={}, extra_files=[], client=default_client):
+def upload(
+    path,
+    file_path=None,
+    data={},
+    extra_files=[],
+    files=None,
+    client=default_client,
+):
     """
     Upload file located at *file_path* to given url *path*.
 
     Args:
         path (str): The url path to upload file.
         file_path (str): The file location on the hard drive.
+        data (dict): The data to send with the file.
+        extra_files (list): List of extra files to upload.
+        files (dict): The dictionary of files to upload.
+        client (KitsuClient): The client to use for the request.
 
     Returns:
         Response: Request response object.
     """
     url = get_full_url(path, client)
-    files = _build_file_dict(file_path, extra_files)
+    if not files:
+        files = _build_file_dict(file_path, extra_files)
     retry = True
     while retry:
         response = client.session.post(
@@ -546,11 +712,23 @@ def upload(path, file_path, data={}, extra_files=[], client=default_client):
 
 
 def _build_file_dict(file_path, extra_files):
+    """
+    Build a dictionary of files to upload.
+
+    Args:
+        file_path (str): The file location on the hard drive.
+        extra_files (list): List of extra files to upload.
+
+    Returns:
+        dict: The dictionary of files to upload.
+    """
+
     files = {"file": open(file_path, "rb")}
-    i = 2
+    i = 0
     for file_path in extra_files:
-        files["file-%s" % i] = open(file_path, "rb")
         i += 1
+        files["file-%s" % i] = open(file_path, "rb")
+
     return files
 
 
@@ -561,6 +739,8 @@ def download(path, file_path, params=None, client=default_client):
     Args:
         path (str): The url path to download file from.
         file_path (str): The location to store the file on the hard drive.
+        params (dict): The parameters to pass to the request.
+        client (KitsuClient): The client to use for the request.
 
     Returns:
         Response: Request response object.
@@ -580,6 +760,14 @@ def download(path, file_path, params=None, client=default_client):
 def get_file_data_from_url(url, full=False, client=default_client):
     """
     Return data found at given url.
+
+    Args:
+        url (str): The url to fetch data from.
+        full (bool): Whether to use full url.
+        client (KitsuClient): The client to use for the request.
+
+    Returns:
+        bytes: The data found at the given url.
     """
     if not full:
         url = get_full_url(url)
@@ -596,15 +784,26 @@ def get_file_data_from_url(url, full=False, client=default_client):
 
 def import_data(model_name, data, client=default_client):
     """
+    Import data for given model.
+
     Args:
-        model_name (str): The data model to import
-        data (dict): The data to import
+        model_name (str): The data model to import.
+        data (dict): The data to import.
+        client (KitsuClient): The client to use for the request.
+
+    Returns:
+        dict: The imported data.
     """
     return post("/import/kitsu/%s" % model_name, data, client=client)
 
 
 def get_api_version(client=default_client):
     """
+    Get the current version of the API.
+
+    Args:
+        client (KitsuClient): The client to use for the request.
+
     Returns:
         str: Current version of the API.
     """
@@ -613,6 +812,11 @@ def get_api_version(client=default_client):
 
 def get_current_user(client=default_client):
     """
+    Get the current user.
+
+    Args:
+        client (KitsuClient): The client to use for the request.
+
     Returns:
         dict: User database information for user linked to auth tokens.
     """

{gazu-0.10.20 → gazu-0.10.22}/gazu/events.py

@@ -52,11 +52,13 @@ def init(
     Returns:
         Event client that will be able to set listeners.
     """
-    params = {"ssl_verify": ssl_verify}
+    params = {
+        "ssl_verify": ssl_verify,
+        "reconnection": reconnection,
+        "logger": logger,
+    }
     params.update(kwargs)
-    event_client = socketio.Client(
-        logger=logger, reconnection=reconnection, **params
-    )
+    event_client = socketio.Client(**params)
     event_client.on("connect_error", connect_error)
     event_client.register_namespace(EventsNamespace("/events"))
     event_client.connect(get_event_host(client), make_auth_header())

{gazu-0.10.20 → gazu-0.10.22}/gazu/sync.py

@@ -17,7 +17,7 @@ default = raw.default_client
 
 
 def get_last_events(
-    page_size=20000,
+    limit=20000,
     project=None,
     after=None,
     before=None,
@@ -29,7 +29,7 @@ def get_last_events(
     Get last events that occured on the machine.
 
     Args:
-        page_size (int): Number of events to retrieve.
+        limit (int): Number of events to retrieve.
         project (dict/id): Get only events related to this project.
         after (dict/id): Get only events occuring after given date.
         before (dict/id): Get only events occuring before given date.
@@ -39,7 +39,7 @@ def get_last_events(
         dict: Last events matching criterions.
     """
     path = "/data/events/last"
-    params = {"page_size": page_size, "only_files": only_files}
+    params = {"limit": limit, "only_files": only_files}
     if project is not None:
         project = normalize_model_parameter(project)
         params["project_id"] = project["id"]

{gazu-0.10.20 → gazu-0.10.22}/gazu/task.py

@@ -839,6 +839,7 @@ def add_comment(
         "task_status_id": task_status["id"],
         "comment": comment,
         "checklist": checklist,
+        "links": links,
     }
 
     if person is not None:
@@ -1014,6 +1015,7 @@ def publish_preview(
     normalize_movie=True,
     revision=None,
     set_thumbnail=False,
+    links=[],
     client=default,
 ):
     """
@@ -1035,6 +1037,7 @@ def publish_preview(
         server side.
         revision (int): Revision number.
         set_thumbnail (bool): Set the preview as thumbnail of the entity.
+        links (list): List of links to add to the comment
     Returns:
         tuple(dict, dict): Created comment model and created preview file
         model.
@@ -1047,6 +1050,7 @@ def publish_preview(
         checklist=checklist,
         attachments=attachments,
         created_at=created_at,
+        links=links,
         client=client,
     )
     preview_file = add_preview(
@@ -1063,6 +1067,39 @@ def publish_preview(
     return new_comment, preview_file
 
 
+def publish_comments_previews(task, comments=[], client=default):
+    """
+    Publish a list of comments (with attachments and previews) for given task.
+    Each dict comments may contain a list of attachment files path and preview
+    files path in the keys "attachment_files" and "preview_files".
+
+    Args:
+        task (str / dict): The task dict or the task ID.
+        comments (list): List of comments to publish.
+
+    Returns:
+        list: List of created comments.
+    """
+    task = normalize_model_parameter(task)
+
+    files = {}
+    for x, comment in enumerate(comments):
+        if comment.get("attachment_files"):
+            for y, file_path in enumerate(comment["attachment_files"]):
+                files["attachment_file-%i-%i" % (x, y)] = open(file_path, "rb")
+        if comment.get("preview_files"):
+            for y, file_path in enumerate(comment["preview_files"]):
+                files["preview_file-%i-%i" % (x, y)] = open(file_path, "rb")
+
+    files["comments"] = (None, json.dumps(comments), "application/json")
+    return raw.upload(
+        "actions/tasks/%s/add-comments-previews" % task["id"],
+        file_path=None,
+        files=files,
+        client=client,
+    )
+
+
 def set_main_preview(preview_file, frame_number=None, client=default):
     """
     Set given preview as thumbnail of given entity.

{gazu-0.10.20 → gazu-0.10.22}/gazu.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gazu
-Version: 0.10.20
+Version: 0.10.22
 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.20 → gazu-0.10.22}/tests/test_client.py

@@ -70,7 +70,10 @@ class BaseFuncTestCase(ClientTestCase):
         pass
 
     def test_make_auth_header(self):
-        pass
+        self.assertEqual(
+            raw.default_client.make_auth_header(),
+            raw.make_auth_header(),
+        )
 
     def test_url_path_join(self):
         root = raw.get_host()
@@ -255,6 +258,7 @@ class BaseFuncTestCase(ClientTestCase):
 
     def test_make_auth_token(self):
         tokens = {"access_token": "token_test"}
+
         raw.set_tokens(tokens)
         self.assertEqual(
             raw.make_auth_header(),
@@ -327,7 +331,7 @@ class BaseFuncTestCase(ClientTestCase):
                     mock,
                     {
                         "file": test_file_read,
-                        "file-2": test_file_read,
+                        "file-1": test_file_read,
                         "test": "True",
                     },
                 )
@@ -437,7 +441,7 @@ class BaseFuncTestCase(ClientTestCase):
                 "auth/refresh-token",
                 text={"access_token": "tokentest1"},
             )
-            gazu.refresh_token()
+            gazu.refresh_access_token()
         self.assertEqual(
             raw.default_client.tokens["access_token"], "tokentest1"
         )

gazu-0.10.20/gazu/__version__.py

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