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

hh_applicant_tool/__main__.py

@@ -1,6 +1,6 @@
 import sys
 
-from . import main
+from .main import main
 
 if __name__ == "__main__":
     sys.exit(main())

hh_applicant_tool/ai/__init__.py

@@ -0,0 +1 @@
+from .openai import ChatOpenAI, OpenAIError

hh_applicant_tool/ai/openai.py

@@ -1,6 +1,5 @@
 import logging
-from dataclasses import dataclass, field
-from typing import ClassVar
+from dataclasses import KW_ONLY, dataclass, field
 
 import requests
 
@@ -9,43 +8,60 @@ from .base import AIError
 logger = logging.getLogger(__package__)
 
 
+DEFAULT_COMPLETION_ENDPOINT = "https://api.openai.com/v1/chat/completions"
+
+
 class OpenAIError(AIError):
     pass
 
 
 @dataclass
 class ChatOpenAI:
-    chat_endpoint: ClassVar[str] = "https://api.openai.com/v1/chat/completions"
-
     token: str
-    model: str
+    _: KW_ONLY
     system_prompt: str | None = None
+    timeout: float = 15.0
     temperature: float = 0.7
     max_completion_tokens: int = 1000
+    model: str | None = None
+    completion_endpoint: str = None
     session: requests.Session = field(default_factory=requests.Session)
 
+    def __post_init__(self) -> None:
+        self.completion_endpoint = (
+            self.completion_endpoint or DEFAULT_COMPLETION_ENDPOINT
+        )
+
     def _default_headers(self) -> dict[str, str]:
         return {
             "Authorization": f"Bearer {self.token}",
         }
 
     def send_message(self, message: str) -> str:
+        messages = []
+
+        # Добавляем системный промпт только если он не пустой и не None
+        if self.system_prompt:
+            messages.append({"role": "system", "content": self.system_prompt})
+
+        # Пользовательское сообщение всегда обязательно
+        messages.append({"role": "user", "content": message})
+
         payload = {
-            "model": self.model,
-            "messages": [
-                {"role": "system", "content": self.system_prompt},
-                {"role": "user", "content": message},
-            ],
+            "messages": messages,
             "temperature": self.temperature,
             "max_completion_tokens": self.max_completion_tokens,
         }
 
+        if self.model:
+            payload["model"] = self.model
+
         try:
             response = self.session.post(
-                self.chat_endpoint,
+                self.completion_endpoint,
                 json=payload,
                 headers=self._default_headers(),
-                timeout=30,
+                timeout=self.timeout,
             )
             response.raise_for_status()
 

hh_applicant_tool/api/client.py

@@ -13,19 +13,20 @@ from urllib.parse import urlencode, urljoin
 import requests
 from requests import Session
 
+from hh_applicant_tool.api.user_agent import generate_android_useragent
+
 from . import errors
 from .client_keys import (
     ANDROID_CLIENT_ID,
     ANDROID_CLIENT_SECRET,
 )
 from .datatypes import AccessToken
-from .user_agent import generate_android_useragent
 
 __all__ = ("ApiClient", "OAuthClient")
 
 HH_API_URL = "https://api.hh.ru/"
 HH_OAUTH_URL = "https://hh.ru/oauth/"
-DEFAULT_DELAY = 0.334
+DEFAULT_DELAY = 0.345
 
 AllowedMethods = Literal["GET", "POST", "PUT", "DELETE"]
 T = TypeVar("T")
@@ -41,18 +42,21 @@ class BaseClient:
     _: dataclasses.KW_ONLY
     user_agent: str | None = None
     session: Session | None = None
-    delay: float = DEFAULT_DELAY
+    delay: float | None = None
     _previous_request_time: float = 0.0
 
     def __post_init__(self) -> None:
         assert self.base_url.endswith("/"), "base_url must ends with /"
-        self.lock = Lock()
+        self.delay = self.delay or DEFAULT_DELAY
+        self.user_agent = self.user_agent or generate_android_useragent()
+
         # logger.debug(f"user agent: {self.user_agent}")
+
         if not self.session:
             logger.debug("create new session")
             self.session = requests.session()
-        # if self.proxies:
-        #     logger.debug(f"client proxies: {self.proxies}")
+
+        self.lock = Lock()
 
     @property
     def proxies(self):
@@ -60,7 +64,7 @@ class BaseClient:
 
     def _default_headers(self) -> dict[str, str]:
         return {
-            "user-agent": self.user_agent or generate_android_useragent(),
+            "user-agent": self.user_agent,
             "x-hh-app-active": "true",
         }
 

hh_applicant_tool/main.py

@@ -17,8 +17,7 @@ from typing import Any, Iterable
 import requests
 import urllib3
 
-from . import utils
-from .api import ApiClient, datatypes
+from . import ai, api, utils
 from .storage import StorageFacade
 from .utils.log import setup_logger
 from .utils.mixins import MegaTool
@@ -29,7 +28,6 @@ DEFAULT_CONFIG_DIR = utils.get_config_path() / (__package__ or "").replace(
 DEFAULT_CONFIG_FILENAME = "config.json"
 DEFAULT_LOG_FILENAME = "log.txt"
 DEFAULT_DATABASE_FILENAME = "data"
-DEFAULT_PROFILE_ID = "."
 
 logger = logging.getLogger(__package__)
 
@@ -54,7 +52,6 @@ class BaseNamespace(argparse.Namespace):
     delay: float
     user_agent: str
     proxy_url: str
-    disable_telemetry: bool
 
 
 class HHApplicantTool(MegaTool):
@@ -89,19 +86,18 @@ class HHApplicantTool(MegaTool):
             "--config",
             help="Путь до директории с конфигом",
             type=Path,
-            default=DEFAULT_CONFIG_DIR,
+            default=None,
         )
         parser.add_argument(
             "--profile-id",
             "--profile",
-            help="Используемый профиль — подкаталог в --config-dir",
-            default=DEFAULT_PROFILE_ID,
+            help="Используемый профиль — подкаталог в --config-dir. Так же можно передать через переменную окружения HH_PROFILE_ID.",
         )
         parser.add_argument(
             "-d",
+            "--api-delay",
             "--delay",
             type=float,
-            default=0.654,
             help="Задержка между запросами к API HH по умолчанию",
         )
         parser.add_argument(
@@ -168,14 +164,20 @@ class HHApplicantTool(MegaTool):
         session.verify = False
 
         if proxies := self._get_proxies():
-            logger.info("Use proxies: %r", proxies)
+            logger.info("Use proxies for requests: %r", proxies)
             session.proxies = proxies
 
         return session
 
-    @property
+    @cached_property
     def config_path(self) -> Path:
-        return (self.args.config_dir / self.args.profile_id).resolve()
+        return (
+            (
+                self.args.config_dir
+                or Path(getenv("CONFIG_DIR", DEFAULT_CONFIG_DIR))
+            )
+            / (self.args.profile_id or getenv("HH_PROFILE_ID", "."))
+        ).resolve()
 
     @cached_property
     def config(self) -> utils.Config:
@@ -199,36 +201,35 @@ class HHApplicantTool(MegaTool):
         return StorageFacade(self.db)
 
     @cached_property
-    def api_client(self) -> ApiClient:
+    def api_client(self) -> api.client.ApiClient:
         args = self.args
         config = self.config
         token = config.get("token", {})
-        api = ApiClient(
+        return api.client.ApiClient(
             client_id=config.get("client_id"),
             client_secret=config.get("client_id"),
             access_token=token.get("access_token"),
             refresh_token=token.get("refresh_token"),
             access_expires_at=token.get("access_expires_at"),
-            delay=args.delay,
-            user_agent=config.get("user_agent"),
+            delay=args.api_delay or config.get("api_delay"),
+            user_agent=args.user_agent or config.get("user_agent"),
             session=self.session,
         )
-        return api
 
-    def get_me(self) -> datatypes.User:
+    def get_me(self) -> api.datatypes.User:
         return self.api_client.get("/me")
 
-    def get_resumes(self) -> list[datatypes.Resume]:
+    def get_resumes(self) -> list[api.datatypes.Resume]:
         return self.api_client.get("/resumes/mine")["items"]
 
     def first_resume_id(self) -> str:
-        resumes = self.get_resumes()
-        return resumes[0]["id"]
+        resume = self.get_resumes()[0]
+        return resume["id"]
 
     def get_blacklisted(self) -> list[str]:
         rv = []
         for page in count():
-            r: datatypes.PaginatedItems[datatypes.EmployerShort] = (
+            r: api.datatypes.PaginatedItems[api.datatypes.EmployerShort] = (
                 self.api_client.get("/employers/blacklisted", page=page)
             )
             rv += [item["id"] for item in r["items"]]
@@ -238,7 +239,7 @@ class HHApplicantTool(MegaTool):
 
     def get_negotiations(
         self, status: str = "active"
-    ) -> Iterable[datatypes.Negotiation]:
+    ) -> Iterable[api.datatypes.Negotiation]:
         for page in count():
             r: dict[str, Any] = self.api_client.get(
                 "/negotiations",
@@ -257,6 +258,8 @@ class HHApplicantTool(MegaTool):
             if page + 1 >= r.get("pages", 0):
                 break
 
+    # TODO: добавить еще методов или те удалить?
+
     def save_token(self) -> bool:
         if self.api_client.access_token != self.config.get("token", {}).get(
             "access_token"
@@ -265,6 +268,20 @@ class HHApplicantTool(MegaTool):
             return True
         return False
 
+    def get_openai_chat(self, system_prompt: str) -> ai.ChatOpenAI:
+        c = self.config.get("openai", {})
+        if not (token := c.get("token")):
+            raise ValueError("Токен для OpenAI не задан")
+        return ai.ChatOpenAI(
+            token=token,
+            model=c.get("model"),
+            temperature=c.get("temperature", 0.7),
+            max_completion_tokens=c.get("max_completion_tokens", 1000),
+            system_prompt=system_prompt,
+            completion_endpoint=c.get("completion_endpoint"),
+            session=self.session,
+        )
+
     def run(self) -> None | int:
         verbosity_level = max(
             logging.DEBUG,
@@ -273,19 +290,25 @@ class HHApplicantTool(MegaTool):
 
         setup_logger(logger, verbosity_level, self.log_file)
 
-        if sys.platform == "win32":
-            utils.setup_terminal()
+        logger.debug("Путь до профиля: %s", self.config_path)
+
+        utils.setup_terminal()
 
         try:
             if self.args.run:
                 try:
-                    res = self.args.run(self)
-                    if self.save_token():
-                        logger.info("Токен был обновлен.")
-                    return res
+                    return self.args.run(self)
                 except KeyboardInterrupt:
                     logger.warning("Выполнение прервано пользователем!")
-                    return 1
+                except api.errors.CaptchaRequired as ex:
+                    logger.error(f"Требуется ввод капчи: {ex.captcha_url}")
+                except api.errors.InternalServerError:
+                    logger.error(
+                        "Сервер HH.RU не смог обработать запрос из-за высокой"
+                        " нагрузки или по иной причине"
+                    )
+                except api.errors.Forbidden:
+                    logger.error("Требуется авторизация")
                 except sqlite3.Error as ex:
                     logger.exception(ex)
 
@@ -295,10 +318,13 @@ class HHApplicantTool(MegaTool):
                         f"Возможно база данных повреждена, попробуйте выполнить команду:\n\n"  # noqa: E501
                         f"  {script_name} migrate-db"
                     )
-                    return 1
                 except Exception as e:
                     logger.exception(e)
-                    return 1
+                finally:
+                    # Токен мог автоматически обновиться
+                    if self.save_token():
+                        logger.info("Токен был сохранен после обновления.")
+                return 1
             self._parser.print_help(file=sys.stderr)
             return 2
         finally:

hh_applicant_tool/operations/apply_similar.py

@@ -12,7 +12,13 @@ from ..api.datatypes import PaginatedItems, SearchVacancy
 from ..api.errors import ApiError, LimitExceeded
 from ..main import BaseNamespace, BaseOperation
 from ..storage.repositories.errors import RepositoryError
-from ..utils import bool2str, list2str, rand_text, shorten
+from ..utils.string import (
+    bool2str,
+    list2str,
+    rand_text,
+    shorten,
+    unescape_string,
+)
 
 if TYPE_CHECKING:
     from ..main import HHApplicantTool
@@ -77,7 +83,7 @@ class Operation(BaseOperation):
             "-L",
             "--message-list-path",
             "--message-list",
-            help="Путь до файла, где хранятся сообщения для отклика на вакансии. Каждое сообщение — с новой строки.",  # noqa: E501
+            help="Путь до файла, где хранятся сообщения для отклика на вакансии. Каждое сообщение — с новой строки. Символы \\n будут заменены на переносы.",  # noqa: E501
             type=Path,
         )
         parser.add_argument(
@@ -412,7 +418,7 @@ class Operation(BaseOperation):
                             resume["alternate_url"],
                         )
                         print(
-                            "⛔  Пришел отказ от",
+                            "⛔ Пришел отказ от",
                             vacancy["alternate_url"],
                             "на резюме",
                             resume["alternate_url"],
@@ -440,7 +446,7 @@ class Operation(BaseOperation):
                         logger.debug("prompt: %s", msg)
                         msg = self.openai_chat.send_message(msg)
                     else:
-                        msg = (
+                        msg = unescape_string(
                             rand_text(random.choice(self.application_messages))
                             % message_placeholders
                         )
@@ -487,9 +493,9 @@ class Operation(BaseOperation):
         params = {
             "page": page,
             "per_page": self.per_page,
-            "order_by": self.order_by,
         }
-
+        if self.order_by:
+            params |= {"order_by": self.order_by}
         if self.search:
             params["text"] = self.search
         if self.schedule:

hh_applicant_tool/operations/authorize.py

@@ -14,7 +14,7 @@ except ImportError:
     pass
 
 from ..main import BaseOperation
-from ..utils.terminal import print_kitty_image
+from ..utils.terminal import print_kitty_image, print_sixel_mage
 
 if TYPE_CHECKING:
     from ..main import HHApplicantTool
@@ -91,7 +91,14 @@ class Operation(BaseOperation):
             "--use-kitty",
             "--kitty",
             action="store_true",
-            help="Использовать kitty protocol для вывода изображения в терминал. Гуглите поддерживает ли ваш терминал его",
+            help="Использовать kitty protocol для вывода капчи в терминал.",
+        )
+        parser.add_argument(
+            "-s",
+            "--use-sixel",
+            "--sixel",
+            action="store_true",
+            help="Использовать sixel protocol для вывода капчи в терминал.",
         )
 
     def run(self, tool: HHApplicantTool) -> None:
@@ -152,7 +159,7 @@ class Operation(BaseOperation):
 
                 #      # Блокировка сканирования локальных портов
                 #      if any(d in url for d in ["localhost", "127.0.0.1", "::1"]):
-                #           logger.debug(f"🛑 Блокировка запроса на локальный порт: {url}")
+                #           logger.debug(f"🛑  Блокировка запроса на локальный порт: {url}")
                 #           return await route.abort()
 
                 #      # Оптимизация трафика в headless
@@ -201,8 +208,11 @@ class Operation(BaseOperation):
                     await page.fill(self.SELECT_LOGIN_INPUT, username)
                     logger.debug("Логин введен")
 
-                    if args.password:
-                        await self._direct_login(page, args.password)
+                    password = args.password or storage.settings.get_value(
+                        "auth.password"
+                    )
+                    if password:
+                        await self._direct_login(page, password)
                     else:
                         await self._onetime_code_login(page)
 
@@ -302,14 +312,11 @@ class Operation(BaseOperation):
             logger.debug("Капчи нет, продолжаем как обычно.")
             return
 
-        if not self._args.use_kitty:
+        if not (self._args.use_kitty or self._args.use_sixel):
             raise RuntimeError(
-                "Используйте флаг --use-kitty/-k для вывода капчи в терминал."
-                "Работает не во всех терминалах!",
+                "Требуется ввод капчи!",
             )
 
-        logger.info("Обнаружена капча!")
-
         # box = await captcha_element.bounding_box()
 
         # width = int(box["width"])
@@ -319,10 +326,15 @@ class Operation(BaseOperation):
 
         print(
             "Если вы не видите картинку ниже, то ваш терминал не поддерживает"
-            " kitty protocol."
+            " вывод изображений."
         )
         print()
-        print_kitty_image(img_bytes)
+
+        if self._args.use_kitty:
+            print_kitty_image(img_bytes)
+
+        if self._args.use_sixel:
+            print_sixel_mage(img_bytes)
 
         captcha_text = (await ainput("Введите текст с картинки: ")).strip()
 

hh_applicant_tool/operations/clear_negotiations.py

@@ -55,7 +55,6 @@ class Operation(BaseOperation):
 
     def clear(self) -> None:
         blacklisted = set(self.tool.get_blacklisted())
-        storage = self.tool.storage
         for negotiation in self.tool.get_negotiations():
             vacancy = negotiation["vacancy"]
 

hh_applicant_tool/operations/query.py

@@ -68,7 +68,7 @@ class Operation(BaseOperation):
                         writer.writerows(cursor.fetchall())
 
                         if tool.args.output:
-                            print(f"✅ Exported to {tool.args.output.name}")
+                            print(f"✅  Exported to {tool.args.output.name}")
                         return
 
                     rows = cursor.fetchmany(MAX_RESULTS + 1)
@@ -93,7 +93,7 @@ class Operation(BaseOperation):
                         print(f"Rows affected: {cursor.rowcount}")
 
             except sqlite3.Error as ex:
-                print(f"❌ SQL Error: {ex}")
+                print(f"❌  SQL Error: {ex}")
                 return 1
 
         if initial_sql := tool.args.sql:

hh_applicant_tool/operations/refresh_token.py

@@ -26,5 +26,12 @@ class Operation(BaseOperation):
         pass
 
     def run(self, tool: HHApplicantTool) -> None:
-        tool.api_client.refresh_access_token()
-        print("✅ Токен обновлен!")
+        if tool.api_client.is_access_expired:
+            tool.api_client.refresh_access_token()
+            if not tool.save_token():
+                print("⚠️ Токен не был обновлен!")
+                return 1
+            print("✅ Токен успешно обновлен.")
+        else:
+            # logger.debug("Токен валиден, игнорируем обновление.")
+            print("ℹ️ Токен не истек, обновление не требуется.")

hh_applicant_tool/operations/whoami.py

@@ -53,6 +53,6 @@ class Operation(BaseOperation):
         print(
             f"🆔 {result['id']} {full_name or 'Анонимный аккаунт'} "
             f"[ 📄 {counters['resumes_count']} "
-            f"| 👁️  {fmt_plus(counters['new_resume_views'])} "
-            f"| ✉️  {fmt_plus(counters['unread_negotiations'])} ]"
+            f"| 👁️ {fmt_plus(counters['new_resume_views'])} "
+            f"| ✉️ {fmt_plus(counters['unread_negotiations'])} ]"
         )

hh_applicant_tool/storage/models/base.py

@@ -13,12 +13,15 @@ MISSING = object()
 
 
 def mapped(
+    *,
+    skip_src: bool = False,
     path: str | None = None,
     transform: Callable[[Any], Any] | None = None,
     store_json: bool = False,
     **kwargs: Any,
 ):
     metadata = kwargs.get("metadata", {})
+    metadata.setdefault("skip_src", skip_src)
     metadata.setdefault("path", path)
     metadata.setdefault("transform", transform)
     metadata.setdefault("store_json", store_json)
@@ -89,6 +92,8 @@ class BaseModel:
         kwargs = {}
         for f in fields(cls):
             if from_source:
+                if f.metadata.get("skip_src") and f.name in data:
+                    continue
                 if path := f.metadata.get("path"):
                     found = True
                     v = data

hh_applicant_tool/storage/models/contacts.py

@@ -1,9 +1,15 @@
+import secrets
+
 from .base import BaseModel, mapped
 
 
 # Из вакансии извлекается
 class VacancyContactsModel(BaseModel):
-    id: int
+    # При вызове from_api на вакансии нужно игнорировать ее id
+    id: str = mapped(
+        skip_src=True,
+        default_factory=lambda: secrets.token_hex(16),
+    )
     vacancy_id: int = mapped(path="id")
 
     vacancy_name: str = mapped(path="name")
@@ -26,3 +32,11 @@ class VacancyContactsModel(BaseModel):
         ),
         default=None,
     )
+
+    def __post_init__(self):
+        self.vacancy_salary_from = (
+            self.vacancy_salary_from or self.vacancy_salary_to or 0
+        )
+        self.vacancy_salary_to = (
+            self.vacancy_salary_to or self.vacancy_salary_from or 0
+        )

hh_applicant_tool/storage/queries/schema.sql

@@ -16,7 +16,7 @@ CREATE TABLE IF NOT EXISTS employers (
 );
 /* ===================== contacts ===================== */
 CREATE TABLE IF NOT EXISTS vacancy_contacts (
-    id TEXT PRIMARY KEY DEFAULT (lower(hex(randomblob(16)))),
+    id TEXT PRIMARY KEY DEFAULT (lower(hex(randomblob(16)))) NOT NULL,
     vacancy_id INTEGER NOT NULL,
     -- Все это избыточные поля
     vacancy_alternate_url TEXT,

hh_applicant_tool/storage/utils.py

@@ -13,10 +13,16 @@ logger: logging.Logger = logging.getLogger(__package__)
 
 def init_db(conn: sqlite3.Connection) -> None:
     """Создает схему БД"""
+    changes_before = conn.total_changes
+
     conn.executescript(
         (QUERIES_PATH / "schema.sql").read_text(encoding="utf-8")
     )
-    logger.debug("Database initialized")
+
+    if conn.total_changes > changes_before:
+        logger.info("Применена схема бд")
+    # else:
+    #     logger.debug("База данных не изменилась.")
 
 
 def list_migrations() -> list[str]:

hh_applicant_tool/utils/log.py

@@ -121,9 +121,9 @@ TS_RE = re.compile(r"^\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}")
 def collect_traceback_logs(
     fp: TextIO,
     after_dt: datetime,
-    maxlen: int = 1000,
+    maxlines: int = 1000,
 ) -> str:
-    error_lines = deque(maxlen=maxlen)
+    error_lines = deque(maxlen=maxlines)
     prev_line = ""
     log_dt = None
     collecting_traceback = False