from typing_extensions import runtime
from nornir import InitNornir
from nornir.core.plugins.inventory import InventoryPluginRegister
# from nornir_netmiko import netmiko_send_command, netmiko_send_config
from nornir_paramiko.plugins.tasks import paramiko_command, paramiko_commands
from nornir_utils.plugins.functions import print_result
from lab2.cmdb import CMDBInventory

from tqdm import tqdm

InventoryPluginRegister.register("CMDBInventory", CMDBInventory)

# device_type改为platform也是兼容的
devices = [
    {'ip': '192.168.11.11', 'username': 'python',
        'password': '123', 'device_type': 'huawei'},
    {'ip': '192.168.11.12', 'username': 'python',
        'password': '123', 'platform': 'huawei'},
    {'ip': '192.168.11.13', 'username': 'python',
        'password': '123', 'platform': 'huawei'},
]

nr = InitNornir(
    runner={
        "plugin": "threaded",
        "options": {
            "num_workers": 100,
        },
    },
    inventory={
        "plugin": "CMDBInventory",
        "options": {
            "devices": devices,
        },
    },
)


commands = ['sys', 'dis this', 'return', 'quit']

# 一个task也可以调用其他task
def multiple_progress_bar(task, current_bar, other_bar):
    task.run(task=paramiko_commands, commands=commands)

    current_bar.update()
    tqdm.write(f"{task.host}: processing...")

    other_bar.update()
    tqdm.write(f"{task.host}: done!")


with tqdm(total=len(nr.inventory.hosts), desc='processing') as current_bar:
    with tqdm(total=len(nr.inventory.hosts), desc='other action') as other_bar:
        nr.run(task=multiple_progress_bar,
               current_bar=current_bar, other_bar=other_bar)