From 2827c09958aa6778e4499d34f5949d6f5677f2c6 Mon Sep 17 00:00:00 2001 From: 简律纯 Date: Fri, 28 Jun 2024 11:33:07 +0800 Subject: refactor: hrc.rules - > hrc.rule --- hrc/config.py | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) (limited to 'hrc/config.py') diff --git a/hrc/config.py b/hrc/config.py index e69de29..7db4efe 100644 --- a/hrc/config.py +++ b/hrc/config.py @@ -0,0 +1,28 @@ +from typing import Set, Union + +from pydantic import BaseModel, ConfigDict, DirectoryPath, Field + +class ConfigModel(BaseModel): + model_config = ConfigDict(extra="allow") + + __config_name__: str = "" + + +class LogConfig(ConfigModel): + level: Union[str, int] = "DEBUG" + verbose_exception: bool = False + + +class CoreConfig(ConfigModel): + rules: Set[str] = Field(default_factory=set) + rule_dirs: Set[DirectoryPath] = Field(default_factory=set) + log: LogConfig = LogConfig() + + +class RuleConfig(ConfigModel): + """rule configuration.""" + + +class MainConfig(ConfigModel): + core: CoreConfig = CoreConfig() + rule: RuleConfig = RuleConfig() \ No newline at end of file -- cgit v1.2.3-70-g09d2