franky-control 1.0.2__cp313-cp313-manylinux_2_28_x86_64.whl

franky/motion.py

@@ -0,0 +1,12 @@
+from typing import Union
+
+from ._franky import BaseCartesianPoseMotion, BaseCartesianVelocityMotion, BaseJointPositionMotion, \
+    BaseJointVelocityMotion, BaseTorqueMotion
+
+Motion = Union[
+    BaseCartesianPoseMotion,
+    BaseCartesianVelocityMotion,
+    BaseJointPositionMotion,
+    BaseJointVelocityMotion,
+    BaseTorqueMotion
+]

franky/reaction.py

@@ -0,0 +1,43 @@
+from ._franky import Condition, BaseCartesianPoseMotion, BaseCartesianVelocityMotion, BaseJointPositionMotion, \
+    BaseJointVelocityMotion, BaseTorqueMotion, \
+    CartesianPoseReaction as _CartesianPoseReaction, \
+    CartesianVelocityReaction as _CartesianVelocityReaction, \
+    JointPositionReaction as _JointPositionReaction, \
+    JointVelocityReaction as _JointVelocityReaction, \
+    TorqueReaction as _TorqueReaction
+
+from .motion import Motion
+
+
+class Reaction:
+    _control_signal_type = None
+
+    def __new__(cls, condition: Condition, motion: Motion):
+        for reaction_type in _REACTION_TYPES:
+            if isinstance(motion, reaction_type._motion_type):
+                return reaction_type.__new__(reaction_type, condition, motion)
+        raise TypeError(f"Unknown motion type {type(motion)}.")
+
+
+class CartesianPoseReaction(_CartesianPoseReaction, Reaction):
+    _motion_type = BaseCartesianPoseMotion
+
+
+class CartesianVelocityReaction(_CartesianVelocityReaction, Reaction):
+    _motion_type = BaseCartesianVelocityMotion
+
+
+class JointPositionReaction(_JointPositionReaction, Reaction):
+    _motion_type = BaseJointPositionMotion
+
+
+class JointVelocityReaction(_JointVelocityReaction, Reaction):
+    _motion_type = BaseJointVelocityMotion
+
+
+class TorqueReaction(_TorqueReaction, Reaction):
+    _motion_type = BaseTorqueMotion
+
+
+_REACTION_TYPES = [
+    CartesianPoseReaction, CartesianVelocityReaction, JointPositionReaction, JointVelocityReaction, TorqueReaction]

franky/robot.py

@@ -0,0 +1,8 @@
+from ._franky import _RobotInternal
+
+from .robot_web_session import RobotWebSession
+
+
+class Robot(_RobotInternal):
+    def create_web_session(self, username: str, password: str):
+        return RobotWebSession(self, username, password)

franky/robot_web_session.py

@@ -0,0 +1,183 @@
+import base64
+import hashlib
+import http.client
+import json
+import ssl
+import time
+import urllib.parse
+from http.client import HTTPSConnection, HTTPResponse
+from typing import Dict, Optional, Any, Literal
+from urllib.error import HTTPError
+
+
+class FrankaAPIError(Exception):
+    def __init__(self, target: str, http_code: int, http_reason: str, headers: Dict[str, str], message: str):
+        super().__init__(
+            f"Franka API returned error {http_code} ({http_reason}) when accessing end-point {target}: {message}")
+        self.target = target
+        self.http_code = http_code
+        self.headers = headers
+        self.message = message
+
+
+class RobotWebSession:
+    def __init__(self, fci_hostname: str, username: str, password: str):
+        self.__fci_hostname = fci_hostname
+        self.__username = username
+        self.__password = password
+
+        self.__client = None
+        self.__token = None
+        self.__control_token = None
+        self.__control_token_id = None
+
+    @staticmethod
+    def __encode_password(user: str, password: str) -> str:
+        bs = ",".join([str(b) for b in hashlib.sha256((password + "#" + user + "@franka").encode("utf-8")).digest()])
+        return base64.encodebytes(bs.encode("utf-8")).decode("utf-8")
+
+    def _send_api_request(self, target: str, headers: Optional[Dict[str, str]] = None, body: Optional[Any] = None,
+                          method: Literal["GET", "POST", "DELETE"] = "POST"):
+        _headers = {
+            "Cookie": f"authorization={self.__token}"
+        }
+        if headers is not None:
+            _headers.update(headers)
+        self.__client.request(method, target, headers=_headers, body=body)
+        res: HTTPResponse = self.__client.getresponse()
+        if res.getcode() != 200:
+            raise FrankaAPIError(target, res.getcode(), res.reason, dict(res.headers), res.read().decode("utf-8"))
+        return res.read()
+
+    def send_api_request(self, target: str, headers: Optional[Dict[str, str]] = None, body: Optional[Any] = None,
+                         method: Literal["GET", "POST", "DELETE"] = "POST"):
+        last_error = None
+        for i in range(3):
+            try:
+                return self._send_api_request(target, headers, body, method)
+            except http.client.RemoteDisconnected as ex:
+                last_error = ex
+        raise last_error
+
+    def send_control_api_request(self, target: str, headers: Optional[Dict[str, str]] = None,
+                                 body: Optional[Any] = None,
+                                 method: Literal["GET", "POST", "DELETE"] = "POST"):
+        if headers is None:
+            headers = {}
+        self.__check_control_token()
+        _headers = {
+            "X-Control-Token": self.__control_token
+        }
+        _headers.update(headers)
+        return self.send_api_request(target, headers=_headers, method=method, body=body)
+
+    def open(self):
+        if self.is_open:
+            raise RuntimeError("Session is already open.")
+        self.__client = HTTPSConnection(self.__fci_hostname, timeout=12, context=ssl._create_unverified_context())
+        self.__client.connect()
+        payload = json.dumps(
+            {"login": self.__username, "password": self.__encode_password(self.__username, self.__password)})
+        self.__token = self.send_api_request(
+            "/admin/api/login", headers={"content-type": "application/json"},
+            body=payload).decode("utf-8")
+        return self
+
+    def close(self):
+        if not self.is_open:
+            raise RuntimeError("Session is not open.")
+        if self.__control_token is not None:
+            self.release_control()
+        self.__token = None
+        self.__client.close()
+
+    def __enter__(self):
+        self.open()
+
+    def __exit__(self, type, value, traceback):
+        self.close()
+
+    def __check_control_token(self):
+        if self.__control_token is None:
+            raise RuntimeError("Client does not have control. Call take_control() first.")
+
+    def take_control(self, wait_timeout: float = 10.0):
+        if self.__control_token is None:
+            res = self.send_api_request(
+                "/admin/api/control-token/request", headers={"content-type": "application/json"},
+                body=json.dumps({"requestedBy": self.__username}))
+            response_dict = json.loads(res)
+            self.__control_token = response_dict["token"]
+            self.__control_token_id = response_dict["id"]
+            # One should probably use websockets here but that would introduce another dependency
+            start = time.time()
+            while time.time() - start < wait_timeout and not self.has_control():
+                time.sleep(1.0)
+        return self.has_control()
+
+    def release_control(self):
+        if self.__control_token is not None:
+            self.send_control_api_request(
+                "/admin/api/control-token", headers={"content-type": "application/json"}, method="DELETE",
+                body=json.dumps({"token": self.__control_token}))
+            self.__control_token = None
+            self.__control_token_id = None
+
+    def enable_fci(self):
+        self.send_control_api_request(
+            "/desk/api/system/fci", headers={"content-type": "application/x-www-form-urlencoded"},
+            body=f"token={urllib.parse.quote(base64.b64encode(self.__control_token.encode('ascii')))}")
+
+    def has_control(self):
+        if self.__control_token_id is not None:
+            status = self.get_system_status()
+            active_token = status["controlToken"]["activeToken"]
+            return active_token is not None and active_token["id"] == self.__control_token_id
+        return False
+
+    def start_task(self, task: str):
+        self.send_api_request(
+            "/desk/api/execution", headers={"content-type": "application/x-www-form-urlencoded"},
+            body=f"id={task}")
+
+    def unlock_brakes(self):
+        self.send_control_api_request(
+            "/desk/api/joints/unlock", headers={"content-type": "application/x-www-form-urlencoded"})
+
+    def lock_brakes(self):
+        self.send_control_api_request(
+            "/desk/api/joints/lock", headers={"content-type": "application/x-www-form-urlencoded"})
+
+    def set_mode_programming(self):
+        self.send_control_api_request(
+            "/desk/api/operating-mode/programming", headers={"content-type": "application/x-www-form-urlencoded"})
+
+    def set_mode_execution(self):
+        self.send_control_api_request(
+            "/desk/api/operating-mode/execution", headers={"content-type": "application/x-www-form-urlencoded"})
+
+    def get_system_status(self):
+        return json.loads(self.send_api_request("/admin/api/system-status", method="GET").decode("utf-8"))
+
+    def execute_self_test(self):
+        if self.get_system_status()["safety"]["recoverableErrors"]["td2Timeout"]:
+            self.send_control_api_request(
+                "/admin/api/safety/recoverable-safety-errors/acknowledge?error_id=TD2Timeout")
+        response = json.loads(self.send_control_api_request(
+            "/admin/api/safety/td2-tests/execute", headers={"content-type": "application/json"}).decode("utf-8"))
+        assert response["code"] == "SuccessResponse"
+        time.sleep(0.5)
+        while self.get_system_status()["safety"]["safetyControllerStatus"] == "SelfTest":
+            time.sleep(0.5)
+
+    @property
+    def client(self) -> HTTPSConnection:
+        return self.__client
+
+    @property
+    def token(self) -> str:
+        return self.__token
+
+    @property
+    def is_open(self) -> bool:
+        return self.__token is not None