Refactor ip APIs

This commit is contained in:
Iam54r1n4
2025-02-08 01:06:43 +00:00
parent 3915f61cfb
commit 304dd51ef0

View File

@ -10,6 +10,15 @@ router = APIRouter()
@router.get('/add')
async def add_ip_api():
"""
Adds the auto-detected IP addresses to the .configs.env file.
Returns:
A DetailResponse with a message indicating the IP addresses were added successfully.
Raises:
HTTPException: if an error occurs while adding the IP addresses.
"""
try:
cli_api.add_ip_address()
return DetailResponse(detail='IP addresses added successfully.')
@ -19,6 +28,18 @@ async def add_ip_api():
@router.post('/edit', response_model=DetailResponse)
async def edit_ip_api(body: EditInputBody):
"""
Edits the IP addresses in the .configs.env file.
Args:
body: An instance of EditInputBody containing the new IPv4 and/or IPv6 addresses.
Returns:
A DetailResponse with a message indicating the IP addresses were edited successfully.
Raises:
HTTPException: if an error occurs while editing the IP addresses.
"""
try:
if not body.ipv4 and not body.ipv6:
raise HTTPException(status_code=400, detail='Error: You must specify either ipv4 or ipv6')