add heleket
This commit is contained in:
@@ -0,0 +1,90 @@
|
||||
"""Heleket payment integration."""
|
||||
|
||||
import logging
|
||||
from typing import Optional
|
||||
import aiohttp
|
||||
|
||||
from config.settings import settings
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class HelketPayment:
|
||||
"""Heleket payment gateway. Принимает сумму в рублях напрямую."""
|
||||
|
||||
BASE_URL = "https://api.heleket.com/v1"
|
||||
|
||||
def __init__(self):
|
||||
self.api_key = settings.HELEKET_API_KEY
|
||||
self.shop_id = settings.HELEKET_SHOP_ID
|
||||
self._mock = not (self.api_key and self.shop_id)
|
||||
|
||||
async def create_invoice(self, amount_rub: int, order_id: str) -> Optional[dict]:
|
||||
"""
|
||||
Создаёт счёт на сумму в рублях.
|
||||
Возвращает dict: payment_id, address, url. None при ошибке.
|
||||
"""
|
||||
if self._mock:
|
||||
return {
|
||||
"payment_id": f"mock_{order_id}",
|
||||
"address": None,
|
||||
"url": None,
|
||||
"mock": True,
|
||||
}
|
||||
|
||||
try:
|
||||
async with aiohttp.ClientSession() as session:
|
||||
payload = {
|
||||
"shop_id": self.shop_id,
|
||||
"amount": str(amount_rub),
|
||||
"currency": "RUB",
|
||||
"order_id": order_id,
|
||||
"url_return": "",
|
||||
"url_callback": "",
|
||||
}
|
||||
headers = {"merchant": self.api_key, "Content-Type": "application/json"}
|
||||
async with session.post(
|
||||
f"{self.BASE_URL}/payment",
|
||||
json=payload,
|
||||
headers=headers,
|
||||
timeout=aiohttp.ClientTimeout(total=15),
|
||||
) as resp:
|
||||
data = await resp.json()
|
||||
if data.get("state") == 0:
|
||||
r = data.get("result", {})
|
||||
return {
|
||||
"payment_id": r.get("uuid"),
|
||||
"address": r.get("address"),
|
||||
"url": r.get("url"),
|
||||
"mock": False,
|
||||
}
|
||||
logger.error(f"Heleket create error: {data}")
|
||||
return None
|
||||
except Exception as e:
|
||||
logger.error(f"Heleket request error: {e}")
|
||||
return None
|
||||
|
||||
async def check_status(self, payment_id: str) -> str:
|
||||
"""Возвращает: 'paid' | 'pending' | 'expired' | 'error'"""
|
||||
if payment_id.startswith("mock_"):
|
||||
return "pending"
|
||||
try:
|
||||
async with aiohttp.ClientSession() as session:
|
||||
headers = {"merchant": self.api_key}
|
||||
async with session.get(
|
||||
f"{self.BASE_URL}/payment/info/{payment_id}",
|
||||
headers=headers,
|
||||
timeout=aiohttp.ClientTimeout(total=15),
|
||||
) as resp:
|
||||
data = await resp.json()
|
||||
if data.get("state") == 0:
|
||||
status = data.get("result", {}).get("status", "")
|
||||
if status == "paid":
|
||||
return "paid"
|
||||
if status in ("cancel", "expired"):
|
||||
return "expired"
|
||||
return "pending"
|
||||
return "error"
|
||||
except Exception as e:
|
||||
logger.error(f"Heleket check error: {e}")
|
||||
return "error"
|
||||
Reference in New Issue
Block a user