Add new file

2026-06-30 09:08:26 +00:00
parent 33f4f18f00
commit d39757ff65
1 changed files with 132 additions and 0 deletions
--- a/core/commands/template.py
+++ b/core/commands/template.py
@@ -0,0 +1,132 @@
+# -*- coding: utf-8 -*-
+import os
+import hashlib
+import shutil
+import logging
+from core.commands.base import Command
+
+logger = logging.getLogger(__name__)
+
+
+class TemplateSaveCommand(Command):
+    name = "!template_save"
+    aliases = ["/template_save"]
+
+    async def execute(self, msg, args: str, user_jid: str, room_jid: str = None):
+        if not room_jid:
+            reply = msg.reply("⚠️ Команда !template_save работает только в комнатах.")
+            if reply:
+                reply.send()
+            return
+
+        if args.startswith(self.name):
+            args = args[len(self.name):].lstrip()
+        name = args.strip()
+        if name.startswith('"') and name.endswith('"'):
+            name = name[1:-1]
+        if not name:
+            reply = msg.reply("Укажите название шаблона: !template_save \"Название\"")
+            if reply:
+                reply.send()
+            return
+
+        key = (user_jid, room_jid) if room_jid is not None else user_jid
+        original_path = self.bot._last_file_path.get(key)
+        if not original_path or not os.path.exists(original_path):
+            reply = msg.reply("❌ Нет загруженного файла. Сначала отправьте DOCX-шаблон.")
+            if reply:
+                reply.send()
+            return
+
+        if not original_path.lower().endswith('.docx'):
+            reply = msg.reply("❌ Шаблон должен быть в формате .docx.")
+            if reply:
+                reply.send()
+            return
+
+        with open(original_path, 'rb') as f:
+            file_hash = hashlib.sha256(f.read()).hexdigest()
+
+        templates_dir = self.bot.config.data_dir / "templates"
+        templates_dir.mkdir(parents=True, exist_ok=True)
+        safe_room = room_jid.replace('/', '_').replace('@', '_')
+        dest_name = f"{safe_room}_{name}.docx"
+        dest_path = templates_dir / dest_name
+        shutil.copy2(original_path, dest_path)
+
+        await self.bot.db.save_template(room_jid, name, str(dest_path), file_hash)
+
+        reply = msg.reply(f"✅ Шаблон '{name}' сохранён.")
+        if reply:
+            reply.send()
+
+    def get_help(self) -> str:
+        return (
+            "Сохраняет текущий DOCX-файл как шаблон с указанным именем. "
+            "Использование: `!template_save \"Имя шаблона\"` (только в комнате)"
+        )
+
+
+class TemplateListCommand(Command):
+    name = "!template_list"
+    aliases = ["/template_list"]
+
+    async def execute(self, msg, args: str, user_jid: str, room_jid: str = None):
+        if not room_jid:
+            reply = msg.reply("⚠️ Команда !template_list работает только в комнатах.")
+            if reply:
+                reply.send()
+            return
+
+        if args.startswith(self.name):
+            args = args[len(self.name):].lstrip()
+        templates = await self.bot.db.list_templates(room_jid)
+        if not templates:
+            reply = msg.reply("📭 В этой комнате нет сохранённых шаблонов.")
+        else:
+            lines = ["📋 Шаблоны комнаты:"] + [f"  • {t}" for t in templates]
+            reply = msg.reply("\n".join(lines))
+        if reply:
+            reply.send()
+
+    def get_help(self) -> str:
+        return (
+            "Показывает список сохранённых шаблонов в текущей комнате. "
+            "Использование: `!template_list` (только в комнате)"
+        )
+
+
+class TemplateDeleteCommand(Command):
+    name = "!template_delete"
+    aliases = ["/template_delete"]
+
+    async def execute(self, msg, args: str, user_jid: str, room_jid: str = None):
+        if not room_jid:
+            reply = msg.reply("⚠️ Команда !template_delete работает только в комнатах.")
+            if reply:
+                reply.send()
+            return
+
+        if args.startswith(self.name):
+            args = args[len(self.name):].lstrip()
+        name = args.strip()
+        if name.startswith('"') and name.endswith('"'):
+            name = name[1:-1]
+        if not name:
+            reply = msg.reply("Укажите название шаблона: !template_delete \"Название\"")
+            if reply:
+                reply.send()
+            return
+
+        if await self.bot.db.delete_template(room_jid, name):
+            reply = msg.reply(f"🗑 Шаблон '{name}' удалён.")
+        else:
+            reply = msg.reply(f"❌ Шаблон '{name}' не найден.")
+        if reply:
+            reply.send()
+
+    def get_help(self) -> str:
+        return (
+            "Удаляет шаблон по имени. "
+            "Использование: `!template_delete \"Имя шаблона\"` (только в комнате)"
+        )