refactor: use model_validator

Closes #74

Author: Ljzd-PRO

ktoolbox/configuration.py

@@ -1,11 +1,12 @@
 import datetime
 import logging
 import os
+import tempfile
 from pathlib import Path
 from typing import Literal, Union, Optional
 
 from loguru import logger
-from pydantic import BaseModel
+from pydantic import BaseModel, model_validator
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
 __all__ = [
@@ -72,6 +73,23 @@ class DownloaderConfiguration(BaseModel):
     use_bucket: bool = False
     bucket_path: Path = Path("./.ktoolbox/bucket_storage")
 
+    @model_validator(mode="after")
+    def check_bucket_path(self) -> "DownloaderConfiguration":
+        if self.use_bucket:
+            # noinspection PyBroadException
+            try:
+                bucket_path = Path(self.bucket_path)
+                bucket_path.mkdir(parents=True, exist_ok=True)
+                with tempfile.TemporaryFile(dir=bucket_path) as temp_file:
+                    temp_link_file_path = f"{bucket_path / temp_file.name}.hlink"
+                    os.link(temp_file.name, temp_link_file_path)
+                    os.remove(temp_link_file_path)
+            except Exception:
+                self.use_bucket = False
+                logger.exception(f"`DownloaderConfiguration.bucket_path` is not available, "
+                                 f"`DownloaderConfiguration.use_bucket` has been disabled.")
+        return self
+
 
 class PostStructureConfiguration(BaseModel):
     # noinspection SpellCheckingInspection
@@ -164,23 +182,3 @@ class Configuration(BaseSettings):
 
 
 config = Configuration(_env_file='prod.env')
-
-
-def config_check_bucket():
-    if config.downloader.use_bucket:
-        import tempfile
-        # noinspection PyBroadException
-        try:
-            bucket_path = Path(config.downloader.bucket_path)
-            bucket_path.mkdir(parents=True, exist_ok=True)
-            with tempfile.TemporaryFile(dir=bucket_path) as temp_file:
-                temp_link_file_path = f"{bucket_path / temp_file.name}.hlink"
-                os.link(temp_file.name, temp_link_file_path)
-                os.remove(temp_link_file_path)
-        except Exception:
-            config.downloader.use_bucket = False
-            logger.exception(f"`DownloaderConfiguration.bucket_path` is not available, "
-                             f"`DownloaderConfiguration.use_bucket` has been disabled.")
-
-
-config_check_bucket()