solution

app/main.py

@@ -2,4 +2,80 @@ class Cargo:
     def __init__(self, weight: int) -> None:
         self.weight = weight
 
-# write your code here
+
+class BaseRobot:
+    def __init__(
+            self, name: str, weight: int, coords: list[int] = None
+    ) -> None:
+        if coords is None:
+            coords = [0, 0]
+        self.name = name
+        self.weight = weight
+        self.coords = coords
+
+    def go_forward(self, step: int = 1) -> list[int]:
+        self.coords[1] += step
+        return self.coords
+
+    def go_back(self, step: int = 1) -> list[int]:
+        self.coords[1] -= step
+        return self.coords
+
+    def go_right(self, step: int = 1) -> list[int]:
+        self.coords[0] += step
+        return self.coords
+
+    def go_left(self, step: int = 1) -> list[int]:
+        self.coords[0] -= step
+        return self.coords
+
+    def get_info(self) -> str:
+        return f"Robot: {self.name}, Weight: {self.weight}"
+
+
+class FlyingRobot(BaseRobot):
+    def __init__(
+            self, name: str, weight: int, coords: list[int] = None
+    ) -> None:
+        if coords is None:
+            coords = [0, 0, 0]
+        super().__init__(name, weight, coords)
+
+    def go_up(self, step: int = 1) -> list[int]:
+        self.coords[2] += step
+        return self.coords
+
+    def go_down(self, step: int = 1) -> list[int]:
+        self.coords[2] -= step
+        return self.coords
+
+
+class DeliveryDrone(FlyingRobot):
+    def __init__(
+            self, name: str, weight: int, max_load_weight: int,
+            coords: list[int] = None, current_load: Cargo = None
+    ) -> None:
+        if coords is None:
+            coords = [0, 0, 0]
+        super().__init__(name, weight, coords)
+        self.max_load_weight = max_load_weight
+        self.current_load = current_load
+
+    def hook_load(self, cargo: Cargo) -> str:
+        if self.current_load is not None:
+            return "Cannot hook load. There is already cargo hooked."
+        elif cargo.weight > self.max_load_weight:
+            return (
+                f"Cannot hook load. The cargo is too heavy: "
+                f"{cargo.weight} > {self.max_load_weight}."
+            )
+        self.current_load = cargo
+        return f"Cargo weighing {cargo.weight} hooked successfully."
+
+    def unhook_load(self) -> str:
+        if self.current_load is None:
+            return "No cargo to unhook."
+
+        unhooked_cargo = self.current_load
+        self.current_load = None
+        return f"Cargo weighing {unhooked_cargo.weight} unhooked successfully."