python - 使用 mako 生成 Cisco 配置。是否可以在模板中使用 netaddr 模块(或任何模块)？

Question

我对 python 和 mako 非常陌生，我可能在基本概念上遇到困难。我有工作模板，但如果我可以使用 netaddr 模块来处理模板中的 IP 地址，我的 CSV 输入可以得到很大的清理。我想做的是传递一个接口(interface) IP 变量，例如:LAN_IP = '192.168.1.1/24'(我从 CSV 中执行此操作)到模板，然后以某种方式使用 netaddr 模块来填写 IP、子网掩码(点分十进制和反掩码)和网络地址，这样我可以在同一配置中将这一 CSV 变量用于 bgp 网络语句、eigrp 掩码和 ACL 等。我可以从 Python shell 执行以下操作:

>>> from netaddr import *
>>> LAN_IP = '192.168.1.1/24'
>>> IP = IPNetwork(LAN_IP)
>>> print(IP.ip)
192.168.1.1
>>> print(IP.network)
192.168.1.0
>>> print(IP.netmask)
255.255.255.0
>>> print(IP.hostmask)
0.0.0.255

编辑。我被告知要尝试内置的 ipaddress 功能。在 Python 提示符下尝试此操作表明它可以工作:

MAKO_TEMPLATE_STRING = """\
<%def name="get_netmask(ip_string)"><%
    import ipaddress
    return ipaddress.IPv4Interface(ip_string).netmask
%></%def>
<%def name="get_address(ip_string)"><%
    import ipaddress
    return ipaddress.IPv4Interface(ip_string).ip
%></%def>
<%def name="get_subnet(ip_string)"><%
    import ipaddress
    return ipaddress.IPv4Interface(ip_string).network
%></%def>
<%def name="get_hostmask(ip_string)"><%
    import ipaddress
    return ipaddress.IPv4Interface(ip_string).hostmask
%></%def>
! Variable Input: ${data}
${get_address(data)} ${get_netmask(data)} ${get_subnet(data)} ${get_hostmask(data)}
"""
print(Template(MAKO_TEMPLATE_STRING).render(data="192.168.1.1/25"))

这给了我以下输出:!变量输入:192.168.1.1/25192.168.1.1 255.255.255.128 192.168.1.0/25 0.0.0.127

现在，一个新问题是 ipaddress.IPv4Interface(ip_string).network 定义返回网络加上子网“192.168.1.0/25”，而 netaddr 没有这样做。我还没有找到任何方法让 ipaddress 仅返回子网部分。

因此，对于模板构建，我走得更远了......如果我仅将其放入模板中，它就会验证:

<%def name="get_netmask(ip_string)"><%
    import ipaddress
    return ipaddress.IPv4Interface(ip_string).netmask
%></%def>
<%def name="get_address(ip_string)"><%
    import ipaddress
    return ipaddress.IPv4Interface(ip_string).ip
%></%def>
<%def name="get_network(ip_string)"><%
    import ipaddress
    return ipaddress.IPv4Interface(ip_string).network
%></%def>

但是当我尝试任何类型的引用时，我都会收到错误。

! Variable Input: ${LAN_IP}  <--I tried with and without this line
${get_address(LAN_IP)}  <--The template does not seem to like these references.
${get_netmask(LAN_IP)}
${get_subnet(LAN_IP)}

在我看来，我在这里遗漏了一些非常简单的东西，但也许我正在以错误的方式处理问题。任何帮助将不胜感激，因为 Mako 文档和 Google 几乎没有提供任何帮助来说明如何做这样的事情。

Answer 1

问题出在验证代码上。在模板的 python 部分中，类 IPv4Interface 将因验证过程中的值无效而引发异常。因此，服务器拒绝表单数据并且不会创建模板。

解决方案:如果将错误处理添加到模板的 python 部分，就会起作用。

<%!
    ## python module-level code
    import ipaddress
%>
<%def name="get_address(ip_string)">
    <%
    try:
        return ipaddress.IPv4Interface(ip_string).ip
    except Exception:
        return "FAIL_OR_EMPTY"
    %>
</%def>
! Variable Input: ${LAN_IP}
${get_address(LAN_IP)}

说明:在 HTML 表单的服务器端验证期间，配置模板会使用虚拟参数集进行呈现，以验证语法(请参阅文件 app/forms.py，类 ConfigTemplateForm)。您的配置模板在表单验证期间使用以下参数集进行验证:

{
    "hostname": "test",
    "LAN_IP": "test"
}