import yaml import netaddr import os import log as logging LOG = logging.getLogger("net-recover") config_path = os.path.join(os.path.dirname(__file__), "network.cfg") def setup_bondings(bond_mappings): print bond_mappings def add_ovs_port(ovs_br, ifname, uplink, vlan_id=None): LOG.info("add_ovs_port enter") cmd = "ovs-vsctl --may-exist add-port %s %s" % (ovs_br, ifname) if vlan_id: cmd += " tag=%s" % vlan_id cmd += " -- set Interface %s type=internal;" % ifname cmd += "ip link set dev %s address `ip link show %s |awk '/link\/ether/{print $2}'`;" \ % (ifname, uplink) cmd += "ip link set %s up;" % ifname LOG.info("add_ovs_port: cmd=%s" % cmd) os.system(cmd) def setup_ips(ip_settings, sys_intf_mappings): LOG.info("setup_ips enter") for intf_info in ip_settings.values(): network = netaddr.IPNetwork(intf_info["cidr"]) if sys_intf_mappings[intf_info["name"]]["type"] == "ovs": intf_name = intf_info["name"] else: intf_name = intf_info["alias"] if "gw" in intf_info: cmd = "ip addr add %s/%s brd %s dev %s;" \ % (intf_info["ip"], intf_info["netmask"], str(network.broadcast),intf_name) cmd += "route del default;" cmd += "ip route add default via %s dev %s" % (intf_info["gw"], intf_name) LOG.info("setup_ips: cmd=%s" % cmd) os.system(cmd) def setup_intfs(sys_intf_mappings, uplink_map): LOG.info("setup_intfs enter") for intf_name, intf_info in sys_intf_mappings.items(): if intf_info["type"] == "ovs": add_ovs_port( intf_info["interface"], intf_name, uplink_map[intf_info["interface"]], vlan_id=intf_info.get("vlan_tag")) else: pass def main(config): uplink_map = {} setup_bondings(config["bond_mappings"]) for provider_net in config["provider_net_mappings"]: uplink_map[provider_net['name']] = provider_net['interface'] setup_intfs(config["sys_intf_mappings"], uplink_map) setup_ips(config["ip_settings"], config["sys_intf_mappings"]) if __name__ == "__main__": os.system("service openvswitch-switch status|| service openvswitch-switch start") config = yaml.load(open(config_path)) main(config)