fast switching between network configurations

netset.py

#! c:\miniconda\python
from collections import namedtuple
import time
import sys
import click
import wmi

TIMEOUT_SECONDS = 20

# static configuration
Config = namedtuple('Config', 'name dhcp ip subnetmask gateway')

configs = {
    'dhcp': Config('dhcp', True, None, None, None),
    'static': Config('static', False, '192.168.1.110', '255.255.255.0', '192.168.1.0')
}


nic_configs = wmi.WMI().Win32_NetworkAdapterConfiguration(IPEnabled=True)
nic = nic_configs[0]


def ip_address():
    ip = nic.wmi_property('IPAddress')
    return ip.value[0]


def subnetmask():
    subnet = nic.wmi_property('IPSubnet')
    return subnet.value[0]


def gateway():
    gateway = nic.wmi_property('DefaultIPGateway')
    return gateway.value[0]

def dhcp_enabled():
    dhcp = nic.wmi_property('DHCPEnabled')
    return dhcp.value


def config_active(config):
    if config.dhcp:
        return dhcp_enabled()
    return (config.ip == ip_address() and
            config.subnetmask == subnetmask() and
            config.gateway == gateway())

@click.group()
def cli():
    pass


@cli.command()
def list():
    """ list available network configurations """
    for key, config in configs.items():
        s = '{x.name}: {x.ip}'.format(x=config)
        click.echo(s)


@cli.command()
def status():
    """ show current network status """
    click.echo('Status')


@cli.command()
@click.pass_context
def load(config):
    print(config)


@click.command()
@click.option('--config', type=click.Choice(configs.keys()))
def set_nic(config):
    config_name = config
    config = configs[config]
    if config.dhcp:
        nic.EnableDHCP()
    else:
        nic.EnableStatic(IPAddress=[config.ip],SubnetMask=[config.subnetmask])
        nic.SetGateways(DefaultIPGateway=[config.gateway])

    t0 = time.perf_counter()

    while not abs(time.perf_counter() - t0) > TIMEOUT_SECONDS:
        if config_active(config):
            click.echo('Successfully changed config to {}.'.format(config_name))
            break
        time.sleep(1)
    else:
        click.echo('Timeout')

    click.echo('IP: {ip}\nSubnetmask: {subnetmask}\n'
               'Gateway: {gateway}'.format(
                    ip=ip_address(),
                    subnetmask=subnetmask(),
                    gateway=gateway()
                ))

if __name__ == '__main__':
    set_nic()