Source code for codegrade.models.io_test_extra

"""The module that defines the ``IOTestExtra`` model.

SPDX-License-Identifier: AGPL-3.0-only OR BSD-3-Clause-Clear
"""

import typing as t
from dataclasses import dataclass, field

import cg_request_args as rqa

from .. import parsers
from ..utils import to_dict
from .io_test_data import IOTestData


[docs]@dataclass class IOTestExtra: """The extra data for an IOTest.""" #: This is an IO Test. type: "t.Literal['io_test']" #: The data for this IO test. data: "IOTestData" raw_data: t.Optional[t.Dict[str, t.Any]] = field(init=False, repr=False) data_parser: t.ClassVar = rqa.Lazy( lambda: rqa.FixedMapping( rqa.RequiredArgument( "type", rqa.StringEnum("io_test"), doc="This is an IO Test.", ), rqa.RequiredArgument( "data", parsers.ParserFor.make(IOTestData), doc="The data for this IO test.", ), ).use_readable_describe(True) ) def to_dict(self) -> t.Dict[str, t.Any]: res: t.Dict[str, t.Any] = { "type": to_dict(self.type), "data": to_dict(self.data), } return res @classmethod def from_dict( cls: t.Type["IOTestExtra"], d: t.Dict[str, t.Any] ) -> "IOTestExtra": parsed = cls.data_parser.try_parse(d) res = cls( type=parsed.type, data=parsed.data, ) res.raw_data = d return res