mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-11-01 03:07:56 +00:00
126 lines
4.0 KiB
Python
Executable File
126 lines
4.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
"""
|
|
Script for modifying VAPs.
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
import argparse
|
|
|
|
if sys.version_info[0] != 3:
|
|
print("This script requires Python 3")
|
|
exit(1)
|
|
|
|
if 'py-json' not in sys.path:
|
|
sys.path.append(os.path.join(os.path.abspath('..'), 'py-json'))
|
|
import LANforge
|
|
from LANforge.lfcli_base import LFCliBase
|
|
from LANforge import LFUtils
|
|
import realm
|
|
from realm import Realm
|
|
import time
|
|
import pprint
|
|
|
|
|
|
class ModifyVAP(Realm):
|
|
def __init__(self,
|
|
_ssid="NA",
|
|
_security="NA",
|
|
_password="NA",
|
|
_mac="NA",
|
|
_host=None,
|
|
_port=None,
|
|
_vap_list=None,
|
|
_enable_flags=None,
|
|
_disable_flags=None,
|
|
_number_template="00000",
|
|
_radio=None,
|
|
_proxy_str=None,
|
|
_debug_on=False,
|
|
_exit_on_error=False,
|
|
_exit_on_fail=False,
|
|
_dhcp=True):
|
|
super().__init__(_host,
|
|
_port)
|
|
self.host = _host
|
|
self.port = _port
|
|
self.ssid = _ssid
|
|
self.security = _security
|
|
self.password = _password
|
|
self.mac = _mac
|
|
self.vap_list = _vap_list
|
|
self.enable_flags = _enable_flags
|
|
self.disable_flags = _disable_flags
|
|
self.radio = _radio
|
|
self.timeout = 120
|
|
self.number_template = _number_template
|
|
self.debug = _debug_on
|
|
self.dhcp = _dhcp
|
|
self.vap_profile = self.new_vap_profile()
|
|
self.vap_profile.vap_name = self.vap_list
|
|
self.vap_profile.ssid = self.ssid
|
|
self.vap_profile.security = self.security
|
|
self.vap_profile.ssid_pass = self.password
|
|
self.vap_profile.mac = self.mac
|
|
self.vap_profile.dhcp = self.dhcp
|
|
self.vap_profile.debug = self.debug
|
|
self.vap_profile.desired_add_vap_flags = self.enable_flags
|
|
self.vap_profile.desired_add_vap_flags_mask = self.enable_flags + self.disable_flags
|
|
|
|
def set_vap(self):
|
|
return self.vap_profile.modify(resource=1,
|
|
radio=self.radio)
|
|
|
|
|
|
def main():
|
|
parser = LFCliBase.create_basic_argparse(
|
|
prog='modify_vap.py',
|
|
formatter_class=argparse.RawTextHelpFormatter,
|
|
epilog='''\
|
|
Modify VAPs on a system. Use the enable_flag to create a flag on a VAP. Turn off a flag with \
|
|
the disable_flag option. A list of available flags are available in the add_vap.py file in \
|
|
py-json/LANforge.
|
|
''',
|
|
|
|
description='''\
|
|
modify_vap.py
|
|
--------------------
|
|
Command example:
|
|
./modify_vap.py
|
|
--radio wiphy0
|
|
--vap 1.1.vap0000
|
|
--security open
|
|
--ssid netgear
|
|
--passwd BLANK
|
|
--enable_flag osen_enable
|
|
--disable_flag ht160_enable
|
|
--debug
|
|
''')
|
|
|
|
optional = parser.add_argument_group('optional arguments')
|
|
optional.add_argument('--enable_flag', help='VAP flags to add', default=list(), action='append')
|
|
optional.add_argument('--disable_flag', help='VAP flags to disable', default=list(), action='append')
|
|
optional.add_argument('--vap', help='VAP to modify', required=True)
|
|
optional.add_argument('--mac', default="NA")
|
|
|
|
args = parser.parse_args()
|
|
|
|
modify_vap = ModifyVAP(_host=args.mgr,
|
|
_port=args.mgr_port,
|
|
_ssid=args.ssid,
|
|
_password=args.passwd,
|
|
_security=args.security,
|
|
_mac=args.mac,
|
|
_vap_list=args.vap,
|
|
_enable_flags=args.enable_flag,
|
|
_disable_flags=args.disable_flag,
|
|
_radio=args.radio,
|
|
_proxy_str=args.proxy,
|
|
_debug_on=args.debug)
|
|
modify_vap.set_vap()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|