Little improvements

rnanosaur · Jan 27, 2025 · 60574c1 · 60574c1
1 parent 6b993c4
commit 60574c1
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 10 deletions.
diff --git a/src/nanosaur/docker.py b/src/nanosaur/docker.py
@@ -52,9 +52,9 @@ def docker_info(platform):
         print(f"{TerminalFormatter.color_text('   Docker compose:', bold=True)} {TerminalFormatter.color_text('not installed', color='red', bold=True)}")
 
     if version := check_nvidia_container_cli():
-        print(f"{TerminalFormatter.color_text('   Nvidia container:', bold=True)} {version}")
+        print(f"{TerminalFormatter.color_text('   NVIDIA container:', bold=True)} {version}")
     else:
-        print(f"{TerminalFormatter.color_text('   Nvidia container:', bold=True)} {TerminalFormatter.color_text('not installed', color='red', bold=True)}")
+        print(f"{TerminalFormatter.color_text('   NVIDIA container:', bold=True)} {TerminalFormatter.color_text('not installed', color='red', bold=True)}")
 
 
 def is_docker_installed():

diff --git a/src/nanosaur/main.py b/src/nanosaur/main.py
@@ -269,19 +269,19 @@ def main():
         parser_workspace = parser_workspace_menu(subparsers)
 
     # Subcommand: simulation (with a sub-menu for simulation types)
-    if device_type == 'desktop' and 'mode' in params:
+    if device_type == 'desktop':
         # Add simulation subcommand
         parser_simulation = parser_simulation_menu(subparsers, params)
 
     # Add robot subcommand
-    parser_robot, parser_config = parser_robot_menu(subparsers, params)
+    parser_robot, parser_config = parser_robot_menu(platform, subparsers, params)
 
     if device_type == 'desktop':
         # Subcommand: swarm (with a sub-menu for swarm operations)
         parser_swarm = parser_swarm_menu(subparsers, params)
 
     # Subcommand: wakeup (with a sub-menu for wakeup operations)
-    if 'mode' in params and 'robots' in params:
+    if 'robots' in params:
         robot_control(params, subparsers)
 
     # Enable tab completion

diff --git a/src/nanosaur/robot.py b/src/nanosaur/robot.py
@@ -39,7 +39,7 @@
 logger = logging.getLogger(__name__)
 
 
-def add_robot_config_subcommands(subparsers: argparse._SubParsersAction, params: Params) -> argparse.ArgumentParser:
+def add_robot_config_subcommands(platform, subparsers: argparse._SubParsersAction, params: Params) -> argparse.ArgumentParser:
     # Get the robot data
     robot_data = RobotList.current_robot(params)
     parser_robot_config = subparsers.add_parser('config', help=f"Configure the robot settings [{robot_data.name}]")
@@ -51,8 +51,10 @@ def add_robot_config_subcommands(subparsers: argparse._SubParsersAction, params:
     parser_robot_domain_id = config_subparsers.add_parser('domain_id', help=f"Set robot domain ID [{robot_data.domain_id}]")
     parser_robot_domain_id.set_defaults(func=robot_set_domain_id)
     # Add robot simulation subcommand
-    parser_robot_simulation = config_subparsers.add_parser('simulation', help=f"Set robot real or simulation [{'simulation' if robot_data.simulation else 'real'}]")
-    parser_robot_simulation.set_defaults(func=robot_set_simulation)
+    device_type = "robot" if platform['Machine'] == 'aarch64' else "desktop"
+    if device_type == 'desktop':
+        parser_robot_simulation = config_subparsers.add_parser('simulation', help=f"Set robot real or simulation [{'simulation' if robot_data.simulation else 'real'}]")
+        parser_robot_simulation.set_defaults(func=robot_set_simulation)
     # Add robot camera subcommand
     parser_robot_camera = config_subparsers.add_parser('camera', help=f"Set robot camera type [{robot_data.camera_type or 'NOT SELECTED'}]")
     parser_robot_camera.add_argument('--new', type=str, help=f"Specify the new camera type (options: {', '.join(CAMERA_CHOICES)})")
@@ -73,7 +75,7 @@ def add_robot_config_subcommands(subparsers: argparse._SubParsersAction, params:
     return parser_robot_config
 
 
-def parser_robot_menu(subparsers: argparse._SubParsersAction, params: Params) -> argparse.ArgumentParser:
+def parser_robot_menu(platform, subparsers: argparse._SubParsersAction, params: Params) -> argparse.ArgumentParser:
     try:
         robot_data = RobotList.current_robot(params)
         parser_robot = subparsers.add_parser('robot', help=f"Manage the Nanosaur robot [{robot_data.name}]")
@@ -96,7 +98,7 @@ def parser_robot_menu(subparsers: argparse._SubParsersAction, params: Params) ->
         parser_robot_stop.set_defaults(func=docker.docker_robot_stop)
 
         # Add robot config subcommand
-        parser_config = add_robot_config_subcommands(robot_subparsers, params)
+        parser_config = add_robot_config_subcommands(platform, robot_subparsers, params)
 
         return parser_robot, parser_config
     except IndexError: