bivashy/Blitz-Proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Implement get_ip_address in cli_api.py

Browse Source
This commit is contained in:
Iam54r1n4
2025-02-08 16:36:12 +00:00
parent a5d607207b
commit fe038b2967
1 changed files with 8 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
core/cli_api.py
View File

@ -4,6 +4,7 @@ from enum import Enum
from datetime import datetime from datetime import datetime
import json import json
from typing import Any from typing import Any
from dotenv import dotenv_values
import traffic import traffic
@ -297,19 +298,15 @@ def server_info() -> str | None:
return run_cmd(['bash', Command.SERVER_INFO.value]) return run_cmd(['bash', Command.SERVER_INFO.value])
def ip_address(edit: bool, ipv4: str, ipv6: str): def get_ip_address() -> tuple[str | None, str | None]:
''' '''
Manages IP address configuration with edit options. Retrieves the IP address from the .configs.env file.
''' '''
if edit:
if ipv4: env_file_path = os.path.join(SCRIPT_DIR, '..', '..', '.configs.env')
run_cmd(['bash', Command.IP_ADD.value, 'edit', '-4', ipv4]) env_vars = dotenv_values(env_file_path)
if ipv6:
run_cmd(['bash', Command.IP_ADD.value, 'edit', '-6', ipv6]) return env_vars.get('IP4'), env_vars.get('IP6')
if not ipv4 and not ipv6:
raise InvalidInputError('Error: --edit requires at least one of --ipv4 or --ipv6.')
else:
run_cmd(['bash', Command.IP_ADD.value, 'add'])
def add_ip_address(): def add_ip_address():