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

hh_applicant_tool/operations/query.py

@@ -0,0 +1,122 @@
+from __future__ import annotations
+
+import argparse
+import csv
+import logging
+import sqlite3
+import sys
+from typing import TYPE_CHECKING
+
+from prettytable import PrettyTable
+
+from ..main import BaseNamespace, BaseOperation
+
+if TYPE_CHECKING:
+    from ..main import HHApplicantTool
+
+try:
+    import readline
+
+    readline.parse_and_bind("tab: complete")
+except ImportError:
+    readline = None
+
+MAX_RESULTS = 10
+
+
+logger = logging.getLogger(__package__)
+
+
+class Namespace(BaseNamespace):
+    pass
+
+
+class Operation(BaseOperation):
+    """Выполняет SQL-запрос. Поддерживает вывод в консоль или CSV файл."""
+
+    __aliases__: list[str] = ["sql"]
+
+    def setup_parser(self, parser: argparse.ArgumentParser) -> None:
+        parser.add_argument("sql", nargs="?", help="SQL запрос")
+        parser.add_argument(
+            "--csv", action="store_true", help="Вывести результат в формате CSV"
+        )
+        parser.add_argument(
+            "-o",
+            "--output",
+            type=argparse.FileType("w", encoding="utf-8"),
+            help="Файл для сохранения",
+        )
+
+    def run(self, tool: HHApplicantTool) -> None:
+        def execute(sql_query: str) -> None:
+            sql_query = sql_query.strip()
+            if not sql_query:
+                return
+            try:
+                cursor = tool.db.cursor()
+                cursor.execute(sql_query)
+
+                if cursor.description:
+                    columns = [d[0] for d in cursor.description]
+
+                    if tool.args.csv or tool.args.output:
+                        # Если -o не задан, используем sys.stdout
+                        output = tool.args.output or sys.stdout
+                        writer = csv.writer(output)
+                        writer.writerow(columns)
+                        writer.writerows(cursor.fetchall())
+
+                        if tool.args.output:
+                            print(f"✅ Exported to {tool.args.output.name}")
+                        return
+
+                    rows = cursor.fetchmany(MAX_RESULTS + 1)
+                    if not rows:
+                        print("No results found.")
+                        return
+
+                    table = PrettyTable()
+                    table.field_names = columns
+                    for row in rows[:MAX_RESULTS]:
+                        table.add_row(row)
+
+                    print(table)
+                    if len(rows) > MAX_RESULTS:
+                        print(
+                            f"⚠️  Warning: Showing only first {MAX_RESULTS} results."
+                        )
+                else:
+                    tool.db.commit()
+
+                    if cursor.rowcount > 0:
+                        print(f"Rows affected: {cursor.rowcount}")
+
+            except sqlite3.Error as ex:
+                print(f"❌ SQL Error: {ex}")
+                return 1
+
+        if initial_sql := tool.args.sql:
+            return execute(initial_sql)
+
+        if not sys.stdin.isatty():
+            return execute(sys.stdin.read())
+
+        print("SQL Console (q or ^D to exit)")
+        try:
+            while True:
+                try:
+                    user_input = input("query> ").strip()
+                    if user_input.lower() in (
+                        "exit",
+                        "quit",
+                        "q",
+                    ):
+                        break
+                    execute(user_input)
+                    print()
+                except KeyboardInterrupt:
+                    print("^C")
+                    continue
+        except EOFError:
+            print()

hh_applicant_tool/operations/refresh_token.py

@@ -1,11 +1,14 @@
-# Этот модуль можно использовать как образец для других
+from __future__ import annotations
+
 import argparse
 import logging
-from typing import Any
-from ..api import ApiError, ApiClient
-from ..main import BaseOperation
-from ..main import Namespace as BaseNamespace
-from ..utils import print_err
+from typing import TYPE_CHECKING
+
+from ..main import BaseNamespace, BaseOperation
+
+if TYPE_CHECKING:
+    from ..main import HHApplicantTool
+
 
 logger = logging.getLogger(__package__)
 
@@ -17,13 +20,11 @@ class Namespace(BaseNamespace):
 class Operation(BaseOperation):
     """Получает новый access_token."""
 
+    __aliases__ = ["refresh"]
+
     def setup_parser(self, parser: argparse.ArgumentParser) -> None:
         pass
 
-    def run(self, _, api_client: ApiClient, *args: Any) -> None:
-        try:
-            api_client.refresh_access_token()
-            print("✅ Токен обновлен!")
-        except ApiError as ex:
-            print_err("❗ Ошибка:", ex)
-            return 1
+    def run(self, tool: HHApplicantTool) -> None:
+        tool.api_client.refresh_access_token()
+        print("✅ Токен обновлен!")