primitive 0.1.95__tar.gz → 0.1.96__tar.gz

{primitive-0.1.95 → primitive-0.1.96}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: primitive
-Version: 0.1.95
+Version: 0.1.96
 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.95 → primitive-0.1.96}/src/primitive/__about__.py

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

{primitive-0.1.95 → primitive-0.1.96}/src/primitive/agent/actions.py

@@ -8,6 +8,7 @@ from primitive.__about__ import __version__
 from primitive.utils.actions import BaseAction
 
 from ..utils.cache import get_sources_cache
+from ..utils.exceptions import P_CLI_100
 from .runner import AgentRunner
 from .uploader import Uploader
 
@@ -181,5 +182,11 @@ class Agent(BaseAction):
                 sleep(5)
         except KeyboardInterrupt:
             logger.info(" [*] Stopping primitive...")
-            self.primitive.hardware.check_in_http(is_available=False, is_online=False)
+            try:
+                self.primitive.hardware.check_in_http(
+                    is_available=False, is_online=False, stopping_agent=True
+                )
+            except P_CLI_100 as exception:
+                logger.error(" [*] Error stopping primitive.")
+                logger.error(str(exception))
             sys.exit()

{primitive-0.1.95 → primitive-0.1.96}/src/primitive/hardware/actions.py

@@ -17,6 +17,7 @@ from primitive.utils.memory_size import MemorySize
 
 from ..utils.auth import guard
 from ..utils.config import update_config_file
+from ..utils.exceptions import P_CLI_100
 from .graphql.mutations import (
     hardware_checkin_mutation,
     hardware_update_mutation,
@@ -384,6 +385,7 @@ class Hardware(BaseAction):
         is_available: bool = False,
         is_online: bool = True,
         fingerprint: Optional[str] = None,
+        stopping_agent: Optional[bool] = False,
     ):
         # if no fingerprint supplied from argument try from the host_config
         if not fingerprint:
@@ -457,9 +459,12 @@ class Hardware(BaseAction):
                 raise Exception(message)
             return result
         except client_exceptions.ClientConnectorError as exception:
-            message = " [*] Failed to check in! "
-            logger.exception(message)
-            raise exception
+            if not stopping_agent:
+                message = " [*] Failed to check in! "
+                logger.error(message)
+                raise exception
+            else:
+                raise P_CLI_100
 
     @guard
     def get_hardware_list(

{primitive-0.1.95 → primitive-0.1.96}/src/primitive/utils/auth.py

@@ -43,6 +43,7 @@ def guard(func):
     @backoff.on_exception(
         backoff.expo,
         (
+            ConnectionRefusedError,
             ClientConnectorError,
             TransportServerError,
         ),

primitive-0.1.96/src/primitive/utils/exceptions.py

@@ -0,0 +1,12 @@
+from dataclasses import dataclass
+
+
+@dataclass
+class P_CLI_100(Exception):
+    """Check In Connection Failure while Stopping Agent"""
+
+    codename: str = "P_CLI_100"
+    message: str = "Check In Connection Failure while Stopping Agent"
+
+    def __str__(self):
+        return f"{self.codename}: {self.message}"

{primitive-0.1.95 → primitive-0.1.96}/uv.lock

@@ -280,7 +280,7 @@ name = "click"
 version = "8.1.8"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
-    { name = "colorama", marker = "platform_system == 'Windows'" },
+    { name = "colorama", marker = "sys_platform == 'win32'" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/b9/2e/0090cbf739cee7d23781ad4b89a9894a41538e4fcf4c31dcdd705b78eb8b/click-8.1.8.tar.gz", hash = "sha256:ed53c9d8990d83c2a27deae68e4ee337473f6330c040a31d4225c9574d16096a", size = 226593 }
 wheels = [
@@ -703,7 +703,6 @@ wheels = [
 
 [[package]]
 name = "primitive"
-version = "0.1.89"
 source = { editable = "." }
 dependencies = [
     { name = "click" },