hh-applicant-tool 1.4.12__py3-none-any.whl → 1.5.7__py3-none-any.whl

hh_applicant_tool/utils/mixins.py

@@ -11,7 +11,6 @@ from typing import TYPE_CHECKING, Literal
 import requests
 from requests.exceptions import RequestException
 
-from ..ai.openai import ChatOpenAI
 from . import binpack
 from .log import collect_traceback_logs
 
@@ -38,9 +37,7 @@ class ErrorReporter:
         error_logs = ""
         if self.log_file.exists():
             with self.log_file.open(encoding="utf-8", errors="ignore") as fp:
-                error_logs = collect_traceback_logs(
-                    fp, last_report, maxlen=10000
-                )
+                error_logs = collect_traceback_logs(fp, last_report)
 
         # Эти данные нужны для воспроизведения ошибок. Среди них ваших
         # персональных данных нет
@@ -49,7 +46,7 @@ class ErrorReporter:
             for c in self.storage.vacancy_contacts.find(
                 updated_at__ge=last_report
             )
-        ][-10000:]
+        ]
 
         for c in vacancy_contacts:
             c.pop("id", 0)
@@ -71,7 +68,7 @@ class ErrorReporter:
                 ]
             }
             for emp in self.storage.employers.find(updated_at__ge=last_report)
-        ][-10000:]
+        ]
 
         vacancies = [
             {
@@ -94,7 +91,7 @@ class ErrorReporter:
                 ]
             }
             for vac in self.storage.vacancies.find(updated_at__ge=last_report)
-        ][-10000:]
+        ]
 
         # log.info("num vacncies: %d", len(vacancies))
 
@@ -106,10 +103,10 @@ class ErrorReporter:
         }
 
         return dict(
-            error_logs=error_logs,
-            vacancy_contacts=vacancy_contacts,
-            employers=employers,
-            vacancies=vacancies,
+            error_logs=error_logs[-100000:],
+            vacancy_contacts=vacancy_contacts[-10000:],
+            employers=employers[-10000:],
+            vacancies=vacancies[-10000:],
             package_version=get_package_version(),
             system_info=system_info,
             report_created=datetime.now(timezone.utc),
@@ -194,23 +191,7 @@ class VersionChecker:
                 )
 
 
-class ChatOpenAISupport:
-    def get_openai_chat(
-        self: HHApplicantTool,
-        system_prompt: str,
-    ) -> ChatOpenAI:
-        c = self.config.get("openai", {})
-        if not (token := c.get("token")):
-            raise ValueError("Токен для OpenAI не задан")
-        return ChatOpenAI(
-            token=token,
-            model=c.get("model", "gpt-5.1"),
-            system_prompt=system_prompt,
-            session=self.session,
-        )
-
-
-class MegaTool(ErrorReporter, VersionChecker, ChatOpenAISupport):
+class MegaTool(ErrorReporter, VersionChecker):
     def _check_system(self: HHApplicantTool):
         if not self.storage.settings.get_value("disable_version_check", False):
             self._check_version()

hh_applicant_tool/utils/string.py

@@ -4,9 +4,11 @@ import random
 import re
 from typing import Any
 
+
 def shorten(s: str, limit: int = 75, ellipsis: str = "…") -> str:
     return s[:limit] + bool(s[limit:]) * ellipsis
 
+
 def rand_text(s: str) -> str:
     while (
         temp := re.sub(
@@ -20,8 +22,21 @@ def rand_text(s: str) -> str:
         s = temp
     return s
 
+
 def bool2str(v: bool) -> str:
     return str(v).lower()
 
+
 def list2str(items: list[Any] | None) -> str:
     return ",".join(f"{v}" for v in items) if items else ""
+
+
+def unescape_string(text: str) -> str:
+    if not text:
+        return ""
+    return (
+        text.replace(r"\n", "\n")
+        .replace(r"\r", "\r")
+        .replace(r"\t", "\t")
+        .replace(r"\\", "\\")
+    )

hh_applicant_tool/utils/terminal.py

@@ -1,8 +1,22 @@
+from __future__ import annotations
+
 import base64
 import ctypes
+import io
+import os
 import platform
 import sys
 
+try:
+    from PIL import Image
+except ImportError:
+
+    class Image:
+        pass
+
+
+ESC = "\x1b"
+
 
 def setup_terminal() -> None:
     if platform.system() != "Windows":
@@ -30,3 +44,91 @@ def print_kitty_image(data: bytes) -> None:
     sys.stdout.write(f"\033_Ga=T,f=100;{b64data}\033\\")
     sys.stdout.flush()
     print()
+
+
+def print_sixel_mage(image_bytes: bytes) -> None:
+    img = Image.open(io.BytesIO(image_bytes))
+
+    # Рекомендуется оставить ограничение размера,
+    # иначе Zellij может "лагать" на огромных картинках
+    # max_size = (800, 600)
+    # img.thumbnail(max_size, Image.Resampling.LANCZOS)
+    img = img.convert("RGB")
+
+    try:
+        img = img.quantize(colors=256, method=Image.Quantize.MAXCOVERAGE)
+    except Exception:
+        img = img.quantize(colors=256)
+
+    palette = img.getpalette()[: 256 * 3]
+    width, height = img.size
+    pixels = img.load()
+
+    is_multiplexer = "ZELLIJ" in os.environ or "TMUX" in os.environ
+
+    # Собираем всё в список строк, чтобы минимизировать количество вызовов write
+    out = []
+
+    # 1. Начало (Обертка для Zellij)
+    if is_multiplexer:
+        out.append(f"{ESC}P+p")
+
+    # 2. Sixel заголовок + Растр
+    out.append(f'{ESC}Pq"1;1;{width};{height}')
+
+    # 3. Палитра
+    for i in range(256):
+        r, g, b = palette[i * 3 : i * 3 + 3]
+        out.append(f"#{i};2;{r * 100 // 255};{g * 100 // 255};{b * 100 // 255}")
+
+    # 4. Отрисовка
+    for y in range(0, height, 6):
+        h_chunk = min(6, height - y)
+
+        # Считаем уникальные цвета в полосе (быстрее, чем перебирать всю палитру)
+        colors_in_band = set()
+        for dy in range(h_chunk):
+            for x in range(width):
+                colors_in_band.add(pixels[x, y + dy])
+
+        for color in colors_in_band:
+            out.append(f"#{color}")
+            last_char = ""
+            count = 0
+
+            for x in range(width):
+                bits = 0
+                for dy in range(h_chunk):
+                    if pixels[x, y + dy] == color:
+                        bits |= 1 << dy
+
+                char = chr(63 + bits)
+                if char == last_char:
+                    count += 1
+                else:
+                    if count > 0:
+                        out.append(
+                            f"!{count}{last_char}"
+                            if count > 3
+                            else last_char * count
+                        )
+                    last_char, count = char, 1
+
+            if count > 0:
+                out.append(
+                    f"!{count}{last_char}" if count > 3 else last_char * count
+                )
+            out.append("$")
+        out.append("-")
+
+    # 5. Конец Sixel
+    out.append(f"{ESC}\\")
+
+    # 6. Конец обертки мультиплексора
+    if is_multiplexer:
+        out.append(f"{ESC}\\")
+
+    # ВЫВОД: Одной строкой БЕЗ лишних переносов в конце
+    sys.stdout.write("".join(out))
+    sys.stdout.flush()
+    print()