set_vendor.py 文件源码

def main():
    """
    Main function doing all the heavy lifting
    """

    django.setup()

    devices = NetworkDevice.objects.all()

    for dev in devices:
        if 'cisco' in dev.device_type:
            dev.vendor = "Cisco"
        elif 'juniper' in dev.device_type:
            dev.vendor = "Juniper"
        elif 'arista' in dev.device_type:
            dev.vendor = "Arista"
        dev.save()

    for dev in devices:
        print dev.device_name, dev.vendor