From e9a780496f7ce067e0d8d51ce1d62e48c9f2a8d9 Mon Sep 17 00:00:00 2001 From: HsiangNianian Date: Thu, 19 Sep 2024 14:44:33 +0800 Subject: feat(core): Implement Service class and related functionalities Co-authored-by: yuzhe --- hrc/config.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) (limited to 'hrc/config.py') diff --git a/hrc/config.py b/hrc/config.py index b6458b2..d179258 100644 --- a/hrc/config.py +++ b/hrc/config.py @@ -1,4 +1,4 @@ -from typing import Set, Union +from typing import Literal, Optional, Set, Union from pydantic import BaseModel, ConfigDict, DirectoryPath, Field @@ -14,16 +14,21 @@ class LogConfig(ConfigModel): verbose_exception: bool = False +class ServiceConfig(ConfigModel): + """Service configuration.""" + + class CoreConfig(ConfigModel): rules: Set[str] = Field(default_factory=set) rule_dirs: Set[DirectoryPath] = Field(default_factory=set) log: LogConfig = LogConfig() - + services: Set[str] = Field(default_factory=set) class RuleConfig(ConfigModel): - """rule configuration.""" - - + """Rule configuration.""" + + class MainConfig(ConfigModel): core: CoreConfig = CoreConfig() rule: RuleConfig = RuleConfig() + service: ServiceConfig = ServiceConfig() -- cgit v1.2.3-70-g09d2