provision_l3vxlan.py

from nornir import InitNornir
from nornir.core.task import Task, Result
from nornir.core.filter import F
from nornir_utils.plugins.functions import print_result
from nornir_jinja2.plugins.tasks import template_file
from nornir_scrapli.tasks import send_commands, send_config

def deploy_l3vxlan(task: Task, data: dict) -> Result:
    task.host["vrf"] = data
    r = task.run(task=template_file, 
                template="l3vxlan.j2",
                path="./templates")
    task.host["config"] = r.result


    task.run(task=send_config, 
            name="Provisioning L3 EVPN VXLAN!", 
            dry_run=False,
            config=task.host["config"])
    
    task.run(task=send_commands, 
            name="Show new config and copy running config to startup config.", 
            commands=["show run", "write memory"])

def get_input() -> dict:
    print("*"*50)
    print("* This script will provision L3 EVPN *")
    print("*"*50)
    data = {}
    data["vlan_id"] = input("Enter the vlan ID (ex: 10): ")
    data["name"] = input("Enter the vrf name (ex: cust-big): ")
    data["vni"] = input("Enter the vni (ex: 100010): ")
    data["a_end"] = input("Enter the A-end switch name (ex: leaf1): ")
    data["a_if"] = input("Enter the A-end switch port (ex: Eth3): ")
    data["a_ipv4"] = input("Enter the A-end IP address (ex: 123.1.1.2/24): ")
    data["a_vipv4"] = input("Enter the A-end Virtual IP address (ex: 123.1.1.1/24): ")
    data["b_end"] = input("Enter the B-end switch name (ex: leaf2): ")
    data["b_if"] = input("Enter the B-end switch port (ex: Eth3): ")
    data["b_ipv4"] = input("Enter the A-end IP address (ex: 123.1.1.2/24): ")
    data["b_vipv4"] = input("Enter the B-end Virtual IP address (ex: 123.1.1.1/24): ")
    
    return data


if __name__ == "__main__":
    user_input = get_input()
    nr = InitNornir(config_file="config.yml")
    try:
        nr = nr.filter(F(name=user_input["a_end"]) | F(name=user_input["b_end"]))
        r = nr.run(task=deploy_l3vxlan, data=user_input)
        print_result(r)
    except KeyError as e:
        print(f"Could not find device: {e}")