1 files changed, 69 insertions, 0 deletions

diff --git a/shared/settings.py b/shared/settings.py
new file mode 100644
index 0000000..20f3f76
--- /dev/null
+++ b/shared/settings.py
@@ -0,0 +1,69 @@
+from json import dump, load
+
+from pydantic import BaseModel, BaseSettings, Field
+
+
+class Settings(BaseSettings):
+    token: str
+
+
+class GenConfig(BaseModel):
+    chance: int = Field(
+        10,
+        description="Шанс с которым бот ответит на сообщение",
+        ge=1,
+        le=100,
+    )
+    min_word_count: int | str | None = Field(
+        None,
+        description="Минимальное количество слов в сгенерированном предложении",
+        ge=1,
+    )
+    max_word_count: int | None = Field(
+        None,
+        description="Максимальное количество слов в сгенерированном предложении",
+        ge=1,
+    )
+
+
+class CommandsConfig(BaseModel):
+    pin_answers_count: int = Field(
+        4,
+        description="Минимальное количество голосов для проверки опроса на закрепление сообщения",
+    )
+    accept_member_answers_count: int = Field(
+        5,
+        description="Минимальное количество голосов для проверки опроса на принятия человека в группу",
+    )
+
+
+class Config(BaseModel):
+    gen: GenConfig = Field(
+        GenConfig(),
+        description="Настройки генерации сообщений",
+    )
+    commands: CommandsConfig = Field(
+        CommandsConfig(),
+        description="Настройки команд бота",
+    )
+
+
+class Chats(BaseModel):
+    chats: dict[int, Config] = {}
+
+    @classmethod
+    def load(cls, file_name: str) -> "Chats":
+        with open(file_name, "r") as file:
+            return cls.parse_obj(load(file))
+
+    def save(self, file_name: str) -> None:
+        with open(file_name, "w") as file:
+            dump(self.schema(), file)
+
+    def get_config(self, chat_id: int) -> Config:
+        if chat_id not in self.chats:
+            self.chats[chat_id] = Config()
+        return self.chats[chat_id]
+
+    def set_config(self, chat_id: int, config: Config) -> None:
+        self.chats[chat_id] = config