Refactor ip-address command(May break functionality)
This commit is contained in:
10
core/cli.py
10
core/cli.py
@ -237,7 +237,15 @@ def ip_address(edit: bool, ipv4: str, ipv6: str):
|
||||
- Use --edit with -4 or -6 to manually update IPs.
|
||||
"""
|
||||
try:
|
||||
cli_api.ip_address(edit, ipv4, ipv6)
|
||||
if not edit:
|
||||
cli_api.add_ip_address()
|
||||
click.echo("IP addresses added successfully.")
|
||||
return
|
||||
|
||||
if not ipv4 and not ipv6:
|
||||
raise click.UsageError('Error: You must specify either -4 or -6')
|
||||
|
||||
cli_api.edit_ip_address(ipv4, ipv6)
|
||||
click.echo("IP address configuration updated successfully.")
|
||||
except Exception as e:
|
||||
click.echo(f'{e}', err=True)
|
||||
|
||||
@ -298,6 +298,30 @@ def ip_address(edit: bool, ipv4: str, ipv6: str):
|
||||
run_cmd(['bash', Command.IP_ADD.value, 'add'])
|
||||
|
||||
|
||||
def add_ip_address():
|
||||
'''
|
||||
Adds IP addresses from the environment to the .configs.env file.
|
||||
'''
|
||||
run_cmd(['bash', Command.IP_ADD.value, 'add'])
|
||||
|
||||
|
||||
def edit_ip_address(ipv4: str, ipv6: str):
|
||||
"""
|
||||
Edits the IP address configuration based on provided IPv4 and/or IPv6 addresses.
|
||||
|
||||
:param ipv4: The new IPv4 address to be configured. If provided, the IPv4 address will be updated.
|
||||
:param ipv6: The new IPv6 address to be configured. If provided, the IPv6 address will be updated.
|
||||
:raises InvalidInputError: If neither ipv4 nor ipv6 is provided.
|
||||
"""
|
||||
|
||||
if not ipv4 and not ipv6:
|
||||
raise InvalidInputError('Error: --edit requires at least one of --ipv4 or --ipv6.')
|
||||
if ipv4:
|
||||
run_cmd(['bash', Command.IP_ADD.value, 'edit', '-4', ipv4])
|
||||
if ipv6:
|
||||
run_cmd(['bash', Command.IP_ADD.value, 'edit', '-6', ipv6])
|
||||
|
||||
|
||||
def update_geo(country: str):
|
||||
'''
|
||||
Updates geographic data files based on the specified country.
|
||||
|
||||
Reference in New Issue
Block a user