Viewing file:  v1.py (1.33 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
from uaclient.api.api import APIEndpoint
from uaclient.api.data_types import AdditionalInfo
from uaclient.config import UAConfig
from uaclient.contract import UAContractClient
from uaclient.data_types import (
DataObject,
Field,
IntDataValue,
StringDataValue,
)
class MagicAttachInitiateResult(DataObject, AdditionalInfo):
fields = [
Field("user_code", StringDataValue),
Field("token", StringDataValue),
Field("expires", StringDataValue),
Field("expires_in", IntDataValue),
]
def __init__(
self,
user_code: str,
token: str,
expires: str,
expires_in: int,
):
self.user_code = user_code
self.token = token
self.expires = expires
self.expires_in = expires_in
def initiate() -> MagicAttachInitiateResult:
return _initiate(UAConfig())
def _initiate(cfg: UAConfig) -> MagicAttachInitiateResult:
contract = UAContractClient(cfg)
initiate_resp = contract.new_magic_attach_token()
return MagicAttachInitiateResult(
user_code=initiate_resp["userCode"],
token=initiate_resp["token"],
expires=initiate_resp["expires"],
expires_in=int(initiate_resp["expiresIn"]),
)
endpoint = APIEndpoint(
version="v1",
name="MagicAttachInitiate",
fn=_initiate,
options_cls=None,
)
|