从 Nornir 2.3.0 隐蔽到 Nornir 3.2.0

Question

需要从 shell 输入获取用户名和密码

def adapt_host_data(host):
    host.username = Gusername
    host.password = Gpassword

if __name__ == "__main__":
    Gusername = input('Username: ')
    Gpassword = getpass('Password: ')

    nr = InitNornir(
    inventory={
                "plugin": "nornir.plugins.inventory.simple.SimpleInventory",
                "options": {
                    "host_file": "./inventory/hostise.yaml",
                    "group_file": "./inventory/groups.yaml"
                            },
                "transform_function": adapt_host_data,
            },)

这在 Nornir 2.3.0 下工作正常。但我无法在 Nornir 3.2.0 下使用它。 请帮助

这是我得到的错误

Traceback (most recent call last):
  File "filesname.py", line 70, in <module>
    "transform_function": adapt_host_data,
  File "/heim/tahir/scripts/Python/nornir/envNornir3/lib64/python3.6/site-packages/nornir/init_nornir.py", line 72, in InitNornir
    inventory=load_inventory(config),
  File "/heim/tahir/scripts/Python/nornir/envNornir3/lib64/python3.6/site-packages/nornir/init_nornir.py", line 19, in load_inventory
    inventory_plugin = InventoryPluginRegister.get_plugin(config.inventory.plugin)
  File "/heim/tahir/scripts/Python/nornir/envNornir3/lib64/python3.6/site-packages/nornir/core/plugins/register.py", line 76, in get_plugin
    raise PluginNotRegistered(f"plugin {name!r} is not registered")
nornir.core.exceptions.PluginNotRegistered: plugin 'nornir.plugins.inventory.simple.SimpleInventory' is not registered

Answer 1

需要注册 InventoryPluginRegister 和 SimpleInventory

def transform_func(host):
    host.username = Gusername
    host.password = Gpassword

if __name__ == "__main__":

    from nornir.core.plugins.inventory import InventoryPluginRegister
    from nornir.plugins.inventory import SimpleInventory
    from nornir.core.plugins.inventory import TransformFunctionRegister
    InventoryPluginRegister.register("SimpleInventory", SimpleInventory)
    TransformFunctionRegister.register("transform_func", transform_func)

    Gusername = input('Username: ')
    Gpassword = getpass('Password: ')
#
    nr = InitNornir(config_file="./inventory/config.yaml")

并且 config.yaml 必须如下所示

inventory:
  plugin: SimpleInventory
  transform_function: transform_func
  options:
    host_file: "./inventory/hostise.yaml"
    group_file: "./inventory/groups.yaml"
    defaults_file: "./inventory/defaults.yaml"

logging:
  enabled: False