1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
|
"""Add or remove specific subnets access for a user."""
# :license: MIT, see LICENSE for more details.
import click
import SoftLayer
from SoftLayer.CLI import environment
from SoftLayer.CLI import helpers
@click.command(cls=SoftLayer.CLI.command.SLCommand, )
@click.option('--add/--remove', default=True,
help="Add or remove access to subnets.")
@click.argument('user', nargs=1, required=True)
@click.argument('subnet', nargs=-1, required=True)
@environment.pass_env
def cli(env, user, add, subnet):
"""Add or remove subnets access for a user."""
mgr = SoftLayer.UserManager(env.client)
user_id = helpers.resolve_id(mgr.resolve_ids, user, 'username')
if add:
result = mgr.vpn_subnet_add(user_id, subnet)
else:
result = mgr.vpn_subnet_remove(user_id, subnet)
if result:
click.secho(f"{user} updated successfully", fg='green')
else:
click.secho(f"Failed to update {user}", fg='red')
|