feat(apply): adds template func to get interface ip

pyproject.toml

@@ -46,7 +46,7 @@ dependencies = [
     "python-magic>=0.4.27",
     "pytz",
     "pyyaml>=5.4.1",
-    "rapyuta-io>=2.0.0",
+    "rapyuta-io>=2.1.0",
     "requests>=2.20.0",
     "semver>=3.0.0",
     "setuptools",

riocli/apply/filters.py

@@ -11,17 +11,17 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
 """
 Filters to use in the manifests.
 """
-
 import os
 
+from riocli.config import new_client
+from riocli.device.util import find_device_guid
+
 
 def getenv(default: str, env_var: str) -> str:
-    """
-    Get the value of an environment variable.
+    """Get the value of an environment variable.
 
     Usage:
         "foo" : {{ "bar" | getenv('FOO') }}
@@ -36,6 +36,41 @@ def getenv(default: str, env_var: str) -> str:
     return os.getenv(env_var, default)
 
 
+def get_interface_ip(device_name: str, interface: str) -> str:
+    """Get the IP address of an interface on a device.
+
+    Usage:
+        "ip" : {{ device_name | get_intf_ip(interface='intf-name') }}
+
+    Args:
+        device_name: The name of the device.
+        interface: The name of the interface.
+
+    Returns:
+        The IP address of the interface
+
+    Raises:
+        Exception: If the interface is not available on the device.
+    """
+    client = new_client(with_project=True)
+    device_id = find_device_guid(client, device_name)
+
+    device = client.get_device(device_id)
+    try:
+        device.poll_till_ready(retry_count=50, sleep_interval=10)
+    except Exception as e:
+        raise e
+
+    device.refresh()
+
+    for name in device.ip_interfaces:
+        if name == interface:
+            return device.ip_interfaces[name][0]
+
+    raise Exception(f'interface "{interface}" not found on device "{device_name}"')
+
+
 FILTERS = {
     "getenv": getenv,
+    "get_intf_ip": get_interface_ip,
 }