mirror of
https://github.com/lrnselfreliance/wrolpi.git
synced 2026-10-26 19:01:39 -04:00
28 lines
1.1 KiB
Python
28 lines
1.1 KiB
Python
|
"""One helper for API -> Controller requests.
|
||
|
|
|
||
|
|
Existing call sites disagree on the base URL (wrolpi/common.py uses http://127.0.0.1, the zim
|
||
|
|
module uses the docker service name); this module resolves it once: the docker service name when
|
||
|
|
DOCKERIZED, else localhost, overridable with the CONTROLLER_URL environment variable.
|
||
|
|
"""
|
||
|
|
import os
|
||
|
|
from typing import Tuple
|
||
|
|
from urllib.parse import urlencode
|
||
|
|
|
||
|
|
from wrolpi.common import aiohttp_get, logger
|
||
|
|
from wrolpi.vars import DOCKERIZED
|
||
|
|
|
||
|
|
logger = logger.getChild(__name__)
|
||
|
|
|
||
|
|
CONTROLLER_URL = os.environ.get('CONTROLLER_URL') or ('http://controller' if DOCKERIZED else 'http://127.0.0.1')
|
||
|
|
|
||
|
|
|
||
|
|
async def controller_get(path: str, params: dict = None, timeout: int = 10) -> Tuple[int, dict]:
|
||
|
|
"""GET a Controller endpoint, returning (status, JSON body). Raises on connection errors."""
|
||
|
|
url = f'{CONTROLLER_URL}{path}'
|
||
|
|
if params:
|
||
|
|
params = {k: v for k, v in params.items() if v is not None}
|
||
|
|
if params:
|
||
|
|
url = f'{url}?{urlencode(params)}'
|
||
|
|
async with aiohttp_get(url, timeout=timeout) as response:
|
||
|
|
return response.status, await response.json()
|