primitive 0.1.73__tar.gz → 0.1.75__tar.gz

{primitive-0.1.73 → primitive-0.1.75}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: primitive
-Version: 0.1.73
+Version: 0.1.75
 Project-URL: Documentation, https://github.com//primitivecorp/primitive-cli#readme
 Project-URL: Issues, https://github.com//primitivecorp/primitive-cli/issues
 Project-URL: Source, https://github.com//primitivecorp/primitive-cli

{primitive-0.1.73 → primitive-0.1.75}/src/primitive/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2024-present Dylan Stein <dylan@primitive.tech>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "0.1.73"
+__version__ = "0.1.75"

{primitive-0.1.73 → primitive-0.1.75}/src/primitive/agent/uploader.py

@@ -45,16 +45,11 @@ class Uploader:
             )
 
         for file in files:
-            # result = self.primitive.files.upload_file_direct(
-            #     path=file,
-            #     key_prefix=str(PurePath(file).relative_to(cache.parent).parent)
-            # )
-            # upload_id = result.data["fileUpdate"]["id"]
-            response = self.primitive.files.upload_file_via_api(
+            result = self.primitive.files.upload_file_direct(
                 path=file,
-                key_prefix=str(PurePath(file).relative_to(cache.parent).parent)
+                key_prefix=str(PurePath(file).relative_to(cache.parent).parent),
             )
-            upload_id = response.json()["data"]["fileUpload"]["id"]
+            upload_id = result.data["fileUpdate"]["id"]
 
             if upload_id:
                 file_ids.append(upload_id)

{primitive-0.1.73 → primitive-0.1.75}/src/primitive/auth/commands.py

@@ -1,11 +1,12 @@
 import os
+import typing
 import webbrowser
+
 import click
-from .actions import Auth
+
 from ..utils.config import PRIMITIVE_CREDENTIALS_FILEPATH
 from ..utils.printer import print_result
-
-import typing
+from .actions import Auth
 
 if typing.TYPE_CHECKING:
     from ..client import Primitive

{primitive-0.1.73 → primitive-0.1.75}/src/primitive/graphql/sdk.py

@@ -1,7 +1,9 @@
 from typing import Dict
+
+import requests
 from gql import Client
 from gql.transport.aiohttp import AIOHTTPTransport
-import requests
+
 from primitive.__about__ import __version__
 
 

{primitive-0.1.73 → primitive-0.1.75}/src/primitive/utils/auth.py

@@ -1,9 +1,23 @@
 import sys
 
+import backoff
+from aiohttp.client_exceptions import ClientConnectorError
+from gql.transport.exceptions import TransportServerError
 from loguru import logger
 
 from ..graphql.sdk import create_session
 
+MAX_TIME_FOR_BACKOFF = 60 * 15  # 15 minutes
+
+
+def connection_backoff_handler(details):
+    logger.enable("primitive")
+    logger.error(
+        "Cannot connect to API. Waiting {wait:0.1f} seconds after {tries} tries.".format(
+            **details
+        )
+    )
+
 
 def create_new_session(primitive):
     token = primitive.host_config.get("token")
@@ -26,6 +40,15 @@ def create_new_session(primitive):
 
 
 def guard(func):
+    @backoff.on_exception(
+        backoff.expo,
+        (
+            ClientConnectorError,
+            TransportServerError,
+        ),
+        on_backoff=connection_backoff_handler,
+        max_time=MAX_TIME_FOR_BACKOFF,
+    )
     def wrapper(self, *args, **kwargs):
         if self.primitive.session is None:
             self.primitive.session = create_new_session(self.primitive)

{primitive-0.1.73 → primitive-0.1.75}/src/primitive/utils/files.py

@@ -19,7 +19,7 @@ def find_files_for_extension(source: Path, extensions: Tuple[str]) -> List[Path]
         for filename in filenames:
             if filename.endswith(extensions):
                 matching_files.append(Path(dirpath).joinpath(filename))
-    
+
     logger.debug(
         f"Found {len(matching_files)} following files that match: {matching_files}"
     )