Create PDF certificates

Author: coder8080

Dockerfile

@@ -1,5 +1,7 @@
 FROM pypy:3
 
+RUN apt update && apt install inkscape -y
+
 WORKDIR /usr/src/app
 
 COPY requirements.txt .

Dockerfile.dev

@@ -1,9 +1,11 @@
 FROM pypy:3
 
+RUN apt update && apt install inkscape -y
+RUN pip install watchdog[watchmedo] --no-cache-dir
+
 WORKDIR /usr/src/app
 
 COPY requirements.txt .
 RUN pip install -r requirements.txt --no-cache-dir
-RUN pip install watchdog[watchmedo] --no-cache-dir
 
 CMD watchmedo auto-restart --patterns="*.py" --recursive --interval 5 -- pypy -u src/main.py

src/entities/constants.py

@@ -50,4 +50,5 @@
     "create-link-cancelled": "👌 Создание ссылки отменено",
     "create-cert-cancelled": "👌 Создание сертификата отменено",
     "check-cert-cancelled": "👌 Проверка сертификата отменена",
+    "start-create-cert": "Генерирую сертификаты. Это может занять ⏱️ некоторое время. Пожалуйста, подождите...",
 }

src/entities/utils.py

@@ -1,5 +1,3 @@
-import subprocess
-
 from aiogram.types import ReplyKeyboardMarkup, Update
 from aiogram.utils.keyboard import ReplyKeyboardBuilder
 
@@ -56,25 +54,3 @@ async def log_error(message: str):
     logger.error(message)
     for admin_chat_id in ADMINS:
         await bot.send_message(chat_id=admin_chat_id, text=message)
-
-
-def replaceInDocument(FIO: str, dateTill: str, pdfFileName: str) -> None:
-    with open("src/static/certificate.svg", "r", encoding="utf-8") as file:
-        svg_content = file.read()
-    svg_content = svg_content.replace("fioHandler", FIO)
-    svg_content = svg_content.replace("dateHandler", dateTill)
-
-    temp_svg_path = "temp_certificate.svg"
-    with open(temp_svg_path, "w", encoding="utf-8") as f:
-        f.write(svg_content)
-    subprocess.run(
-        [
-            "inkscape",
-            temp_svg_path,
-            "--export-type=pdf",
-            f"--export-filename={pdfFileName}",
-        ],
-        check=True,
-    )
-
-    logger.info(f"PDF успешно создан: {pdfFileName}")

src/models/cert.py

@@ -1,4 +1,7 @@
+import subprocess
+from datetime import datetime, timedelta
 from random import randint
+from typing import cast
 
 from peewee import BooleanField, TextField
 
@@ -9,16 +12,42 @@ def cert_code():
     return "".join([str(randint(1, 9)) for _ in range(8)])
 
 
+async def replace_in_document(
+    name: str, expires: str, result_filename: str, temp_filename: str
+) -> None:
+    with open("src/static/certificate.svg", "r", encoding="utf-8") as file:
+        svg_content = file.read()
+    svg_content = svg_content.replace("fioHandler", name)
+    svg_content = svg_content.replace("dateHandler", expires)
+
+    with open(temp_filename, "w", encoding="utf-8") as f:
+        f.write(svg_content)
+    subprocess.run(
+        [
+            "inkscape",
+            temp_filename,
+            "--export-type=pdf",
+            f"--export-filename={result_filename}",
+        ],
+        check=True,
+    )
+
+
 class Cert(BaseModel):
     name = TextField()
     code = TextField(default=cert_code)
     used = BooleanField(default=False)
 
-    def generate_file(self):
-        path = f"certs/{self.code}.txt"
-        f = open(path, "w")
-        print(self.name, self.code, file=f)
-        f.close()
+    async def generate_file(self):
+        path = f"certs/{self.code}.pdf"
+        temp_path = f"certs/{self.code}.svg"
+
+        expire = datetime.now() + timedelta(days=365)
+        expire_str = f"{expire.year}/{expire.month}/{expire.day}"
+
+        await replace_in_document(
+            cast(str, self.name), expire_str, path, temp_path
+        )
         return path
 
 

src/routers/create_cert.py

@@ -41,9 +41,11 @@ async def create_cert(message: Message, user: User, state: FSMContext):
 
     await state.set_state(None)
 
+    await message.answer(TEXT["start-create-cert"])
+
     for name in names:
         cert = await Cert.aio_create(name=name)
-        path = cert.generate_file()
+        path = await cert.generate_file()
         await message.answer_document(FSInputFile(path))
 
     if user.chat_id not in ADMINS: