main.py

"""
Клиент API ЕГРЮЛ/ЕГРИП Atlorium — проверка контрагента по ИНН, ОГРН или ОГРНИП.

Запуск (работает сразу, без регистрации — на демо-ключе):
    pip install -r requirements.txt
    python main.py

Боевой ключ: получить на https://atlorium.com и положить в переменную окружения
ATLORIUM_API_KEY. Код при этом не меняется.
"""

import os
import sys
from dataclasses import dataclass

import requests

# Публичный демо-ключ. С ним API отвечает правдоподобными МОКАМИ (не реальными
# данными) — чтобы можно было встроить и протестировать интеграцию до оплаты.
# Ответы детерминированы: один и тот же запрос всегда даёт один и тот же результат,
# поэтому на них можно писать стабильные тесты.
SANDBOX_KEY = "ak_sandbox_demo_mockdata_v1"

API_KEY = os.environ.get("ATLORIUM_API_KEY", SANDBOX_KEY)
BASE_URL = os.environ.get("ATLORIUM_BASE_URL", "https://atlorium.com")

TIMEOUT = 30


class AtloriumError(RuntimeError):
    """Ошибка API. Код HTTP разложен в человекочитаемую причину."""

    REASONS = {
        400: "Неверный формат идентификатора (ожидается ИНН 10/12, ОГРН 13 или ОГРНИП 15 цифр)",
        401: "API-ключ отсутствует, просрочен или недействителен",
        402: "Недостаточно кредитов на балансе — пополните на https://atlorium.com",
        404: "Организация не найдена в реестре",
        429: "Превышен лимит запросов — повторите позже",
        503: "Реестр временно недоступен (за сбой на своей стороне мы не списываем деньги)",
    }

    def __init__(self, status: int, body: str):
        reason = self.REASONS.get(status, "Неизвестная ошибка")
        super().__init__(f"HTTP {status}: {reason}. Ответ сервера: {body[:200]}")
        self.status = status


def _get(path: str, params: dict | None = None) -> requests.Response:
    response = requests.get(
        f"{BASE_URL}{path}",
        params=params,
        headers={
            "Authorization": f"Bearer {API_KEY}",
            "Accept": "application/json",
        },
        timeout=TIMEOUT,
    )
    if not response.ok:
        raise AtloriumError(response.status_code, response.text)
    return response


def get_company(query: str, extended: bool = False) -> dict:
    """Карточка организации по ИНН / ОГРН / ОГРНИП.

    extended=True добавляет данные из официальной выписки: состав участников и их
    доли для ЮЛ, e-mail и пол для ИП. Тарифицируется отдельно и отвечает медленнее.
    """
    return _get(f"/api/egrul/{query}", {"extended": str(extended).lower()}).json()


def search_companies(query: str, limit: int = 10) -> dict:
    """Поиск по наименованию организации или ФИО индивидуального предпринимателя."""
    return _get("/api/egrul/search", {"query": query, "limit": limit}).json()


def download_excerpt(query: str, path: str) -> None:
    """Официальная выписка из реестра в PDF."""
    response = _get(f"/api/egrul/{query}/excerpt")
    with open(path, "wb") as file:
        file.write(response.content)


# ── Применение данных: экспресс-проверка контрагента ──────────────────────────
# Карточка сама по себе — просто JSON. Ценность появляется, когда из неё делают
# вывод. Ниже — набор проверок, которые обычно делают перед сделкой.


@dataclass
class Verdict:
    risks: list[str]
    notes: list[str]

    @property
    def is_risky(self) -> bool:
        return bool(self.risks)


def assess_counterparty(card: dict) -> Verdict:
    risks: list[str] = []
    notes: list[str] = []

    if not card.get("found"):
        risks.append("Субъект отсутствует в реестре")
        return Verdict(risks, notes)

    # Главный стоп-фактор: организация ликвидирована или в процессе ликвидации.
    if not card.get("active"):
        risks.append(f"Организация недействующая: {card.get('status')}")

    # Адрес массовой регистрации — классический признак фирмы-однодневки.
    if card.get("massAddress"):
        risks.append("Адрес массовой регистрации")

    # Минимальный уставный капитал ООО — 10 000 ₽. Ровно минимум — не нарушение,
    # но в связке с другими флагами усиливает подозрение.
    capital = card.get("authorizedCapital")
    if card.get("kind") == "LegalEntity" and capital is not None and capital <= 10_000:
        risks.append(f"Уставный капитал на минимуме: {capital:,.0f} ₽".replace(",", " "))

    # Публикации в «Вестнике государственной регистрации»: сообщения о ликвидации,
    # реорганизации, уменьшении капитала и т.п. Сам факт публикации — повод прочитать её.
    # Каждый элемент — объект {code, count, url}, а не строка.
    for publication in card.get("publications") or []:
        risks.append(
            "Публикация в «Вестнике госрегистрации» "
            f"(код {publication.get('code')}, записей: {publication.get('count')})"
        )

    for warning in card.get("warnings") or []:
        notes.append(warning)

    if card.get("smeCategory"):
        notes.append(f"Реестр МСП: {card['smeCategory']}")
    if card.get("specialTaxRegimes"):
        notes.append(f"Налоговые режимы: {', '.join(card['specialTaxRegimes'])}")

    return Verdict(risks, notes)


def main() -> int:
    if API_KEY == SANDBOX_KEY:
        print("Демо-ключ: ответы сгенерированы (моки), не реальные данные.\n")

    inn = sys.argv[1] if len(sys.argv) > 1 else "7707083893"

    try:
        card = get_company(inn)
    except AtloriumError as error:
        print(f"Ошибка: {error}", file=sys.stderr)
        return 1

    if not card.get("found"):
        print(f"ИНН {inn}: в реестре не найден.")
        return 0

    print(f"{card['fullName']}")
    print(f"  ИНН {card['inn']} · ОГРН {card['ogrn']}")
    print(f"  Статус: {card['status']}")
    print(f"  Адрес: {card['address']}")

    okved = card.get("primaryOkved") or {}
    if okved:
        print(f"  Основной ОКВЭД: {okved.get('code')} — {okved.get('name')}")

    for manager in card.get("managers") or []:
        print(f"  Руководитель: {manager['position']} — {manager['fullName']}")

    verdict = assess_counterparty(card)
    print()
    if verdict.is_risky:
        print("РИСКИ:")
        for risk in verdict.risks:
            print(f"  [!] {risk}")
    else:
        print("Стоп-факторов не обнаружено.")

    for note in verdict.notes:
        print(f"  [i] {note}")

    print(f"\nДанные актуальны на {card.get('dataDate')}.")
    return 0


if __name__ == "__main__":
    raise SystemExit(main())