gazu 0.10.10__py2.py3-none-any.whl → 0.10.12__py2.py3-none-any.whl

gazu/__init__.py

@@ -120,9 +120,8 @@ def set_token(token, client=raw.default_client):
     Args:
         new_tokens (dict): Tokens to use for authentication.
     """
-    tokens = {}
+
     if isinstance(token, dict):
-        tokens["access_token"] = token["access_token"]
+        return raw.set_tokens(token, client=client)
     else:
-        tokens["access_token"] = token
-    return raw.set_tokens(tokens, client=client)
+        return raw.set_tokens({"access_token": token}, client=client)

gazu/__version__.py

@@ -1 +1 @@
-__version__ = "0.10.10"
+__version__ = "0.10.12"

gazu/client.py

@@ -20,6 +20,7 @@ from .exception import (
     UploadFailedException,
 )
 
+
 if sys.version_info[0] == 3:
     from json import JSONDecodeError
 else:
@@ -29,17 +30,38 @@ DEBUG = os.getenv("GAZU_DEBUG", "false").lower() == "true"
 
 
 class KitsuClient(object):
-    def __init__(self, host, ssl_verify=True, cert=None):
+    def __init__(
+        self,
+        host,
+        ssl_verify=True,
+        cert=None,
+        automatic_refresh_token=False,
+        callback_not_authenticated=None,
+    ):
         self.tokens = {"access_token": "", "refresh_token": ""}
         self.session = requests.Session()
         self.session.verify = ssl_verify
         self.session.cert = cert
         self.host = host
         self.event_host = host
-
-
-def create_client(host, ssl_verify=True, cert=None):
-    return KitsuClient(host, ssl_verify, cert=None)
+        self.automatic_refresh_token = automatic_refresh_token
+        self.callback_not_authenticated = callback_not_authenticated
+
+
+def create_client(
+    host,
+    ssl_verify=True,
+    cert=None,
+    automatic_refresh_token=False,
+    callback_not_authenticated=None,
+):
+    return KitsuClient(
+        host,
+        ssl_verify,
+        cert=cert,
+        automatic_refresh_token=automatic_refresh_token,
+        callback_not_authenticated=callback_not_authenticated,
+    )
 
 
 default_client = None
@@ -199,11 +221,13 @@ def get(path, json_response=True, params=None, client=default_client):
     if DEBUG:
         print("GET", get_full_url(path, client))
     path = build_path_with_params(path, params)
-    response = client.session.get(
-        get_full_url(path, client=client),
-        headers=make_auth_header(client=client),
-    )
-    check_status(response, path)
+    retry = True
+    while retry:
+        response = client.session.get(
+            get_full_url(path, client=client),
+            headers=make_auth_header(client=client),
+        )
+        _, retry = check_status(response, path, client=client)
 
     if json_response:
         return response.json()
@@ -222,12 +246,14 @@ def post(path, data, client=default_client):
         print("POST", get_full_url(path, client))
         if not "password" in data:
             print("Body:", data)
-    response = client.session.post(
-        get_full_url(path, client),
-        json=data,
-        headers=make_auth_header(client=client),
-    )
-    check_status(response, path)
+    retry = True
+    while retry:
+        response = client.session.post(
+            get_full_url(path, client),
+            json=data,
+            headers=make_auth_header(client=client),
+        )
+        _, retry = check_status(response, path, client=client)
     try:
         result = response.json()
     except JSONDecodeError:
@@ -246,12 +272,14 @@ def put(path, data, client=default_client):
     if DEBUG:
         print("PUT", get_full_url(path, client))
         print("Body:", data)
-    response = client.session.put(
-        get_full_url(path, client),
-        json=data,
-        headers=make_auth_header(client=client),
-    )
-    check_status(response, path)
+    retry = True
+    while retry:
+        response = client.session.put(
+            get_full_url(path, client),
+            json=data,
+            headers=make_auth_header(client=client),
+        )
+        _, retry = check_status(response, path, client=client)
     return response.json()
 
 
@@ -266,14 +294,16 @@ def delete(path, params=None, client=default_client):
         print("DELETE", get_full_url(path, client))
     path = build_path_with_params(path, params)
 
-    response = client.session.delete(
-        get_full_url(path, client), headers=make_auth_header(client=client)
-    )
-    check_status(response, path)
+    retry = True
+    while retry:
+        response = client.session.delete(
+            get_full_url(path, client), headers=make_auth_header(client=client)
+        )
+        _, retry = check_status(response, path, client=client)
     return response.text
 
 
-def check_status(request, path):
+def check_status(request, path, client=None):
     """
     Raise an exception related to status code, if the status code does not
     match a success code. Print error message when it's relevant.
@@ -309,7 +339,24 @@ def check_status(request, path):
             "Change your proxy configuration to allow bigger files." % path
         )
     elif status_code in [401, 422]:
-        raise NotAuthenticatedException(path)
+        try:
+            if client is not None and client.automatic_refresh_token:
+                from . import refresh_token
+
+                refresh_token(client=client)
+
+                return status_code, True
+            else:
+                raise NotAuthenticatedException(path)
+        except NotAuthenticatedException:
+            if (
+                client is not None
+                and client.callback_not_authenticated is not None
+            ):
+                retry = client.callback_not_authenticated(client, path)
+                if retry:
+                    return status_code, True
+            raise
     elif status_code in [500, 502]:
         try:
             stacktrace = request.json().get(
@@ -324,7 +371,7 @@ def check_status(request, path):
         except Exception:
             print(request.text)
         raise ServerErrorException(path)
-    return status_code
+    return status_code, False
 
 
 def fetch_all(path, params=None, client=default_client):
@@ -413,10 +460,15 @@ def upload(path, file_path, data={}, extra_files=[], client=default_client):
     """
     url = get_full_url(path, client)
     files = _build_file_dict(file_path, extra_files)
-    response = client.session.post(
-        url, data=data, headers=make_auth_header(client=client), files=files
-    )
-    check_status(response, path)
+    retry = True
+    while retry:
+        response = client.session.post(
+            url,
+            data=data,
+            headers=make_auth_header(client=client),
+            files=files,
+        )
+        _, retry = check_status(response, path, client=client)
     try:
         result = response.json()
     except JSONDecodeError:
@@ -465,12 +517,14 @@ def get_file_data_from_url(url, full=False, client=default_client):
     """
     if not full:
         url = get_full_url(url)
-    response = requests.get(
-        url,
-        stream=True,
-        headers=make_auth_header(client=client),
-    )
-    check_status(response, url)
+    retry = True
+    while retry:
+        response = requests.get(
+            url,
+            stream=True,
+            headers=make_auth_header(client=client),
+        )
+        _, retry = check_status(response, url, client=client)
     return response.content
 
 

gazu/task.py

@@ -582,6 +582,40 @@ def remove_task_status(task_status, client=default):
     )
 
 
+def update_task_type(task_type, client=default):
+    """
+    Update given task type into the API.
+
+    Args:
+        task_type (dict): The task type dict to update.
+
+    Returns:
+        dict: Updated task type.
+    """
+    return raw.put(
+        "data/task-types/%s" % task_type["id"],
+        task_type,
+        client=client,
+    )
+
+
+def update_task_status(task_status, client=default):
+    """
+    Update given task status into the API.
+
+    Args:
+        task_status (dict): The task status dict to update.
+
+    Returns:
+        dict: Updated task status.
+    """
+    return raw.put(
+        "data/task-status/%s" % task_status["id"],
+        task_status,
+        client=client,
+    )
+
+
 @cache
 def get_task(task_id, client=default):
     """
@@ -1166,7 +1200,7 @@ def new_task_status(name, short_name, color, client=default):
 
 def update_task(task, client=default):
     """
-    Save given task data into the API. Metadata are fully replaced by the ones
+    Update given task into the API. Metadata are fully replaced by the ones
     set on given task.
 
     Args:

{gazu-0.10.10.dist-info → gazu-0.10.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gazu
-Version: 0.10.10
+Version: 0.10.12
 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.10.dist-info → gazu-0.10.12.dist-info}/RECORD

@@ -1,9 +1,9 @@
-gazu/__init__.py,sha256=cek7Vq4FrzBIzMvtAjlL-3af5TOVLWywYBSMQQWuYec,3056
-gazu/__version__.py,sha256=Aktsk85UlRwldAMwi5D3o9k6UTyqVYYJbSK_i8y26sE,24
+gazu/__init__.py,sha256=gVhtpZsvDiPuqvsEbCrgm6fKw8hluUhqspmlgGNIAvQ,3020
+gazu/__version__.py,sha256=8HmqcalKYluoRM1nCmpntcjsFDeqXAb5Cl0wcMuiFY0,24
 gazu/asset.py,sha256=GWwam0yqHna_b9_5g5r0-O2zhMb7KlhfbIC17IOXKkY,14488
 gazu/cache.py,sha256=MnxrnfYN7wHNTTL7qzkEpYCYzWcolT56fqQ0_RegMbE,5879
 gazu/casting.py,sha256=5xmGMoU3MFgv1br8_5xyTx-Z7wo9r51wRisiblqcr94,3945
-gazu/client.py,sha256=wgilPeAAF2D3heOfRkZdMUjR1QzWaFs7vmlEKSjNxsg,13032
+gazu/client.py,sha256=YRIxX7u095XTe28JM0tMZ05WoTcYexJGS2jrxXSAQvs,14640
 gazu/concept.py,sha256=GcOPEmkbtZcSwlX8tnUj9Q5DTPBprSxtmXhlq7ioPwk,3727
 gazu/context.py,sha256=iUyug8EUz3kkF-kmYlH5JuLp66TUqR3uhAq7CouVd_U,4349
 gazu/edit.py,sha256=sPSsnzykGr1Htl6ceKulUSVHGhoQLGLeWDni3Pul7BE,4609
@@ -20,10 +20,10 @@ gazu/scene.py,sha256=bYsy7zKdQfDTfeqNAlgaqGMs2Hhxdy40NGt6TX_FBdA,5307
 gazu/shot.py,sha256=4fmMj4vWb4VMg92tfRDP5lqjSwip3Yv9lPrKerv_oUM,18850
 gazu/sorting.py,sha256=qSIO0pOHkj0Tl4gm9BJrYrcifWGGGmsW68Pl86zB_bg,266
 gazu/sync.py,sha256=OMsbQDROTdEwlukazUXPGnlC29QcaaHvsmqv3WsSexo,21582
-gazu/task.py,sha256=4aO8Ggk4lagVaNkFIdFE0-HnCcWg6d38BTFuwlqbyRY,35193
+gazu/task.py,sha256=7A3A_Zms4WlFl72Jqi2u8dWcoVY3d81leFWAE8wOdQQ,35883
 gazu/user.py,sha256=GyJf6mrynHvLllw3Hsiv-6wjaYTHO_PBNkJzyJjJA1A,9556
-gazu-0.10.10.dist-info/LICENSE,sha256=2n6rt7r999OuXp8iOqW9we7ORaxWncIbOwN1ILRGR2g,7651
-gazu-0.10.10.dist-info/METADATA,sha256=L-Y-tTqDlvxkwmzjgby8xq_iWk8s1K_83oinzrXTuDg,5182
-gazu-0.10.10.dist-info/WHEEL,sha256=DZajD4pwLWue70CAfc7YaxT1wLUciNBvN_TTcvXpltE,110
-gazu-0.10.10.dist-info/top_level.txt,sha256=nv7fRIVpYYyIlk_66hBmMyvWcSC7UU-r-GE8uC1u1Go,5
-gazu-0.10.10.dist-info/RECORD,,
+gazu-0.10.12.dist-info/LICENSE,sha256=2n6rt7r999OuXp8iOqW9we7ORaxWncIbOwN1ILRGR2g,7651
+gazu-0.10.12.dist-info/METADATA,sha256=khWS9mumUPINWXh353r625F8C_Z-xgmro-Jrk0KZ2v4,5182
+gazu-0.10.12.dist-info/WHEEL,sha256=XRxW4r1PNiVhMpP4bT9oWtu3HyndxpJ84SkubFgzp_Y,109
+gazu-0.10.12.dist-info/top_level.txt,sha256=nv7fRIVpYYyIlk_66hBmMyvWcSC7UU-r-GE8uC1u1Go,5
+gazu-0.10.12.dist-info/RECORD,,

{gazu-0.10.10.dist-info → gazu-0.10.12.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: setuptools (72.1.0)
 Root-Is-Purelib: true
 Tag: py2-none-any
 Tag: py3-none-any