refactor: setup generate schema

.pre-commit-config.yaml

@@ -41,7 +41,8 @@ repos:
         name: Generate Config Schemas
         language: system
         entry: uv run generate-config-schema # Require venv activated
-        files: '^src/check_phat_nguoi/models/.*$'
+        files: '^src/check_phat_nguoi/models/'
         types: [file, python]
+        verbose: true
         stages:
           - pre-commit

README.md

@@ -25,6 +25,9 @@ uv sync # no need uv venv
 uv run pre-commit install
 
 uv run check-phat-nguoi
+uv run generate-schema
+
+# Just for single schema generation
 uv run generate-config-schema
 ```
 

pyproject.toml

@@ -11,7 +11,8 @@ dependencies = ["pydantic>=2.10.4", "requests>=2.32.3"]
 
 [project.scripts]
 check-phat-nguoi = "check_phat_nguoi:main"
-generate-config-schema = "generate_config_schema:main"
+generate-config-schema = "generate_schemas:generate_config_schema"
+generate-schema = "generate_schemas:main"
 
 [build-system]
 requires = ["hatchling"]

src/check_phat_nguoi/models/config/config.py

@@ -15,4 +15,7 @@ class ConfigModel(BaseModel):
     log_level: LogLevelModel = LogLevelModel.warning
 
 
+print("coo")
+
+
 __all__ = ["ConfigModel"]

src/generate_schemas/__init__.py

@@ -0,0 +1,9 @@
+from generate_schemas.modules import generate_config_schema
+
+
+def main() -> None:
+    print("Generating config schema...")
+    generate_config_schema()
+
+
+__all__ = ["generate_config_schema"]

src/generate_config_schema/__main__.py → src/generate_schemas/__main__.py

@@ -1,4 +1,4 @@
-from generate_config_schema import main
+from generate_schemas import main
 
 if __name__ == "__main__":
     main()

src/generate_schemas/modules/__init__.py

@@ -0,0 +1,3 @@
+from generate_schemas.modules.config import generate_config_schema
+
+__all__ = ["generate_config_schema"]

src/generate_schemas/modules/config.py

@@ -0,0 +1,13 @@
+from json import dumps
+
+from pydantic import TypeAdapter
+
+from check_phat_nguoi.models.config import ConfigModel
+from generate_schemas.utils.constant import CONFIG_SCHEMA_PATH
+
+
+def generate_config_schema():
+    adapter = TypeAdapter(ConfigModel)
+    with open(CONFIG_SCHEMA_PATH, "w", encoding="utf8") as file:
+        file.write(dumps(adapter.json_schema(), indent=2))
+    print(f"Created {CONFIG_SCHEMA_PATH} successfully!")

src/generate_schemas/utils/constant.py

@@ -0,0 +1 @@
+CONFIG_SCHEMA_PATH: str = "schemas/config.json"