1a978b79f0
* add websockets lib, bump version Signed-off-by: Jess Frazelle <github@jessfraz.com> * use template Signed-off-by: Jess Frazelle <github@jessfraz.com> * fixes Signed-off-by: Jess Frazelle <github@jessfraz.com> * I have generated the latest API! --------- Signed-off-by: Jess Frazelle <github@jessfraz.com> Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
64 lines
1.6 KiB
Python
64 lines
1.6 KiB
Python
from typing import Any, Dict
|
|
|
|
from websockets.client import WebSocketClientProtocol, connect as ws_connect_async
|
|
from websockets.sync.client import ClientConnection, connect as ws_connect
|
|
|
|
from ...client import Client
|
|
from ...models.error import Error
|
|
|
|
|
|
def _get_kwargs(
|
|
*,
|
|
client: Client,
|
|
) -> Dict[str, Any]:
|
|
url = "{}/ws/executor/term".format(client.base_url) # noqa: E501
|
|
|
|
headers: Dict[str, Any] = client.get_headers()
|
|
cookies: Dict[str, Any] = client.get_cookies()
|
|
|
|
return {
|
|
"url": url,
|
|
"headers": headers,
|
|
"cookies": cookies,
|
|
"timeout": client.get_timeout(),
|
|
}
|
|
|
|
|
|
def sync(
|
|
*,
|
|
client: Client,
|
|
) -> ClientConnection:
|
|
"""Attach to a docker container to create an interactive terminal.""" # noqa: E501
|
|
|
|
kwargs = _get_kwargs(
|
|
client=client,
|
|
)
|
|
|
|
with ws_connect(
|
|
kwargs["url"].replace("https://", "wss://"),
|
|
additional_headers=kwargs["headers"],
|
|
) as websocket:
|
|
return websocket # type: ignore
|
|
|
|
# Return an error if we got here.
|
|
return Error(message="An error occurred while connecting to the websocket.")
|
|
|
|
|
|
async def asyncio(
|
|
*,
|
|
client: Client,
|
|
) -> WebSocketClientProtocol:
|
|
"""Attach to a docker container to create an interactive terminal.""" # noqa: E501
|
|
|
|
kwargs = _get_kwargs(
|
|
client=client,
|
|
)
|
|
|
|
async with ws_connect_async(
|
|
kwargs["url"].replace("https://", "wss://"), extra_headers=kwargs["headers"]
|
|
) as websocket:
|
|
return websocket
|
|
|
|
# Return an error if we got here.
|
|
return Error(message="An error occurred while connecting to the websocket.")
|