One Hat Cyber Team

Your IP: 216.73.216.144
Server IP: 157.245.143.252
Server: Linux www 6.11.0-9-generic #9-Ubuntu SMP PREEMPT_DYNAMIC Mon Oct 14 13:19:59 UTC 2024 x86_64
Server Software: nginx/1.26.0
PHP Version: 8.3.11
Buat File | Buat Folder

Dir : ~/lib/python3/dist-packages/uaclient/api/

View File Name : data_types.py

from typing import Any, Dict, List, Union  # noqa: F401

from uaclient.data_types import DataObject, Field, StringDataValue, data_list
from uaclient.util import get_pro_environment
from uaclient.version import get_version


class AdditionalInfo:
    meta = {}  # type: Dict[str, Any]
    warnings = []  # type: List[ErrorWarningObject]


class ErrorWarningObject(DataObject):
    fields = [
        Field("title", StringDataValue),
        Field("code", StringDataValue),
        Field("meta", DataObject),
    ]

    def __init__(self, *, title: str, code: str, meta: dict):
        self.title = title
        self.code = code
        self.meta = meta


class APIData(DataObject):
    fields = [
        Field("type", StringDataValue),
        Field("attributes", DataObject),
        Field("meta", DataObject),
    ]

    def __init__(self, *, type: str, attributes: DataObject, meta: dict):
        self.type = type
        self.attributes = attributes
        self.meta = {
            "environment_vars": [
                {"name": name, "value": value}
                for name, value in sorted(get_pro_environment().items())
            ],
            **meta,
        }


class APIResponse(DataObject):
    fields = [
        Field("_schema_version", StringDataValue),
        Field("result", StringDataValue),
        Field("version", StringDataValue),
        Field("errors", data_list(ErrorWarningObject)),
        Field("warnings", data_list(ErrorWarningObject)),
        Field("data", APIData),
    ]

    def __init__(
        self,
        *,
        _schema_version: str,
        result: str = "success",
        errors: List[ErrorWarningObject] = [],
        warnings: List[ErrorWarningObject] = [],
        data: Union[APIData, dict]
    ):
        self._schema_version = _schema_version
        self.result = result
        self.version = get_version()
        self.errors = errors
        self.warnings = warnings
        self.data = data