Add pydantic as dependency. Make Language and Config classes inherit BaseModel.

Author: fkdosilovic

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "judge0"
-version = "0.0.1"
+version = "0.0.2dev"
 description = "The official Python library for Judge0."
 readme = "README.md"
 requires-python = ">=3.9"
@@ -25,7 +25,7 @@ classifiers = [
     "Topic :: Software Development :: Libraries :: Python Modules",
     "Typing :: Typed",
 ]
-dependencies = ["requests>=2.28.0,<3.0.0"]
+dependencies = ["requests>=2.28.0,<3.0.0", "pydantic>=2.0.0,<3.0.0"]
 
 [build-system]
 requires = ["setuptools>=70.0"]

src/judge0/base_types.py

@@ -2,6 +2,8 @@
 from enum import IntEnum
 from typing import Optional, Protocol, runtime_checkable, Sequence, Union
 
+from pydantic import BaseModel
+
 Iterable = Sequence
 
 TestCaseType = Union["TestCase", list, tuple, dict]
@@ -15,6 +17,7 @@ class TestCase:
 
     @staticmethod
     def from_record(test_case: Optional[TestCaseType] = None) -> "TestCase":
+        """Create a TestCase from built-in types."""
         if isinstance(test_case, (tuple, list)):
             test_case = {
                 field: value
@@ -39,8 +42,7 @@ def encode(self) -> bytes:
         ...
 
 
-@dataclass(frozen=True)
-class Language:
+class Language(BaseModel):
     id: int
     name: str
 
@@ -85,8 +87,9 @@ def __str__(self):
         return self.name.lower().replace("_", " ").title()
 
 
-@dataclass(frozen=True)
-class Config:
+class Config(BaseModel):
+    """Client config data."""
+
     allow_enable_network: bool
     allow_enable_per_process_and_thread_memory_limit: bool
     allow_enable_per_process_and_thread_time_limit: bool