diff --git a/modules/vagrant_generator.py b/modules/vagrant_generator.py index f6211bbfd49b13306c9c2be60bfa4af6d0952f62..12f1fb00f7fd1c6f918241f0e60b290961132561 100644 --- a/modules/vagrant_generator.py +++ b/modules/vagrant_generator.py @@ -1,5 +1,7 @@ """Contains functions for generating a Vagrantfile from input definitions.""" +import netaddr + from modules.file_manager import generate_file, open_yaml VAGRANT_MAPPING = open_yaml("conf/vagrant_mapping.yml") @@ -173,8 +175,10 @@ def _add_networks_to_device(definition, mappings, input_definitions): network["network_type"] = "private_network" network["name"] = mapping["network"] network["ip"] = mapping["ip"] - network["netmask"] = _find_netmask(mapping["network"], - input_definitions["networks"]) + netmask = netaddr.IPNetwork("0.0.0.0/" + + _find_netmask(mapping["network"], + input_definitions["networks"])).netmask + network["netmask"] = str(netmask) definition["commands"].append(network)