primitive 0.1.9__py3-none-any.whl → 0.1.11__py3-none-any.whl

primitive/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2024-present Dylan Stein <dylan@steins.studio>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "0.1.9"
+__version__ = "0.1.11"

primitive/agent/actions.py

@@ -14,7 +14,11 @@ class Agent(BaseAction):
         logger.info(f" [*] Version: {__version__}")
 
         # self.primitive.hardware.update_hardware_system_info()
-        self.primitive.hardware.check_in_http(is_available=True, is_online=True)
+        try:
+            self.primitive.hardware.check_in_http(is_available=True, is_online=True)
+        except Exception as ex:
+            logger.error(f"Error checking in hardware: {ex}")
+            sys.exit(1)
 
         try:
             while True:

primitive/hardware/actions.py

@@ -171,12 +171,6 @@ class Hardware(BaseAction):
 
                 gpu_config.append(gpu_info)
 
-        # Raise an error if there is no valid gpu config
-        if not gpu_config:
-            message = "No valid gpu configuration"
-            logger.error(message)
-            raise NotImplementedError(message)
-
         return gpu_config
 
     def _get_windows_computer_service_product_values(self) -> Dict[str, str]:
@@ -417,29 +411,43 @@ class Hardware(BaseAction):
         input = new_state
         variables = {"input": input}
         try:
-            result = self.primitive.session.execute(mutation, variable_values=variables)
-            previous_state = self.previous_state
-            self.previous_state = new_state.copy()
-
-            message = " [*] Checked in successfully: "
-            for key, value in new_state.items():
-                if value != previous_state.get(key, None):
-                    if value is True:
-                        message = (
-                            message
-                            + click.style(f"{key}: ")
-                            + click.style("💤")
-                            + click.style(" ==> ✅ ", fg="green")
-                        )
+            result = self.primitive.session.execute(
+                mutation, variable_values=variables, get_execution_result=True
+            )
+            checkin_success = result.data.get("checkIn").get("lastCheckIn")
+            if messages := result.data.get("checkIn").get("messages"):
+                for message in messages:
+                    logger.enable("primitive")
+                    if message.get("kind") == "ERROR":
+                        logger.error(message.get("message"))
                     else:
-                        message = (
-                            message
-                            + click.style(f"{key}: ")
-                            + click.style("✅")
-                            + click.style(" ==> 💤 ", fg="yellow")
-                        )
-            logger.info(message)
-
+                        logger.debug(message.get("message"))
+
+            if checkin_success:
+                previous_state = self.previous_state
+                self.previous_state = new_state.copy()
+
+                message = " [*] Checked in successfully: "
+                for key, value in new_state.items():
+                    if value != previous_state.get(key, None):
+                        if value is True:
+                            message = (
+                                message
+                                + click.style(f"{key}: ")
+                                + click.style("💤")
+                                + click.style(" ==> ✅ ", fg="green")
+                            )
+                        else:
+                            message = (
+                                message
+                                + click.style(f"{key}: ")
+                                + click.style("✅")
+                                + click.style(" ==> 💤 ", fg="yellow")
+                            )
+                logger.info(message)
+            else:
+                message = "Failed to check in!"
+                raise Exception(message)
             return result
         except client_exceptions.ClientConnectorError as exception:
             message = " [*] Failed to check in! "

{primitive-0.1.9.dist-info → primitive-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: primitive
-Version: 0.1.9
+Version: 0.1.11
 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.9.dist-info → primitive-0.1.11.dist-info}/RECORD

@@ -1,8 +1,8 @@
-primitive/__about__.py,sha256=ZISDHEwYuDZ5fKFtcsW-ewhfzmkOO8yp5qFnYsNgT08,128
+primitive/__about__.py,sha256=cEU7i5k8S0WhpV8qykCw-GDDdmS5X4Rl6vZ5cCkC7EE,129
 primitive/__init__.py,sha256=bwKdgggKNVssJFVPfKSxqFMz4IxSr54WWbmiZqTMPNI,106
 primitive/cli.py,sha256=LXSpPtqQLxkOXsnsbs4UN3n7G_pHCz8AkBFCrk7Is4g,1796
 primitive/client.py,sha256=3RMRNY4ZiGT2OKnZtquWyrXjRSgDe5HlZvtFNKZErLQ,2083
-primitive/agent/actions.py,sha256=9y8_zKg9fUV4VT_rx9geSo7sS0oBR5jWWMdXMJwMmFk,2758
+primitive/agent/actions.py,sha256=f6bv2va1YrDfrzOVK2kTnelRQ2oEvmrLAmsPrxMCh7E,2893
 primitive/agent/commands.py,sha256=-dVDilELfkGfbZB7qfEPs77Dm1oT62qJj4tsIk4KoxI,254
 primitive/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/auth/actions.py,sha256=N2bGcwXNsB89pzs66gF9A5_WzUScY5fhfOyWixqo2y8,1054
@@ -11,7 +11,7 @@ primitive/files/actions.py,sha256=f4JN3QFB2WXw-0JpnE-4-movnqtvXIpCrGd_9pdkeW4,26
 primitive/files/commands.py,sha256=DDizo3xJnU3KLUBTMeeM72viVpnJinLwxs84tmqKhqo,810
 primitive/graphql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/graphql/sdk.py,sha256=BhCGmDtc4sNnH8CxbQSJyFwOZ-ZSqMtjsxMB3JRBhPw,1456
-primitive/hardware/actions.py,sha256=UL2Wyh_XFXKVNEvlVCe1ltCZHpNaRSXPNwSR8hGJyR0,18343
+primitive/hardware/actions.py,sha256=Ea3_2E3F_3WapV60g_mOIcpXhadoknwihR7slXyUWtk,18840
 primitive/hardware/commands.py,sha256=QE7LLeFdfOqlvz3JwdwJJRZAY3fHI1zB9kYmmDajpq0,1477
 primitive/lint/actions.py,sha256=FVigksFrADPDD3zHS0g0riyVrge7eEtwWk5PG2aZv9A,2352
 primitive/lint/commands.py,sha256=3CZvkOEMpJspJWmaQzA5bpPKx0_VCijQIXA9l-eTnZE,487
@@ -27,8 +27,8 @@ primitive/utils/memory_size.py,sha256=4xfha21kW82nFvOTtDFx9Jk2ZQoEhkfXii-PGNTpIU
 primitive/utils/printer.py,sha256=f1XUpqi5dkTL3GWvYRUGlSwtj2IxU1q745T4Fxo7Tn4,370
 primitive/utils/shell.py,sha256=-7UjQaBqSGHzEEyX8pNjeYFFP0P3lVnDV0OkgPz1qHU,1050
 primitive/utils/verible.py,sha256=QYczN1IvxODfj4jeq0nqjFuF0Oi0Zdx-Q32ySOJgcw8,2205
-primitive-0.1.9.dist-info/METADATA,sha256=MZhFwacKp68g2ffhknZfVy84VRSTgCsJu7zMttrE1V8,1817
-primitive-0.1.9.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-primitive-0.1.9.dist-info/entry_points.txt,sha256=p1K8DMCWka5FqLlqP1sPek5Uovy9jq8u51gUsP-z334,48
-primitive-0.1.9.dist-info/licenses/LICENSE.txt,sha256=B8kmQMJ2sxYygjCLBk770uacaMci4mPSoJJ8WoDBY_c,1098
-primitive-0.1.9.dist-info/RECORD,,
+primitive-0.1.11.dist-info/METADATA,sha256=-KZVIgcT4_othtd9SQqixiI7PhsJNcgUzeO7wQxRw7s,1818
+primitive-0.1.11.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+primitive-0.1.11.dist-info/entry_points.txt,sha256=p1K8DMCWka5FqLlqP1sPek5Uovy9jq8u51gUsP-z334,48
+primitive-0.1.11.dist-info/licenses/LICENSE.txt,sha256=B8kmQMJ2sxYygjCLBk770uacaMci4mPSoJJ8WoDBY_c,1098
+primitive-0.1.11.dist-info/RECORD,,