standardbots 2.0.0.dev1772493130__tar.gz → 2.0.0.dev1772570485__tar.gz

{standardbots-2.0.0.dev1772493130 → standardbots-2.0.0.dev1772570485}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: standardbots
-Version: 2.0.0.dev1772493130
+Version: 2.0.0.dev1772570485
 Summary: Standard Bots RO1 Robotics API
 Home-page: 
 Author: Standard Bots Support

{standardbots-2.0.0.dev1772493130 → standardbots-2.0.0.dev1772570485}/setup.py

@@ -13,7 +13,7 @@
 from setuptools import setup, find_packages  # noqa: H301
 
 NAME = "standardbots"
-VERSION = "2.0.0-dev1772493130"
+VERSION = "2.0.0-dev1772570485"
 # To install the library, run the following
 #
 # python setup.py install

{standardbots-2.0.0.dev1772493130 → standardbots-2.0.0.dev1772570485}/standardbots/auto_generated/apis.py

@@ -1270,6 +1270,51 @@ class Default:
                 if parsed is None and (is_user_error or is_unavailable):
                     parsed = models.parse_error_response(json.loads(response.data))
 
+                return Response(
+                    parsed,
+                    response.status,
+                    response
+                )
+            except urllib3.exceptions.MaxRetryError:
+                return Response(
+                    models.ErrorResponse(
+                        error=models.ErrorEnum.InternalServerError,
+                        message="Connection Refused"
+                    ),
+                    503,
+                    None
+                )
+        def set_bolt_head_position(
+            self,
+            body: models.SetBoltHeadPositionRequest,
+        ) -> Response[
+            Union[
+                models.TeleopState,
+                models.ErrorResponse,
+                None
+            ],
+            models.TeleopState
+        ]:
+            """
+            Set static bolt head position
+            """
+            path = "/api/v1/teleop/set-bolt-head-position"
+            try:
+                response = self._request_manager.request(
+                    "POST",
+                    path,
+                    headers=self._request_manager.json_headers(),
+                    body=json.dumps(models.serialize_set_bolt_head_position_request(body)),
+                )
+                parsed = None
+                if response.status == 200:
+                    parsed = models.parse_teleop_state(json.loads(response.data))
+
+                is_user_error = response.status >= 400 and response.status <= 500
+                is_unavailable = response.status == 503
+                if parsed is None and (is_user_error or is_unavailable):
+                    parsed = models.parse_error_response(json.loads(response.data))
+
                 return Response(
                     parsed,
                     response.status,

{standardbots-2.0.0.dev1772493130 → standardbots-2.0.0.dev1772570485}/standardbots/auto_generated/models.py

@@ -240,6 +240,66 @@ def parse_arm_position_update_request_response_kind_enum(data: object) -> ArmPos
 def serialize_arm_position_update_request_response_kind_enum(data: Union[ArmPositionUpdateRequestResponseKindEnum, str]) -> object:
   return ArmPositionUpdateRequestResponseKindEnum(data).value
 
+@dataclass
+class BoltHeadPosition:
+  """Bolt head position in pitch, yaw, roll"""
+  pitch: Union[float, None] = None
+  yaw: Union[float, None] = None
+  roll: Union[float, None] = None
+
+  def validate_pitch(self, value: float) -> Tuple[bool, str]:
+    if value is None:
+      return [False, "pitch is required for BoltHeadPosition"]
+
+    if not isinstance(value, float):
+      return [False, "pitch must be of type float for BoltHeadPosition, got " + type(value).__name__]
+
+    return [True, ""]
+
+  def validate_yaw(self, value: float) -> Tuple[bool, str]:
+    if value is None:
+      return [False, "yaw is required for BoltHeadPosition"]
+
+    if not isinstance(value, float):
+      return [False, "yaw must be of type float for BoltHeadPosition, got " + type(value).__name__]
+
+    return [True, ""]
+
+  def validate_roll(self, value: float) -> Tuple[bool, str]:
+    if value is None:
+      return [False, "roll is required for BoltHeadPosition"]
+
+    if not isinstance(value, float):
+      return [False, "roll must be of type float for BoltHeadPosition, got " + type(value).__name__]
+
+    return [True, ""]
+
+  def __post_init__(self):
+    # Type check incoming model - raise error if invalid (required or wrong type)
+    is_valid, error_str = self.validate_pitch(self.pitch)
+    if not is_valid:
+      raise TypeError(error_str)
+    is_valid, error_str = self.validate_yaw(self.yaw)
+    if not is_valid:
+      raise TypeError(error_str)
+    is_valid, error_str = self.validate_roll(self.roll)
+    if not is_valid:
+      raise TypeError(error_str)
+
+def parse_bolt_head_position(data: object):
+  return BoltHeadPosition(
+    pitch=parse_f_64(data["pitch"]) if "pitch" in data and data.get("pitch") is not None else None,
+    yaw=parse_f_64(data["yaw"]) if "yaw" in data and data.get("yaw") is not None else None,
+    roll=parse_f_64(data["roll"]) if "roll" in data and data.get("roll") is not None else None,
+  )
+
+def serialize_bolt_head_position(data: BoltHeadPosition) -> object:
+  return {
+    "pitch": serialize_f_64(data.pitch),
+    "yaw": serialize_f_64(data.yaw),
+    "roll": serialize_f_64(data.roll),
+  }
+
 @dataclass
 class BotIdentityData:
   """Data about the robot's identity"""
@@ -3859,6 +3919,36 @@ def parse_arm_position_update_request_response_event_stream_subscriptions_list(d
 def serialize_arm_position_update_request_response_event_stream_subscriptions_list(data: ArmPositionUpdateRequestResponseEventStreamSubscriptionsList) -> List[object]:
   return [serialize_arm_position_update_request_response_event_stream_subscription_kind_enum(item) for item in data]
 
+@dataclass
+class SetBoltHeadPositionRequest:
+  """Request to set static bolt head position"""
+  position: Union[BoltHeadPosition, None] = None
+
+  def validate_position(self, value: BoltHeadPosition) -> Tuple[bool, str]:
+    if value is None:
+      return [False, "position is required for SetBoltHeadPositionRequest"]
+
+    if not isinstance(value, BoltHeadPosition):
+      return [False, "position must be of type BoltHeadPosition for SetBoltHeadPositionRequest, got " + type(value).__name__]
+
+    return [True, ""]
+
+  def __post_init__(self):
+    # Type check incoming model - raise error if invalid (required or wrong type)
+    is_valid, error_str = self.validate_position(self.position)
+    if not is_valid:
+      raise TypeError(error_str)
+
+def parse_set_bolt_head_position_request(data: object):
+  return SetBoltHeadPositionRequest(
+    position=parse_bolt_head_position(data["position"]) if "position" in data and data.get("position") is not None else None,
+  )
+
+def serialize_set_bolt_head_position_request(data: SetBoltHeadPositionRequest) -> object:
+  return {
+    "position": serialize_bolt_head_position(data.position),
+  }
+
 @dataclass
 class BrakesState:
   """State of the robot joint brakes. Each joint contains a brake which can be engaged when the robot is at a standstill to prevent the robot from moving.
@@ -9031,6 +9121,7 @@ class TeleopConfig:
   """Config of the teleop"""
   isSecondary: Union[bool, None] = None
   bots: Union[BotTeleopDetailsList, None] = None
+  enableBoltHeadTracking: Union[bool, None] = None
 
   def validate_isSecondary(self, value: bool) -> Tuple[bool, str]:
     if value is None:
@@ -9050,6 +9141,15 @@ class TeleopConfig:
 
     return [True, ""]
 
+  def validate_enableBoltHeadTracking(self, value: bool) -> Tuple[bool, str]:
+    if value is None:
+      return [True, ""]
+
+    if not isinstance(value, bool):
+      return [False, "enableBoltHeadTracking must be of type bool for TeleopConfig, got " + type(value).__name__]
+
+    return [True, ""]
+
   def __post_init__(self):
     # Type check incoming model - raise error if invalid (required or wrong type)
     is_valid, error_str = self.validate_isSecondary(self.isSecondary)
@@ -9058,17 +9158,22 @@ class TeleopConfig:
     is_valid, error_str = self.validate_bots(self.bots)
     if not is_valid:
       raise TypeError(error_str)
+    is_valid, error_str = self.validate_enableBoltHeadTracking(self.enableBoltHeadTracking)
+    if not is_valid:
+      raise TypeError(error_str)
 
 def parse_teleop_config(data: object):
   return TeleopConfig(
     isSecondary=parse_bool(data["isSecondary"]) if "isSecondary" in data and data.get("isSecondary") is not None else None,
     bots=parse_bot_teleop_details_list(data["bots"]) if "bots" in data and data.get("bots") is not None else None,
+    enableBoltHeadTracking=parse_bool(data["enableBoltHeadTracking"]) if "enableBoltHeadTracking" in data and data.get("enableBoltHeadTracking") is not None else None,
   )
 
 def serialize_teleop_config(data: TeleopConfig) -> object:
   return {
     "isSecondary": None if data.isSecondary is None else serialize_bool(data.isSecondary),
     "bots": None if data.bots is None else serialize_bot_teleop_details_list(data.bots),
+    "enableBoltHeadTracking": None if data.enableBoltHeadTracking is None else serialize_bool(data.enableBoltHeadTracking),
   }
 
 @dataclass

{standardbots-2.0.0.dev1772493130 → standardbots-2.0.0.dev1772570485}/standardbots.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: standardbots
-Version: 2.0.0.dev1772493130
+Version: 2.0.0.dev1772570485
 Summary: Standard Bots RO1 Robotics API
 Home-page: 
 Author: Standard Bots Support