mirror of
				https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
				synced 2025-11-04 12:48:00 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			149 lines
		
	
	
		
			4.6 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			149 lines
		
	
	
		
			4.6 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
#!/usr/bin/env python3
 | 
						|
 | 
						|
"""create_bond.py Script to create a bond
 | 
						|
 | 
						|
This script can be used to create a bond, only one can be created at a time. Network devices must be specified
 | 
						|
as a list of comma-separated items with no spaces.
 | 
						|
 | 
						|
Use './create_bond.py --help' to see command line usage and options
 | 
						|
Copyright 2021 Candela Technologies Inc
 | 
						|
License: Free to distribute and modify. LANforge systems must be licensed.
 | 
						|
"""
 | 
						|
import sys
 | 
						|
import os
 | 
						|
import importlib
 | 
						|
import argparse
 | 
						|
import time
 | 
						|
import logging
 | 
						|
from time import sleep
 | 
						|
 | 
						|
logger = logging.getLogger(__name__)
 | 
						|
 | 
						|
if sys.version_info[0] != 3:
 | 
						|
    logger.critical("This script requires Python 3")
 | 
						|
    exit(1)
 | 
						|
 | 
						|
sys.path.append(os.path.join(os.path.abspath(__file__ + "../../../")))
 | 
						|
 | 
						|
lfcli_base = importlib.import_module("py-json.LANforge.lfcli_base")
 | 
						|
LFCliBase = lfcli_base.LFCliBase
 | 
						|
LFUtils = importlib.import_module("py-json.LANforge.LFUtils")
 | 
						|
realm = importlib.import_module("py-json.realm")
 | 
						|
Realm = realm.Realm
 | 
						|
lf_logger_config = importlib.import_module("py-scripts.lf_logger_config")
 | 
						|
 | 
						|
 | 
						|
class CreateBond(Realm):
 | 
						|
    def __init__(self, _network_dev_list=None,
 | 
						|
                 _host=None,
 | 
						|
                 _port=None,
 | 
						|
                 _eid=None,
 | 
						|
                 _debug_on=False):
 | 
						|
        super().__init__(_host, _port)
 | 
						|
        self.host = _host
 | 
						|
        self.timeout = 120
 | 
						|
        self.debug = _debug_on
 | 
						|
        self.network_dev_list = _network_dev_list
 | 
						|
 | 
						|
        eid = self.name_to_eid(_eid)
 | 
						|
        self.shelf = eid[0]
 | 
						|
        self.resource = eid[1]
 | 
						|
        self.bond_name = eid[2]
 | 
						|
 | 
						|
    def build(self):
 | 
						|
        eid = "%s.%s.%s" % (self.shelf, self.resource, self.bond_name)
 | 
						|
        nd = False
 | 
						|
        for td in self.network_dev_list.split(","):
 | 
						|
            e2 = self.name_to_eid(td)
 | 
						|
            if not nd:
 | 
						|
                nd = e2[2]
 | 
						|
            else:
 | 
						|
                nd += ","
 | 
						|
                nd += e2[2]
 | 
						|
 | 
						|
        data = {
 | 
						|
            'shelf': self.shelf,
 | 
						|
            'resource': self.resource,
 | 
						|
            'port': self.bond_name,
 | 
						|
            'network_devs': nd
 | 
						|
        }
 | 
						|
        self.json_post("cli-json/add_bond", data)
 | 
						|
        #time.sleep(3)
 | 
						|
        bond_set_port = {
 | 
						|
            "shelf": self.shelf,
 | 
						|
            "resource": self.resource,
 | 
						|
            "port": self.bond_name,
 | 
						|
            "current_flags": 0x80000000,
 | 
						|
            # (0x2 + 0x4000 + 0x800000)  # current, dhcp, down
 | 
						|
            "interest": 0x4000
 | 
						|
        }
 | 
						|
        self.json_post("cli-json/set_port", bond_set_port)
 | 
						|
 | 
						|
        if LFUtils.wait_until_ports_admin_up(base_url=self.lfclient_url,
 | 
						|
                                             port_list=[eid],
 | 
						|
                                             debug_=self.debug):
 | 
						|
            self._pass("Bond interface went admin up.")
 | 
						|
        else:
 | 
						|
            self._fail("Bond interface did NOT go admin up.")
 | 
						|
 | 
						|
    def cleanup(self):
 | 
						|
        eid = "%s.%s.%s" % (self.shelf, self.resource, self.bond_name)
 | 
						|
        self.rm_port(eid, check_exists=False, debug_=self.debug)
 | 
						|
        if LFUtils.wait_until_ports_disappear(base_url=self.lfclient_url, port_list=[eid], debug=self.debug):
 | 
						|
            self._pass("Ports successfully cleaned up.")
 | 
						|
        else:
 | 
						|
            self._fail("Ports NOT successfully cleaned up.")
 | 
						|
 | 
						|
def main():
 | 
						|
    parser = LFCliBase.create_basic_argparse(
 | 
						|
        prog='create_bond.py',
 | 
						|
        formatter_class=argparse.RawTextHelpFormatter,
 | 
						|
        epilog='''\
 | 
						|
         Create bonds
 | 
						|
            ''',
 | 
						|
 | 
						|
        description='''\
 | 
						|
        create_bond.py
 | 
						|
--------------------
 | 
						|
Command example:
 | 
						|
./create_bond.py
 | 
						|
    --bond_name bond0
 | 
						|
    --network_dev_list eth0,eth1
 | 
						|
    --noclean
 | 
						|
    --debug
 | 
						|
            ''')
 | 
						|
 | 
						|
    required = parser.add_argument_group('required arguments')
 | 
						|
 | 
						|
    required.add_argument('--bond_name', help='Name of the bridge to create', required=True)
 | 
						|
    required.add_argument('--network_dev_list', help='list of network devices in the bond, must be comma separated '
 | 
						|
                          'with no spaces', required=True)
 | 
						|
 | 
						|
    args = parser.parse_args()
 | 
						|
 | 
						|
    logger_config = lf_logger_config.lf_logger_config()
 | 
						|
    # set the logger level to requested value
 | 
						|
    logger_config.set_level(level=args.log_level)
 | 
						|
    logger_config.set_json(json_file=args.lf_logger_config_json)
 | 
						|
 | 
						|
    create_bond = CreateBond(_host=args.mgr,
 | 
						|
                             _port=args.mgr_port,
 | 
						|
                             _eid = args.bond_name,
 | 
						|
                             _network_dev_list=args.network_dev_list,
 | 
						|
                             _debug_on=args.debug
 | 
						|
                             )
 | 
						|
    create_bond.build()
 | 
						|
 | 
						|
    if not args.noclean:
 | 
						|
        sleep(5)
 | 
						|
 | 
						|
        create_bond.cleanup()
 | 
						|
 | 
						|
    if create_bond.passes():
 | 
						|
        create_bond.exit_success()
 | 
						|
    else:
 | 
						|
        create_bond.exit_fail()
 | 
						|
 | 
						|
if __name__ == "__main__":
 | 
						|
    main()
 |