from dataclasses import dataclass, field
from lerobot.cameras.configs import CameraConfig
from lerobot.cameras.realsense.configuration_realsense import RealSenseCameraConfig
from lerobot.motors.motors_bus import Motor, MotorCalibration
from lerobot.robots.config import RobotConfig

@RobotConfig.register_subclass("realman")
@dataclass
class RealmanRobotConfig(RobotConfig):
    # Port to connect to the arm
    port: str
    gripper_range: list[int] = field(default_factory=list)
    disable_torque_on_disconnect: bool = True
    # cameras
    cameras: dict[str, RealSenseCameraConfig] = field(default_factory=dict)
    joint: list=field(default_factory=list)
    motors: dict[str, Motor] = field(default_factory=dict)
    calibration: dict[str, MotorCalibration] | None = None