primitive 0.2.61__py3-none-any.whl → 0.2.64__py3-none-any.whl

primitive/network/commands.py

@@ -0,0 +1,40 @@
+from typing import TYPE_CHECKING
+
+import click
+from primitive.utils.printer import print_result
+from primitive.network.ui import render_ports_table
+
+if TYPE_CHECKING:
+    from ..client import Primitive
+
+
+@click.group()
+@click.pass_context
+def cli(context):
+    """Network"""
+    pass
+
+
+@cli.command("switch")
+@click.pass_context
+def switch(context):
+    """Switch"""
+    primitive: Primitive = context.obj.get("PRIMITIVE")
+    switch_info = primitive.network.get_switch_info()
+    if context.obj["JSON"]:
+        message = switch_info
+    else:
+        message = f"Vendor: {switch_info.get('vendor')}. Model: {switch_info.get('model')}. IP: {switch_info.get('ip_address')}"
+    print_result(message=message, context=context)
+
+
+@cli.command("ports")
+@click.pass_context
+def ports(context):
+    """Ports"""
+    primitive: Primitive = context.obj.get("PRIMITIVE")
+    ports_info = primitive.network.get_interfaces_info()
+    if context.obj["JSON"]:
+        print_result(message=ports_info, context=context)
+    else:
+        render_ports_table(ports_info.get("interfaces"))

primitive/network/redfish.py

@@ -0,0 +1,121 @@
+import requests
+from typing import Literal
+
+
+class RedfishClient:
+    def __init__(self, host: str, username: str, password: str):
+        self.host = host
+        self.username = username
+        self.password = password
+
+    def create_session_token(self):
+        response = requests.post(
+            f"https://{self.host}/redfish/v1/SessionService/Sessions",
+            json={"UserName": self.username, "Password": self.password},
+            verify=False,
+        )
+        response.raise_for_status()
+        return response.headers["X-Auth-Token"]
+
+    def get_systems(self):
+        token = self.create_session_token()
+        headers = {"X-Auth-Token": token}
+        response = requests.get(
+            f"https://{self.host}/redfish/v1/Systems",
+            headers=headers,
+            verify=False,
+        )
+        response.raise_for_status()
+        return response.json()
+
+    def get_system(self, system_id: str):
+        token = self.create_session_token()
+        headers = {"X-Auth-Token": token}
+        response = requests.get(
+            f"https://{self.host}/redfish/v1/Systems/{system_id}",
+            headers=headers,
+            verify=False,
+        )
+        response.raise_for_status()
+        return response.json()
+
+    def update_boot_options(
+        self,
+        system_id: str,
+        boot_source_override_enabled: Literal["Once", "Continuous", "Disabled"],
+        boot_source_override_target: Literal[
+            "None", "Pxe", "Cd", "Usb", "Hdd", "BiosSetup", "Diags", "UefiTarget"
+        ],
+        boot_source_override_mode: Literal["UEFI", "Legacy"] = "UEFI",
+    ):
+        token = self.create_session_token()
+        headers = {"X-Auth-Token": token}
+        body = {
+            "Boot": {
+                "BootSourceOverrideTarget": boot_source_override_target,
+                "BootSourceOverrideEnabled": boot_source_override_enabled,
+                "BootSourceOverrideMode": boot_source_override_mode,
+            }
+        }
+        response = requests.patch(
+            f"https://{self.host}/redfish/v1/Systems/{system_id}",
+            headers=headers,
+            verify=False,
+            json=body,
+        )
+        response.raise_for_status()
+        return response.json()
+
+    def get_boot_options(self, system_id: str):
+        token = self.create_session_token()
+        headers = {"X-Auth-Token": token}
+        response = requests.get(
+            f"https://{self.host}/redfish/v1/Systems/{system_id}/BootOptions",
+            headers=headers,
+            verify=False,
+        )
+        response.raise_for_status()
+        return response.json()
+
+    def get_boot_option(self, system_id: str, option_id: str):
+        # option_id typically looks like Boot000X
+        token = self.create_session_token()
+        headers = {"X-Auth-Token": token}
+        response = requests.get(
+            f"https://{self.host}/redfish/v1/Systems/{system_id}/BootOptions/{option_id}",
+            headers=headers,
+            verify=False,
+        )
+        response.raise_for_status()
+        return response.json()
+
+    def get_pxe_boot_option(self, system_id: str):
+        boot_options = self.get_boot_options(system_id)
+        for option in boot_options.get("Members", []):
+            option_data = self.get_boot_option(
+                system_id, option["@odata.id"].split("/")[-1]
+            )
+            boot_option_enabled = option_data.get("BootOptionEnabled", False)
+            display_name = option_data.get("DisplayName", "")
+            if "PXE" in display_name and "IPv4" in display_name and boot_option_enabled:
+                return option_data
+
+        return None
+
+    def compute_system_reset(
+        self,
+        system_id: str,
+        reset_type: Literal[
+            "ForceRestart", "GracefulRestart", "ForceOff", "PowerCycle"
+        ],
+    ) -> bool:
+        token = self.create_session_token()
+        headers = {"X-Auth-Token": token}
+        response = requests.post(
+            f"https://{self.host}/redfish/v1/Systems/{system_id}/Actions/ComputerSystem.Reset",
+            json={"ResetType": reset_type},
+            headers=headers,
+            verify=False,
+        )
+        response.raise_for_status()
+        return response.status_code == 200

primitive/network/ui.py

@@ -0,0 +1,19 @@
+from rich.console import Console
+from rich.table import Table
+
+
+def render_ports_table(ports_dict) -> None:
+    console = Console()
+
+    table = Table(show_header=True, header_style="bold #FFA800")
+    table.add_column("Port")
+    table.add_column("Status")
+    table.add_column("MAC Address")
+    table.add_column("IP Address")
+
+    for k, v in ports_dict.items():
+        table.add_row(
+            k, v.get("link_status"), v.get("mac_address"), v.get("ip_address")
+        )
+
+    console.print(table)

primitive/provisioning/actions.py

@@ -25,7 +25,14 @@ class Provisioning(BaseAction):
         return result.data["authorizedKeys"]
 
     def add_reservation_authorized_keys(self, reservation_id: str) -> None:
-        SSH_DIR.mkdir(parents=True, exist_ok=True)
+        if SSH_DIR.exists() is False:
+            SSH_DIR.mkdir(parents=True, exist_ok=True)
+            SSH_DIR.chmod(0o700)
+
+        if AUTHORIZED_KEYS_FILEPATH.exists() is False:
+            AUTHORIZED_KEYS_FILEPATH.touch()
+            AUTHORIZED_KEYS_FILEPATH.chmod(0o600)
+
         AUTHORIZED_KEYS_BACKUP_FILEPATH = SSH_DIR.joinpath(
             f"authorized_keys.bak-{reservation_id}"
         )

primitive/provisioning/graphql/queries.py

@@ -1,5 +1,5 @@
 authorized_keys_query = """
-query authorizedKeys($reservationId: GlobalID!) {
+query authorizedKeys($reservationId: ID!) {
     authorizedKeys(reservationId: $reservationId)
 }
 """

primitive/reservations/graphql/queries.py

@@ -38,7 +38,7 @@ query reservations(
 reservation_query = (
     reservation_fragment
     + """
-query reservation($id: GlobalID!) {
+query reservation($id: ID!) {
   reservation(id: $id) {
     ...ReservationFragment
   }

{primitive-0.2.61.dist-info → primitive-0.2.64.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: primitive
-Version: 0.2.61
+Version: 0.2.64
 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
@@ -24,6 +24,7 @@ Requires-Dist: loguru
 Requires-Dist: paramiko[invoke]
 Requires-Dist: pika>=1.3.2
 Requires-Dist: psutil>=7.0.0
+Requires-Dist: pyserial>=3.5
 Requires-Dist: rich>=13.9.4
 Requires-Dist: speedtest-cli
 Description-Content-Type: text/markdown

{primitive-0.2.61.dist-info → primitive-0.2.64.dist-info}/RECORD

@@ -1,11 +1,11 @@
-primitive/__about__.py,sha256=44bjP-EhQtVr_s6llvOiTQ56lIx7zqJ9wHAGO3s_eCw,130
+primitive/__about__.py,sha256=hRAAesFWrmf9g-0XnQagAF61QBoVVA8Jtq5ovjMar4U,130
 primitive/__init__.py,sha256=bwKdgggKNVssJFVPfKSxqFMz4IxSr54WWbmiZqTMPNI,106
-primitive/cli.py,sha256=g7EtHI9MATAB0qQu5w-WzbXtxz_8zu8z5E7sETmMkKU,2509
-primitive/client.py,sha256=gyZIj61qMtOi_s8y0WG3gDehGT-Ms3BtbDP2J_2lajU,3753
+primitive/cli.py,sha256=kLjqyNijkbElbH8XhEtR6vjPc5uwOoh9mKEf2RkcfKk,2608
+primitive/client.py,sha256=n0eQAft3lQyTT_tVE1_1BWSsh8G7gJI2kHBuXLvZVEY,3836
 primitive/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/agent/actions.py,sha256=f1sSDdFHOflfGQCwNJ5P51YWUTKs8EK5YXbMAvZ-KMM,8018
+primitive/agent/actions.py,sha256=ohY81bflsYSTsNgo2o2WP_tZr66haX4z319zf5oib-4,9362
 primitive/agent/commands.py,sha256=o847pK7v7EWQGG67tky6a33qtwoutX6LZrP2FIS_NOk,388
-primitive/agent/runner.py,sha256=vymeuIDiah6J9SHjTbsNQGi74PpnJw1MxWPd3TWjc-w,11512
+primitive/agent/runner.py,sha256=oKMCkZpeHGENLXgLIFMgOitrjIpUGYUQmcfTWwSsqSM,15598
 primitive/agent/uploader.py,sha256=DT_Nzt5eOTm_uRcYKm1sjBBaQZzp5iNZ_uN5XktfQ30,3382
 primitive/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/auth/actions.py,sha256=9NIEXJ1BNJutJs6AMMSjMN_ziONUAUhY_xHwojYJCLA,942
@@ -19,7 +19,7 @@ primitive/daemons/launch_agents.py,sha256=cCxsvCBFmGlKOiBINsKi_NxcozLC8yPw3w6pxq
 primitive/daemons/launch_service.py,sha256=iuklHeuEqadlf8U1n9xFg4ZG1EKdK2jyaPI-VTlpJ4I,7907
 primitive/daemons/ui.py,sha256=Af3OJWJ0jdGlb1nfA5yaGYdhBEqqpM8zP2U2vUQdCbw,1236
 primitive/exec/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/exec/actions.py,sha256=4d_TCjNDcVFoZ9Zw7ZuBa6hKMv2Xzm7_UX_8wcX1aSk,4124
+primitive/exec/actions.py,sha256=rcAMP7hxJZhQUwx-orduCUratzMwVRknX14QSyQLdcA,4348
 primitive/exec/commands.py,sha256=66LO2kkJC-ynNZQpUCXv4Ol15QoacdSZAHblePDcmLo,510
 primitive/exec/interactive.py,sha256=TscY6s2ZysijidKPheq6y-fCErUVLS0zcdTW8XyFWGI,2435
 primitive/files/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -40,25 +40,30 @@ primitive/graphql/relay.py,sha256=bmij2AjdpURQ6GGVCxwWhauF-r_SxuAU2oJ4sDbLxpI,72
 primitive/graphql/sdk.py,sha256=dE4TD8KiTKw3Y0uiw5XrIcuZGqexE47eSlPaPD6jDGo,1545
 primitive/graphql/utility_fragments.py,sha256=uIjwILC4QtWNyO5vu77VjQf_p0jvP3A9q_6zRq91zqs,303
 primitive/hardware/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/hardware/actions.py,sha256=4xswcLaHk4IhEwRK5Pw_rpEgn6XciT5K2apnHi8_cI4,29204
+primitive/hardware/actions.py,sha256=kSQ1DufDCwxY5fs3zNMj7evAunEF8vt36a7rgtjI7Uw,37323
 primitive/hardware/android.py,sha256=tu7pBPxWFrIwb_mm5CEdFFf1_veNDOKjOCQg13i_Lh4,2758
-primitive/hardware/commands.py,sha256=lwO-cm3doGtuGvR30fib-nALJyNCScdCTXOVVbYVpkY,4604
+primitive/hardware/commands.py,sha256=XU365RByl75ECAV_GdA4xxLrmZkwo1axlw9ZH0k3yRk,4751
 primitive/hardware/ui.py,sha256=12rucuZ2s-w5R4bKyxON5dEbrdDnVf5sbj3K_nbdo44,2473
 primitive/hardware/graphql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/hardware/graphql/fragments.py,sha256=H315uv-ujOsZEbQR5tmstbRU_9OCa5SErZMwIckSiLc,383
-primitive/hardware/graphql/mutations.py,sha256=wOKNJN-1x_DGxvdHt2Bm3AUqyW8P1lNPkxLIQThbXUg,1874
-primitive/hardware/graphql/queries.py,sha256=I86uLuOSjHSph11Y5MVCYko5Js7hoiEZ-cEoPTc4J-k,1392
+primitive/hardware/graphql/fragments.py,sha256=FRPAKB3KBfVxVm69HeYCxvvrdkW-O8CR46Inbk3wgc8,481
+primitive/hardware/graphql/mutations.py,sha256=MqhjICTouKPgtxXy1CCyLZFa4SZbJQGk9jJMcIJDQW8,2251
+primitive/hardware/graphql/queries.py,sha256=D7C32AKizZ_4KaqFCSffjMpnLBTnCgr5npvyrU_om-o,2099
 primitive/jobs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/jobs/actions.py,sha256=-IRNO7KM63cPAbGRNQycyFsugW9hAyg-TDuK0g5AiSw,6542
 primitive/jobs/commands.py,sha256=MxPCkBEYW_eLNqgCRYeyj7ZcLOFAWfpVZlqDR2Y_S0o,830
 primitive/jobs/graphql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/jobs/graphql/fragments.py,sha256=rWZWxZs9ZWjWal0eiY_XPWUS7i7f3fwSM_w8ZvDs2JQ,614
+primitive/jobs/graphql/fragments.py,sha256=I-Ly0TGARt_TdenLQ7877pGLWBzono_IMj7NCzsRtYA,654
 primitive/jobs/graphql/mutations.py,sha256=8ASvCmwQh7cMeeiykOdYaYVryG8FRIuVF6v_J8JJZuw,219
-primitive/jobs/graphql/queries.py,sha256=ZxNmm-WovytbggNuKRnwa0kc26T34_0yhqkoqx-2uj0,1736
+primitive/jobs/graphql/queries.py,sha256=57B5mSnAYjNEFFk1P69odON0fqkf7FyhsA316pcxb6g,1712
 primitive/messaging/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/messaging/provider.py,sha256=NCgl1KiV4Ow49Laz2grNGsP7Ij4vUUeorUjkhAKF1H4,4182
-primitive/monitor/actions.py,sha256=AZOoNzASorvye5309h6BJbsup_jR8YunQ19EiG5SAQg,9643
+primitive/messaging/provider.py,sha256=NkR4hnQzEZ0SItno1TgYNvzTwgKZq03iNrnIpLK7MTg,4400
+primitive/monitor/actions.py,sha256=GqwnTibVwAtiSU40TBexBruZI_V-7ZS8MoYZpFxq3mI,10833
 primitive/monitor/commands.py,sha256=VDlEL_Qpm_ysHxug7VpI0cVAZ0ny6AS91Y58D7F1zkU,409
+primitive/network/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+primitive/network/actions.py,sha256=qpkn9U7LTDzdoOntiFP2sBPCQYEraRjAYrkcPzhmw78,17021
+primitive/network/commands.py,sha256=F84KmwkG0DHNLIa7Em8LMm2XDOaKMCWRpcFej8olxM0,1071
+primitive/network/redfish.py,sha256=uOtAS_Dwc4I4bnWKNSTCQ_xsj5LTtRzW5v2P_fWaSJM,4248
+primitive/network/ui.py,sha256=_i4lJ3hhTrPc_KS5EjXPkqqKkzdLCOdxxtk5MiDWsfU,504
 primitive/organizations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/organizations/actions.py,sha256=kVHOhG1oS2sI5p8uldSo5L-RUZsnG36eaulVuKLyZ-M,1863
 primitive/organizations/commands.py,sha256=_dwgVEJCqMa5VgB_7P1wLPFc0AuT1p9dtyR9JRr4kpw,487
@@ -74,16 +79,16 @@ primitive/projects/graphql/fragments.py,sha256=02F5nyI8i-ML_bV5FFHUgFWM5bBBfjmz_
 primitive/projects/graphql/mutations.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/projects/graphql/queries.py,sha256=nFaVf6YOHA2L_FTgIUdRK-80hYTmv1a1X5ac7QPMp1k,646
 primitive/provisioning/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/provisioning/actions.py,sha256=oM19p1JyqwPhfJtyy23RqhuIKinorTo78yOYWVLC98c,2165
+primitive/provisioning/actions.py,sha256=o4JA-CNJPfMXwhqSLAQ5SI9o1TLNPKbn5GpJn4ML_ds,2392
 primitive/provisioning/graphql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-primitive/provisioning/graphql/queries.py,sha256=cBtuKa6shoatYZfKSnQoPJP6B8g8y3QhFqJ_pkvMcG0,134
+primitive/provisioning/graphql/queries.py,sha256=QxXkn9jEzfpevXu9iYCPFEJLs8SXgS0mABwJM98dEGg,128
 primitive/reservations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/reservations/actions.py,sha256=vxJ62oriy3Km1d_tiFGMbSclkMJf66eaUgVuhjGmuyA,7021
 primitive/reservations/commands.py,sha256=LFRoV59QGgWIjBdrGjJdffHugg8TLe0Fwlcyu_JaTkk,2369
 primitive/reservations/graphql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/reservations/graphql/fragments.py,sha256=h3edd0Es38SNldhHiRU3hAJNYwrUIJMWxiP8av3BfrI,431
 primitive/reservations/graphql/mutations.py,sha256=IqzwQL7OclN7RpIcidrTQo9cGYofY7wqoBOdnY0pwN8,651
-primitive/reservations/graphql/queries.py,sha256=x31wTRelskX2fc0fx2qrY7XT1q74nvzLv_Xef3o9weg,746
+primitive/reservations/graphql/queries.py,sha256=BP7ON6QfChZRlY_YTyUbxM_tEPSfJqc8DknfpTvQ7hk,740
 primitive/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 primitive/utils/actions.py,sha256=Gm84YgXFSpilpYqjG5nxcQTdRSp3waN7ZorzIzMBLtc,208
 primitive/utils/auth.py,sha256=uBIZNPF2CpbaPV2UMi6eWVUKghV6WIm-pG3-UM29bNs,1465
@@ -99,8 +104,8 @@ primitive/utils/psutil.py,sha256=xa7ef435UL37jyjmUPbEqCO2ayQMpCs0HCrxVEvLcuM,763
 primitive/utils/shell.py,sha256=Z4zxmOaSyGCrS0D6I436iQci-ewHLt4UxVg1CD9Serc,2171
 primitive/utils/text.py,sha256=XiESMnlhjQ534xE2hMNf08WehE1SKaYFRNih0MmnK0k,829
 primitive/utils/x509.py,sha256=HwHRPqakTHWd40ny-9O_yNknSL1Cxo50O0UCjXHFq04,3796
-primitive-0.2.61.dist-info/METADATA,sha256=VMscEolFKYI028E5OFCQY8wouk_Zw0eBzU1vXSCjwBA,3540
-primitive-0.2.61.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-primitive-0.2.61.dist-info/entry_points.txt,sha256=p1K8DMCWka5FqLlqP1sPek5Uovy9jq8u51gUsP-z334,48
-primitive-0.2.61.dist-info/licenses/LICENSE.txt,sha256=B8kmQMJ2sxYygjCLBk770uacaMci4mPSoJJ8WoDBY_c,1098
-primitive-0.2.61.dist-info/RECORD,,
+primitive-0.2.64.dist-info/METADATA,sha256=BJiBuQqlDVUUeO7_00Gq0IvKM1983vxjyp2w7x_dP7k,3569
+primitive-0.2.64.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+primitive-0.2.64.dist-info/entry_points.txt,sha256=p1K8DMCWka5FqLlqP1sPek5Uovy9jq8u51gUsP-z334,48
+primitive-0.2.64.dist-info/licenses/LICENSE.txt,sha256=B8kmQMJ2sxYygjCLBk770uacaMci4mPSoJJ8WoDBY_c,1098
+primitive-0.2.64.dist-info/RECORD,,