Use Primary IP Address in a custom script

[bogdancordos]

Hi,

I am building a simple custom script to run an SSH command using netmiko.
The below crude version works just fine, however, I would like to choose an IP based on Device Model instead of passing it myself as a StringVar:

from netmiko import ConnectHandler
from extras.scripts import *

class RunCommand(Script):
    class Meta:
        name = "Run command on Cisco Devices via SSH"
        description = "Run command on Cisco Devices via SSH"
        field_order = ['input_ip', 'input_username', 'input_pass', 'input_command']

    input_ip = StringVar(
        description="Enter the IP Address:"
    )
    input_username = StringVar(
        description="Enter username:"
    )
    input_pass = StringVar(
        description="Enter your password:"
    )
    input_command = StringVar(
        description="Enter the SSH command:"
    )



    def run(self, data, commit):

        cisco1 = { 
            "device_type": "cisco_ios",
            "host": data['input_ip'],
            "username": data['input_username'],
            "password": data['input_pass'],
        }

        command = data['input_command']

        with ConnectHandler(**cisco1) as net_connect:
            output = net_connect.send_command(command)


            self.log_success(output)
        return "Success"

Any suggestions on how can I replace the " input_ip" with a model that will let me choose a device and use its primary IP?

Thank you

[candlerb]

Look at ObjectVar. The user can select a Device object, and once your script has a model instance, it can use its primary_ip attribute.