houseassassin/Solo_bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Docker/ API 0.5.0: Partner Statistics/Coupons for New Customers and More

Browse Source
This commit is contained in:
Vladless
2026-02-01 09:58:58 +03:00
parent 6478495e6e
commit 6477afd00f
46 changed files with 820 additions and 495 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.gitignore
+2 -2
View File
@@ -43,7 +43,6 @@ database.db
bot_old.py
bot_old_2.py
backup_pg.sh
docker-compose.yml
config copy.py
handlers/texts.py
@@ -53,7 +52,6 @@ Thumbs.db
nginx.conf
scripts
Dockerfile
.csv
/logs
setup.py
@@ -61,3 +59,5 @@ setup.py
.github/workflows/
modules/
storage/
.license_state
Dockerfile
+23
View File
@@ -0,0 +1,23 @@
FROM python:3.12-slim
ENV PYTHONDONTWRITEBYTECODE=1
ENV PYTHONUNBUFFERED=1
ENV PIP_NO_CACHE_DIR=1
WORKDIR /app
RUN apt-get update && apt-get install -y --no-install-recommends \
ca-certificates \
tzdata \
git \
&& rm -rf /var/lib/apt/lists/*
COPY requirements.txt .
COPY . .
RUN rm -rf /app/venv \
&& python -m venv /app/venv \
&& /app/venv/bin/pip install --upgrade pip \
&& /app/venv/bin/pip install -r requirements.txt
CMD ["/app/venv/bin/python", "main.py"]
api/main.py
+3 -3
View File
@@ -1,9 +1,9 @@
from fastapi import FastAPI
from api.routes import coupons, gifts, keys, misc, referrals, servers, settings, tariffs, users
from api.routes import users, keys, coupons, servers, tariffs, gifts, referrals, misc, partners
app = FastAPI(
title="SoloBot API (Alpha)",
version="0.4.0",
version="0.5.0",
docs_url="/api/docs",
redoc_url="/api/redoc",
openapi_url="/api/openapi.json",
@@ -16,7 +16,7 @@ app.include_router(servers.router, prefix="/api/servers", tags=["Servers"])
app.include_router(tariffs.router, prefix="/api/tariffs", tags=["Tariffs"])
app.include_router(gifts.router, prefix="/api/gifts", tags=["Gifts"])
app.include_router(referrals.router, prefix="/api/referrals", tags=["Referrals"])
app.include_router(settings.router, prefix="/api/settings", tags=["Settings"])
app.include_router(partners.router, prefix="/api/partners", tags=["Partners"])
app.include_router(misc.router, prefix="/api")
api/routes/base_crud.py
+34 -15
View File
@@ -1,3 +1,5 @@
from typing import Any
from fastapi import APIRouter, Depends, HTTPException, Path, Query
from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession
@@ -7,13 +9,23 @@ from api.depends import get_session, verify_admin_token
from database.models import Admin
def _cast_identifier_type(field: InstrumentedAttribute, value: int | str):
def cast_identifier_type(field: InstrumentedAttribute, value: int | str):
column_type = type(field.property.columns[0].type).__name__
if column_type in ("Integer", "BigInteger"):
return int(value)
return value
def normalize_outgoing_object(obj: object) -> None:
if hasattr(obj, "vless") and getattr(obj, "vless") is None:
setattr(obj, "vless", False)
def to_schema(schema_response: type, obj: object):
normalize_outgoing_object(obj)
return schema_response.model_validate(obj, from_attributes=True)
def generate_crud_router(
*,
model: type,
@@ -35,7 +47,10 @@ def generate_crud_router(
session: AsyncSession = Depends(get_session),
):
result = await session.execute(select(model))
return result.scalars().all()
items = result.scalars().all()
for item in items:
normalize_outgoing_object(item)
return [schema_response.model_validate(item, from_attributes=True) for item in items]
if "get_by_email" in enabled_methods and extra_get_by_email:
@@ -49,7 +64,7 @@ def generate_crud_router(
obj = result.scalar_one_or_none()
if not obj:
raise HTTPException(status_code=404, detail="Not found by email")
return obj
return to_schema(schema_response, obj)
if "get_one" in enabled_methods:
@@ -60,12 +75,12 @@ def generate_crud_router(
session: AsyncSession = Depends(get_session),
):
field = getattr(model, identifier_field)
casted = _cast_identifier_type(field, value)
casted = cast_identifier_type(field, value)
result = await session.execute(select(model).where(field == casted))
obj = result.scalar_one_or_none()
if not obj:
raise HTTPException(status_code=404, detail=f"{model.__name__} not found")
return obj
return to_schema(schema_response, obj)
if "get_all_by_field" in enabled_methods:
@@ -76,52 +91,56 @@ def generate_crud_router(
session: AsyncSession = Depends(get_session),
):
field = getattr(model, identifier_field)
casted = _cast_identifier_type(field, value)
casted = cast_identifier_type(field, value)
result = await session.execute(select(model).where(field == casted))
objs = result.scalars().all()
if not objs:
raise HTTPException(status_code=404, detail=f"{model.__name__} not found")
return objs
for obj in objs:
normalize_outgoing_object(obj)
return [schema_response.model_validate(obj, from_attributes=True) for obj in objs]
if "create" in enabled_methods:
@router.post("/", response_model=schema_response)
async def create(
payload: schema_create, # type: ignore
payload: Any,
admin: Admin = Depends(verify_admin_token),
session: AsyncSession = Depends(get_session),
):
data = payload.dict(exclude_unset=True)
validated = schema_create.model_validate(payload)
data = validated.model_dump(exclude_unset=True)
if "days" in data and data["days"] == 0:
data["days"] = None
obj = model(**data)
session.add(obj)
await session.commit()
await session.refresh(obj)
return obj
return to_schema(schema_response, obj)
if "update" in enabled_methods:
@router.patch(f"/{{{parameter_name}}}", response_model=schema_response)
async def update(
payload: schema_update, # type: ignore
payload: Any,
value: int | str = Path(..., alias=parameter_name),
admin: Admin = Depends(verify_admin_token),
session: AsyncSession = Depends(get_session),
):
field = getattr(model, identifier_field)
casted = _cast_identifier_type(field, value)
casted = cast_identifier_type(field, value)
result = await session.execute(select(model).where(field == casted))
obj = result.scalar_one_or_none()
if not obj:
raise HTTPException(status_code=404, detail=f"{model.__name__} not found")
for k, v in payload.dict(exclude_unset=True).items():
validated = schema_update.model_validate(payload)
for k, v in validated.model_dump(exclude_unset=True).items():
setattr(obj, k, v)
await session.commit()
await session.refresh(obj)
return obj
return to_schema(schema_response, obj)
if "delete" in enabled_methods:
@@ -132,7 +151,7 @@ def generate_crud_router(
session: AsyncSession = Depends(get_session),
):
field = getattr(model, identifier_field)
casted = _cast_identifier_type(field, value)
casted = cast_identifier_type(field, value)
result = await session.execute(select(model).where(field == casted))
obj = result.scalar_one_or_none()
if not obj:
api/routes/keys.py
+4 -20
View File
@@ -93,31 +93,15 @@ async def edit_key_by_email(
setattr(db_key, field, value)
try:
tariff = None
if db_key.tariff_id is not None:
tariff_result = await session.execute(select(Tariff).where(Tariff.id == db_key.tariff_id))
tariff = tariff_result.scalar_one_or_none()
total_gb = db_key.current_traffic_limit
if total_gb is None:
total_gb = db_key.selected_traffic_limit
if total_gb is None and tariff is not None:
total_gb = tariff.traffic_limit
hwid_device_limit = db_key.current_device_limit
if hwid_device_limit is None:
hwid_device_limit = db_key.selected_device_limit
if hwid_device_limit is None and tariff is not None:
hwid_device_limit = tariff.device_limit
new_expiry_time = db_key.expiry_time
await renew_key_in_cluster(
cluster_id=db_key.server_id,
email=db_key.email,
client_id=db_key.client_id,
new_expiry_time=db_key.expiry_time,
total_gb=total_gb,
new_expiry_time=new_expiry_time,
total_gb=getattr(db_key, "traffic_limit", None),
session=session,
hwid_device_limit=hwid_device_limit,
hwid_device_limit=getattr(db_key, "device_limit", None),
reset_traffic=True,
)
await session.commit()
api/routes/partners.py
+325
View File
@@ -0,0 +1,325 @@
from datetime import datetime
from fastapi import APIRouter, Depends, Path, Query
from fastapi.responses import JSONResponse
from sqlalchemy import text
from sqlalchemy.ext.asyncio import AsyncSession
from api.depends import get_session, verify_admin_token
router = APIRouter()
@router.get("/all")
async def get_all_partners(
limit: int = Query(1000, ge=1, le=10000, description="Лимит результатов"),
offset: int = Query(0, ge=0, description="Смещение"),
admin=Depends(verify_admin_token),
session: AsyncSession = Depends(get_session),
):
"""Возвращает список всех партнёров со статистикой.
Структура ответа:
{
"total": int,
"items": [
{
"tg_id": int,
"balance": float,
"percent": float,
"code": str | None,
"method": str | None,
"referred_count": int
}
]
}
"""
partners_sql = text(
"""
SELECT
p.partner_tg_id AS tg_id,
COALESCE(u.partner_balance, 0) AS partner_balance,
COALESCE(u.partner_percent, 0) AS partner_percent,
u.partner_code,
u.payout_method,
COUNT(p.joined_tg_id) as joined_count
FROM partners p
LEFT JOIN users u ON u.tg_id = p.partner_tg_id
WHERE p.partner_tg_id IS NOT NULL
GROUP BY p.partner_tg_id, u.partner_balance, u.partner_percent, u.partner_code, u.payout_method
ORDER BY partner_balance DESC
LIMIT :limit OFFSET :offset
"""
)
count_sql = text(
"""
SELECT COUNT(DISTINCT partner_tg_id) FROM partners
WHERE partner_tg_id IS NOT NULL
"""
)
result = await session.execute(partners_sql, {"limit": limit, "offset": offset})
partners = result.fetchall()
count_result = await session.execute(count_sql)
total = count_result.scalar() or 0
partners_list = [
{
"tg_id": int(partner[0]),
"balance": float(partner[1] or 0),
"percent": float(partner[2] or 0),
"code": partner[3] or None,
"method": partner[4] or None,
"referred_count": int(partner[5] or 0),
}
for partner in partners
]
return JSONResponse(content={"total": total, "items": partners_list})
@router.get("/stats/all")
async def get_partners_stats(
admin=Depends(verify_admin_token),
session: AsyncSession = Depends(get_session),
):
"""Возвращает общую статистику партнёрской программы.
Структура ответа:
{
"total_partners": int,
"total_referred": int,
"total_balance": float,
"top_partner_tg_id": int,
"top_partner_refs": int
}
"""
stats_sql = text(
"""
WITH partner_refs AS (
SELECT partner_tg_id, COUNT(DISTINCT joined_tg_id) AS ref_count
FROM partners
WHERE partner_tg_id IS NOT NULL
GROUP BY partner_tg_id
)
SELECT
(SELECT COUNT(*) FROM partner_refs) AS total_partners,
(SELECT COUNT(DISTINCT joined_tg_id) FROM partners WHERE partner_tg_id IS NOT NULL) AS total_referred,
(
SELECT COALESCE(SUM(u.partner_balance), 0.0)
FROM users u
WHERE u.tg_id IN (SELECT partner_tg_id FROM partner_refs)
) AS total_balance,
(SELECT partner_tg_id FROM partner_refs ORDER BY ref_count DESC LIMIT 1) AS top_partner_tg_id,
(SELECT ref_count FROM partner_refs ORDER BY ref_count DESC LIMIT 1) AS top_partner_refs
"""
)
stats_result = await session.execute(stats_sql)
stats_row = stats_result.fetchone()
if stats_row:
stats = {
"total_partners": int(stats_row[0] or 0),
"total_referred": int(stats_row[1] or 0),
"total_balance": float(stats_row[2] or 0.0),
"top_partner_tg_id": int(stats_row[3] or 0),
"top_partner_refs": int(stats_row[4] or 0),
}
else:
stats = {
"total_partners": 0,
"total_referred": 0,
"total_balance": 0.0,
"top_partner_tg_id": 0,
"top_partner_refs": 0,
}
return JSONResponse(content=stats)
@router.patch("/{tg_id}")
async def update_partner(
tg_id: int = Path(..., description="Telegram ID партнёра"),
balance: float = Query(..., description="Новый баланс партнёра"),
percent: float = Query(..., description="Новый процент партнёра"),
admin=Depends(verify_admin_token),
session: AsyncSession = Depends(get_session),
):
"""Обновляет данные партнёра (баланс и процент).
Структура ответа:
{
"success": bool,
"message": str
}
"""
try:
stmt = text(
"""
UPDATE users
SET partner_balance = :balance, partner_percent = :percent
WHERE tg_id = :tg_id
"""
)
result = await session.execute(stmt, {"tg_id": tg_id, "balance": balance, "percent": percent})
await session.commit()
if result.rowcount > 0:
return JSONResponse(
content={"success": True, "message": f"Партнёр {tg_id} успешно обновлён"},
status_code=200,
)
else:
return JSONResponse(
content={"success": False, "message": "Партнёр не найден"},
status_code=404,
)
except Exception as e:
await session.rollback()
return JSONResponse(
content={"success": False, "message": str(e)},
status_code=500,
)
@router.get("/{tg_id}")
async def get_partner_data(
tg_id: int = Path(..., description="Telegram ID партнёра"),
admin=Depends(verify_admin_token),
session: AsyncSession = Depends(get_session),
):
"""Возвращает партнёрские данные для указанного `tg_id`.
Структура ответа:
{
"tg_id": int,
"partner_balance": float,
"partner_percent": float,
"partner_code": str | None,
"payout_method": str | None,
"invited": [
{ "tg_id": int, "joined_at": str | None, "balance": float, "subs_count": int, "payments_count": int }
]
}
"""
meta_sql = text(
"""
SELECT
COALESCE(u.partner_balance, 0) AS partner_balance,
COALESCE(u.partner_percent, 0) AS partner_percent,
u.partner_code,
u.payout_method
FROM users u
WHERE u.tg_id = :tg_id
"""
)
invited_sql = text(
"""
SELECT
pr.joined_tg_id,
pr.created_at,
COALESCE(u.balance, 0) AS user_balance,
(
SELECT COUNT(*) FROM keys k
WHERE k.tg_id = pr.joined_tg_id
) AS subs_count,
(
SELECT COUNT(*) FROM payments pay
WHERE pay.tg_id = pr.joined_tg_id
AND lower(pay.status) = 'success'
) AS payments_count
FROM partners pr
LEFT JOIN users u ON u.tg_id = pr.joined_tg_id
WHERE pr.partner_tg_id = :tg_id
ORDER BY pr.created_at DESC
"""
)
meta_res = await session.execute(meta_sql, {"tg_id": tg_id})
meta_row = meta_res.fetchone()
invited_res = await session.execute(invited_sql, {"tg_id": tg_id})
invited_rows = invited_res.fetchall()
response = {
"tg_id": tg_id,
"partner_balance": float(meta_row[0] or 0) if meta_row else 0.0,
"partner_percent": float(meta_row[1] or 0) if meta_row else 0.0,
"partner_code": meta_row[2] if meta_row else None,
"payout_method": meta_row[3] if meta_row else None,
"invited": [
{
"tg_id": row[0],
"joined_at": row[1].isoformat() if isinstance(row[1], datetime) else None,
"balance": float(row[2] or 0),
"subs_count": int(row[3] or 0),
"payments_count": int(row[4] or 0),
}
for row in invited_rows
],
}
return JSONResponse(content=response)
@router.get("/{tg_id}/invited")
async def get_partner_invited(
tg_id: int = Path(..., description="Telegram ID партнёра"),
admin=Depends(verify_admin_token),
session: AsyncSession = Depends(get_session),
):
"""Возвращает список приглашённых пользователей конкретного партнёра.
Структура ответа:
[
{ "tg_id": int, "joined_at": str | None, "balance": float, "subs_count": int, "payments_count": int }
]
"""
invited_sql = text(
"""
SELECT
pr.joined_tg_id,
pr.created_at,
COALESCE(u.balance, 0) AS user_balance,
(
SELECT COUNT(*) FROM keys k
WHERE k.tg_id = pr.joined_tg_id
) AS subs_count,
(
SELECT COUNT(*) FROM payments pay
WHERE pay.tg_id = pr.joined_tg_id
AND lower(pay.status) = 'success'
) AS payments_count
FROM partners pr
LEFT JOIN users u ON u.tg_id = pr.joined_tg_id
WHERE pr.partner_tg_id = :tg_id
ORDER BY pr.created_at DESC
"""
)
invited_res = await session.execute(invited_sql, {"tg_id": tg_id})
invited_rows = invited_res.fetchall()
invited_list = [
{
"tg_id": row[0],
"joined_at": row[1].isoformat() if isinstance(row[1], datetime) else None,
"balance": float(row[2] or 0),
"subs_count": int(row[3] or 0),
"payments_count": int(row[4] or 0),
}
for row in invited_rows
]
return JSONResponse(content=invited_list)
database/coupons.py
+9 -1
View File
@@ -8,7 +8,14 @@ from database.models import Coupon, CouponUsage
from logger import logger
async def create_coupon(session: AsyncSession, code: str, amount: int, usage_limit: int, days: int = None) -> bool:
async def create_coupon(
session: AsyncSession,
code: str,
amount: int,
usage_limit: int,
days: int = None,
new_users_only: bool = False,
) -> bool:
try:
exists = await session.scalar(select(Coupon.id).where(Coupon.code == code))
if exists:
@@ -23,6 +30,7 @@ async def create_coupon(session: AsyncSession, code: str, amount: int, usage_lim
usage_count=0,
is_used=False,
days=days,
new_users_only=new_users_only,
)
)
await session.commit()
database/models.py
+1
View File
@@ -183,6 +183,7 @@ class Coupon(DictLikeMixin, Base):
usage_count = Column(Integer, default=0)
is_used = Column(Boolean, default=False)
days = Column(Integer, nullable=True)
new_users_only = Column(Boolean, default=False, nullable=False)
class CouponUsage(DictLikeMixin, Base):
database/servers.py
+2 -5
View File
@@ -345,9 +345,7 @@ async def filter_cluster_by_subgroup(
return cluster
async def filter_cluster_by_tariff(
session: AsyncSession, cluster: list, tariff_id: int, cluster_id: str
) -> list:
async def filter_cluster_by_tariff(session: AsyncSession, cluster: list, tariff_id: int, cluster_id: str) -> list:
names = [s.get("server_name") for s in cluster if s.get("server_name")]
if not names:
return []
@@ -396,8 +394,7 @@ async def has_legacy_subgroup_bindings(session: AsyncSession, server_ids: list[i
return False
result = await session.execute(
select(ServerSubgroup.subgroup_title)
.where(ServerSubgroup.server_id.in_(server_ids))
select(ServerSubgroup.subgroup_title).where(ServerSubgroup.server_id.in_(server_ids))
)
for (title,) in result.all():
if title and not title.isdigit():
database/users.py
+1 -3
View File
@@ -225,9 +225,7 @@ async def mark_trial_extended(tg_id: int, session: AsyncSession):
async def get_user_snapshot(session: AsyncSession, tg_id: int) -> tuple[int, int] | None:
keys_count_sq = select(func.count(Key.client_id)).where(Key.tg_id == tg_id).scalar_subquery()
res = await session.execute(
select(func.coalesce(User.trial, 0), keys_count_sq).where(User.tg_id == tg_id)
)
res = await session.execute(select(func.coalesce(User.trial, 0), keys_count_sq).where(User.tg_id == tg_id))
row = res.first()
if row is None:
return None
docker-compose.yml
+10
View File
@@ -0,0 +1,10 @@
services:
bot:
container_name: solobot
build: .
restart: unless-stopped
network_mode: host
volumes:
- /etc/machine-id:/etc/machine-id:ro
- /var/lib/dbus/machine-id:/var/lib/dbus/machine-id:ro
handlers/admin/clusters/cluster_manage.py
+2 -4
View File
@@ -155,10 +155,7 @@ async def handle_days_input(message: Message, state: FSMContext, session: AsyncS
await state.clear()
return
is_full_remnawave = all(
str(s.get("panel_type", "")).lower() == "remnawave"
for s in cluster_servers
)
is_full_remnawave = all(str(s.get("panel_type", "")).lower() == "remnawave" for s in cluster_servers)
if is_full_remnawave:
uuids = [key.client_id for key in keys if key.client_id]
@@ -175,6 +172,7 @@ async def handle_days_input(message: Message, state: FSMContext, session: AsyncS
return
from panels.remnawave import RemnawaveAPI
remna = RemnawaveAPI(api_url)
try:
handlers/admin/clusters/cluster_sync.py
+2 -1
View File
@@ -565,7 +565,8 @@ async def handle_sync_cluster(
if subgroup_title or (tariff and tariff.get("id")):
tid = tariff.get("id") if tariff else None
filtered_servers = [
s for s in cluster_servers
s
for s in cluster_servers
if (tid and tid in (s.get("tariff_ids") or []))
or (subgroup_title and subgroup_title in (s.get("tariff_subgroups") or []))
]
handlers/admin/clusters/cluster_tariffs.py
+9 -20
View File
@@ -100,8 +100,7 @@ async def show_servers_for_tariffs(
data = await state.get_data()
selected = set(data.get(f"subgrp_sel:{cluster_name}", []))
await callback.message.edit_text(
f"<b>📋 Выберите серверы для привязки тарифов</b>\n"
f"<i>Кластер: {cluster_name}</i>",
f"<b>📋 Выберите серверы для привязки тарифов</b>\n<i>Кластер: {cluster_name}</i>",
reply_markup=build_select_subgroup_servers_kb(cluster_name, cluster_servers, selected),
)
@@ -135,8 +134,7 @@ async def toggle_server_for_tariffs(
selected.add(server_name)
await state.update_data({key: list(selected)})
await callback.message.edit_text(
f"<b>📋 Выберите серверы для привязки тарифов</b>\n"
f"<i>Кластер: {cluster_name}</i>",
f"<b>📋 Выберите серверы для привязки тарифов</b>\n<i>Кластер: {cluster_name}</i>",
reply_markup=build_select_subgroup_servers_kb(cluster_name, cluster_servers, selected),
)
@@ -153,8 +151,7 @@ async def reset_tariff_selection(
f"tariff_sel:{cluster_name}": [],
})
await callback.message.edit_text(
f"<b>📋 Выберите серверы для привязки тарифов</b>\n"
f"<i>Кластер: {cluster_name}</i>",
f"<b>📋 Выберите серверы для привязки тарифов</b>\n<i>Кластер: {cluster_name}</i>",
reply_markup=build_select_subgroup_servers_kb(cluster_name, cluster_servers, set()),
)
@@ -193,8 +190,7 @@ async def choose_tariffs(
selected_tariffs = set(data.get(f"tariff_sel:{cluster_name}", []))
await callback.message.edit_text(
f"<b>📋 Выберите тарифы для {len(selected_servers)} сервер(а/ов)</b>\n"
f"<i>Кластер: {cluster_name}</i>",
f"<b>📋 Выберите тарифы для {len(selected_servers)} сервер(а/ов)</b>\n<i>Кластер: {cluster_name}</i>",
reply_markup=build_tariff_selection_kb(cluster_name, tariffs, selected_tariffs),
)
@@ -232,8 +228,7 @@ async def toggle_tariff_selection(
selected_servers = set(data.get(f"subgrp_sel:{cluster_name}", []))
await callback.message.edit_text(
f"<b>📋 Выберите тарифы для {len(selected_servers)} сервер(а/ов)</b>\n"
f"<i>Кластер: {cluster_name}</i>",
f"<b>📋 Выберите тарифы для {len(selected_servers)} сервер(а/ов)</b>\n<i>Кластер: {cluster_name}</i>",
reply_markup=build_tariff_selection_kb(cluster_name, tariffs, selected_tariffs),
)
@@ -258,8 +253,7 @@ async def apply_tariffs(
return
servers_q = await session.execute(
select(Server.id, Server.server_name, Server.tariff_group)
.where(Server.server_name.in_(selected_servers))
select(Server.id, Server.server_name, Server.tariff_group).where(Server.server_name.in_(selected_servers))
)
servers_data = servers_q.fetchall()
server_ids = [row[0] for row in servers_data]
@@ -274,7 +268,7 @@ async def apply_tariffs(
await session.execute(
delete(ServerSubgroup)
.where(ServerSubgroup.server_id.in_(server_ids))
.where(ServerSubgroup.subgroup_title.regexp_match(r'^\d+$'))
.where(ServerSubgroup.subgroup_title.regexp_match(r"^\d+$"))
.where(ServerSubgroup.subgroup_title.notin_(selected_tariff_strs))
)
@@ -359,11 +353,8 @@ async def reset_cluster_subgroups(callback: CallbackQuery, callback_data: AdminC
def render_attach_tariff_menu_text(
cluster_name: str,
cluster_servers: list[dict],
tariffs_cache: dict[int, dict] | None = None
cluster_name: str, cluster_servers: list[dict], tariffs_cache: dict[int, dict] | None = None
) -> str:
tariff_map: dict[int, list[str]] = {}
legacy_map: dict[str, list[str]] = {}
@@ -456,9 +447,7 @@ async def handle_attach_tariff_menu(callback: CallbackQuery, session: AsyncSessi
tariffs_cache = {}
if all_tariff_ids:
result = await session.execute(
select(Tariff).where(Tariff.id.in_(all_tariff_ids))
)
result = await session.execute(select(Tariff).where(Tariff.id.in_(all_tariff_ids)))
for t in result.scalars().all():
tariffs_cache[t.id] = {
"id": t.id,
handlers/admin/clusters/keyboard.py
+1 -4
View File
@@ -208,10 +208,7 @@ def build_tariff_selection_kb(cluster_name: str, tariffs: list, selected: set[in
builder.row(
InlineKeyboardButton(
text=f"{mark} {t.name}",
callback_data=AdminClusterCallback(
action="toggle_tariff",
data=f"{cluster_name}|{t.id}"
).pack(),
callback_data=AdminClusterCallback(action="toggle_tariff", data=f"{cluster_name}|{t.id}").pack(),
)
)
handlers/admin/coupons/coupons_handler.py
+76 -12
View File
@@ -34,6 +34,7 @@ router = Router()
class AdminCouponsState(StatesGroup):
waiting_for_coupon_type = State()
waiting_for_coupon_audience = State()
waiting_for_balance_data = State()
waiting_for_days_data = State()
@@ -62,30 +63,59 @@ async def handle_coupons_create(callback_query: CallbackQuery, state: FSMContext
await state.set_state(AdminCouponsState.waiting_for_coupon_type)
@router.callback_query(F.data == "coupon_type_balance")
async def show_coupon_audience_step(callback_query: CallbackQuery, state: FSMContext, coupon_type: str):
await state.update_data(coupon_type=coupon_type)
text = "🎯 <b>Кому доступен купон?</b>"
kb = InlineKeyboardBuilder()
kb.button(text="👤 Всем", callback_data="coupon_audience_all")
kb.button(text="🆕 Только новым", callback_data="coupon_audience_new")
kb.button(text=BACK, callback_data=AdminPanelCallback(action="coupons").pack())
kb.adjust(1)
await callback_query.message.edit_text(text=text, reply_markup=kb.as_markup())
await state.set_state(AdminCouponsState.waiting_for_coupon_audience)
@router.callback_query(F.data == "coupon_type_balance", IsAdminFilter())
async def handle_balance_coupon_selection(callback_query: CallbackQuery, state: FSMContext):
await show_coupon_audience_step(callback_query, state, "balance")
@router.callback_query(F.data == "coupon_type_days", IsAdminFilter())
async def handle_days_coupon_selection(callback_query: CallbackQuery, state: FSMContext):
await show_coupon_audience_step(callback_query, state, "days")
@router.callback_query(F.data.in_(("coupon_audience_all", "coupon_audience_new")), IsAdminFilter())
async def handle_coupon_audience(callback_query: CallbackQuery, state: FSMContext):
data = await state.get_data()
coupon_type = data.get("coupon_type")
if coupon_type not in ("balance", "days"):
await callback_query.answer("Ошибка: тип купона не найден", show_alert=True)
return
await state.update_data(new_users_only=callback_query.data == "coupon_audience_new")
kb = InlineKeyboardBuilder()
kb.button(text=BACK, callback_data=AdminPanelCallback(action="coupons").pack())
kb.adjust(1)
if coupon_type == "balance":
text = (
"🎫 <b>Введите данные для создания купона в формате:</b>\n\n"
"📝 <i>код</i> 💰 <i>сумма</i> 🔢 <i>лимит</i>\n\n"
"Пример: <b>'COUPON1 50 5'</b> 👈\n\n"
)
kb = InlineKeyboardBuilder()
kb.button(text=BACK, callback_data=AdminPanelCallback(action="coupons").pack())
await callback_query.message.edit_text(text=text, reply_markup=kb.as_markup())
await state.set_state(AdminCouponsState.waiting_for_balance_data)
return
@router.callback_query(F.data == "coupon_type_days")
async def handle_days_coupon_selection(callback_query: CallbackQuery, state: FSMContext):
text = (
"🎫 <b>Введите данные для создания купона в формате:</b>\n\n"
"📝 <i>код</i> ⏳ <i>дни</i> 🔢 <i>лимит</i>\n\n"
"Пример: <b>'DAYS10 10 50'</b> 👈\n\n"
)
kb = InlineKeyboardBuilder()
kb.button(text=BACK, callback_data=AdminPanelCallback(action="coupons").pack())
await callback_query.message.edit_text(text=text, reply_markup=kb.as_markup())
await state.set_state(AdminCouponsState.waiting_for_days_data)
@@ -97,6 +127,7 @@ async def handle_balance_coupon_input(message: Message, state: FSMContext, sessi
kb = InlineKeyboardBuilder()
kb.button(text=BACK, callback_data=AdminPanelCallback(action="coupons").pack())
kb.adjust(1)
if len(parts) != 3:
text = (
@@ -119,13 +150,29 @@ async def handle_balance_coupon_input(message: Message, state: FSMContext, sessi
return
try:
await create_coupon(session, coupon_code, coupon_amount, usage_limit, days=None)
data = await state.get_data()
new_users_only = bool(data.get("new_users_only"))
ok = await create_coupon(
session,
coupon_code,
coupon_amount,
usage_limit,
days=None,
new_users_only=new_users_only,
)
if not ok:
await message.answer("❌ Купон с таким кодом уже существует.", reply_markup=kb.as_markup())
return
coupon_link = f"https://t.me/{USERNAME_BOT}?start=coupons_{coupon_code}"
audience_txt = "🆕 Только новым" if new_users_only else "👤 Всем"
text = (
f"✅ Купон с кодом <b>{coupon_code}</b> успешно создан!\n"
f"💰 Сумма: <b>{coupon_amount} рублей</b>\n"
f"🔢 Лимит использования: <b>{usage_limit} раз</b>\n"
f"🎯 Доступ: <b>{audience_txt}</b>\n"
f"🔗 <b>Ссылка:</b> <code>{coupon_link}</code>\n"
)
@@ -150,6 +197,7 @@ async def handle_days_coupon_input(message: Message, state: FSMContext, session:
kb = InlineKeyboardBuilder()
kb.button(text=BACK, callback_data=AdminPanelCallback(action="coupons").pack())
kb.adjust(1)
if len(parts) != 3:
text = (
@@ -172,13 +220,29 @@ async def handle_days_coupon_input(message: Message, state: FSMContext, session:
return
try:
await create_coupon(session, coupon_code, 0, usage_limit, days=days)
data = await state.get_data()
new_users_only = bool(data.get("new_users_only"))
ok = await create_coupon(
session,
coupon_code,
0,
usage_limit,
days=days,
new_users_only=new_users_only,
)
if not ok:
await message.answer("❌ Купон с таким кодом уже существует.", reply_markup=kb.as_markup())
return
coupon_link = f"https://t.me/{USERNAME_BOT}?start=coupons_{coupon_code}"
audience_txt = "🆕 Только новым" if new_users_only else "👤 Всем"
text = (
f"✅ Купон с кодом <b>{coupon_code}</b> успешно создан!\n"
f"⏳ <b>{format_days(days)}</b>\n"
f"🔢 Лимит использования: <b>{usage_limit} раз</b>\n"
f"🎯 Доступ: <b>{audience_txt}</b>\n"
f"🔗 <b>Ссылка:</b> <code>{coupon_link}</code>\n"
)
handlers/admin/users/keyboard.py
+1 -3
View File
@@ -481,9 +481,7 @@ def build_user_gifts_kb(tg_id: int, gifts: list, page: int = 0) -> InlineKeyboar
callback_data=f"user_gift_page|{tg_id}|{page - 1}",
)
)
nav_buttons.append(
InlineKeyboardButton(text=f"{page + 1}/{total_pages}", callback_data="noop")
)
nav_buttons.append(InlineKeyboardButton(text=f"{page + 1}/{total_pages}", callback_data="noop"))
if page < total_pages - 1:
nav_buttons.append(
InlineKeyboardButton(
handlers/admin/users/users_gifts.py
+3 -9
View File
@@ -29,10 +29,7 @@ async def show_gifts_list(message: types.Message, session: AsyncSession, tg_id:
gifts = await get_user_gifts(session, tg_id)
if not gifts:
text = (
f"🎁 <b>Подарки пользователя</b> <code>{tg_id}</code>\n\n"
f"У пользователя нет созданных подарков."
)
text = f"🎁 <b>Подарки пользователя</b> <code>{tg_id}</code>\n\nУ пользователя нет созданных подарков."
await message.edit_text(
text=text,
reply_markup=build_user_gifts_kb(tg_id, [], page),
@@ -40,6 +37,7 @@ async def show_gifts_list(message: types.Message, session: AsyncSession, tg_id:
return
from .keyboard import GIFTS_PER_PAGE
start_idx = page * GIFTS_PER_PAGE
end_idx = start_idx + GIFTS_PER_PAGE
page_gifts = gifts[start_idx:end_idx]
@@ -61,11 +59,7 @@ async def show_gifts_list(message: types.Message, session: AsyncSession, tg_id:
created_str = gift.created_at.replace(tzinfo=pytz.UTC).astimezone(MOSCOW_TZ).strftime("%d.%m.%Y %H:%M")
lines.append(
f"\n<b>{i}.🎁 </b> {gift.selected_months} мес.\n"
f" 📅 Создан: {created_str}\n"
f" {status}"
)
lines.append(f"\n<b>{i}.🎁 </b> {gift.selected_months} мес.\n 📅 Создан: {created_str}\n {status}")
lines.append("\n\n<i>Нажмите кнопку для удаления:</i>")
handlers/admin/users/users_keys.py
+6 -1
View File
@@ -110,7 +110,12 @@ async def handle_key_edit(
if key_obj.tariff_id:
result = await session.execute(
select(
Tariff.name, Tariff.subgroup_title, Tariff.group_code, Tariff.device_limit, Tariff.traffic_limit, Tariff.configurable
Tariff.name,
Tariff.subgroup_title,
Tariff.group_code,
Tariff.device_limit,
Tariff.traffic_limit,
Tariff.configurable,
).where(Tariff.id == key_obj.tariff_id)
)
row = result.first()
handlers/admin/users/users_manage.py
+3 -1
View File
@@ -342,7 +342,9 @@ async def process_user_search(
) -> None:
await state.clear()
stmt_user = select(User.username, User.balance, User.created_at, User.updated_at, User.trial).where(User.tg_id == tg_id)
stmt_user = select(User.username, User.balance, User.created_at, User.updated_at, User.trial).where(
User.tg_id == tg_id
)
result_user = await session.execute(stmt_user)
user_data = result_user.first()
handlers/admin/users/users_tariffs.py
+18 -6
View File
@@ -207,10 +207,14 @@ async def handle_user_renew_confirm(
base_traffic_gb = int(base_traffic_gb) if base_traffic_gb is not None else None
selected_devices = (
base_device_limit if base_device_limit is not None else (device_int_options[0] if device_int_options else None)
base_device_limit
if base_device_limit is not None
else (device_int_options[0] if device_int_options else None)
)
selected_traffic_gb = (
base_traffic_gb if base_traffic_gb is not None else (traffic_int_options[0] if traffic_int_options else None)
base_traffic_gb
if base_traffic_gb is not None
else (traffic_int_options[0] if traffic_int_options else None)
)
await state.update_data(
@@ -343,7 +347,9 @@ async def handle_user_renew_confirm(
plan=tariff_id,
)
except Exception as e:
logger.error(f"[AdminRenew] renew_key_in_cluster failed: tg_id={tg_id} email={email} tariff_id={tariff_id}: {e}")
logger.error(
f"[AdminRenew] renew_key_in_cluster failed: tg_id={tg_id} email={email} tariff_id={tariff_id}: {e}"
)
ok = False
await state.clear()
@@ -353,7 +359,9 @@ async def handle_user_renew_confirm(
callback_data_back = AdminUserEditorCallback(action="users_key_edit", data=email, tg_id=tg_id)
await handle_key_edit(callback_query=callback_query, callback_data=callback_data_back, session=session, update=False)
await handle_key_edit(
callback_query=callback_query, callback_data=callback_data_back, session=session, update=False
)
@router.callback_query(F.data.startswith("cfg_renew_devices|"), IsAdminFilter())
@@ -662,7 +670,9 @@ async def handle_cfg_renew_apply(callback_query: CallbackQuery, session: AsyncSe
plan=tariff_id,
)
except Exception as e:
logger.error(f"[AdminRenewCfg] renew_key_in_cluster failed: tg_id={tg_id} email={email} tariff_id={tariff_id}: {e}")
logger.error(
f"[AdminRenewCfg] renew_key_in_cluster failed: tg_id={tg_id} email={email} tariff_id={tariff_id}: {e}"
)
ok = False
await state.clear()
@@ -672,7 +682,9 @@ async def handle_cfg_renew_apply(callback_query: CallbackQuery, session: AsyncSe
callback_data_back = AdminUserEditorCallback(action="users_key_edit", data=email, tg_id=int(tg_id))
await handle_key_edit(callback_query=callback_query, callback_data=callback_data_back, session=session, update=False)
await handle_key_edit(
callback_query=callback_query, callback_data=callback_data_back, session=session, update=False
)
@router.callback_query(F.data == "back:group", IsAdminFilter())
handlers/coupons.py
+17 -1
View File
@@ -107,6 +107,15 @@ async def activate_coupon(
await state.clear()
return
from database.models import User
if getattr(coupon, "new_users_only", False):
exists = await session.scalar(select(User.tg_id).where(User.tg_id == user_id))
if exists is not None:
await message.answer("❌ Этот купон доступен только для новых пользователей.")
await state.clear()
return
if isinstance(user, dict):
await add_user(session=session, **user)
else:
@@ -184,7 +193,7 @@ async def handle_key_extension(
session: AsyncSession,
admin: bool = False,
):
from database.models import Coupon, Key
from database.models import Coupon, Key, User
parts = callback_query.data.split("|")
client_id = parts[1]
@@ -205,6 +214,13 @@ async def handle_key_extension(
await state.clear()
return
if getattr(coupon, "new_users_only", False):
exists = await session.scalar(select(User.tg_id).where(User.tg_id == tg_id))
if exists is not None:
await callback_query.message.edit_text("❌ Этот купон доступен только для новых пользователей.")
await state.clear()
return
result = await session.execute(select(Key).where(Key.tg_id == tg_id, Key.client_id == client_id))
key = result.scalar_one_or_none()
if not key or key.is_frozen:
handlers/keys/key_view.py
+3 -1
View File
@@ -340,7 +340,9 @@ async def build_key_view_payload(session: AsyncSession, key_name: str):
remna_server = None
for cluster_name, cluster_servers in servers.items():
for srv in cluster_servers:
if (srv.get("server_name") == server_name or cluster_name == server_name) and srv.get("panel_type") == "remnawave":
if (srv.get("server_name") == server_name or cluster_name == server_name) and srv.get(
"panel_type"
) == "remnawave":
remna_server = srv
break
if remna_server:
handlers/keys/operations/creation.py
+1 -3
View File
@@ -97,9 +97,7 @@ async def create_key_on_cluster(
device_limit_value = int(hwid_limit or 0)
if plan is not None:
filtered = await filter_cluster_by_tariff(
session, enabled_servers, plan, cluster_id
)
filtered = await filter_cluster_by_tariff(session, enabled_servers, plan, cluster_id)
if filtered is not enabled_servers:
enabled_servers = filtered
elif subgroup_title:
handlers/keys/operations/renewal.py
+5 -7
View File
@@ -78,7 +78,9 @@ async def renew_on_remnawave(
if old_device_limit is not None and hwid_device_limit < old_device_limit:
try:
await remna.clear_all_hwid_devices(client_id)
logger.info(f"{PANEL_REMNA} HWID устройства сброшены для {client_id} (лимит {old_device_limit} → {hwid_device_limit})")
logger.info(
f"{PANEL_REMNA} HWID устройства сброшены для {client_id} (лимит {old_device_limit} → {hwid_device_limit})"
)
except Exception as e:
logger.warning(f"{PANEL_REMNA} Ошибка сброса HWID: {e}")
@@ -298,9 +300,7 @@ async def renew_key_in_cluster(
cluster_scope = [single_server]
else:
if plan is not None:
filtered = await filter_cluster_by_tariff(
session, cluster, plan, cluster_id
)
filtered = await filter_cluster_by_tariff(session, cluster, plan, cluster_id)
if filtered is not cluster:
cluster_scope = filtered
elif target_subgroup:
@@ -310,9 +310,7 @@ async def renew_key_in_cluster(
else:
cluster_scope = cluster
elif target_subgroup:
target = await filter_cluster_by_subgroup(
session, cluster, target_subgroup, cluster_id, tariff_id=plan
)
target = await filter_cluster_by_subgroup(session, cluster, target_subgroup, cluster_id, tariff_id=plan)
cluster_scope = target if target else cluster
else:
cluster_scope = cluster
handlers/keys/operations/update.py
+4 -12
View File
@@ -53,9 +53,7 @@ async def update_key_on_cluster(
raise ValueError(f"Кластер или сервер с ID/именем {cluster_id} не найден.")
if tariff_id is not None:
filtered = await filter_cluster_by_tariff(
session, cluster, tariff_id, cluster_id
)
filtered = await filter_cluster_by_tariff(session, cluster, tariff_id, cluster_id)
if filtered is not cluster:
cluster = filtered
elif subgroup_code:
@@ -63,9 +61,7 @@ async def update_key_on_cluster(
session, cluster, subgroup_code, cluster_id, tariff_id=tariff_id
)
elif subgroup_code:
cluster = await filter_cluster_by_subgroup(
session, cluster, subgroup_code, cluster_id, tariff_id=tariff_id
)
cluster = await filter_cluster_by_subgroup(session, cluster, subgroup_code, cluster_id, tariff_id=tariff_id)
if not cluster:
logger.warning(f"[Update] Нет серверов после фильтрации по привязкам в кластере {cluster_id}")
@@ -276,9 +272,7 @@ async def update_subscription(
cluster_servers = []
if tariff_id is not None:
filtered = await filter_cluster_by_tariff(
session, cluster_servers, tariff_id, new_cluster_id
)
filtered = await filter_cluster_by_tariff(session, cluster_servers, tariff_id, new_cluster_id)
if filtered is not cluster_servers:
cluster_servers = filtered
elif subgroup_code:
@@ -291,9 +285,7 @@ async def update_subscription(
)
if not cluster_servers:
logger.warning(
f"[Update] Пересоздание пропущено: нет серверов после фильтрации в {new_cluster_id}."
)
logger.warning(f"[Update] Пересоздание пропущено: нет серверов после фильтрации в {new_cluster_id}.")
return
if tariff:
handlers/notifications/general_notifications.py
+21 -19
View File
@@ -139,26 +139,18 @@ async def execute_bulk_updates(session: AsyncSession, bulk_updates: dict[str, An
for tg_id, balance_change in bulk_updates["balance_changes"].items():
await session.execute(
text("UPDATE users SET balance = balance + :change WHERE tg_id = :tg_id"),
{"change": balance_change, "tg_id": tg_id}
{"change": balance_change, "tg_id": tg_id},
)
logger.info(f"Bulk: обновлено {len(bulk_updates['balance_changes'])} балансов")
if bulk_updates["key_expiry_updates"]:
for client_id, new_expiry in bulk_updates["key_expiry_updates"]:
await session.execute(
update(Key)
.where(Key.client_id == client_id)
.values(expiry_time=new_expiry)
)
await session.execute(update(Key).where(Key.client_id == client_id).values(expiry_time=new_expiry))
logger.info(f"Bulk: обновлено {len(bulk_updates['key_expiry_updates'])} сроков действия ключей")
if bulk_updates["key_tariff_updates"]:
for client_id, new_tariff_id in bulk_updates["key_tariff_updates"]:
await session.execute(
update(Key)
.where(Key.client_id == client_id)
.values(tariff_id=new_tariff_id)
)
await session.execute(update(Key).where(Key.client_id == client_id).values(tariff_id=new_tariff_id))
logger.info(f"Bulk: обновлено {len(bulk_updates['key_tariff_updates'])} тарифов ключей")
for tg_id, notification_type in bulk_updates["notifications_to_add"]:
@@ -487,7 +479,9 @@ async def notify_expiring_keys(
await add_notification(ctx.session, msg["tg_id"], msg["notification_id"])
if result:
sent_count += 1
logger.info(f"Отправлено уведомление об истекающей подписке {msg['email']} пользователю {msg['tg_id']}.")
logger.info(
f"Отправлено уведомление об истекающей подписке {msg['email']} пользователю {msg['tg_id']}."
)
logger.info(f"Отправлено {sent_count} уведомлений типа {notify_type}.")
logger.info(f"Обработка уведомлений {notify_type} завершена.")
@@ -633,18 +627,21 @@ async def periodic_notifications(bot: Bot, *, sessionmaker: async_sessionmaker):
notify_10_enabled = bool(NOTIFICATIONS_CONFIG.get("EXPIRY_10H_ENABLED", NOTIFY_10H_ENABLED))
notify_10_hours = int(NOTIFICATIONS_CONFIG.get("EXPIRY_10H_BEFORE_HOURS", NOTIFY_10H_HOURS))
notify_renew_enabled = bool(NOTIFICATIONS_CONFIG.get("RENEW_ENABLED", NOTIFY_RENEW))
inactive_traffic_enabled = bool(NOTIFICATIONS_CONFIG.get("INACTIVE_TRAFFIC_ENABLED", NOTIFY_INACTIVE_TRAFFIC))
inactive_traffic_enabled = bool(
NOTIFICATIONS_CONFIG.get("INACTIVE_TRAFFIC_ENABLED", NOTIFY_INACTIVE_TRAFFIC)
)
notify_hot_leads_enabled = bool(NOTIFICATIONS_CONFIG.get("HOT_LEADS_ENABLED", NOTIFY_HOT_LEADS))
if notify_24_enabled:
try:
await notify_expiring_keys(
ctx, keys,
ctx,
keys,
min_hours=notify_10_hours if notify_10_enabled else 0,
max_hours=notify_24_hours,
notify_type="key_24h",
photo="notify_24h.jpg",
notify_renew_enabled=notify_renew_enabled
notify_renew_enabled=notify_renew_enabled,
)
except Exception as error:
logger.error(f"Ошибка в notify_expiring_keys (24h): {error}")
@@ -652,12 +649,13 @@ async def periodic_notifications(bot: Bot, *, sessionmaker: async_sessionmaker):
if notify_10_enabled:
try:
await notify_expiring_keys(
ctx, keys,
ctx,
keys,
min_hours=0,
max_hours=notify_10_hours,
notify_type="key_10h",
photo="notify_10h.jpg",
notify_renew_enabled=notify_renew_enabled
notify_renew_enabled=notify_renew_enabled,
)
except Exception as error:
logger.error(f"Ошибка в notify_expiring_keys (10h): {error}")
@@ -689,8 +687,12 @@ async def periodic_notifications(bot: Bot, *, sessionmaker: async_sessionmaker):
await execute_bulk_updates(session, bulk_updates)
bulk_time = (datetime.now() - bulk_start).total_seconds()
total_renewals = len(bulk_updates["balance_changes"])
total_key_updates = len(bulk_updates["key_expiry_updates"]) + len(bulk_updates["key_tariff_updates"])
total_notification_updates = len(bulk_updates["notifications_to_add"]) + len(bulk_updates["notifications_to_delete"])
total_key_updates = len(bulk_updates["key_expiry_updates"]) + len(
bulk_updates["key_tariff_updates"]
)
total_notification_updates = len(bulk_updates["notifications_to_add"]) + len(
bulk_updates["notifications_to_delete"]
)
logger.info(
f"Bulk-операции выполнены за {bulk_time:.2f}s. Обработано: {total_renewals} продлений, {total_key_updates} ключей, {total_notification_updates} уведомлений"
)
handlers/notifications/notify_utils.py
+1
View File
@@ -152,6 +152,7 @@ class FastNotificationSender:
try:
from sqlalchemy.dialects.postgresql import insert
from database.models import BlockedUser
values = [{"tg_id": tg_id} for tg_id in self.blocked_users]
stmt = insert(BlockedUser).values(values).on_conflict_do_nothing(index_elements=[BlockedUser.tg_id])
await self.session.execute(stmt)
handlers/notifications/special_notifications.py
+2 -6
View File
@@ -119,9 +119,7 @@ async def notify_inactive_trial_users(bot: Bot, session: AsyncSession):
extend_ids = [tg_id for tg_id in users_to_extend if tg_id in sent_tg_ids]
if extend_ids:
await session.execute(
update(User).where(User.tg_id.in_(extend_ids)).values(trial=-1)
)
await session.execute(update(User).where(User.tg_id.in_(extend_ids)).values(trial=-1))
await session.commit()
logger.info(f"Bulk: отмечено {len(extend_ids)} пользователей с расширенным триалом")
@@ -232,9 +230,7 @@ async def notify_users_no_traffic(bot: Bot, session: AsyncSession, current_time:
if keys_to_mark_notified:
try:
await session.execute(
update(Key).where(Key.client_id.in_(keys_to_mark_notified)).values(notified=True)
)
await session.execute(update(Key).where(Key.client_id.in_(keys_to_mark_notified)).values(notified=True))
await session.commit()
logger.info(f"Bulk: отмечено {len(keys_to_mark_notified)} ключей как notified")
except Exception as error:
handlers/payments/currency_flow.py
+1 -1
View File
@@ -18,7 +18,7 @@ def build_currency_choice_kb(
kb.row(InlineKeyboardButton(text=RUB_CURRENCY, callback_data=f"{prefix}|RUB"))
kb.row(InlineKeyboardButton(text=USD_CURRENCY, callback_data=f"{prefix}|USD"))
trib = (TRIBUTE_LINK or "").strip()
trib_enabled = (bool(trib) if show_tribute is None else bool(show_tribute) and bool(trib))
trib_enabled = bool(trib) if show_tribute is None else bool(show_tribute) and bool(trib)
if show_stars:
row = [InlineKeyboardButton(text=STARS, callback_data=f"{prefix}|STARS")]
handlers/payments/fast_payment_flow.py
+2 -10
View File
@@ -119,18 +119,10 @@ async def try_fast_payment_flow(
tribute_cfg = providers_map.get("TRIBUTE") or {}
tribute_link = (TRIBUTE_LINK or "").strip()
tribute_enabled = (
"TRIBUTE" in configured_set
and tribute_cfg.get("enabled", True)
and bool(tribute_link)
)
tribute_enabled = "TRIBUTE" in configured_set and tribute_cfg.get("enabled", True) and bool(tribute_link)
stars_cfg = providers_map.get("STARS") or {}
stars_enabled_for_fast = (
"STARS" in configured_set
and stars_cfg.get("fast")
and stars_cfg.get("enabled", True)
)
stars_enabled_for_fast = "STARS" in configured_set and stars_cfg.get("fast") and stars_cfg.get("enabled", True)
if multicurrency_mode and not one_screen:
show_stars = stars_enabled_for_fast
handlers/payments/gift.cpython-312-x86_64-linux-gnu.so
BIN
View File
Binary file not shown.
handlers/payments/heleket/handlers.py
+7 -26
View File
@@ -31,9 +31,7 @@ async def handle_pay_heleket_crypto(
session: AsyncSession,
):
"""Обработчик оплаты через Heleket криптовалютой."""
await process_callback_pay_heleket(
callback_query, state, session, method_name="crypto"
)
await process_callback_pay_heleket(callback_query, state, session, method_name="crypto")
async def handle_custom_amount_input_heleket(
@@ -85,34 +83,23 @@ async def handle_custom_amount_input_heleket(
method = enabled_methods[0]
try:
payment_url = await generate_heleket_payment_link(
amount, tg_id, method
)
payment_url = await generate_heleket_payment_link(amount, tg_id, method)
if not payment_url or payment_url == "https://heleket.com/":
await edit_or_send_message(
target_message=message,
text=(
"❌ Произошла ошибка при создании платежа. "
"Попробуйте позже или выберите другой способ оплаты."
),
text=("❌ Произошла ошибка при создании платежа. Попробуйте позже или выберите другой способ оплаты."),
)
return
markup = InlineKeyboardMarkup(
inline_keyboard=[
[InlineKeyboardButton(text=pay_button_text, url=payment_url)],
[
InlineKeyboardButton(
text=main_menu_text, callback_data="profile"
)
],
[InlineKeyboardButton(text=main_menu_text, callback_data="profile")],
]
)
result = await session.execute(
select(User.language_code).where(User.tg_id == tg_id)
)
result = await session.execute(select(User.language_code).where(User.tg_id == tg_id))
language_code = result.scalar_one_or_none()
amount_text = await format_for_user(
session,
@@ -123,17 +110,11 @@ async def handle_custom_amount_input_heleket(
)
text_out = DEFAULT_PAYMENT_MESSAGE.format(amount=amount_text)
await edit_or_send_message(
target_message=message, text=text_out, reply_markup=markup
)
await edit_or_send_message(target_message=message, text=text_out, reply_markup=markup)
except Exception as e:
logger.error(
f"Ошибка при создании платежа Heleket "
f"для пользователя {tg_id}: {e}"
)
logger.error(f"Ошибка при создании платежа Heleket для пользователя {tg_id}: {e}")
await edit_or_send_message(
target_message=message,
text="Произошла ошибка при создании платежа. Попробуйте позже.",
reply_markup=InlineKeyboardMarkup(inline_keyboard=[]),
)
handlers/payments/heleket/service.py
+13 -22
View File
@@ -51,9 +51,7 @@ router = Router()
async def get_user_language(session: AsyncSession, tg_id: int) -> str | None:
"""Получает язык пользователя из базы данных"""
result = await session.execute(
select(User.language_code).where(User.tg_id == tg_id)
)
result = await session.execute(select(User.language_code).where(User.tg_id == tg_id))
return result.scalar_one_or_none()
@@ -65,7 +63,13 @@ class ReplenishBalanceHeleket(StatesGroup):
HELEKET_METHODS = {
"crypto": {"enable": PROVIDERS_ENABLED.get("HELEKET", False), "currency": "USD", "to_currency": None, "button": HELEKET, "desc": HELEKET_CRYPTO_DESCRIPTION},
"crypto": {
"enable": PROVIDERS_ENABLED.get("HELEKET", False),
"currency": "USD",
"to_currency": None,
"button": HELEKET,
"desc": HELEKET_CRYPTO_DESCRIPTION,
},
}
@@ -93,15 +97,13 @@ async def process_callback_pay_heleket(
return
language_code = await get_user_language(session, tg_id)
opts = await payment_options_for_user(
session, tg_id, language_code, force_currency="USD"
)
opts = await payment_options_for_user(session, tg_id, language_code, force_currency="USD")
builder = build_amounts_keyboard(
prefix=f"heleket_{method_name}",
pattern="{prefix}_amount|{price}",
back_cb="balance",
custom_cb=f"heleket_custom_amount|{method_name}",
opts=opts
opts=opts,
)
await edit_or_send_message(
@@ -120,9 +122,7 @@ async def process_callback_pay_heleket(
builder = InlineKeyboardBuilder()
for name, method in HELEKET_METHODS.items():
if method["enable"]:
builder.row(
InlineKeyboardButton(text=method["button"], callback_data=f"heleket_method|{name}")
)
builder.row(InlineKeyboardButton(text=method["button"], callback_data=f"heleket_method|{name}"))
builder.row(InlineKeyboardButton(text=BACK, callback_data="balance"))
await edit_or_send_message(
@@ -158,15 +158,13 @@ async def process_method_selection(callback_query: types.CallbackQuery, state: F
tg_id = callback_query.from_user.id
language_code = await get_user_language(session, tg_id)
opts = await payment_options_for_user(
session, tg_id, language_code, force_currency="USD"
)
opts = await payment_options_for_user(session, tg_id, language_code, force_currency="USD")
builder = build_amounts_keyboard(
prefix=f"heleket_{method_name}",
pattern="{prefix}_amount|{price}",
back_cb="pay",
custom_cb=f"heleket_custom_amount|{method_name}",
opts=opts
opts=opts,
)
await edit_or_send_message(
@@ -407,10 +405,3 @@ async def generate_heleket_payment_link(amount: int, tg_id: int, method: dict) -
except Exception as e:
logger.error(f"Error creating Heleket payment: {e}")
return "https://heleket.com/"
handlers/payments/heleket/webhook.py
+14 -37
View File
@@ -34,24 +34,16 @@ def verify_heleket_signature(data: dict) -> bool:
data_without_sign = data.copy()
del data_without_sign["sign"]
json_data = json.dumps(
data_without_sign, ensure_ascii=False, separators=(",", ":")
)
json_data = json.dumps(data_without_sign, ensure_ascii=False, separators=(",", ":"))
json_data = json_data.replace("/", "\\/")
base64_data = base64.b64encode(json_data.encode("utf-8")).decode(
"utf-8"
)
base64_data = base64.b64encode(json_data.encode("utf-8")).decode("utf-8")
sign_string = base64_data + HELEKET_API_KEY
calculated_signature = hashlib.md5(
sign_string.encode("utf-8")
).hexdigest()
calculated_signature = hashlib.md5(sign_string.encode("utf-8")).hexdigest()
is_valid = calculated_signature.lower() == received_signature.lower()
if not is_valid:
logger.error(
f"Heleket webhook: неверная подпись. "
f"Ожидалось: {calculated_signature}, "
f"получено: {received_signature}"
f"Heleket webhook: неверная подпись. Ожидалось: {calculated_signature}, получено: {received_signature}"
)
logger.error(f"Heleket webhook: строка для подписи: {sign_string}")
else:
@@ -82,18 +74,12 @@ async def process_heleket_webhook(data: dict) -> bool:
payer_currency = data.get("payer_currency")
additional_data = data.get("additional_data")
logger.info(
f"Heleket webhook - Type: {webhook_type}, "
f"Order: {order_id}, Status: {status}"
)
logger.info(f"Heleket webhook - Type: {webhook_type}, Order: {order_id}, Status: {status}")
if webhook_type != "payment":
logger.warning(f"Heleket webhook: неизвестный тип {webhook_type}")
return False
if status in ["paid", "paid_over"]:
logger.info(
f"Heleket: успешный платёж {order_id} на сумму "
f"{payment_amount} {payer_currency}"
)
logger.info(f"Heleket: успешный платёж {order_id} на сумму {payment_amount} {payer_currency}")
tg_id = None
rub_amount = None
if additional_data:
@@ -109,24 +95,20 @@ async def process_heleket_webhook(data: dict) -> bool:
try:
tg_id = int(order_id.split("_")[1])
except Exception as e:
logger.error(
f"Ошибка извлечения tg_id из order_id: {e}"
)
logger.error(f"Ошибка извлечения tg_id из order_id: {e}")
if not tg_id:
logger.error(
f"Не удалось извлечь tg_id из Heleket webhook: {data}"
)
logger.error(f"Не удалось извлечь tg_id из Heleket webhook: {data}")
return False
balance_amount = (
rub_amount if rub_amount else float(merchant_amount)
)
balance_amount = rub_amount if rub_amount else float(merchant_amount)
async with async_session_maker() as session:
payment = await get_payment_by_payment_id(session, order_id)
if payment:
if payment.get("status") == "success":
logger.info(f"Heleket: платёж {order_id} уже обработан")
return True
ok = await update_payment_status(session=session, internal_id=int(payment["id"]), new_status="success")
ok = await update_payment_status(
session=session, internal_id=int(payment["id"]), new_status="success"
)
if not ok:
logger.error(f"Heleket: не удалось обновить статус платежа {order_id}")
return False
@@ -150,9 +132,7 @@ async def process_heleket_webhook(data: dict) -> bool:
)
return True
elif status in ["fail", "wrong_amount", "cancel", "system_fail"]:
logger.warning(
f"Heleket: неудачный платёж {order_id}, статус: {status}"
)
logger.warning(f"Heleket: неудачный платёж {order_id}, статус: {status}")
async with async_session_maker() as session:
payment = await get_payment_by_payment_id(session, order_id)
@@ -165,10 +145,7 @@ async def process_heleket_webhook(data: dict) -> bool:
await session.commit()
return True
else:
logger.info(
f"Heleket: промежуточный статус {status} "
f"для платежа {order_id}"
)
logger.info(f"Heleket: промежуточный статус {status} для платежа {order_id}")
return True
except Exception as e:
logger.error(f"Ошибка обработки Heleket webhook: {e}")
handlers/payments/kassai/handlers.py
+11 -39
View File
@@ -31,9 +31,7 @@ async def handle_pay_kassai_cards(
session: AsyncSession,
):
"""Обработчик оплаты через KassaI картами."""
await process_callback_pay_kassai(
callback_query, state, session, method_name="cards"
)
await process_callback_pay_kassai(callback_query, state, session, method_name="cards")
@router.callback_query(F.data == "pay_kassai_sbp")
@@ -43,9 +41,7 @@ async def handle_pay_kassai_sbp(
session: AsyncSession,
):
"""Обработчик оплаты через KassaI СБП."""
await process_callback_pay_kassai(
callback_query, state, session, method_name="sbp"
)
await process_callback_pay_kassai(callback_query, state, session, method_name="sbp")
async def _handle_custom_amount_input_kassai(
@@ -94,10 +90,7 @@ async def _handle_custom_amount_input_kassai(
if amount < min_amount:
method_label = method_labels.get(method_name, "")
error_msg = (
f"❌ Минимальная сумма для оплаты {method_label} — "
f"{min_amount}₽."
)
error_msg = f"❌ Минимальная сумма для оплаты {method_label} — {min_amount}₽."
await edit_or_send_message(
target_message=message,
text=error_msg,
@@ -114,34 +107,23 @@ async def _handle_custom_amount_input_kassai(
return
try:
payment_url = await generate_kassai_payment_link(
amount, tg_id, method
)
payment_url = await generate_kassai_payment_link(amount, tg_id, method)
if not payment_url or payment_url == "https://fk.life/":
await edit_or_send_message(
target_message=message,
text=(
"❌ Произошла ошибка при создании платежа. "
"Попробуйте позже или выберите другой способ оплаты."
),
text=("❌ Произошла ошибка при создании платежа. Попробуйте позже или выберите другой способ оплаты."),
)
return
markup = InlineKeyboardMarkup(
inline_keyboard=[
[InlineKeyboardButton(text=pay_button_text, url=payment_url)],
[
InlineKeyboardButton(
text=main_menu_text, callback_data="profile"
)
],
[InlineKeyboardButton(text=main_menu_text, callback_data="profile")],
]
)
result = await session.execute(
select(User.language_code).where(User.tg_id == tg_id)
)
result = await session.execute(select(User.language_code).where(User.tg_id == tg_id))
language_code = result.scalar_one_or_none()
amount_text = await format_for_user(
session,
@@ -152,15 +134,10 @@ async def _handle_custom_amount_input_kassai(
)
text_out = DEFAULT_PAYMENT_MESSAGE.format(amount=amount_text)
await edit_or_send_message(
target_message=message, text=text_out, reply_markup=markup
)
await edit_or_send_message(target_message=message, text=text_out, reply_markup=markup)
except Exception as e:
method_label = "Cards" if method_name == "cards" else "SBP"
logger.error(
f"Ошибка при создании платежа KassaAI {method_label} "
f"для пользователя {tg_id}: {e}"
)
logger.error(f"Ошибка при создании платежа KassaAI {method_label} для пользователя {tg_id}: {e}")
await edit_or_send_message(
target_message=message,
text="Произошла ошибка при создании платежа. Попробуйте позже.",
@@ -175,9 +152,7 @@ async def handle_custom_amount_input_kassai_cards(
main_menu_text: str = MAIN_MENU,
):
"""Функция быстрого потока для KassaI Cards."""
await _handle_custom_amount_input_kassai(
event, session, "cards", pay_button_text, main_menu_text
)
await _handle_custom_amount_input_kassai(event, session, "cards", pay_button_text, main_menu_text)
async def handle_custom_amount_input_kassai_sbp(
@@ -187,7 +162,4 @@ async def handle_custom_amount_input_kassai_sbp(
main_menu_text: str = MAIN_MENU,
):
"""Функция быстрого потока для KassaI SBP."""
await _handle_custom_amount_input_kassai(
event, session, "sbp", pay_button_text, main_menu_text
)
await _handle_custom_amount_input_kassai(event, session, "sbp", pay_button_text, main_menu_text)
handlers/payments/kassai/service.py
+17 -16
View File
@@ -49,9 +49,7 @@ router = Router()
async def get_user_language(session: AsyncSession, tg_id: int) -> str | None:
"""Получает язык пользователя из базы данных."""
result = await session.execute(
select(User.language_code).where(User.tg_id == tg_id)
)
result = await session.execute(select(User.language_code).where(User.tg_id == tg_id))
return result.scalar_one_or_none()
@@ -65,8 +63,18 @@ class ReplenishBalanceKassaiState(StatesGroup):
KASSAI_METHODS = {
"cards": {"enable": PROVIDERS_ENABLED.get("KASSAI_CARDS", False), "method": 36, "button": KASSAI_CARDS, "desc": KASSAI_CARDS_DESCRIPTION},
"sbp": {"enable": PROVIDERS_ENABLED.get("KASSAI_SBP", False), "method": 44, "button": KASSAI_SBP, "desc": KASSAI_SBP_DESCRIPTION},
"cards": {
"enable": PROVIDERS_ENABLED.get("KASSAI_CARDS", False),
"method": 36,
"button": KASSAI_CARDS,
"desc": KASSAI_CARDS_DESCRIPTION,
},
"sbp": {
"enable": PROVIDERS_ENABLED.get("KASSAI_SBP", False),
"method": 44,
"button": KASSAI_SBP,
"desc": KASSAI_SBP_DESCRIPTION,
},
}
@@ -94,9 +102,7 @@ async def process_callback_pay_kassai(
return
language_code = await get_user_language(session, tg_id)
opts = await payment_options_for_user(
session, tg_id, language_code, force_currency="RUB"
)
opts = await payment_options_for_user(session, tg_id, language_code, force_currency="RUB")
builder = build_amounts_keyboard(
prefix=f"kassai_{method_name}",
pattern="{prefix}_amount|{price}",
@@ -141,10 +147,7 @@ async def process_callback_pay_kassai(
await state.set_state(ReplenishBalanceKassaiState.choosing_method)
except Exception as e:
logger.error(
f"Error in process_callback_pay_kassai for user "
f"{callback_query.message.chat.id}: {e}"
)
logger.error(f"Error in process_callback_pay_kassai for user {callback_query.message.chat.id}: {e}")
await callback_query.answer(
"Произошла ошибка при инициализации платежа. Попробуйте позже.",
show_alert=True,
@@ -168,15 +171,13 @@ async def process_method_selection(callback_query: types.CallbackQuery, state: F
tg_id = callback_query.from_user.id
language_code = await get_user_language(session, tg_id)
opts = await payment_options_for_user(
session, tg_id, language_code, force_currency="RUB"
)
opts = await payment_options_for_user(session, tg_id, language_code, force_currency="RUB")
builder = build_amounts_keyboard(
prefix=f"kassai_{method_name}",
pattern="{prefix}_amount|{price}",
back_cb="pay_kassai",
custom_cb=f"kassai_custom_amount|{method_name}",
opts=opts
opts=opts,
)
await edit_or_send_message(
handlers/payments/kassai/webhook.py
+7 -22
View File
@@ -26,18 +26,12 @@ def verify_kassai_signature(data: dict, signature: str) -> bool:
"""
try:
sign_string = (
f"{KASSAI_SHOP_ID}:{data.get('AMOUNT', '')}:"
f"{KASSAI_SECRET_KEY}:{data.get('MERCHANT_ORDER_ID', '')}"
f"{KASSAI_SHOP_ID}:{data.get('AMOUNT', '')}:{KASSAI_SECRET_KEY}:{data.get('MERCHANT_ORDER_ID', '')}"
)
expected_signature = hashlib.md5(
sign_string.encode("utf-8")
).hexdigest()
expected_signature = hashlib.md5(sign_string.encode("utf-8")).hexdigest()
result = signature.upper() == expected_signature.upper()
if not result:
logger.error(
f"KassaAI signature mismatch. "
f"Expected: {expected_signature}, Got: {signature}"
)
logger.error(f"KassaAI signature mismatch. Expected: {expected_signature}, Got: {signature}")
logger.error(f"Sign string: {sign_string}")
else:
logger.info("KassaAI webhook: подпись успешно проверена")
@@ -64,9 +58,7 @@ async def kassai_webhook(request: web.Request):
order_id = data.get("MERCHANT_ORDER_ID")
if not amount_raw or not order_id:
logger.error(
"KassaAI webhook: отсутствуют обязательные параметры"
)
logger.error("KassaAI webhook: отсутствуют обязательные параметры")
return web.Response(status=400)
amount = float(amount_raw)
@@ -74,16 +66,10 @@ async def kassai_webhook(request: web.Request):
try:
tg_id = int(order_id.split("_")[1])
except (IndexError, ValueError) as e:
logger.error(
f"KassaAI webhook: не удалось извлечь tg_id "
f"из order_id {order_id}: {e}"
)
logger.error(f"KassaAI webhook: не удалось извлечь tg_id из order_id {order_id}: {e}")
return web.Response(status=400)
logger.info(
f"KassaAI: успешный платёж {order_id} на сумму {amount} RUB "
f"для пользователя {tg_id}"
)
logger.info(f"KassaAI: успешный платёж {order_id} на сумму {amount} RUB для пользователя {tg_id}")
async with async_session_maker() as session:
payment = await get_payment_by_payment_id(session, order_id)
@@ -110,8 +96,7 @@ async def kassai_webhook(request: web.Request):
await update_balance(session, tg_id, amount)
await send_payment_success_notification(tg_id, amount, session)
logger.info(
f"KassaAI: платёж {order_id} успешно обработан, "
f"баланс пользователя {tg_id} пополнен на {amount} RUB"
f"KassaAI: платёж {order_id} успешно обработан, баланс пользователя {tg_id} пополнен на {amount} RUB"
)
return web.Response(text=KASSAI_WEBHOOK_RESPONSE)
except Exception as e:
handlers/payments/pay.py
+1 -3
View File
@@ -217,9 +217,7 @@ async def balance_history_handler(callback_query: CallbackQuery, session: Any):
payment_system = record["payment_system"]
status = record["status"]
date = record["created_at"].strftime("%Y-%m-%d %H:%M:%S")
history_text += (
f"Сумма: {formatted_amount}\nОплата: {payment_system}\nСтатус: {status}\nДата: {date}\n\n"
)
history_text += f"Сумма: {formatted_amount}\nОплата: {payment_system}\nСтатус: {status}\nДата: {date}\n\n"
history_text += "</blockquote>"
else:
history_text = "❌ У вас пока нет операций с балансом."
handlers/tariffs/addons/key_addons_pack.py
+3 -1
View File
@@ -788,7 +788,9 @@ async def handle_addons_confirm(callback: CallbackQuery, state: FSMContext, sess
if has_device_option and selected_devices is not None:
pack_devices_val = int(selected_devices)
if pack_devices_val <= 0 or (new_device_limit_effective is not None and new_device_limit_effective <= 0):
if pack_devices_val <= 0 or (
new_device_limit_effective is not None and new_device_limit_effective <= 0
):
new_device_limit_effective = 0
else:
if new_device_limit_effective is None:
handlers/tariffs/subgroup_migration.py
+1 -3
View File
@@ -244,9 +244,7 @@ async def migrate_between_subgroups(
external_squad_uuid: str | None = None,
tariff_id: int | None = None,
) -> tuple[str, str | None]:
target = await filter_cluster_by_subgroup(
session, cluster_all, target_subgroup, cluster_id, tariff_id=tariff_id
)
target = await filter_cluster_by_subgroup(session, cluster_all, target_subgroup, cluster_id, tariff_id=tariff_id)
xui_tgt, remna_tgt = split_by_panel(target)
old_set = await filter_cluster_by_subgroup(session, cluster_all, old_subgroup, cluster_id)
img/store.png
BIN
View File
Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 143 KiB

main.py
+4 -1
View File
File diff suppressed because one or more lines are too long
middlewares/user.py
+3 -8
View File
@@ -70,10 +70,7 @@ class UserMiddleware(BaseMiddleware):
async def _touch_user(self, tg_id: int, session: AsyncSession) -> dict | None:
now = datetime.utcnow()
res = await session.execute(
update(DbUser)
.where(DbUser.tg_id == tg_id)
.values(updated_at=now)
.returning(DbUser)
update(DbUser).where(DbUser.tg_id == tg_id).values(updated_at=now).returning(DbUser)
)
obj = res.scalar_one_or_none()
if obj is None:
@@ -84,13 +81,11 @@ class UserMiddleware(BaseMiddleware):
return d
def _fingerprint(self, user: User) -> str:
return "|".join(
[
return "|".join([
str(user.id),
user.username or "",
user.first_name or "",
user.last_name or "",
user.language_code or "",
"1" if user.is_bot else "0",
]
)
])