tests: replace pycodestyle with black
[vpp.git] / src / plugins / nat / extras / nat_static_gen_cfg.py
1 #!/usr/bin/env python3
2 import ipaddress
3 import argparse
4
5 parser = argparse.ArgumentParser(description="Generate NAT plugin config.")
6 parser.add_argument(
7     "static_map_num", metavar="N", type=int, nargs=1, help="number of static mappings"
8 )
9 args = parser.parse_args()
10
11 file_name = "nat_static_%s" % (args.static_map_num[0])
12 outfile = open(file_name, "w")
13
14 outfile.write("set int ip address TenGigabitEthernet4/0/0 172.16.2.1/24\n")
15 outfile.write("set int ip address TenGigabitEthernet4/0/1 173.16.1.1/24\n")
16 outfile.write("set int state TenGigabitEthernet4/0/0 up\n")
17 outfile.write("set int state TenGigabitEthernet4/0/1 up\n")
18 outfile.write("ip route add 2.2.0.0/16 via 173.16.1.2 TenGigabitEthernet4/0/1\n")
19 outfile.write("ip route add 10.0.0.0/24 via 172.16.2.2 TenGigabitEthernet4/0/0\n")
20 outfile.write("set int nat44 in TenGigabitEthernet4/0/0 out TenGigabitEthernet4/0/1\n")
21
22 for i in range(0, args.static_map_num[0]):
23     local = str(ipaddress.IPv4Address("10.0.0.3") + i)
24     external = str(ipaddress.IPv4Address("173.16.1.3") + i)
25     outfile.write("nat44 add static mapping local %s external %s\n" % (local, external))