Compare commits
66 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 50dc5a0fd1 | |||
| 4dc8b4c091 | |||
| 4db9e85062 | |||
| fca8d6da97 | |||
| 5009703676 | |||
| e88a5989b6 | |||
| 02747381dc | |||
| e74fda954c | |||
| 8587f03f67 | |||
| 4707cdf60c | |||
| 0879b8b218 | |||
| 1d91382b8e | |||
| 3768b18a39 | |||
| 1eeeb39779 | |||
| 570af82dfd | |||
| bc3893b934 | |||
| 16d91638bc | |||
| eb18b3a0f9 | |||
| a8e2b62f4b | |||
| fb8d2b3ee4 | |||
| 2321667ecb | |||
| 113304b212 | |||
| 0300044b00 | |||
| 931abfe7a5 | |||
| 1d96f80f60 | |||
| 4e50419171 | |||
| 7208a52c94 | |||
| 63fdfe4a42 | |||
| e0e2edf816 | |||
| 522a8779d6 | |||
| be32010d63 | |||
| 7e920fa30f | |||
| 1b376baeca | |||
| 91a756a33e | |||
| 970dc549df | |||
| 65120f0bad | |||
| 9cb559ff39 | |||
| 8f1882f24c | |||
| 8542a39305 | |||
| 646ac4cfa1 | |||
| abdf296767 | |||
| a1b6d9bb61 | |||
| cf19e4e1f7 | |||
| 35412e9f21 | |||
| 6aed7d355b | |||
| 9c08ce6948 | |||
| 862352139e | |||
| d465ccb3ac | |||
| b57f185258 | |||
| ffbb3fb8be | |||
| f01dbff000 | |||
| 31adcfded4 | |||
| 78f963bf5e | |||
| 357d94d1b0 | |||
| 0fb4a2c235 | |||
| 2f7184627a | |||
| d55e9db62a | |||
| 57adfaf4f3 | |||
| 4165eaea7a | |||
| 3b5d5a18a1 | |||
| eef41c4bca | |||
| 987c3c93c2 | |||
| 7d24e8d704 | |||
| 819f09a68e | |||
| 2f9d00343b | |||
| 9b7ac47f16 |
+5
-1
@@ -13,11 +13,15 @@ SUPPORT_USERNAME=@support
|
||||
# Имя пользователя бота (опционально, автоопределяется)
|
||||
# BOT_USERNAME=
|
||||
|
||||
# ===== SOCKS5 ПРОКСИ =====
|
||||
# ===== СЕТЬ И ПРОКСИ =====
|
||||
# URL SOCKS5 прокси-сервера для маршрутизации трафика бота к Telegram API
|
||||
# Формат: socks5://user:password@host:port или socks5://host:port
|
||||
# PROXY_URL=socks5://127.0.0.1:1080
|
||||
|
||||
# Альтернативный URL сервера Telegram Bot API (для регионов где api.telegram.org заблокирован)
|
||||
# Примеры: Cloudflare Worker, self-hosted telegram-bot-api (tdlib), любой совместимый прокси
|
||||
# TELEGRAM_API_URL=https://your-telegram-proxy.workers.dev
|
||||
|
||||
# ===== СИСТЕМА ПОДДЕРЖКИ =====
|
||||
# Включить меню поддержки в интерфейсе
|
||||
SUPPORT_MENU_ENABLED=true
|
||||
|
||||
@@ -26,36 +26,38 @@ jobs:
|
||||
- name: Get version info
|
||||
id: version
|
||||
run: |
|
||||
echo "short_sha=$(git rev-parse --short HEAD)" >> $GITHUB_OUTPUT
|
||||
|
||||
SHORT_SHA=$(git rev-parse --short HEAD)
|
||||
echo "short_sha=$SHORT_SHA" >> $GITHUB_OUTPUT
|
||||
echo "build_date=$(date -u +'%Y-%m-%dT%H:%M:%SZ')" >> $GITHUB_OUTPUT
|
||||
|
||||
# Определяем версию и теги
|
||||
|
||||
# Read base version from release-please manifest (single source of truth)
|
||||
BASE_VERSION=$(jq -r '."."' .release-please-manifest.json)
|
||||
|
||||
if [[ $GITHUB_REF == refs/tags/* ]]; then
|
||||
VERSION=${GITHUB_REF#refs/tags/}
|
||||
TAGS="fr1ngg/remnawave-bedolaga-telegram-bot:latest,fr1ngg/remnawave-bedolaga-telegram-bot:${VERSION}"
|
||||
echo "🏷️ Собираем релизную версию: $VERSION"
|
||||
elif [[ $GITHUB_REF == refs/heads/main ]]; then
|
||||
VERSION="v3.7.0-$(git rev-parse --short HEAD)" # x-release-please-version
|
||||
VERSION="v${BASE_VERSION}-${SHORT_SHA}"
|
||||
TAGS="fr1ngg/remnawave-bedolaga-telegram-bot:latest,fr1ngg/remnawave-bedolaga-telegram-bot:${VERSION}"
|
||||
echo "🚀 Собираем версию из main: $VERSION"
|
||||
elif [[ $GITHUB_REF == refs/heads/dev ]]; then
|
||||
VERSION="v3.7.0-dev-$(git rev-parse --short HEAD)" # x-release-please-version
|
||||
VERSION="v${BASE_VERSION}-dev-${SHORT_SHA}"
|
||||
TAGS="fr1ngg/remnawave-bedolaga-telegram-bot:dev,fr1ngg/remnawave-bedolaga-telegram-bot:${VERSION}"
|
||||
echo "🧪 Собираем dev версию: $VERSION"
|
||||
else
|
||||
VERSION="v3.7.0-pr-$(git rev-parse --short HEAD)" # x-release-please-version
|
||||
TAGS="fr1ngg/remnawave-bedolaga-telegram-bot:pr-$(git rev-parse --short HEAD)"
|
||||
VERSION="v${BASE_VERSION}-pr-${SHORT_SHA}"
|
||||
TAGS="fr1ngg/remnawave-bedolaga-telegram-bot:pr-${SHORT_SHA}"
|
||||
echo "🔀 Собираем PR версию: $VERSION"
|
||||
fi
|
||||
|
||||
|
||||
echo "version=$VERSION" >> $GITHUB_OUTPUT
|
||||
echo "tags=$TAGS" >> $GITHUB_OUTPUT
|
||||
echo "should_push=${{ github.event_name != 'pull_request' }}" >> $GITHUB_OUTPUT
|
||||
|
||||
|
||||
echo "=== Информация о сборке ==="
|
||||
echo "Версия: $VERSION"
|
||||
echo "Коммит: $(git rev-parse --short HEAD)"
|
||||
echo "Коммит: $SHORT_SHA"
|
||||
echo "Теги: $TAGS"
|
||||
echo "Push: ${{ github.event_name != 'pull_request' }}"
|
||||
echo "==========================="
|
||||
|
||||
@@ -42,25 +42,28 @@ jobs:
|
||||
- name: Get version info
|
||||
id: version
|
||||
run: |
|
||||
echo "short_sha=$(git rev-parse --short HEAD)" >> $GITHUB_OUTPUT
|
||||
SHORT_SHA=$(git rev-parse --short HEAD)
|
||||
echo "short_sha=$SHORT_SHA" >> $GITHUB_OUTPUT
|
||||
echo "build_date=$(date -u +'%Y-%m-%dT%H:%M:%SZ')" >> $GITHUB_OUTPUT
|
||||
|
||||
|
||||
# Read base version from release-please manifest (single source of truth)
|
||||
BASE_VERSION=$(jq -r '."."' .release-please-manifest.json)
|
||||
|
||||
if [[ $GITHUB_REF == refs/tags/* ]]; then
|
||||
VERSION=${GITHUB_REF#refs/tags/}
|
||||
echo "🏷️ Building release version: $VERSION"
|
||||
elif [[ $GITHUB_REF == refs/heads/main ]]; then
|
||||
VERSION="v3.7.0-$(git rev-parse --short HEAD)" # x-release-please-version
|
||||
VERSION="v${BASE_VERSION}-${SHORT_SHA}"
|
||||
echo "🚀 Building main version: $VERSION"
|
||||
elif [[ $GITHUB_REF == refs/heads/dev ]]; then
|
||||
VERSION="v3.7.0-dev-$(git rev-parse --short HEAD)" # x-release-please-version
|
||||
VERSION="v${BASE_VERSION}-dev-${SHORT_SHA}"
|
||||
echo "🧪 Building dev version: $VERSION"
|
||||
else
|
||||
VERSION="v3.7.0-pr-$(git rev-parse --short HEAD)" # x-release-please-version
|
||||
VERSION="v${BASE_VERSION}-pr-${SHORT_SHA}"
|
||||
echo "🔀 Building PR version: $VERSION"
|
||||
fi
|
||||
echo "version=$VERSION" >> $GITHUB_OUTPUT
|
||||
|
||||
# Определяем, нужно ли пушить образ
|
||||
|
||||
if [[ "${{ github.event_name }}" == "pull_request" ]]; then
|
||||
echo "should_push=false" >> $GITHUB_OUTPUT
|
||||
echo "⚠️ PR - only build without push"
|
||||
|
||||
@@ -1,3 +1,3 @@
|
||||
{
|
||||
".": "3.44.0"
|
||||
".": "3.47.0"
|
||||
}
|
||||
|
||||
@@ -1,5 +1,98 @@
|
||||
# Changelog
|
||||
|
||||
## [3.47.0](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/compare/v3.46.1...v3.47.0) (2026-04-15)
|
||||
|
||||
|
||||
### New Features
|
||||
|
||||
* multi-tariff sync fix, daily discount fix, campaign links, TELEGRAM_API_URL ([4db9e85](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/4db9e850629f25cbcb11bb5ba0e0de5c580ca115))
|
||||
|
||||
## [3.46.1](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/compare/v3.46.0...v3.46.1) (2026-04-13)
|
||||
|
||||
|
||||
### Bug Fixes
|
||||
|
||||
* add checkfirst guards to cabinet_refresh_tokens migration ([8587f03](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/8587f03f67d7a451b07a4d9c450bc4524f4ac0e7))
|
||||
* add missing migration for cabinet_refresh_tokens table ([4707cdf](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/4707cdf60c9d163c1191719b3b1fc4a17ae993d2))
|
||||
* cabinet_refresh_tokens migration + notification_settings jsonb ([0274738](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/02747381dce2b96af7f97b5f41d6acff5d9d8fd3))
|
||||
* change notification_settings from json to jsonb for DISTINCT compatibility ([e74fda9](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/e74fda954ccc63e2ac7a30933d9f9ac26772b25d))
|
||||
|
||||
## [3.46.0](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/compare/v3.45.2...v3.46.0) (2026-04-13)
|
||||
|
||||
|
||||
### New Features
|
||||
|
||||
* add broadcast category (system/news/promo) + filter recipients by user prefs ([931abfe](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/931abfe7a5a7fb70e9638fbf6b566fa8d1a837e4))
|
||||
* add category field to broadcast API schemas and routes ([0300044](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/0300044b009f3e4b3aa3928652dfaf261a387dbc))
|
||||
* add RemnaWave retry queue for failed API calls (BUG-2, BUG-10) ([abdf296](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/abdf2967675975e90f0c4d834f129281c1c28e7b))
|
||||
* add remnawave_resync_service for identity-change sync ([b57f185](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/b57f185258be050d945cec5989ad6dc710980a6a))
|
||||
* add traffic % warning check using user's threshold preference ([1d96f80](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/1d96f80f60ca445eb5108e7bc54e00d022a4cc9e))
|
||||
* add user notification preferences helper utility ([e0e2edf](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/e0e2edf81659fbeea361046d1bb2718c2149d884))
|
||||
* implement low balance alert + respect user notification preferences ([4e50419](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/4e50419171176ee452371ff095bdfead3879e554))
|
||||
* respect user subscription_expiry notification preferences ([63fdfe4](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/63fdfe4a421942b26caca35d8bd9b1d65f1fe7e2))
|
||||
* respect user traffic_warning notification preference in webhook handler ([7208a52](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/7208a52c9424d39757187fc86eec2c3460a2cdbb))
|
||||
* save campaign_slug during standalone email registration ([a8e2b62](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/a8e2b62f4bb0833ca32446b34ad4e0c5615fcd2a))
|
||||
* start RemnaWave retry queue on app startup ([8f1882f](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/8f1882f24c7d066e2d0fc756f38ce23bf082687e))
|
||||
|
||||
|
||||
### Bug Fixes
|
||||
|
||||
* add retry queue to all remaining RemnaWave error handlers ([7e920fa](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/7e920fa30fc8e61ed2dd31e7a09151d57b7361ca))
|
||||
* add retry queue to cabinet subscription operation RemnaWave errors ([1b376ba](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/1b376baeca120970b1ffcd406b1bbf7d9d43cee0))
|
||||
* add retry queue to classic mode bot purchase handler ([970dc54](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/970dc549dfa06ba9945d5bd861374058acdca86b))
|
||||
* add retry queue to daily subscription service RemnaWave errors ([65120f0](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/65120f0badc9a4581ba0a127acdae8a0b23e8501))
|
||||
* add retry queue to payment webhook and renewal service RemnaWave errors ([91a756a](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/91a756a33ed4ce685bdf485cdb4e91c3e08799dd))
|
||||
* add TRAFFIC_WARNING_ALERT and LOW_BALANCE_ALERT localization keys to all locales ([2321667](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/2321667ecbe76bfe8dd37213e0bb4e45104e0fc5))
|
||||
* always sync squads in auto-purchase renewal (BUG-4) ([8542a39](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/8542a393055a93d320d5c8c6d3aa7cc291cf8def))
|
||||
* default sync_squads=True in update_remnawave_user (BUG-4) ([6aed7d3](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/6aed7d355bc47c4dbd2aa761d78a5e5421c32edf))
|
||||
* enforce max_attempts limit in NaloGO receipt queue ([16d9163](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/16d91638bc149c5eee8f4cdd266cbd195c411030))
|
||||
* enqueue retry on RemnaWave API failure in all purchase flows (BUG-2, BUG-10) ([9cb559f](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/9cb559ff3994c0f1c6ba48a4ec09dec9b391e48b))
|
||||
* exclude users with active subscriptions from expired broadcast ([1eeeb39](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/1eeeb39779982ebb2700cb7523760631229407da))
|
||||
* handle TelegramBadRequest when deleting old ticket notifications ([eb18b3a](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/eb18b3a0f9ac3a617a1b21d3293e1619980a2a71))
|
||||
* match tariff_id when creating subscriptions from panel sync (BUG-11) ([646ac4c](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/646ac4cfa18f738040fbc6498c5d86c1546e2b9a))
|
||||
* protect OAuth users with remnawave_uuid from sync deactivation (BUG-6) ([cf19e4e](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/cf19e4e1f7148b21d9a8072f7a0b4ae97fd04e8a))
|
||||
* raise MAX_BUTTONS_PER_ROW to 8 and allow tg:// deep links in menu editor ([570af82](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/570af82dfdec980f42be96c8f816e1677f896f81))
|
||||
* resync RemnaWave after account merge (BUG-7) ([9c08ce6](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/9c08ce69485b78f8fe818500e05ab6995115166a))
|
||||
* resync RemnaWave after Telegram account linking (BUG-1) ([d465ccb](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/d465ccb3ac3a86add6313d6bb61d53d0fe143d5e))
|
||||
* sync connected_squads from panel during sync (BUG-5) ([35412e9](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/35412e9f215680c0fdf1c55b5bf23496f662935c))
|
||||
* trial activation fallback to trial-eligible servers when tariff has no squads (BUG-12) + fix misleading button text ([be32010](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/be32010d63966498bc6216823a75d328346d9a37))
|
||||
* upsert refresh tokens (ON CONFLICT) + periodic cleanup of expired/revoked tokens ([fb8d2b3](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/fb8d2b3ee4566823840100b96fe2f3bc7d41edb7))
|
||||
* use 'is not None' for telegram_id in create_user API (BUG-9) ([8623521](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/862352139e8a3545e144b0618fed5011470a9a67))
|
||||
* use MAX_DEVICES_LIMIT instead of hardcoded 10 for device buttons ([bc3893b](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/bc3893b934f0d4e5059eedbd03cdfbc628852ac1))
|
||||
* use update_remnawave_user when UUID exists in tariff_purchase (BUG-3) ([a1b6d9b](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/a1b6d9bb619ec3de038647fe6f2e5d38979298a8))
|
||||
|
||||
## [3.45.2](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/compare/v3.45.1...v3.45.2) (2026-04-08)
|
||||
|
||||
|
||||
### Bug Fixes
|
||||
|
||||
* batch bug fixes from user complaints ([31adcfd](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/31adcfded4b161bf515d4d6b25b4395e543208f4))
|
||||
* batch bug fixes from user complaints ([78f963b](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/78f963bf5e7b3439d7614584c4041f88be5beb4a))
|
||||
* исправление парсинга черного списка (поддержка '#' и извлечение username) ([357d94d](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/357d94d1b0d7fc8036b00cbb6b75175c29821751))
|
||||
* исправление парсинга черного списка (поддержка '#' и извлечение username) ([2f71846](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/2f7184627a0fb598a8c0208905cdecf0e4bb04a7))
|
||||
|
||||
## [3.45.1](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/compare/v3.45.0...v3.45.1) (2026-04-03)
|
||||
|
||||
|
||||
### Bug Fixes
|
||||
|
||||
* add missing WEBHOOK_TORRENT_DETECTED mapping + dedup before uniq… ([4165eae](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/4165eaea7adfdaf683b1ece16c8c93a9c4ed216d))
|
||||
* add missing WEBHOOK_TORRENT_DETECTED mapping + dedup before unique index in migration 0053 ([3b5d5a1](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/3b5d5a18a1122ef50868fd038a09109d17795a74))
|
||||
|
||||
## [3.45.0](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/compare/v3.44.0...v3.45.0) (2026-04-03)
|
||||
|
||||
|
||||
### New Features
|
||||
|
||||
* send torrent blocker notification to user (not just admin) ([2f9d003](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/2f9d00343bee2980cc89bd24361259073b97127a))
|
||||
|
||||
|
||||
### Bug Fixes
|
||||
|
||||
* resolve multiple subscription bugs — LIMITED status, trial tariff blocking, traffic reset strategy, classic mode pricing, 100% discount support ([9b7ac47](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/9b7ac47f16076e546da62062ff7ce18d7c308988))
|
||||
* restore missing import + rewrite user.deleted webhook to properly deactivate all subscriptions ([819f09a](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/819f09a68ec95237294bae97f31c644044a3623f))
|
||||
* subscription system bugfixes + torrent notifications + user deletion cleanup ([7d24e8d](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/commit/7d24e8d7047c7a3a1c417e655a6fbccbe5ae577d))
|
||||
|
||||
## [3.44.0](https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot/compare/v3.43.1...v3.44.0) (2026-04-02)
|
||||
|
||||
|
||||
|
||||
+1
-1
@@ -19,7 +19,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
|
||||
|
||||
FROM python:3.13-slim
|
||||
|
||||
ARG VERSION="v3.44.0" # x-release-please-version
|
||||
ARG VERSION="v3.47.0" # x-release-please-version
|
||||
ARG BUILD_DATE
|
||||
ARG VCS_REF
|
||||
|
||||
|
||||
+16
@@ -280,12 +280,28 @@ async def setup_bot() -> tuple[Bot, Dispatcher]:
|
||||
except Exception as e:
|
||||
logger.warning('Failed to load menu layout cache', error=e)
|
||||
|
||||
try:
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
await remnawave_retry_queue.start()
|
||||
logger.info('RemnaWave retry queue запущен')
|
||||
except Exception as e:
|
||||
logger.error('Ошибка запуска RemnaWave retry queue', error=e)
|
||||
|
||||
logger.info('Бот успешно настроен')
|
||||
|
||||
return bot, dp
|
||||
|
||||
|
||||
async def shutdown_bot():
|
||||
try:
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
await remnawave_retry_queue.stop()
|
||||
logger.info('RemnaWave retry queue остановлен')
|
||||
except Exception as e:
|
||||
logger.error('Ошибка остановки RemnaWave retry queue', error=e)
|
||||
|
||||
try:
|
||||
await maintenance_service.stop_monitoring()
|
||||
logger.info('Мониторинг техработ остановлен')
|
||||
|
||||
+12
-4
@@ -1,4 +1,4 @@
|
||||
"""Factory for creating Bot instances with proxy support."""
|
||||
"""Factory for creating Bot instances with proxy and custom API server support."""
|
||||
|
||||
from aiogram import Bot
|
||||
from aiogram.client.default import DefaultBotProperties
|
||||
@@ -8,13 +8,21 @@ from app.config import settings
|
||||
|
||||
|
||||
def create_bot(token: str | None = None, **kwargs) -> Bot:
|
||||
"""Create a Bot instance with SOCKS5 proxy session if PROXY_URL is configured."""
|
||||
"""Create a Bot instance with SOCKS5 proxy and/or custom Telegram API server."""
|
||||
proxy_url = settings.get_proxy_url()
|
||||
telegram_api_url = settings.get_telegram_api_url()
|
||||
session = None
|
||||
if proxy_url:
|
||||
if proxy_url or telegram_api_url:
|
||||
from aiogram.client.session.aiohttp import AiohttpSession
|
||||
from aiogram.client.telegram import TelegramAPIServer
|
||||
|
||||
session = AiohttpSession(proxy=proxy_url)
|
||||
session_kwargs: dict = {}
|
||||
if proxy_url:
|
||||
session_kwargs['proxy'] = proxy_url
|
||||
if telegram_api_url:
|
||||
session_kwargs['api'] = TelegramAPIServer.from_base(telegram_api_url)
|
||||
|
||||
session = AiohttpSession(**session_kwargs)
|
||||
|
||||
kwargs.setdefault('default', DefaultBotProperties(parse_mode=ParseMode.HTML))
|
||||
return Bot(token=token or settings.BOT_TOKEN, session=session, **kwargs)
|
||||
|
||||
@@ -195,7 +195,8 @@ async def _get_jwks(force: bool = False) -> dict[str, Any]:
|
||||
if not force and _jwks_cache and _jwks_cache_expiry and now < _jwks_cache_expiry:
|
||||
return _jwks_cache
|
||||
|
||||
async with httpx.AsyncClient(timeout=10) as client:
|
||||
proxy = settings.PROXY_URL if hasattr(settings, 'PROXY_URL') and settings.PROXY_URL else None
|
||||
async with httpx.AsyncClient(timeout=10, proxy=proxy) as client:
|
||||
response = await client.get(_JWKS_URL)
|
||||
response.raise_for_status()
|
||||
_jwks_cache = response.json()
|
||||
|
||||
@@ -622,6 +622,24 @@ async def link_telegram(
|
||||
telegram_id=telegram_id,
|
||||
user_id=user.id,
|
||||
)
|
||||
# BUG-1 fix: Sync all subscriptions with RemnaWave panel so it knows the new telegram_id
|
||||
try:
|
||||
from app.services.remnawave_resync_service import resync_user_subscriptions_with_panel
|
||||
|
||||
resync_result = await resync_user_subscriptions_with_panel(db, user)
|
||||
logger.info(
|
||||
'Post-TG-link resync completed',
|
||||
user_id=user.id,
|
||||
telegram_id=telegram_id,
|
||||
synced=resync_result['synced'],
|
||||
failed=resync_result['failed'],
|
||||
)
|
||||
except Exception as resync_error:
|
||||
logger.error(
|
||||
'Post-TG-link resync failed (non-fatal)',
|
||||
user_id=user.id,
|
||||
error=resync_error,
|
||||
)
|
||||
return LinkCallbackResponse(success=True, message='linked')
|
||||
|
||||
|
||||
@@ -867,6 +885,25 @@ async def execute_merge_endpoint(
|
||||
detail='Failed to load merged user',
|
||||
)
|
||||
|
||||
# BUG-7 fix: Resync merged user's subscriptions with RemnaWave panel
|
||||
try:
|
||||
from app.services.remnawave_resync_service import resync_user_subscriptions_with_panel
|
||||
|
||||
resync_result = await resync_user_subscriptions_with_panel(db, merged_user)
|
||||
logger.info(
|
||||
'Post-merge resync completed',
|
||||
primary_user_id=primary_user_id,
|
||||
secondary_user_id=secondary_user_id,
|
||||
synced=resync_result['synced'],
|
||||
failed=resync_result['failed'],
|
||||
)
|
||||
except Exception as resync_error:
|
||||
logger.error(
|
||||
'Post-merge resync failed (non-fatal)',
|
||||
primary_user_id=primary_user_id,
|
||||
error=resync_error,
|
||||
)
|
||||
|
||||
# 5. Create auth tokens for the merged user
|
||||
try:
|
||||
auth_response = await _create_auth_response(merged_user, db)
|
||||
|
||||
@@ -141,6 +141,7 @@ def _serialize_broadcast(broadcast: BroadcastHistory) -> BroadcastResponse:
|
||||
created_at=broadcast.created_at,
|
||||
completed_at=broadcast.completed_at,
|
||||
progress_percent=progress,
|
||||
category=getattr(broadcast, 'category', 'system') or 'system',
|
||||
channel=getattr(broadcast, 'channel', 'telegram') or 'telegram',
|
||||
email_subject=getattr(broadcast, 'email_subject', None),
|
||||
email_html_content=getattr(broadcast, 'email_html_content', None),
|
||||
@@ -432,6 +433,7 @@ async def create_broadcast(
|
||||
status='queued',
|
||||
admin_id=admin.id,
|
||||
admin_name=admin.username or f'Admin #{admin.id}',
|
||||
category=request.category,
|
||||
)
|
||||
db.add(broadcast)
|
||||
await db.commit()
|
||||
@@ -454,6 +456,7 @@ async def create_broadcast(
|
||||
media=media_config,
|
||||
initiator_name=admin.username or f'Admin #{admin.id}',
|
||||
custom_buttons=[btn.model_dump() for btn in request.custom_buttons] if request.custom_buttons else None,
|
||||
category=request.category,
|
||||
)
|
||||
|
||||
# Start broadcast
|
||||
@@ -626,6 +629,7 @@ async def create_combined_broadcast(
|
||||
status='queued',
|
||||
admin_id=admin.id,
|
||||
admin_name=admin_name,
|
||||
category=request.category,
|
||||
channel=request.channel,
|
||||
email_subject=request.email_subject.strip() if request.email_subject else None,
|
||||
email_html_content=request.email_html_content.strip() if request.email_html_content else None,
|
||||
@@ -653,6 +657,7 @@ async def create_combined_broadcast(
|
||||
media=media_config,
|
||||
initiator_name=admin_name,
|
||||
custom_buttons=[btn.model_dump() for btn in request.custom_buttons] if request.custom_buttons else None,
|
||||
category=request.category,
|
||||
)
|
||||
|
||||
await broadcast_service.start_broadcast(broadcast.id, telegram_config)
|
||||
|
||||
@@ -42,9 +42,9 @@ router = APIRouter(prefix='/admin/menu-layout', tags=['Admin Menu Layout'])
|
||||
# ---- Constants ---------------------------------------------------------------
|
||||
|
||||
MAX_ROWS = 20
|
||||
MAX_BUTTONS_PER_ROW = 3
|
||||
MAX_BUTTONS_PER_ROW = 8 # Telegram inline keyboard limit
|
||||
MAX_LABEL_LENGTH = 100
|
||||
URL_PATTERN = re.compile(r'^https?://')
|
||||
URL_PATTERN = re.compile(r'^(https?://|tg://)')
|
||||
|
||||
|
||||
# ---- Schemas -----------------------------------------------------------------
|
||||
@@ -275,7 +275,7 @@ def _validate_update_payload(rows: list[RowConfig]) -> None:
|
||||
if not btn.url or not URL_PATTERN.match(btn.url):
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_400_BAD_REQUEST,
|
||||
detail=f'Custom button "{btn.id}" must have a URL starting with http:// or https://.',
|
||||
detail=f'Custom button "{btn.id}" must have a URL starting with http://, https://, or tg://.',
|
||||
)
|
||||
if btn.open_in == 'webapp' and not btn.url.startswith('https://'):
|
||||
raise HTTPException(
|
||||
|
||||
@@ -4,14 +4,14 @@ from __future__ import annotations
|
||||
|
||||
import asyncio
|
||||
from datetime import datetime
|
||||
from typing import Any
|
||||
from typing import Any, ClassVar
|
||||
|
||||
import structlog
|
||||
from aiogram import Bot
|
||||
from aiogram.exceptions import TelegramBadRequest, TelegramForbiddenError
|
||||
from aiogram.types import InlineKeyboardButton, InlineKeyboardMarkup
|
||||
from fastapi import APIRouter, Depends, HTTPException, Query, status
|
||||
from pydantic import BaseModel, Field
|
||||
from pydantic import BaseModel, Field, validator
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from app.bot_factory import create_bot
|
||||
@@ -128,6 +128,20 @@ class PromoOfferBroadcastRequest(BaseModel):
|
||||
message_text: str | None = Field(None, description='Custom message text (HTML)')
|
||||
button_text: str | None = Field(None, description='Button text')
|
||||
|
||||
_TARGET_ALIASES: ClassVar[dict[str, str]] = {
|
||||
'no_sub': 'no',
|
||||
'all_users': 'all',
|
||||
'active_subscribers': 'active',
|
||||
'trial_users': 'trial',
|
||||
}
|
||||
|
||||
@validator('target')
|
||||
def normalize_target(cls, value: str | None) -> str | None:
|
||||
if value is None:
|
||||
return None
|
||||
normalized = value.strip().lower()
|
||||
return cls._TARGET_ALIASES.get(normalized, normalized)
|
||||
|
||||
|
||||
class PromoOfferBroadcastResponse(BaseModel):
|
||||
created_offers: int
|
||||
|
||||
@@ -236,8 +236,9 @@ async def _sync_subscription_to_panel(
|
||||
"""
|
||||
try:
|
||||
from app.config import settings
|
||||
from app.external.remnawave_api import TrafficLimitStrategy, UserStatus as PanelUserStatus
|
||||
from app.external.remnawave_api import UserStatus as PanelUserStatus
|
||||
from app.services.remnawave_service import RemnaWaveService
|
||||
from app.services.subscription_service import get_traffic_reset_strategy
|
||||
from app.utils.subscription_utils import resolve_hwid_device_limit_for_payload
|
||||
|
||||
service = RemnaWaveService()
|
||||
@@ -323,7 +324,7 @@ async def _sync_subscription_to_panel(
|
||||
'uuid': panel_uuid,
|
||||
'status': panel_status,
|
||||
'traffic_limit_bytes': traffic_limit_bytes,
|
||||
'traffic_limit_strategy': TrafficLimitStrategy.MONTH,
|
||||
'traffic_limit_strategy': get_traffic_reset_strategy(subscription.tariff),
|
||||
'description': description,
|
||||
}
|
||||
if expire_at:
|
||||
@@ -358,7 +359,7 @@ async def _sync_subscription_to_panel(
|
||||
'expire_at': expire_at or (datetime.now(UTC) + timedelta(days=30)),
|
||||
'status': panel_status,
|
||||
'traffic_limit_bytes': traffic_limit_bytes,
|
||||
'traffic_limit_strategy': TrafficLimitStrategy.MONTH,
|
||||
'traffic_limit_strategy': get_traffic_reset_strategy(subscription.tariff),
|
||||
'telegram_id': user.telegram_id,
|
||||
'email': user.email,
|
||||
'description': description,
|
||||
@@ -3022,6 +3023,11 @@ async def sync_user_from_panel(
|
||||
changes['remnawave_short_uuid'] = {'old': sub.remnawave_short_uuid, 'new': panel_user.short_uuid}
|
||||
sub.remnawave_short_uuid = panel_user.short_uuid
|
||||
|
||||
# Update crypto link
|
||||
if panel_user.happ_crypto_link and sub.subscription_crypto_link != panel_user.happ_crypto_link:
|
||||
changes['subscription_crypto_link'] = {'old': sub.subscription_crypto_link, 'new': '***'}
|
||||
sub.subscription_crypto_link = panel_user.happ_crypto_link
|
||||
|
||||
# Update traffic usage if requested
|
||||
if request.update_traffic and sync_sub:
|
||||
panel_traffic_used = panel_user.used_traffic_bytes / (1024**3) if panel_user.used_traffic_bytes else 0
|
||||
@@ -3118,8 +3124,9 @@ async def sync_user_to_panel(
|
||||
|
||||
try:
|
||||
from app.config import settings
|
||||
from app.external.remnawave_api import TrafficLimitStrategy, UserStatus as PanelUserStatus
|
||||
from app.external.remnawave_api import UserStatus as PanelUserStatus
|
||||
from app.services.remnawave_service import RemnaWaveService
|
||||
from app.services.subscription_service import get_traffic_reset_strategy
|
||||
from app.utils.subscription_utils import resolve_hwid_device_limit_for_payload
|
||||
|
||||
service = RemnaWaveService()
|
||||
@@ -3218,7 +3225,7 @@ async def sync_user_to_panel(
|
||||
|
||||
if request.update_traffic_limit:
|
||||
update_kwargs['traffic_limit_bytes'] = traffic_limit_bytes
|
||||
update_kwargs['traffic_limit_strategy'] = TrafficLimitStrategy.MONTH
|
||||
update_kwargs['traffic_limit_strategy'] = get_traffic_reset_strategy(sub.tariff)
|
||||
changes['traffic_limit_gb'] = sub.traffic_limit_gb
|
||||
|
||||
if request.update_squads and sub.connected_squads:
|
||||
@@ -3252,7 +3259,7 @@ async def sync_user_to_panel(
|
||||
'expire_at': expire_at or (datetime.now(UTC) + timedelta(days=30)),
|
||||
'status': panel_status,
|
||||
'traffic_limit_bytes': traffic_limit_bytes,
|
||||
'traffic_limit_strategy': TrafficLimitStrategy.MONTH,
|
||||
'traffic_limit_strategy': get_traffic_reset_strategy(sub.tariff),
|
||||
'telegram_id': user.telegram_id,
|
||||
'email': user.email,
|
||||
'description': description,
|
||||
|
||||
+29
-10
@@ -144,22 +144,28 @@ async def _store_refresh_token(
|
||||
refresh_token: str,
|
||||
device_info: str | None = None,
|
||||
) -> None:
|
||||
"""Store refresh token hash in database."""
|
||||
"""Store refresh token hash in database using upsert to avoid duplicate key errors."""
|
||||
from sqlalchemy.dialects.postgresql import insert as pg_insert
|
||||
|
||||
token_hash = hashlib.sha256(refresh_token.encode()).hexdigest()
|
||||
expires_at = get_refresh_token_expires_at()
|
||||
|
||||
token_record = CabinetRefreshToken(
|
||||
stmt = pg_insert(CabinetRefreshToken).values(
|
||||
user_id=user_id,
|
||||
token_hash=token_hash,
|
||||
device_info=device_info,
|
||||
expires_at=expires_at,
|
||||
)
|
||||
db.add(token_record)
|
||||
try:
|
||||
await db.commit()
|
||||
except IntegrityError:
|
||||
await db.rollback()
|
||||
logger.debug('Refresh token already exists (duplicate)', user_id=user_id)
|
||||
stmt = stmt.on_conflict_do_update(
|
||||
index_elements=['token_hash'],
|
||||
set_={
|
||||
'expires_at': expires_at,
|
||||
'device_info': device_info,
|
||||
'revoked_at': None,
|
||||
},
|
||||
)
|
||||
await db.execute(stmt)
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def _process_campaign_bonus(
|
||||
@@ -1052,6 +1058,10 @@ async def register_email_standalone(
|
||||
referred_by_id=referrer.id if referrer else None,
|
||||
)
|
||||
|
||||
# Сохранить campaign_slug для обработки при верификации email
|
||||
if request.campaign_slug:
|
||||
user.pending_campaign_slug = request.campaign_slug
|
||||
|
||||
# Для тестового email или отключённой верификации - автоматически верифицировать
|
||||
if is_test_email or not settings.is_cabinet_email_verification_enabled():
|
||||
user.email_verified = True
|
||||
@@ -1063,6 +1073,11 @@ async def register_email_standalone(
|
||||
await _sync_subscription_from_panel_by_email(db, user)
|
||||
except Exception:
|
||||
logger.warning('Failed to sync panel subscription after auto-verify', user_id=user.id, exc_info=True)
|
||||
# Process campaign bonus immediately for auto-verified users
|
||||
if request.campaign_slug:
|
||||
await _process_campaign_bonus(db, user, request.campaign_slug)
|
||||
user.pending_campaign_slug = None
|
||||
await db.commit()
|
||||
else:
|
||||
# Сгенерировать токен верификации
|
||||
verification_token = generate_verification_token()
|
||||
@@ -1173,8 +1188,12 @@ async def verify_email(
|
||||
response = await _create_auth_response(user, db)
|
||||
await _store_refresh_token(db, user.id, response.refresh_token)
|
||||
|
||||
# Process campaign bonus
|
||||
response.campaign_bonus = await _process_campaign_bonus(db, user, request.campaign_slug)
|
||||
# Process campaign bonus (prefer request param, fallback to saved slug from registration)
|
||||
effective_campaign_slug = request.campaign_slug or user.pending_campaign_slug
|
||||
response.campaign_bonus = await _process_campaign_bonus(db, user, effective_campaign_slug)
|
||||
if user.pending_campaign_slug:
|
||||
user.pending_campaign_slug = None
|
||||
await db.commit()
|
||||
if response.campaign_bonus:
|
||||
response.user = _user_to_response(user)
|
||||
|
||||
|
||||
@@ -425,8 +425,8 @@ async def create_gift_purchase(
|
||||
warning=recipient_warning,
|
||||
)
|
||||
|
||||
# Balance mode
|
||||
if user.balance_kopeks < price_kopeks:
|
||||
# Balance mode (skip for 100% discount)
|
||||
if price_kopeks > 0 and user.balance_kopeks < price_kopeks:
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_400_BAD_REQUEST,
|
||||
detail='Insufficient balance',
|
||||
|
||||
@@ -79,6 +79,8 @@ async def activate_promocode(
|
||||
error_messages = {
|
||||
'not_found': 'Promo code not found',
|
||||
'expired': 'Promo code has expired',
|
||||
'inactive': 'Promo code is deactivated',
|
||||
'not_yet_valid': 'Promo code is not yet active',
|
||||
'used': 'Promo code has been fully used',
|
||||
'already_used_by_user': 'You have already used this promo code',
|
||||
'active_discount_exists': 'You already have an active discount. Deactivate it first via /deactivate-discount',
|
||||
|
||||
@@ -168,6 +168,13 @@ async def toggle_subscription_pause(
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error('Error syncing RemnaWave user on resume', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=user.id,
|
||||
action='create',
|
||||
)
|
||||
|
||||
if new_paused_state:
|
||||
message = 'Daily subscription paused'
|
||||
|
||||
@@ -134,8 +134,8 @@ async def purchase_devices_legacy(
|
||||
detail=f'Максимальное количество устройств: {max_device_limit}',
|
||||
)
|
||||
|
||||
# Check balance
|
||||
if user.balance_kopeks < total_price:
|
||||
# Check balance (skip for 100% discount)
|
||||
if total_price > 0 and user.balance_kopeks < total_price:
|
||||
missing = total_price - user.balance_kopeks
|
||||
|
||||
# Сохраняем корзину для автопокупки после пополнения
|
||||
@@ -228,12 +228,24 @@ async def purchase_devices_legacy(
|
||||
# Sync with RemnaWave
|
||||
try:
|
||||
service = SubscriptionService()
|
||||
if _resolve_panel_uuid(subscription, user):
|
||||
await service.update_remnawave_user(db, subscription)
|
||||
if settings.is_multi_tariff_enabled():
|
||||
_should_create = not subscription.remnawave_uuid
|
||||
else:
|
||||
_should_create = not getattr(user, 'remnawave_uuid', None)
|
||||
|
||||
if _should_create:
|
||||
await service.create_remnawave_user(db, subscription)
|
||||
else:
|
||||
await service.update_remnawave_user(db, subscription)
|
||||
except Exception as e:
|
||||
logger.error('Failed to sync devices with RemnaWave (legacy endpoint)', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=user.id,
|
||||
action='create' if _should_create else 'update',
|
||||
)
|
||||
|
||||
# Отправляем уведомление админам
|
||||
try:
|
||||
@@ -375,8 +387,8 @@ async def purchase_devices(
|
||||
if devices_discount_percent < 100:
|
||||
price_kopeks = max(100, price_kopeks)
|
||||
|
||||
# Check balance
|
||||
if user.balance_kopeks < price_kopeks:
|
||||
# Check balance (skip for 100% discount)
|
||||
if price_kopeks > 0 and user.balance_kopeks < price_kopeks:
|
||||
missing = price_kopeks - user.balance_kopeks
|
||||
|
||||
# Сохраняем корзину для автопокупки после пополнения
|
||||
@@ -469,12 +481,24 @@ async def purchase_devices(
|
||||
# Sync with RemnaWave
|
||||
service = SubscriptionService()
|
||||
try:
|
||||
if _resolve_panel_uuid(subscription, user):
|
||||
await service.update_remnawave_user(db, subscription)
|
||||
if settings.is_multi_tariff_enabled():
|
||||
_should_create = not subscription.remnawave_uuid
|
||||
else:
|
||||
_should_create = not getattr(user, 'remnawave_uuid', None)
|
||||
|
||||
if _should_create:
|
||||
await service.create_remnawave_user(db, subscription)
|
||||
else:
|
||||
await service.update_remnawave_user(db, subscription)
|
||||
except Exception as e:
|
||||
logger.error('Failed to sync devices with RemnaWave', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=user.id,
|
||||
action='create' if _should_create else 'update',
|
||||
)
|
||||
|
||||
await db.refresh(user)
|
||||
|
||||
|
||||
@@ -304,9 +304,7 @@ async def get_purchase_options(
|
||||
from app.database.crud.subscription import get_active_subscriptions_by_user_id
|
||||
|
||||
active_subs = await get_active_subscriptions_by_user_id(db, user.id)
|
||||
purchased_tariff_ids = {
|
||||
s.tariff_id for s in active_subs if s.tariff_id and s.status in ('active', 'trial')
|
||||
}
|
||||
purchased_tariff_ids = {s.tariff_id for s in active_subs if s.tariff_id and not s.is_trial}
|
||||
|
||||
if subscription_id:
|
||||
from app.database.crud.subscription import get_subscription_by_id_for_user
|
||||
@@ -678,15 +676,17 @@ async def purchase_tariff(
|
||||
promo_offer_discount_value = result.promo_offer_discount
|
||||
price_before_promo_offer = price_kopeks + promo_offer_discount_value
|
||||
|
||||
# Safety guard: reject zero-price purchases for non-daily tariffs (defense in depth)
|
||||
if price_kopeks <= 0 and result.base_price <= 0 and not is_daily_tariff:
|
||||
# Safety guard: reject zero-price purchases for non-daily tariffs (defense in depth).
|
||||
# Use original_total (pre-discount price) — base_price is already discounted,
|
||||
# so a 100% group discount legitimately makes it 0.
|
||||
if price_kopeks <= 0 and result.original_total <= 0 and not is_daily_tariff:
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_400_BAD_REQUEST,
|
||||
detail='Invalid tariff period or pricing configuration',
|
||||
)
|
||||
|
||||
# Check balance
|
||||
if user.balance_kopeks < price_kopeks:
|
||||
if price_kopeks > 0 and user.balance_kopeks < price_kopeks:
|
||||
missing = price_kopeks - user.balance_kopeks
|
||||
|
||||
# Save cart for auto-purchase after balance top-up
|
||||
@@ -911,6 +911,13 @@ async def purchase_tariff(
|
||||
)
|
||||
except Exception as remnawave_error:
|
||||
logger.error('Failed to sync subscription with RemnaWave', remnawave_error=remnawave_error)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=user.id,
|
||||
action='create' if not subscription.remnawave_uuid else 'update',
|
||||
)
|
||||
|
||||
# Save cart for auto-renewal (not for daily tariffs - they have their own charging)
|
||||
if not is_daily_tariff:
|
||||
@@ -1158,7 +1165,7 @@ async def activate_trial(
|
||||
from app.database.crud.user import subtract_user_balance
|
||||
|
||||
price_kopeks = settings.TRIAL_ACTIVATION_PRICE
|
||||
if user.balance_kopeks < price_kopeks:
|
||||
if price_kopeks > 0 and user.balance_kopeks < price_kopeks:
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_400_BAD_REQUEST,
|
||||
detail=f'Insufficient balance. Need {price_kopeks / 100:.2f} RUB',
|
||||
@@ -1228,6 +1235,13 @@ async def activate_trial(
|
||||
except Exception as e:
|
||||
logger.error('Error getting trial tariff', error=e)
|
||||
|
||||
# BUG-12 fix: If no squads from tariff, fallback to trial-eligible servers
|
||||
if not trial_squads:
|
||||
from app.database.crud.server_squad import get_random_trial_squad_uuid
|
||||
|
||||
trial_squad_uuid = await get_random_trial_squad_uuid(db)
|
||||
trial_squads = [trial_squad_uuid] if trial_squad_uuid else []
|
||||
|
||||
# Create trial subscription
|
||||
subscription = await create_trial_subscription(
|
||||
db=db,
|
||||
@@ -1249,6 +1263,13 @@ async def activate_trial(
|
||||
await db.refresh(subscription)
|
||||
except Exception as e:
|
||||
logger.error('Failed to create RemnaWave user for trial', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=user.id,
|
||||
action='create',
|
||||
)
|
||||
|
||||
# Send admin notification about trial activation
|
||||
try:
|
||||
|
||||
@@ -67,7 +67,7 @@ async def get_renewal_options(
|
||||
for period in periods:
|
||||
pricing = await pricing_engine.calculate_renewal_price(db, subscription, period, user=user)
|
||||
|
||||
if pricing.final_total <= 0 and pricing.base_price <= 0:
|
||||
if pricing.final_total <= 0 and pricing.original_total <= 0:
|
||||
continue
|
||||
|
||||
original_price = pricing.original_total
|
||||
@@ -155,7 +155,7 @@ async def renew_subscription(
|
||||
promo_offer_discount_value = pricing.promo_offer_discount
|
||||
promo_offer_discount_percent = pricing.breakdown.get('offer_discount_pct', 0)
|
||||
|
||||
if price_kopeks <= 0 and pricing.base_price <= 0:
|
||||
if price_kopeks <= 0 and pricing.original_total <= 0:
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_400_BAD_REQUEST,
|
||||
detail='Invalid renewal period',
|
||||
@@ -168,8 +168,8 @@ async def renew_subscription(
|
||||
|
||||
tariff = subscription.tariff if subscription.tariff_id else None
|
||||
|
||||
# Check balance
|
||||
if user.balance_kopeks < price_kopeks:
|
||||
# Check balance (skip for 100% discount)
|
||||
if price_kopeks > 0 and user.balance_kopeks < price_kopeks:
|
||||
missing = price_kopeks - user.balance_kopeks
|
||||
|
||||
# Get tariff info for cart
|
||||
|
||||
@@ -249,6 +249,13 @@ async def update_countries(
|
||||
await subscription_service.create_remnawave_user(db, subscription)
|
||||
except Exception as e:
|
||||
logger.error('Failed to sync countries with RemnaWave', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=user.id,
|
||||
action='update' if _has_panel else 'create',
|
||||
)
|
||||
|
||||
await db.refresh(subscription)
|
||||
|
||||
|
||||
@@ -428,6 +428,13 @@ async def switch_tariff(
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error('Failed to sync tariff switch with RemnaWave', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=user.id,
|
||||
action='update' if _has_panel else 'create',
|
||||
)
|
||||
|
||||
# Reset all devices on tariff switch
|
||||
devices_reset = False
|
||||
|
||||
@@ -254,7 +254,7 @@ async def purchase_traffic(
|
||||
final_price = max(100, final_price)
|
||||
|
||||
# Проверяем баланс
|
||||
if user.balance_kopeks < final_price:
|
||||
if final_price > 0 and user.balance_kopeks < final_price:
|
||||
missing = final_price - user.balance_kopeks
|
||||
|
||||
# Save cart for auto-purchase after balance top-up
|
||||
@@ -316,19 +316,32 @@ async def purchase_traffic(
|
||||
# Синхронизируем с RemnaWave
|
||||
try:
|
||||
subscription_service = SubscriptionService()
|
||||
_panel_uuid = (
|
||||
subscription.remnawave_uuid
|
||||
if settings.is_multi_tariff_enabled() and subscription.remnawave_uuid
|
||||
else getattr(user, 'remnawave_uuid', None)
|
||||
)
|
||||
if _panel_uuid:
|
||||
if settings.is_multi_tariff_enabled():
|
||||
_should_create = not subscription.remnawave_uuid
|
||||
else:
|
||||
_should_create = not getattr(user, 'remnawave_uuid', None)
|
||||
|
||||
if _should_create:
|
||||
await subscription_service.create_remnawave_user(db, subscription)
|
||||
else:
|
||||
await subscription_service.update_remnawave_user(db, subscription)
|
||||
if subscription.status == 'active':
|
||||
await subscription_service.enable_remnawave_user(_panel_uuid)
|
||||
else:
|
||||
await subscription_service.create_remnawave_user(db, subscription)
|
||||
_enable_uuid = (
|
||||
subscription.remnawave_uuid
|
||||
if settings.is_multi_tariff_enabled()
|
||||
else getattr(user, 'remnawave_uuid', None)
|
||||
)
|
||||
if _enable_uuid:
|
||||
await subscription_service.enable_remnawave_user(_enable_uuid)
|
||||
except Exception as e:
|
||||
logger.error('Failed to sync traffic with RemnaWave', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=user.id,
|
||||
action='create' if _should_create else 'update',
|
||||
)
|
||||
|
||||
# Создаём транзакцию
|
||||
await create_transaction(
|
||||
@@ -560,7 +573,7 @@ async def switch_traffic_package(
|
||||
# Prorated calculation
|
||||
final_price, days_charged = calculate_prorated_price(price_diff, subscription.end_date)
|
||||
|
||||
if user.balance_kopeks < final_price:
|
||||
if final_price > 0 and user.balance_kopeks < final_price:
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_402_PAYMENT_REQUIRED,
|
||||
detail=f'Insufficient balance. Need {final_price / 100:.2f} RUB',
|
||||
@@ -604,17 +617,25 @@ async def switch_traffic_package(
|
||||
# Sync with RemnaWave
|
||||
try:
|
||||
subscription_service = SubscriptionService()
|
||||
_panel_uuid2 = (
|
||||
subscription.remnawave_uuid
|
||||
if settings.is_multi_tariff_enabled() and subscription.remnawave_uuid
|
||||
else getattr(user, 'remnawave_uuid', None)
|
||||
)
|
||||
if _panel_uuid2:
|
||||
await subscription_service.update_remnawave_user(db, subscription)
|
||||
if settings.is_multi_tariff_enabled():
|
||||
_should_create = not subscription.remnawave_uuid
|
||||
else:
|
||||
_should_create = not getattr(user, 'remnawave_uuid', None)
|
||||
|
||||
if _should_create:
|
||||
await subscription_service.create_remnawave_user(db, subscription)
|
||||
else:
|
||||
await subscription_service.update_remnawave_user(db, subscription)
|
||||
except Exception as e:
|
||||
logger.error('Failed to sync traffic switch with RemnaWave', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(subscription, 'id') and hasattr(subscription, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='create' if _should_create else 'update',
|
||||
)
|
||||
|
||||
await db.refresh(user)
|
||||
await db.refresh(subscription)
|
||||
|
||||
@@ -138,6 +138,9 @@ class EmailRegisterStandaloneRequest(BaseModel):
|
||||
referral_code: str | None = Field(
|
||||
None, max_length=32, pattern=r'^[a-zA-Z0-9_-]+$', description='Referral code of inviter'
|
||||
)
|
||||
campaign_slug: str | None = Field(
|
||||
None, min_length=1, max_length=64, pattern=r'^[a-zA-Z0-9_-]+$', description='Campaign slug from web link'
|
||||
)
|
||||
|
||||
|
||||
class CampaignBonusInfo(BaseModel):
|
||||
|
||||
@@ -118,6 +118,7 @@ class BroadcastCreateRequest(BaseModel):
|
||||
selected_buttons: list[str] = Field(default_factory=lambda: ['home'])
|
||||
custom_buttons: list[CustomBroadcastButton] = Field(default_factory=list, max_length=10)
|
||||
media: BroadcastMediaRequest | None = None
|
||||
category: str = Field(default='system', pattern='^(system|news|promo)$')
|
||||
|
||||
|
||||
# ============ Response ============
|
||||
@@ -144,6 +145,9 @@ class BroadcastResponse(BaseModel):
|
||||
completed_at: datetime | None = None
|
||||
progress_percent: float = 0.0
|
||||
|
||||
# Category for user notification preference filtering
|
||||
category: str = 'system' # system|news|promo
|
||||
|
||||
# Email/channel fields
|
||||
channel: str = 'telegram' # telegram|email|both
|
||||
email_subject: str | None = None
|
||||
@@ -212,6 +216,9 @@ class CombinedBroadcastCreateRequest(BaseModel):
|
||||
custom_buttons: list[CustomBroadcastButton] = Field(default_factory=list, max_length=10)
|
||||
media: BroadcastMediaRequest | None = None
|
||||
|
||||
# Broadcast category for user notification preference filtering
|
||||
category: str = Field(default='system', pattern='^(system|news|promo)$')
|
||||
|
||||
# Email-specific fields
|
||||
email_subject: str | None = Field(default=None, max_length=255)
|
||||
email_html_content: str | None = Field(default=None, max_length=100000)
|
||||
|
||||
@@ -12,7 +12,16 @@ def get_campaign_deep_link(start_parameter: str) -> str:
|
||||
|
||||
|
||||
def get_campaign_web_link(start_parameter: str) -> str | None:
|
||||
"""Generate a web app link for a campaign."""
|
||||
"""Generate a web app link for a campaign.
|
||||
|
||||
Prefers CABINET_URL (where the auth flow captures ?campaign= param),
|
||||
falls back to MINIAPP_CUSTOM_URL for backwards compatibility.
|
||||
"""
|
||||
cabinet_url = settings._normalized_cabinet_url()
|
||||
if cabinet_url:
|
||||
sep = '&' if '?' in cabinet_url else '?'
|
||||
return f'{cabinet_url}{sep}campaign={start_parameter}'
|
||||
|
||||
base_url = (settings.MINIAPP_CUSTOM_URL or '').rstrip('/')
|
||||
if base_url:
|
||||
return f'{base_url}/?campaign={start_parameter}'
|
||||
|
||||
@@ -133,6 +133,7 @@ class Settings(BaseSettings):
|
||||
WEBHOOK_NOTIFY_NOT_CONNECTED: bool = True
|
||||
WEBHOOK_NOTIFY_BANDWIDTH_THRESHOLD: bool = True
|
||||
WEBHOOK_NOTIFY_DEVICES: bool = True
|
||||
WEBHOOK_NOTIFY_TORRENT_DETECTED: bool = True
|
||||
|
||||
TRIAL_DURATION_DAYS: int = 3
|
||||
TRIAL_TRAFFIC_LIMIT_GB: int = 10
|
||||
@@ -435,6 +436,7 @@ class Settings(BaseSettings):
|
||||
MULENPAY_LANGUAGE: str = 'ru'
|
||||
MULENPAY_VAT_CODE: int = 0
|
||||
|
||||
DISPLAY_NAME_RESTRICTION_ENABLED: bool = True
|
||||
DISPLAY_NAME_BANNED_KEYWORDS: str = '\n'.join(DEFAULT_DISPLAY_NAME_BANNED_KEYWORDS)
|
||||
MULENPAY_PAYMENT_SUBJECT: int = 4
|
||||
MULENPAY_PAYMENT_MODE: int = 4
|
||||
@@ -824,6 +826,10 @@ class Settings(BaseSettings):
|
||||
# Format: socks5://user:password@host:port or socks5://host:port
|
||||
PROXY_URL: str | None = None
|
||||
|
||||
# Custom Telegram Bot API server URL (for regions where api.telegram.org is blocked)
|
||||
# Examples: Cloudflare Worker proxy, self-hosted telegram-bot-api (tdlib), nginx reverse proxy
|
||||
TELEGRAM_API_URL: str | None = None
|
||||
|
||||
@field_validator('PROXY_URL', 'NALOGO_PROXY_URL', mode='before')
|
||||
@classmethod
|
||||
def validate_proxy_url(cls, value: str | None) -> str | None:
|
||||
@@ -971,6 +977,10 @@ class Settings(BaseSettings):
|
||||
"""Return SOCKS5 proxy URL or None."""
|
||||
return self.PROXY_URL if self.PROXY_URL else None
|
||||
|
||||
def get_telegram_api_url(self) -> str | None:
|
||||
"""Return custom Telegram Bot API server URL or None."""
|
||||
return self.TELEGRAM_API_URL if self.TELEGRAM_API_URL else None
|
||||
|
||||
def get_nalogo_proxy_url(self) -> str | None:
|
||||
"""Return SOCKS proxy URL for nalogo or None.
|
||||
|
||||
|
||||
@@ -166,7 +166,7 @@ async def update_promo_group(
|
||||
group.device_discount_percent = max(0, min(100, device_discount_percent))
|
||||
if period_discounts is not None:
|
||||
normalized_period_discounts = _normalize_period_discounts(period_discounts)
|
||||
group.period_discounts = normalized_period_discounts or None
|
||||
group.period_discounts = normalized_period_discounts if normalized_period_discounts else None
|
||||
if auto_assign_total_spent_kopeks is not None:
|
||||
value = max(0, auto_assign_total_spent_kopeks)
|
||||
group.auto_assign_total_spent_kopeks = value if value > 0 else None
|
||||
|
||||
@@ -96,12 +96,13 @@ async def get_subscription_by_user_id(db: AsyncSession, user_id: int) -> Subscri
|
||||
)
|
||||
.where(Subscription.user_id == user_id)
|
||||
.order_by(
|
||||
# Active/trial subscriptions first, then by creation date
|
||||
# Active/trial subscriptions first, then by end_date (most remaining time)
|
||||
case(
|
||||
(Subscription.status == SubscriptionStatus.ACTIVE.value, 0),
|
||||
(Subscription.status == SubscriptionStatus.TRIAL.value, 1),
|
||||
else_=2,
|
||||
),
|
||||
Subscription.end_date.desc().nulls_last(),
|
||||
Subscription.created_at.desc(),
|
||||
)
|
||||
.limit(1)
|
||||
@@ -2075,7 +2076,12 @@ async def toggle_daily_subscription_pause(
|
||||
|
||||
|
||||
async def get_active_subscriptions_by_user_id(db: AsyncSession, user_id: int) -> list[Subscription]:
|
||||
"""Get all active/trial subscriptions for a user."""
|
||||
"""Get all active/trial/limited subscriptions for a user.
|
||||
|
||||
Includes LIMITED status because those subscriptions still have time remaining
|
||||
(just ran out of traffic) and should be treated as "alive" for renewal,
|
||||
duplicate prevention, and display purposes.
|
||||
"""
|
||||
result = await db.execute(
|
||||
select(Subscription)
|
||||
.options(
|
||||
@@ -2084,7 +2090,13 @@ async def get_active_subscriptions_by_user_id(db: AsyncSession, user_id: int) ->
|
||||
)
|
||||
.where(
|
||||
Subscription.user_id == user_id,
|
||||
Subscription.status.in_([SubscriptionStatus.ACTIVE.value, SubscriptionStatus.TRIAL.value]),
|
||||
Subscription.status.in_(
|
||||
[
|
||||
SubscriptionStatus.ACTIVE.value,
|
||||
SubscriptionStatus.TRIAL.value,
|
||||
SubscriptionStatus.LIMITED.value,
|
||||
]
|
||||
),
|
||||
)
|
||||
.order_by(Subscription.created_at.desc())
|
||||
)
|
||||
@@ -2121,7 +2133,11 @@ async def get_subscription_by_id(db: AsyncSession, subscription_id: int) -> Subs
|
||||
|
||||
|
||||
async def get_subscription_by_user_and_tariff(db: AsyncSession, user_id: int, tariff_id: int) -> Subscription | None:
|
||||
"""Get active/trial subscription for a specific user+tariff combination."""
|
||||
"""Get active/trial/limited subscription for a specific user+tariff combination.
|
||||
|
||||
Includes LIMITED status because those subscriptions still have time remaining
|
||||
(just ran out of traffic) and should be extended rather than duplicated.
|
||||
"""
|
||||
result = await db.execute(
|
||||
select(Subscription)
|
||||
.options(
|
||||
@@ -2131,7 +2147,13 @@ async def get_subscription_by_user_and_tariff(db: AsyncSession, user_id: int, ta
|
||||
.where(
|
||||
Subscription.user_id == user_id,
|
||||
Subscription.tariff_id == tariff_id,
|
||||
Subscription.status.in_([SubscriptionStatus.ACTIVE.value, SubscriptionStatus.TRIAL.value]),
|
||||
Subscription.status.in_(
|
||||
[
|
||||
SubscriptionStatus.ACTIVE.value,
|
||||
SubscriptionStatus.TRIAL.value,
|
||||
SubscriptionStatus.LIMITED.value,
|
||||
]
|
||||
),
|
||||
)
|
||||
.order_by(Subscription.created_at.desc())
|
||||
.limit(1)
|
||||
|
||||
@@ -122,6 +122,12 @@ async def clear_trial_tariff(db: AsyncSession) -> None:
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def get_all_active_tariffs(db: AsyncSession) -> list[Tariff]:
|
||||
"""Get all active tariffs."""
|
||||
result = await db.execute(select(Tariff).where(Tariff.is_active.is_(True)).order_by(Tariff.tier_level))
|
||||
return list(result.scalars().all())
|
||||
|
||||
|
||||
async def get_tariffs_for_user(
|
||||
db: AsyncSession,
|
||||
promo_group_id: int | None = None,
|
||||
|
||||
@@ -235,7 +235,12 @@ async def get_user_total_spent_kopeks(db: AsyncSession, user_id: int) -> int:
|
||||
and_(
|
||||
Transaction.user_id == user_id,
|
||||
Transaction.is_completed.is_(True),
|
||||
Transaction.type == TransactionType.SUBSCRIPTION_PAYMENT.value,
|
||||
Transaction.type.in_(
|
||||
[
|
||||
TransactionType.SUBSCRIPTION_PAYMENT.value,
|
||||
TransactionType.GIFT_PAYMENT.value,
|
||||
]
|
||||
),
|
||||
)
|
||||
)
|
||||
)
|
||||
|
||||
@@ -1204,6 +1204,8 @@ class User(Base):
|
||||
password_reset_token = Column(String(255), nullable=True)
|
||||
password_reset_expires = Column(AwareDateTime(), nullable=True)
|
||||
cabinet_last_login = Column(AwareDateTime(), nullable=True)
|
||||
# Campaign slug saved at registration, consumed at email verification
|
||||
pending_campaign_slug = Column(String(64), nullable=True)
|
||||
# Email change fields
|
||||
email_change_new = Column(String(255), nullable=True) # New email pending verification
|
||||
email_change_code = Column(String(6), nullable=True) # 6-digit verification code
|
||||
@@ -1255,7 +1257,7 @@ class User(Base):
|
||||
user_promo_groups = relationship('UserPromoGroup', back_populates='user', cascade='all, delete-orphan')
|
||||
poll_responses = relationship('PollResponse', back_populates='user')
|
||||
admin_roles_rel = relationship('UserRole', foreign_keys='[UserRole.user_id]', back_populates='user')
|
||||
notification_settings = Column(JSON, nullable=True, default=dict)
|
||||
notification_settings = Column(JSONB, nullable=True, default=dict)
|
||||
last_pinned_message_id = Column(Integer, nullable=True)
|
||||
|
||||
# Ограничения пользователя
|
||||
@@ -1357,7 +1359,7 @@ class Subscription(Base):
|
||||
'user_id',
|
||||
'tariff_id',
|
||||
unique=True,
|
||||
postgresql_where=text("tariff_id IS NOT NULL AND status IN ('active', 'trial')"),
|
||||
postgresql_where=text("tariff_id IS NOT NULL AND status IN ('active', 'trial', 'limited')"),
|
||||
),
|
||||
)
|
||||
|
||||
@@ -2238,6 +2240,9 @@ class BroadcastHistory(Base):
|
||||
created_at = Column(AwareDateTime(), server_default=func.now())
|
||||
completed_at = Column(AwareDateTime(), nullable=True)
|
||||
|
||||
# Broadcast category for user notification preferences filtering
|
||||
category = Column(String(20), default='system', nullable=False) # system|news|promo
|
||||
|
||||
# Email broadcast fields
|
||||
channel = Column(String(20), default='telegram', nullable=False) # telegram|email|both
|
||||
email_subject = Column(String(255), nullable=True)
|
||||
|
||||
Vendored
+9
-8
@@ -376,15 +376,16 @@ class RemnaWaveAPI:
|
||||
except json.JSONDecodeError:
|
||||
response_data = {'raw_response': response_text}
|
||||
|
||||
if response.status == 429 and attempt < max_retries:
|
||||
if response.status in (429, 502, 503, 504) and attempt < max_retries:
|
||||
retry_after = float(response.headers.get('Retry-After', base_delay * (2**attempt)))
|
||||
logger.warning(
|
||||
'Rate limited (429) on , retry / after s',
|
||||
method=method,
|
||||
endpoint=endpoint,
|
||||
attempt=attempt + 1,
|
||||
max_retries=max_retries,
|
||||
retry_after=retry_after,
|
||||
'Retryable %s on %s %s, retry %s/%s after %ss',
|
||||
response.status,
|
||||
method,
|
||||
endpoint,
|
||||
attempt + 1,
|
||||
max_retries,
|
||||
retry_after,
|
||||
)
|
||||
await asyncio.sleep(retry_after)
|
||||
continue
|
||||
@@ -445,7 +446,7 @@ class RemnaWaveAPI:
|
||||
'trafficLimitStrategy': traffic_limit_strategy.value,
|
||||
}
|
||||
|
||||
if telegram_id:
|
||||
if telegram_id is not None:
|
||||
data['telegramId'] = telegram_id
|
||||
if email:
|
||||
data['email'] = email
|
||||
|
||||
@@ -1599,42 +1599,28 @@ async def get_target_users_count(db: AsyncSession, target: str) -> int:
|
||||
result = await db.execute(query)
|
||||
return result.scalar() or 0
|
||||
|
||||
if target == 'expired':
|
||||
# Истекшие подписки
|
||||
if target in ('expired', 'expired_subscribers'):
|
||||
# Истекшие подписки — исключаем юзеров с хотя бы одной активной
|
||||
now = datetime.now(UTC)
|
||||
expired_statuses = [
|
||||
SubscriptionStatus.EXPIRED.value,
|
||||
SubscriptionStatus.DISABLED.value,
|
||||
SubscriptionStatus.LIMITED.value,
|
||||
]
|
||||
query = (
|
||||
select(sql_func.count(distinct(User.id)))
|
||||
.outerjoin(Subscription, User.id == Subscription.user_id)
|
||||
has_active_sub = (
|
||||
select(Subscription.id)
|
||||
.where(
|
||||
base_filter,
|
||||
or_(
|
||||
Subscription.status.in_(expired_statuses),
|
||||
and_(Subscription.end_date <= now, Subscription.status != SubscriptionStatus.ACTIVE.value),
|
||||
and_(Subscription.id == None, User.has_had_paid_subscription == True),
|
||||
),
|
||||
Subscription.user_id == User.id,
|
||||
Subscription.status == SubscriptionStatus.ACTIVE.value,
|
||||
)
|
||||
.exists()
|
||||
)
|
||||
result = await db.execute(query)
|
||||
return result.scalar() or 0
|
||||
|
||||
if target == 'expired_subscribers':
|
||||
# То же что и expired
|
||||
now = datetime.now(UTC)
|
||||
expired_statuses = [
|
||||
SubscriptionStatus.EXPIRED.value,
|
||||
SubscriptionStatus.DISABLED.value,
|
||||
SubscriptionStatus.LIMITED.value,
|
||||
]
|
||||
query = (
|
||||
select(sql_func.count(distinct(User.id)))
|
||||
.outerjoin(Subscription, User.id == Subscription.user_id)
|
||||
.where(
|
||||
base_filter,
|
||||
~has_active_sub,
|
||||
or_(
|
||||
Subscription.status.in_(expired_statuses),
|
||||
and_(Subscription.end_date <= now, Subscription.status != SubscriptionStatus.ACTIVE.value),
|
||||
@@ -1785,6 +1771,9 @@ async def get_target_users(db: AsyncSession, target: str) -> list:
|
||||
for user in users:
|
||||
subs = getattr(user, 'subscriptions', None) or []
|
||||
if subs:
|
||||
has_active = any(s.is_active for s in subs)
|
||||
if has_active:
|
||||
continue # Skip users who have at least one active subscription
|
||||
has_expired = any(s.status in expired_statuses or (s.end_date <= now and not s.is_active) for s in subs)
|
||||
if has_expired:
|
||||
expired_users.append(user)
|
||||
@@ -1833,6 +1822,9 @@ async def get_target_users(db: AsyncSession, target: str) -> list:
|
||||
for user in users:
|
||||
subs = getattr(user, 'subscriptions', None) or []
|
||||
if subs:
|
||||
has_active = any(s.is_active for s in subs)
|
||||
if has_active:
|
||||
continue # Skip users who have at least one active subscription
|
||||
has_expired = any(s.status in expired_statuses or (s.end_date <= now and not s.is_active) for s in subs)
|
||||
if has_expired:
|
||||
expired_users.append(user)
|
||||
|
||||
@@ -77,7 +77,7 @@ def _build_server_edit_view(server):
|
||||
],
|
||||
[
|
||||
types.InlineKeyboardButton(
|
||||
text='🎁 Выдавать сквад' if not server.is_trial_eligible else '🚫 Не выдавать сквад',
|
||||
text='🎁 Выдавать в триал' if not server.is_trial_eligible else '🚫 Не выдавать в триал',
|
||||
callback_data=f'admin_server_trial_{server.id}',
|
||||
),
|
||||
],
|
||||
|
||||
+122
-31
@@ -990,13 +990,14 @@ async def _render_user_subscription_overview(
|
||||
]
|
||||
)
|
||||
else:
|
||||
keyboard.append(
|
||||
[
|
||||
types.InlineKeyboardButton(
|
||||
text='✅ Активировать', callback_data=f'admin_sub_activate_{user_id}{_sid}'
|
||||
)
|
||||
]
|
||||
)
|
||||
row = [
|
||||
types.InlineKeyboardButton(text='✅ Активировать', callback_data=f'admin_sub_activate_{user_id}{_sid}'),
|
||||
]
|
||||
if settings.is_multi_tariff_enabled() and subscription_id:
|
||||
row.append(
|
||||
types.InlineKeyboardButton(text='🗑 Удалить', callback_data=f'admin_sub_delete_{user_id}{_sid}')
|
||||
)
|
||||
keyboard.append(row)
|
||||
else:
|
||||
text += '❌ <b>Подписка отсутствует</b>\n\n'
|
||||
text += 'Пользователь еще не активировал подписку.'
|
||||
@@ -3302,6 +3303,76 @@ async def confirm_subscription_deactivation(callback: types.CallbackQuery, db_us
|
||||
await callback.answer()
|
||||
|
||||
|
||||
@admin_required
|
||||
@error_handler
|
||||
async def delete_user_subscription(callback: types.CallbackQuery, db_user: User, db: AsyncSession):
|
||||
"""Show confirmation for deleting a subscription (multi-tariff only)."""
|
||||
user_id, subscription_id = _extract_admin_sub_context(callback.data)
|
||||
|
||||
if not subscription_id or not settings.is_multi_tariff_enabled():
|
||||
await callback.answer('Удаление доступно только в мультитарифном режиме', show_alert=True)
|
||||
return
|
||||
|
||||
back_cb = f'admin_user_sub_select_{user_id}_{subscription_id}'
|
||||
_sid = f'_s{subscription_id}'
|
||||
|
||||
await callback.message.edit_text(
|
||||
'🗑 <b>Удаление подписки</b>\n\n⚠️ Подписка будет полностью удалена из системы.\nЭто действие необратимо!',
|
||||
reply_markup=get_confirmation_keyboard(f'admin_sub_delete_confirm_{user_id}{_sid}', back_cb, db_user.language),
|
||||
)
|
||||
await callback.answer()
|
||||
|
||||
|
||||
@admin_required
|
||||
@error_handler
|
||||
async def confirm_subscription_deletion(callback: types.CallbackQuery, db_user: User, db: AsyncSession):
|
||||
"""Delete a subscription permanently (multi-tariff only)."""
|
||||
user_id, subscription_id = _extract_admin_sub_context(callback.data)
|
||||
|
||||
if not subscription_id or not settings.is_multi_tariff_enabled():
|
||||
await callback.answer('Удаление доступно только в мультитарифном режиме', show_alert=True)
|
||||
return
|
||||
|
||||
from app.database.crud.subscription import get_subscription_by_id_for_user
|
||||
|
||||
subscription = await get_subscription_by_id_for_user(db, subscription_id, user_id)
|
||||
if not subscription:
|
||||
await callback.answer('Подписка не найдена', show_alert=True)
|
||||
return
|
||||
|
||||
# Disable on Remnawave side first
|
||||
_uuid = getattr(subscription, 'remnawave_uuid', None)
|
||||
if _uuid:
|
||||
subscription_service = SubscriptionService()
|
||||
await subscription_service.disable_remnawave_user(_uuid)
|
||||
|
||||
# Delete traffic purchases
|
||||
from sqlalchemy import delete as sql_delete
|
||||
|
||||
from app.database.models import TrafficPurchase
|
||||
|
||||
await db.execute(sql_delete(TrafficPurchase).where(TrafficPurchase.subscription_id == subscription.id))
|
||||
|
||||
await db.delete(subscription)
|
||||
await db.commit()
|
||||
|
||||
logger.info(
|
||||
'Админ удалил подписку пользователя',
|
||||
admin_id=db_user.id,
|
||||
user_id=user_id,
|
||||
subscription_id=subscription_id,
|
||||
)
|
||||
|
||||
back_cb = f'admin_user_subscription_{user_id}'
|
||||
await callback.message.edit_text(
|
||||
'✅ Подписка удалена',
|
||||
reply_markup=types.InlineKeyboardMarkup(
|
||||
inline_keyboard=[[types.InlineKeyboardButton(text='📱 К подпискам', callback_data=back_cb)]]
|
||||
),
|
||||
)
|
||||
await callback.answer()
|
||||
|
||||
|
||||
@admin_required
|
||||
@error_handler
|
||||
async def activate_user_subscription(callback: types.CallbackQuery, db_user: User, db: AsyncSession):
|
||||
@@ -3750,26 +3821,38 @@ async def start_devices_edit(callback: types.CallbackQuery, db_user: User, state
|
||||
else f'admin_user_subscription_{user_id}'
|
||||
)
|
||||
|
||||
await callback.message.edit_text(
|
||||
'📱 <b>Изменение количества устройств</b>\n\n'
|
||||
'Введите новое количество устройств (от 1 до 10):\n'
|
||||
'• Текущее значение будет заменено\n'
|
||||
'• Примеры: 1, 2, 5, 10\n\n'
|
||||
'Или нажмите /cancel для отмены',
|
||||
reply_markup=types.InlineKeyboardMarkup(
|
||||
max_dev = settings.MAX_DEVICES_LIMIT
|
||||
# Build device buttons dynamically: rows of 4, respect Telegram 100 button limit (99 + cancel)
|
||||
if max_dev <= 99:
|
||||
device_buttons: list[list[types.InlineKeyboardButton]] = []
|
||||
row: list[types.InlineKeyboardButton] = []
|
||||
for i in range(1, max_dev + 1):
|
||||
row.append(
|
||||
types.InlineKeyboardButton(
|
||||
text=str(i),
|
||||
callback_data=f'admin_user_devices_set_{user_id}{_sid}_{i}',
|
||||
)
|
||||
)
|
||||
if len(row) == 4:
|
||||
device_buttons.append(row)
|
||||
row = []
|
||||
if row:
|
||||
device_buttons.append(row)
|
||||
device_buttons.append([types.InlineKeyboardButton(text='❌ Отмена', callback_data=back_cb)])
|
||||
markup = types.InlineKeyboardMarkup(inline_keyboard=device_buttons)
|
||||
else:
|
||||
markup = types.InlineKeyboardMarkup(
|
||||
inline_keyboard=[
|
||||
[
|
||||
types.InlineKeyboardButton(text='1', callback_data=f'admin_user_devices_set_{user_id}{_sid}_1'),
|
||||
types.InlineKeyboardButton(text='2', callback_data=f'admin_user_devices_set_{user_id}{_sid}_2'),
|
||||
types.InlineKeyboardButton(text='3', callback_data=f'admin_user_devices_set_{user_id}{_sid}_3'),
|
||||
],
|
||||
[
|
||||
types.InlineKeyboardButton(text='5', callback_data=f'admin_user_devices_set_{user_id}{_sid}_5'),
|
||||
types.InlineKeyboardButton(text='10', callback_data=f'admin_user_devices_set_{user_id}{_sid}_10'),
|
||||
],
|
||||
[types.InlineKeyboardButton(text='❌ Отмена', callback_data=back_cb)],
|
||||
]
|
||||
),
|
||||
)
|
||||
|
||||
await callback.message.edit_text(
|
||||
'📱 <b>Изменение количества устройств</b>\n\n'
|
||||
f'Введите новое количество устройств (от 1 до {max_dev}):\n'
|
||||
'• Текущее значение будет заменено\n\n'
|
||||
'Или нажмите /cancel для отмены',
|
||||
reply_markup=markup,
|
||||
)
|
||||
|
||||
await state.set_state(AdminStates.editing_user_devices)
|
||||
@@ -3839,8 +3922,8 @@ async def process_devices_edit_text(message: types.Message, db_user: User, state
|
||||
try:
|
||||
devices = int(message.text.strip())
|
||||
|
||||
if devices <= 0 or devices > 10:
|
||||
await message.answer('❌ Количество устройств должно быть от 1 до 10')
|
||||
if devices <= 0 or devices > settings.MAX_DEVICES_LIMIT:
|
||||
await message.answer(f'❌ Количество устройств должно быть от 1 до {settings.MAX_DEVICES_LIMIT}')
|
||||
return
|
||||
|
||||
success = await _update_user_devices(db, user_id, devices, db_user.id, subscription_id=subscription_id)
|
||||
@@ -4162,14 +4245,16 @@ async def _update_user_traffic(
|
||||
) or getattr(user, 'remnawave_uuid', None)
|
||||
if _uuid:
|
||||
try:
|
||||
from app.external.remnawave_api import TrafficLimitStrategy
|
||||
from app.services.subscription_service import get_traffic_reset_strategy
|
||||
|
||||
remnawave_service = RemnaWaveService()
|
||||
async with remnawave_service.get_api_client() as api:
|
||||
await api.update_user(
|
||||
uuid=_uuid,
|
||||
traffic_limit_bytes=traffic_gb * (1024**3) if traffic_gb > 0 else 0,
|
||||
traffic_limit_strategy=TrafficLimitStrategy.MONTH,
|
||||
traffic_limit_strategy=get_traffic_reset_strategy(
|
||||
subscription.tariff if subscription else None
|
||||
),
|
||||
description=settings.format_remnawave_user_description(
|
||||
full_name=user.full_name, username=user.username, telegram_id=user.telegram_id
|
||||
),
|
||||
@@ -4877,8 +4962,9 @@ async def admin_buy_subscription_execute(callback: types.CallbackQuery, db_user:
|
||||
)
|
||||
|
||||
try:
|
||||
from app.external.remnawave_api import TrafficLimitStrategy, UserStatus
|
||||
from app.external.remnawave_api import UserStatus
|
||||
from app.services.remnawave_service import RemnaWaveService
|
||||
from app.services.subscription_service import get_traffic_reset_strategy
|
||||
|
||||
remnawave_service = RemnaWaveService()
|
||||
|
||||
@@ -4903,7 +4989,7 @@ async def admin_buy_subscription_execute(callback: types.CallbackQuery, db_user:
|
||||
traffic_limit_bytes=subscription.traffic_limit_gb * (1024**3)
|
||||
if subscription.traffic_limit_gb > 0
|
||||
else 0,
|
||||
traffic_limit_strategy=TrafficLimitStrategy.MONTH,
|
||||
traffic_limit_strategy=get_traffic_reset_strategy(subscription.tariff),
|
||||
description=settings.format_remnawave_user_description(
|
||||
full_name=target_user.full_name,
|
||||
username=target_user.username,
|
||||
@@ -4939,7 +5025,7 @@ async def admin_buy_subscription_execute(callback: types.CallbackQuery, db_user:
|
||||
traffic_limit_bytes=subscription.traffic_limit_gb * (1024**3)
|
||||
if subscription.traffic_limit_gb > 0
|
||||
else 0,
|
||||
traffic_limit_strategy=TrafficLimitStrategy.MONTH,
|
||||
traffic_limit_strategy=get_traffic_reset_strategy(subscription.tariff),
|
||||
telegram_id=target_user.telegram_id,
|
||||
email=target_user.email,
|
||||
description=settings.format_remnawave_user_description(
|
||||
@@ -5939,6 +6025,11 @@ def register_handlers(dp: Dispatcher):
|
||||
|
||||
dp.callback_query.register(activate_user_subscription, F.data.startswith('admin_sub_activate_'))
|
||||
|
||||
dp.callback_query.register(
|
||||
delete_user_subscription, F.data.startswith('admin_sub_delete_') & ~F.data.contains('confirm')
|
||||
)
|
||||
dp.callback_query.register(confirm_subscription_deletion, F.data.startswith('admin_sub_delete_confirm_'))
|
||||
|
||||
dp.callback_query.register(grant_trial_subscription, F.data.startswith('admin_sub_grant_trial_'))
|
||||
|
||||
dp.callback_query.register(
|
||||
|
||||
@@ -197,6 +197,8 @@ async def process_promocode(message: types.Message, db_user: User, state: FSMCon
|
||||
error_messages = {
|
||||
'not_found': texts.PROMOCODE_INVALID,
|
||||
'expired': texts.PROMOCODE_EXPIRED,
|
||||
'inactive': texts.t('PROMOCODE_INACTIVE', '❌ Промокод деактивирован'),
|
||||
'not_yet_valid': texts.t('PROMOCODE_NOT_YET_VALID', '❌ Промокод ещё не начал действовать'),
|
||||
'used': texts.PROMOCODE_USED,
|
||||
'already_used_by_user': texts.PROMOCODE_USED,
|
||||
'not_first_purchase': texts.t(
|
||||
|
||||
@@ -441,7 +441,7 @@ async def handle_simple_subscription_pay_with_balance(
|
||||
# Проверяем баланс пользователя
|
||||
user_balance_kopeks = getattr(db_user, 'balance_kopeks', 0)
|
||||
|
||||
if user_balance_kopeks < total_required:
|
||||
if total_required > 0 and user_balance_kopeks < total_required:
|
||||
await callback.answer('❌ Недостаточно средств на балансе для оплаты подписки', show_alert=True)
|
||||
return
|
||||
|
||||
@@ -549,6 +549,14 @@ async def handle_simple_subscription_pay_with_balance(
|
||||
sync_error=sync_error,
|
||||
exc_info=True,
|
||||
)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(subscription, 'id') and hasattr(subscription, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='create',
|
||||
)
|
||||
|
||||
# Отправляем уведомление об успешной покупке
|
||||
server_label = _get_simple_subscription_server_label(
|
||||
@@ -2181,7 +2189,7 @@ async def confirm_simple_subscription_purchase(
|
||||
# Проверяем баланс пользователя
|
||||
user_balance_kopeks = getattr(db_user, 'balance_kopeks', 0)
|
||||
|
||||
if user_balance_kopeks < total_required:
|
||||
if total_required > 0 and user_balance_kopeks < total_required:
|
||||
await callback.answer('❌ Недостаточно средств на балансе для оплаты подписки', show_alert=True)
|
||||
return
|
||||
|
||||
@@ -2289,6 +2297,14 @@ async def confirm_simple_subscription_purchase(
|
||||
sync_error=sync_error,
|
||||
exc_info=True,
|
||||
)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(subscription, 'id') and hasattr(subscription, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='create',
|
||||
)
|
||||
|
||||
# Отправляем уведомление об успешной покупке
|
||||
server_label = _get_simple_subscription_server_label(
|
||||
|
||||
@@ -253,6 +253,14 @@ async def _handle_trial_payment(
|
||||
except Exception as rw_error:
|
||||
logger.error('Ошибка создания пользователя RemnaWave для триала', rw_error=rw_error)
|
||||
# Не откатываем подписку, просто логируем - RemnaWave может быть временно недоступен
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(subscription, 'id') and hasattr(subscription, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='create',
|
||||
)
|
||||
|
||||
await db.commit()
|
||||
await db.refresh(user)
|
||||
|
||||
@@ -234,6 +234,14 @@ async def _claim_phantom_user(
|
||||
subscription_id=phantom_sub.id,
|
||||
error=str(exc),
|
||||
)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(phantom_sub, 'id') and hasattr(phantom_sub, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=phantom_sub.id,
|
||||
user_id=phantom_sub.user_id,
|
||||
action='update',
|
||||
)
|
||||
|
||||
return True, phantom
|
||||
|
||||
@@ -2443,6 +2451,18 @@ async def required_sub_channel_check(
|
||||
telegram_id=user.telegram_id if user else query.from_user.id,
|
||||
api_error=api_error,
|
||||
)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
for sub in _subs:
|
||||
if sub.is_trial and sub.status == SubscriptionStatus.ACTIVE.value:
|
||||
if hasattr(sub, 'id') and hasattr(sub, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=sub.id,
|
||||
user_id=sub.user_id,
|
||||
action='update'
|
||||
if (getattr(sub, 'remnawave_uuid', None) or user.remnawave_uuid)
|
||||
else 'create',
|
||||
)
|
||||
|
||||
await query.answer(
|
||||
texts.t('CHANNEL_SUBSCRIBE_THANKS', '✅ Спасибо за подписку'),
|
||||
|
||||
@@ -412,7 +412,18 @@ async def apply_countries_changes(callback: types.CallbackQuery, db_user: User,
|
||||
await db.commit()
|
||||
|
||||
subscription_service = SubscriptionService()
|
||||
await subscription_service.update_remnawave_user(db, subscription, sync_squads=True)
|
||||
try:
|
||||
await subscription_service.update_remnawave_user(db, subscription, sync_squads=True)
|
||||
except Exception as rw_err:
|
||||
logger.error('Ошибка синхронизации с RemnaWave при смене стран', error=rw_err)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(subscription, 'id') and hasattr(subscription, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='update',
|
||||
)
|
||||
|
||||
await db.refresh(subscription)
|
||||
|
||||
@@ -861,7 +872,7 @@ async def confirm_add_countries_to_subscription(
|
||||
if country['uuid'] in removed_countries:
|
||||
removed_countries_names.append(html.escape(country['name']))
|
||||
|
||||
if new_countries and db_user.balance_kopeks < total_price:
|
||||
if new_countries and total_price > 0 and db_user.balance_kopeks < total_price:
|
||||
missing_kopeks = total_price - db_user.balance_kopeks
|
||||
message_text = texts.t(
|
||||
'ADDON_INSUFFICIENT_FUNDS_MESSAGE',
|
||||
|
||||
@@ -1273,7 +1273,7 @@ async def confirm_add_devices(callback: types.CallbackQuery, db_user: User, db:
|
||||
total_discount=total_discount / 100,
|
||||
)
|
||||
|
||||
if db_user.balance_kopeks < price:
|
||||
if price > 0 and db_user.balance_kopeks < price:
|
||||
missing_kopeks = price - db_user.balance_kopeks
|
||||
required_text = f'{texts.format_price(price)} (за {period_label})'
|
||||
message_text = texts.t(
|
||||
|
||||
@@ -937,6 +937,13 @@ async def activate_trial(callback: types.CallbackQuery, db_user: User, db: Async
|
||||
except Exception as e:
|
||||
logger.error('Ошибка получения триального тарифа', error=e)
|
||||
|
||||
# BUG-12 fix: If no squads from tariff, fallback to trial-eligible servers
|
||||
if not trial_squads:
|
||||
from app.database.crud.server_squad import get_random_trial_squad_uuid
|
||||
|
||||
trial_squad_uuid = await get_random_trial_squad_uuid(db)
|
||||
trial_squads = [trial_squad_uuid] if trial_squad_uuid else []
|
||||
|
||||
subscription = await create_trial_subscription(
|
||||
db,
|
||||
db_user.id,
|
||||
@@ -1537,7 +1544,7 @@ async def return_to_saved_cart(callback: types.CallbackQuery, state: FSMContext,
|
||||
|
||||
total_price = prepared_cart_data.get('total_price', 0)
|
||||
|
||||
if db_user.balance_kopeks < total_price:
|
||||
if total_price > 0 and db_user.balance_kopeks < total_price:
|
||||
missing_amount = total_price - db_user.balance_kopeks
|
||||
insufficient_keyboard = get_insufficient_balance_keyboard_with_cart(
|
||||
db_user.language,
|
||||
@@ -1635,7 +1642,7 @@ async def handle_extend_subscription(
|
||||
else:
|
||||
subscription = db_user.subscription
|
||||
|
||||
if not subscription or subscription.is_trial:
|
||||
if not subscription:
|
||||
await callback.message.edit_text(
|
||||
'🎯 <b>Пробный период заканчивается</b>\n\nЧтобы продолжить пользоваться VPN, выберите подходящий тариф.',
|
||||
reply_markup=types.InlineKeyboardMarkup(
|
||||
@@ -1654,24 +1661,53 @@ async def handle_extend_subscription(
|
||||
await callback.answer()
|
||||
return
|
||||
|
||||
# В режиме тарифов проверяем наличие tariff_id
|
||||
if settings.is_tariffs_mode():
|
||||
if subscription.tariff_id:
|
||||
# Проверяем, суточный ли тариф — у суточных нет period_prices, продление через resume
|
||||
from app.database.crud.tariff import get_tariff_by_id
|
||||
# Триальная подписка с тарифом — направляем на покупку этого тарифа
|
||||
if subscription.is_trial:
|
||||
if subscription.tariff_id and settings.is_tariffs_mode():
|
||||
from .tariff_purchase import show_tariff_extend
|
||||
|
||||
tariff = await get_tariff_by_id(db, subscription.tariff_id)
|
||||
if tariff and getattr(tariff, 'is_daily', False):
|
||||
# Суточный тариф: перенаправляем на страницу подписки (там кнопка «Возобновить»)
|
||||
await show_subscription_info(callback, db_user, db)
|
||||
return
|
||||
await show_tariff_extend(callback, db_user, db)
|
||||
return
|
||||
# Триал без тарифа — предлагаем выбрать
|
||||
await callback.message.edit_text(
|
||||
'🎯 <b>Пробный период заканчивается</b>\n\nЧтобы продолжить пользоваться VPN, выберите подходящий тариф.',
|
||||
reply_markup=types.InlineKeyboardMarkup(
|
||||
inline_keyboard=[
|
||||
[types.InlineKeyboardButton(text=texts.MENU_BUY_SUBSCRIPTION, callback_data='menu_buy')],
|
||||
[
|
||||
types.InlineKeyboardButton(
|
||||
text=texts.t('WEBHOOK_CLOSE_BUTTON', '✖️ Закрыть'),
|
||||
callback_data='webhook:close',
|
||||
)
|
||||
],
|
||||
]
|
||||
),
|
||||
parse_mode='HTML',
|
||||
)
|
||||
await callback.answer()
|
||||
return
|
||||
|
||||
# Подписка с тарифом — всегда используем тарифный flow,
|
||||
# даже если бот в классическом режиме (подписка могла быть куплена через кабинет)
|
||||
if subscription.tariff_id:
|
||||
# Проверяем, суточный ли тариф — у суточных нет period_prices, продление через resume
|
||||
from app.database.crud.tariff import get_tariff_by_id
|
||||
|
||||
tariff = await get_tariff_by_id(db, subscription.tariff_id)
|
||||
if tariff and getattr(tariff, 'is_daily', False):
|
||||
# Суточный тариф: перенаправляем на страницу подписки (там кнопка «Возобновить»)
|
||||
await show_subscription_info(callback, db_user, db)
|
||||
return
|
||||
|
||||
if tariff:
|
||||
# У подписки есть тариф - перенаправляем на продление по тарифу
|
||||
from .tariff_purchase import show_tariff_extend
|
||||
|
||||
await show_tariff_extend(callback, db_user, db)
|
||||
return
|
||||
# У подписки нет тарифа - предлагаем выбрать тариф
|
||||
|
||||
if settings.is_tariffs_mode():
|
||||
# У подписки нет тарифа, но режим тарифов включён - предлагаем выбрать тариф
|
||||
await callback.message.edit_text(
|
||||
'📦 <b>Выберите тариф для продления</b>\n\n'
|
||||
'Ваша текущая подписка была создана до введения тарифов.\n'
|
||||
@@ -1706,6 +1742,10 @@ async def handle_extend_subscription(
|
||||
# original = price before ALL discounts, final = price with all discounts
|
||||
total_original_price = pricing.original_total
|
||||
|
||||
# Пропускаем периоды с нулевой ценой (если оригинальная цена тоже 0 — не настроен)
|
||||
if pricing.final_total <= 0 and pricing.original_total <= 0:
|
||||
continue
|
||||
|
||||
renewal_prices[days] = {
|
||||
'final': pricing.final_total,
|
||||
'original': total_original_price,
|
||||
@@ -1899,7 +1939,7 @@ async def confirm_extend_subscription(
|
||||
await callback.answer('⚠ Ошибка расчета стоимости', show_alert=True)
|
||||
return
|
||||
|
||||
if db_user.balance_kopeks < price:
|
||||
if price > 0 and db_user.balance_kopeks < price:
|
||||
missing_kopeks = price - db_user.balance_kopeks
|
||||
required_text = texts.format_price(price)
|
||||
message_text = texts.t(
|
||||
@@ -2307,7 +2347,7 @@ async def confirm_purchase(callback: types.CallbackQuery, state: FSMContext, db_
|
||||
)
|
||||
logger.info('ИТОГО: ₽', final_price=final_price / 100)
|
||||
|
||||
if db_user.balance_kopeks < final_price:
|
||||
if final_price > 0 and db_user.balance_kopeks < final_price:
|
||||
missing_kopeks = final_price - db_user.balance_kopeks
|
||||
message_text = texts.t(
|
||||
'ADDON_INSUFFICIENT_FUNDS_MESSAGE',
|
||||
@@ -2546,17 +2586,19 @@ async def confirm_purchase(callback: types.CallbackQuery, state: FSMContext, db_
|
||||
|
||||
subscription_service = SubscriptionService()
|
||||
# При покупке подписки ВСЕГДА сбрасываем трафик в панели
|
||||
_purchase_uuid = (
|
||||
subscription.remnawave_uuid
|
||||
if settings.is_multi_tariff_enabled() and subscription.remnawave_uuid
|
||||
else db_user.remnawave_uuid
|
||||
)
|
||||
if settings.is_multi_tariff_enabled() and not getattr(subscription, 'remnawave_uuid', None):
|
||||
logger.warning(
|
||||
'Multi-tariff: subscription missing remnawave_uuid, using user fallback',
|
||||
subscription_id=getattr(subscription, 'id', None),
|
||||
if settings.is_multi_tariff_enabled():
|
||||
_should_create = not subscription.remnawave_uuid
|
||||
else:
|
||||
_should_create = not getattr(db_user, 'remnawave_uuid', None)
|
||||
|
||||
if _should_create:
|
||||
remnawave_user = await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=True,
|
||||
reset_reason='покупка подписки',
|
||||
)
|
||||
if _purchase_uuid:
|
||||
else:
|
||||
remnawave_user = await subscription_service.update_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
@@ -2564,22 +2606,25 @@ async def confirm_purchase(callback: types.CallbackQuery, state: FSMContext, db_
|
||||
reset_reason='покупка подписки',
|
||||
sync_squads=True,
|
||||
)
|
||||
else:
|
||||
remnawave_user = await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=True,
|
||||
reset_reason='покупка подписки',
|
||||
)
|
||||
|
||||
if not remnawave_user:
|
||||
logger.error('Не удалось создать/обновить RemnaWave пользователя для', telegram_id=db_user.telegram_id)
|
||||
remnawave_user = await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=True,
|
||||
reset_reason='покупка подписки (повторная попытка)',
|
||||
)
|
||||
logger.error('Не удалось создать/обновить RemnaWave пользователя', telegram_id=db_user.telegram_id)
|
||||
try:
|
||||
remnawave_user = await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=True,
|
||||
reset_reason='покупка подписки (повторная попытка)',
|
||||
)
|
||||
except Exception as retry_error:
|
||||
logger.error('Повторная попытка создания RemnaWave пользователя не удалась', error=retry_error)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=db_user.id,
|
||||
action='create',
|
||||
)
|
||||
|
||||
transaction = await create_transaction(
|
||||
db=db,
|
||||
@@ -3129,6 +3174,13 @@ async def handle_toggle_daily_subscription_pause(callback: types.CallbackQuery,
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error('Ошибка синхронизации с Remnawave при возобновлении', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=db_user.id,
|
||||
action='update',
|
||||
)
|
||||
|
||||
# Отправляем уведомление администраторам о возобновлении суточной подписки
|
||||
if resume_transaction is not None:
|
||||
@@ -3270,6 +3322,13 @@ async def handle_trial_pay_with_balance(callback: types.CallbackQuery, db_user:
|
||||
except Exception as e:
|
||||
logger.error('Ошибка получения триального тарифа для платного триала', error=e)
|
||||
|
||||
# BUG-12 fix: If no squads from tariff, fallback to trial-eligible servers
|
||||
if not trial_squads:
|
||||
from app.database.crud.server_squad import get_random_trial_squad_uuid
|
||||
|
||||
trial_squad_uuid = await get_random_trial_squad_uuid(db)
|
||||
trial_squads = [trial_squad_uuid] if trial_squad_uuid else []
|
||||
|
||||
subscription = await create_trial_subscription(
|
||||
db,
|
||||
db_user.id,
|
||||
@@ -4415,8 +4474,8 @@ async def _extend_existing_subscription(
|
||||
device_limit=device_limit,
|
||||
)
|
||||
|
||||
# Проверяем баланс пользователя
|
||||
if db_user.balance_kopeks < price_kopeks:
|
||||
# Проверяем баланс пользователя (при 100% скидке — пропускаем)
|
||||
if price_kopeks > 0 and db_user.balance_kopeks < price_kopeks:
|
||||
missing_kopeks = price_kopeks - db_user.balance_kopeks
|
||||
message_text = texts.t(
|
||||
'ADDON_INSUFFICIENT_FUNDS_MESSAGE',
|
||||
@@ -4561,6 +4620,13 @@ async def _extend_existing_subscription(
|
||||
logger.error('⚠ ОШИБКА ОБНОВЛЕНИЯ REMNAWAVE')
|
||||
except Exception as e:
|
||||
logger.error('⚠ ИСКЛЮЧЕНИЕ ПРИ ОБНОВЛЕНИИ REMNAWAVE', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=current_subscription.id,
|
||||
user_id=db_user.id,
|
||||
action='update',
|
||||
)
|
||||
|
||||
# Создаём транзакцию
|
||||
transaction = await create_transaction(
|
||||
|
||||
@@ -576,7 +576,7 @@ async def show_tariffs_list(
|
||||
from app.database.crud.subscription import get_active_subscriptions_by_user_id
|
||||
|
||||
active_subs = await get_active_subscriptions_by_user_id(db, db_user.id)
|
||||
purchased_tariff_ids = {s.tariff_id for s in active_subs if s.tariff_id and s.status in ('active', 'trial')}
|
||||
purchased_tariff_ids = {s.tariff_id for s in active_subs if s.tariff_id and not s.is_trial}
|
||||
|
||||
# Проверяем есть ли у пользователя скидки по периодам
|
||||
promo_group = db_user.get_primary_promo_group() if hasattr(db_user, 'get_primary_promo_group') else None
|
||||
@@ -619,7 +619,7 @@ async def select_tariff(
|
||||
from app.database.crud.subscription import get_active_subscriptions_by_user_id
|
||||
|
||||
_active = await get_active_subscriptions_by_user_id(db, db_user.id)
|
||||
_existing = next((s for s in _active if s.tariff_id == tariff_id and s.status in ('active', 'trial')), None)
|
||||
_existing = next((s for s in _active if s.tariff_id == tariff_id and not s.is_trial), None)
|
||||
if _existing:
|
||||
days_left = max(0, (_existing.end_date - datetime.now(UTC)).days) if _existing.end_date else 0
|
||||
await callback.answer(
|
||||
@@ -928,14 +928,14 @@ async def handle_custom_confirm(
|
||||
)
|
||||
total_price = result.final_total
|
||||
|
||||
# Проверяем, что цена за период валидна
|
||||
if result.base_price == 0 and not tariff.can_purchase_custom_days():
|
||||
# Проверяем, что цена за период валидна (original_total — цена до скидок)
|
||||
if result.original_total == 0 and not tariff.can_purchase_custom_days():
|
||||
await callback.answer('Выбранный период недоступен для этого тарифа', show_alert=True)
|
||||
return
|
||||
|
||||
# Проверяем баланс (user already locked, balance is fresh)
|
||||
# Проверяем баланс (при 100% скидке — пропускаем)
|
||||
user_balance = db_user.balance_kopeks or 0
|
||||
if user_balance < total_price:
|
||||
if total_price > 0 and user_balance < total_price:
|
||||
await callback.answer('Недостаточно средств на балансе', show_alert=True)
|
||||
return
|
||||
|
||||
@@ -1057,14 +1057,34 @@ async def handle_custom_confirm(
|
||||
# При покупке тарифа ВСЕГДА сбрасываем трафик в панели
|
||||
try:
|
||||
subscription_service = SubscriptionService()
|
||||
await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=True,
|
||||
reset_reason='покупка тарифа',
|
||||
)
|
||||
if settings.is_multi_tariff_enabled():
|
||||
_should_create = not subscription.remnawave_uuid
|
||||
else:
|
||||
_should_create = not getattr(db_user, 'remnawave_uuid', None)
|
||||
|
||||
if _should_create:
|
||||
await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=True,
|
||||
reset_reason='покупка тарифа',
|
||||
)
|
||||
else:
|
||||
await subscription_service.update_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=True,
|
||||
reset_reason='покупка тарифа',
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error('Ошибка обновления Remnawave', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=db_user.id,
|
||||
action='create',
|
||||
)
|
||||
|
||||
# Создаем транзакцию
|
||||
await create_transaction(
|
||||
@@ -1353,7 +1373,7 @@ async def confirm_tariff_purchase(
|
||||
|
||||
# Проверяем баланс (user already locked, balance is fresh)
|
||||
user_balance = db_user.balance_kopeks or 0
|
||||
if user_balance < final_price:
|
||||
if final_price > 0 and user_balance < final_price:
|
||||
await callback.answer('Недостаточно средств на балансе', show_alert=True)
|
||||
return
|
||||
|
||||
@@ -1568,14 +1588,37 @@ async def confirm_tariff_purchase(
|
||||
# При покупке тарифа ВСЕГДА сбрасываем трафик в панели
|
||||
try:
|
||||
subscription_service = SubscriptionService()
|
||||
await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=True,
|
||||
reset_reason='покупка тарифа',
|
||||
)
|
||||
# In multi-tariff mode, each subscription has its own panel user.
|
||||
# A new subscription has no remnawave_uuid yet, so always CREATE.
|
||||
# In single-tariff mode, reuse the user-level UUID if available.
|
||||
if settings.is_multi_tariff_enabled():
|
||||
_should_create = not subscription.remnawave_uuid
|
||||
else:
|
||||
_should_create = not getattr(db_user, 'remnawave_uuid', None)
|
||||
|
||||
if _should_create:
|
||||
await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=True,
|
||||
reset_reason='покупка тарифа',
|
||||
)
|
||||
else:
|
||||
await subscription_service.update_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=True,
|
||||
reset_reason='покупка тарифа',
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error('Ошибка обновления Remnawave', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=db_user.id,
|
||||
action='create',
|
||||
)
|
||||
|
||||
# Создаем транзакцию
|
||||
try:
|
||||
@@ -1694,7 +1737,7 @@ async def confirm_daily_tariff_purchase(
|
||||
|
||||
# Проверяем баланс (user already locked, balance is fresh)
|
||||
user_balance = db_user.balance_kopeks or 0
|
||||
if user_balance < final_daily_price:
|
||||
if final_daily_price > 0 and user_balance < final_daily_price:
|
||||
await callback.answer('Недостаточно средств на балансе', show_alert=True)
|
||||
return
|
||||
|
||||
@@ -1828,14 +1871,34 @@ async def confirm_daily_tariff_purchase(
|
||||
# При покупке тарифа ВСЕГДА сбрасываем трафик в панели
|
||||
try:
|
||||
subscription_service = SubscriptionService()
|
||||
await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=True,
|
||||
reset_reason='покупка суточного тарифа',
|
||||
)
|
||||
if settings.is_multi_tariff_enabled():
|
||||
_should_create = not subscription.remnawave_uuid
|
||||
else:
|
||||
_should_create = not getattr(db_user, 'remnawave_uuid', None)
|
||||
|
||||
if _should_create:
|
||||
await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=True,
|
||||
reset_reason='покупка суточного тарифа',
|
||||
)
|
||||
else:
|
||||
await subscription_service.update_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=True,
|
||||
reset_reason='покупка суточного тарифа',
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error('Ошибка обновления Remnawave', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=db_user.id,
|
||||
action='create',
|
||||
)
|
||||
|
||||
# Создаем транзакцию
|
||||
await create_transaction(
|
||||
@@ -2013,8 +2076,6 @@ async def show_tariff_extend(
|
||||
# Show subscription picker for extending
|
||||
keyboard = []
|
||||
for sub in sorted(active_subs, key=lambda s: s.id):
|
||||
if sub.is_trial:
|
||||
continue
|
||||
tariff_name = ''
|
||||
if sub.tariff_id:
|
||||
_t = await get_tariff_by_id(db, sub.tariff_id)
|
||||
@@ -2246,7 +2307,7 @@ async def confirm_tariff_extend(
|
||||
|
||||
# Проверяем баланс
|
||||
user_balance = db_user.balance_kopeks or 0
|
||||
if user_balance < final_price:
|
||||
if final_price > 0 and user_balance < final_price:
|
||||
await callback.answer('Недостаточно средств на балансе', show_alert=True)
|
||||
return
|
||||
|
||||
@@ -2266,24 +2327,50 @@ async def confirm_tariff_extend(
|
||||
await callback.answer('Ошибка списания баланса', show_alert=True)
|
||||
return
|
||||
|
||||
# Продлеваем подписку (параметры тарифа не меняются, только добавляется время)
|
||||
# Запоминаем, был ли триал ДО продления
|
||||
was_trial = subscription.is_trial
|
||||
|
||||
# Продлеваем подписку; для триала передаём tariff_id чтобы сбросить is_trial
|
||||
subscription = await extend_subscription(
|
||||
db,
|
||||
subscription,
|
||||
days=period,
|
||||
tariff_id=tariff.id if was_trial else None,
|
||||
traffic_limit_gb=tariff.traffic_limit_gb if was_trial else None,
|
||||
device_limit=actual_device_limit if was_trial else None,
|
||||
)
|
||||
|
||||
# Обновляем пользователя в Remnawave
|
||||
try:
|
||||
subscription_service = SubscriptionService()
|
||||
await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=settings.RESET_TRAFFIC_ON_PAYMENT,
|
||||
reset_reason='продление тарифа',
|
||||
)
|
||||
if settings.is_multi_tariff_enabled():
|
||||
_should_create = not subscription.remnawave_uuid
|
||||
else:
|
||||
_should_create = not getattr(db_user, 'remnawave_uuid', None)
|
||||
|
||||
if _should_create:
|
||||
await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=settings.RESET_TRAFFIC_ON_PAYMENT or was_trial,
|
||||
reset_reason='конвертация триала' if was_trial else 'продление тарифа',
|
||||
)
|
||||
else:
|
||||
await subscription_service.update_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=settings.RESET_TRAFFIC_ON_PAYMENT or was_trial,
|
||||
reset_reason='конвертация триала' if was_trial else 'продление тарифа',
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error('Ошибка обновления Remnawave', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=db_user.id,
|
||||
action='create',
|
||||
)
|
||||
|
||||
# Создаем транзакцию
|
||||
await create_transaction(
|
||||
@@ -2303,7 +2390,7 @@ async def confirm_tariff_extend(
|
||||
subscription,
|
||||
None, # Транзакция отсутствует, оплата с баланса
|
||||
period,
|
||||
was_trial_conversion=False,
|
||||
was_trial_conversion=was_trial,
|
||||
amount_kopeks=final_price,
|
||||
purchase_type='renewal',
|
||||
)
|
||||
@@ -2836,7 +2923,7 @@ async def confirm_tariff_switch(
|
||||
|
||||
# Проверяем баланс
|
||||
user_balance = db_user.balance_kopeks or 0
|
||||
if user_balance < final_price:
|
||||
if final_price > 0 and user_balance < final_price:
|
||||
await callback.answer('Недостаточно средств на балансе', show_alert=True)
|
||||
return
|
||||
|
||||
@@ -2889,14 +2976,34 @@ async def confirm_tariff_switch(
|
||||
# Обновляем пользователя в Remnawave
|
||||
try:
|
||||
subscription_service = SubscriptionService()
|
||||
await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=settings.RESET_TRAFFIC_ON_TARIFF_SWITCH,
|
||||
reset_reason='переключение тарифа',
|
||||
)
|
||||
if settings.is_multi_tariff_enabled():
|
||||
_should_create = not subscription.remnawave_uuid
|
||||
else:
|
||||
_should_create = not getattr(db_user, 'remnawave_uuid', None)
|
||||
|
||||
if _should_create:
|
||||
await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=settings.RESET_TRAFFIC_ON_TARIFF_SWITCH,
|
||||
reset_reason='переключение тарифа',
|
||||
)
|
||||
else:
|
||||
await subscription_service.update_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=settings.RESET_TRAFFIC_ON_TARIFF_SWITCH,
|
||||
reset_reason='переключение тарифа',
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error('Ошибка обновления Remnawave при переключении тарифа', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=db_user.id,
|
||||
action='create',
|
||||
)
|
||||
|
||||
# Гарантированный сброс устройств при смене тарифа
|
||||
await db.refresh(db_user)
|
||||
@@ -3042,7 +3149,7 @@ async def confirm_daily_tariff_switch(
|
||||
|
||||
# Проверяем баланс (user already locked, balance is fresh)
|
||||
user_balance = db_user.balance_kopeks or 0
|
||||
if user_balance < final_daily_price:
|
||||
if final_daily_price > 0 and user_balance < final_daily_price:
|
||||
await callback.answer('Недостаточно средств на балансе', show_alert=True)
|
||||
return
|
||||
|
||||
@@ -3117,14 +3224,34 @@ async def confirm_daily_tariff_switch(
|
||||
# Обновляем пользователя в Remnawave (сброс трафика по админ-настройке)
|
||||
try:
|
||||
subscription_service = SubscriptionService()
|
||||
await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=settings.RESET_TRAFFIC_ON_TARIFF_SWITCH,
|
||||
reset_reason='смена на суточный тариф',
|
||||
)
|
||||
if settings.is_multi_tariff_enabled():
|
||||
_should_create = not subscription.remnawave_uuid
|
||||
else:
|
||||
_should_create = not getattr(db_user, 'remnawave_uuid', None)
|
||||
|
||||
if _should_create:
|
||||
await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=settings.RESET_TRAFFIC_ON_TARIFF_SWITCH,
|
||||
reset_reason='смена на суточный тариф',
|
||||
)
|
||||
else:
|
||||
await subscription_service.update_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=settings.RESET_TRAFFIC_ON_TARIFF_SWITCH,
|
||||
reset_reason='смена на суточный тариф',
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error('Ошибка обновления Remnawave', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=db_user.id,
|
||||
action='create',
|
||||
)
|
||||
|
||||
# Гарантированный сброс устройств при смене тарифа
|
||||
await db.refresh(db_user)
|
||||
@@ -3791,14 +3918,34 @@ async def confirm_instant_switch(
|
||||
# Обновляем пользователя в Remnawave (сброс трафика по админ-настройке)
|
||||
try:
|
||||
subscription_service = SubscriptionService()
|
||||
await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=settings.RESET_TRAFFIC_ON_TARIFF_SWITCH,
|
||||
reset_reason='мгновенное переключение тарифа',
|
||||
)
|
||||
if settings.is_multi_tariff_enabled():
|
||||
_should_create = not subscription.remnawave_uuid
|
||||
else:
|
||||
_should_create = not getattr(db_user, 'remnawave_uuid', None)
|
||||
|
||||
if _should_create:
|
||||
await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=settings.RESET_TRAFFIC_ON_TARIFF_SWITCH,
|
||||
reset_reason='мгновенное переключение тарифа',
|
||||
)
|
||||
else:
|
||||
await subscription_service.update_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=settings.RESET_TRAFFIC_ON_TARIFF_SWITCH,
|
||||
reset_reason='мгновенное переключение тарифа',
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error('Ошибка обновления Remnawave при мгновенном переключении', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=db_user.id,
|
||||
action='create',
|
||||
)
|
||||
|
||||
# Гарантированный сброс устройств при смене тарифа
|
||||
await db.refresh(db_user)
|
||||
@@ -3946,8 +4093,8 @@ async def return_to_saved_tariff_cart(
|
||||
user_balance = db_user.balance_kopeks or 0
|
||||
traffic = format_traffic(tariff.traffic_limit_gb)
|
||||
|
||||
# Проверяем баланс
|
||||
if user_balance < total_price:
|
||||
# Проверяем баланс (при 100% скидке — пропускаем)
|
||||
if total_price > 0 and user_balance < total_price:
|
||||
missing = total_price - user_balance
|
||||
|
||||
if cart_mode == 'daily_tariff_purchase':
|
||||
|
||||
@@ -332,7 +332,7 @@ async def confirm_reset_traffic(
|
||||
|
||||
reset_price = _calculate_traffic_reset_price(subscription)
|
||||
|
||||
if db_user.balance_kopeks < reset_price:
|
||||
if reset_price > 0 and db_user.balance_kopeks < reset_price:
|
||||
missing_kopeks = reset_price - db_user.balance_kopeks
|
||||
message_text = texts.t(
|
||||
'ADDON_INSUFFICIENT_FUNDS_MESSAGE',
|
||||
@@ -574,7 +574,7 @@ async def add_traffic(callback: types.CallbackQuery, db_user: User, db: AsyncSes
|
||||
|
||||
total_discount_value = int(discount_per_month * charged_days / 30)
|
||||
|
||||
if db_user.balance_kopeks < price:
|
||||
if price > 0 and db_user.balance_kopeks < price:
|
||||
missing_kopeks = price - db_user.balance_kopeks
|
||||
|
||||
# Save cart for auto-purchase after balance top-up
|
||||
@@ -830,7 +830,7 @@ async def confirm_switch_traffic(
|
||||
total_price_difference = int(price_difference_per_month * days_remaining / 30)
|
||||
total_price_difference = max(100, total_price_difference)
|
||||
|
||||
if db_user.balance_kopeks < total_price_difference:
|
||||
if total_price_difference > 0 and db_user.balance_kopeks < total_price_difference:
|
||||
missing_kopeks = total_price_difference - db_user.balance_kopeks
|
||||
message_text = texts.t(
|
||||
'ADDON_INSUFFICIENT_FUNDS_MESSAGE',
|
||||
|
||||
@@ -983,7 +983,14 @@ async def close_ticket_notification(callback: types.CallbackQuery, db_user: User
|
||||
await callback.answer()
|
||||
return
|
||||
|
||||
await callback.message.delete()
|
||||
try:
|
||||
await callback.message.delete()
|
||||
except TelegramBadRequest:
|
||||
# Message is too old to delete (>48h) — edit it instead
|
||||
try:
|
||||
await callback.message.edit_text(texts.t('NOTIFICATION_CLOSED', 'Уведомление закрыто.'))
|
||||
except TelegramBadRequest:
|
||||
pass
|
||||
await callback.answer(texts.t('NOTIFICATION_CLOSED', 'Уведомление закрыто.'))
|
||||
|
||||
|
||||
|
||||
@@ -1755,5 +1755,8 @@
|
||||
"WEBHOOK_USER_NOT_CONNECTED": "📡 <b>Not connected yet</b>\n\nYour subscription{tariff_label} is active but no VPN connection has been made. Connect to start using the service.",
|
||||
"WEBHOOK_DEVICE_ADDED": "📱 <b>New device</b>\n\nA new device has been added to your subscription{tariff_label}: <code>{device}</code>",
|
||||
"WEBHOOK_DEVICE_DELETED": "📱 <b>Device removed</b>\n\nA device has been removed from your subscription{tariff_label}: <code>{device}</code>",
|
||||
"WEBHOOK_CLOSE_BUTTON": "✖️ Close"
|
||||
"WEBHOOK_TORRENT_DETECTED": "🚫 <b>Torrent detected</b>\n\nTorrent traffic was detected on your connection{tariff_label}. Using torrents may result in subscription restrictions.",
|
||||
"WEBHOOK_CLOSE_BUTTON": "✖️ Close",
|
||||
"TRAFFIC_WARNING_ALERT": "⚠️ <b>Traffic Warning</b>\n\nUsed: {used:.1f} / {limit} GB ({percent:.0f}%)\n\nYour traffic limit is almost reached.",
|
||||
"LOW_BALANCE_ALERT": "⚠️ <b>Low Balance</b>\n\nYour balance: {balance} ₽\nNotification threshold: {threshold} ₽\n\nTop up your balance to ensure automatic subscription renewal."
|
||||
}
|
||||
+1781
-1778
File diff suppressed because it is too large
Load Diff
@@ -1749,19 +1749,16 @@
|
||||
"MODEM_PRICE_WITH_DISCOUNT": "Стоимость: <s>{base_price}</s> <b>{final_price}</b> (за {months} мес)\n🎁 Скидка {discount}%: -{discount_amount}",
|
||||
"MODEM_PRICE_NO_DISCOUNT": "Стоимость: {price} (за {months} мес)",
|
||||
"MODEM_CONFIRM_ENABLE_BASE": "📡 <b>Подтверждение подключения модема</b>\n\n{price_text}\n\nПри подключении модема:\n• К подписке добавится дополнительное устройство\n• Ежемесячная плата увеличится на {monthly_price}\n\nПодтвердить подключение?",
|
||||
|
||||
"ADMIN_USER_RESTRICTIONS": "⚠️ Ограничить",
|
||||
"USER_RESTRICTION_TOPUP_BLOCKED": "🚫 <b>Пополнение ограничено</b>\n\n{reason}\n\nЕсли вы считаете это ошибкой, вы можете обжаловать решение.",
|
||||
"USER_RESTRICTION_SUBSCRIPTION_BLOCKED": "🚫 <b>Покупка/продление подписки ограничено</b>\n\n{reason}\n\nЕсли вы считаете это ошибкой, вы можете обжаловать решение.",
|
||||
"USER_RESTRICTION_APPEAL_BUTTON": "🆘 Обжаловать",
|
||||
|
||||
"PAUSE_DAILY_BUTTON": "⏸️ Приостановить подписку",
|
||||
"RESUME_DAILY_BUTTON": "▶️ Возобновить подписку",
|
||||
"DAILY_SWITCH_WARNING": "⚠️ <b>Внимание!</b> У вас осталось {days} дн. подписки.\nПри смене на суточный тариф они будут утеряны!",
|
||||
"DAILY_SUBSCRIPTION_PAUSED": "⏸️ Подписка приостановлена",
|
||||
"DAILY_SUBSCRIPTION_RESUMED": "▶️ Подписка возобновлена!",
|
||||
"DAILY_SUBSCRIPTION_RESUMED_AFTER_TOPUP": "✅ <b>Подписка возобновлена!</b>\n\nВаш суточный тариф «{tariff_name}» возобновлён после пополнения баланса.\n\n💳 Списано: {amount}\n💰 Остаток: {balance}",
|
||||
|
||||
"WEBHOOK_SUB_EXPIRED": "❌ <b>Подписка{tariff_label} истекла</b>\n\nВаша подписка завершена. Продлите подписку, чтобы восстановить доступ к VPN.",
|
||||
"WEBHOOK_SUB_DISABLED": "🚫 <b>Подписка{tariff_label} отключена</b>\n\nВаша подписка была отключена администратором.",
|
||||
"WEBHOOK_SUB_ENABLED": "✅ <b>Подписка{tariff_label} активирована</b>\n\nВаша подписка снова активна. Приятного использования!",
|
||||
@@ -1779,5 +1776,8 @@
|
||||
"WEBHOOK_USER_NOT_CONNECTED": "📡 <b>Вы ещё не подключились</b>\n\nВаша подписка{tariff_label} активна, но VPN-соединение не было установлено. Подключитесь, чтобы начать пользоваться.",
|
||||
"WEBHOOK_DEVICE_ADDED": "📱 <b>Новое устройство</b>\n\nК подписке{tariff_label} подключено новое устройство: <code>{device}</code>",
|
||||
"WEBHOOK_DEVICE_DELETED": "📱 <b>Устройство удалено</b>\n\nУстройство отключено от подписки{tariff_label}: <code>{device}</code>",
|
||||
"WEBHOOK_CLOSE_BUTTON": "✖️ Закрыть"
|
||||
}
|
||||
"WEBHOOK_TORRENT_DETECTED": "🚫 <b>Обнаружен торрент</b>\n\nВ вашем подключении{tariff_label} обнаружен торрент-трафик. Использование торрентов может привести к ограничению подписки.",
|
||||
"WEBHOOK_CLOSE_BUTTON": "✖️ Закрыть",
|
||||
"TRAFFIC_WARNING_ALERT": "⚠️ <b>Предупреждение о трафике</b>\n\nИспользовано: {used:.1f} / {limit} ГБ ({percent:.0f}%)\n\nВаш лимит трафика почти исчерпан.",
|
||||
"LOW_BALANCE_ALERT": "⚠️ <b>Низкий баланс</b>\n\nВаш баланс: {balance} ₽\nПорог уведомления: {threshold} ₽\n\nПополните баланс, чтобы автопродление подписки прошло успешно."
|
||||
}
|
||||
+1529
-1526
File diff suppressed because it is too large
Load Diff
+1651
-1648
File diff suppressed because it is too large
Load Diff
@@ -85,6 +85,9 @@ class DisplayNameRestrictionMiddleware(BaseMiddleware):
|
||||
if not user or user.is_bot:
|
||||
return await handler(event, data)
|
||||
|
||||
if not settings.DISPLAY_NAME_RESTRICTION_ENABLED:
|
||||
return await handler(event, data)
|
||||
|
||||
display_name = self._build_display_name(user)
|
||||
username = user.username or ''
|
||||
|
||||
|
||||
@@ -86,33 +86,38 @@ class BlacklistService:
|
||||
if not line or line.startswith('#'):
|
||||
continue # Пропускаем пустые строки и комментарии
|
||||
|
||||
# В формате '7021477105 #@MAMYT_PAXAL2016, перепродажа подписок'
|
||||
# В формате '7021477105 # @MAMYT_PAXAL2016, перепродажа подписок'
|
||||
# только первая часть до пробела - это Telegram ID, всё остальное комментарий
|
||||
parts = line.split()
|
||||
if not parts:
|
||||
continue
|
||||
|
||||
try:
|
||||
telegram_id = int(parts[0]) # Первое число - это Telegram ID
|
||||
# Всё остальное - просто комментарий, не используем его для логики
|
||||
# Но можем использовать первую часть после ID как username для отображения
|
||||
username = ''
|
||||
if len(parts) > 1:
|
||||
# Берем вторую часть как username (если начинается с @)
|
||||
if parts[1].startswith('@'):
|
||||
username = parts[1]
|
||||
# 1. Разделяем строку на ID и всё остальное по символу '#'
|
||||
if '#' in line:
|
||||
id_part, content_part = line.split('#', 1)
|
||||
telegram_id = int(id_part.strip())
|
||||
content = content_part.strip()
|
||||
else:
|
||||
# Если решётки нет, пробуем просто взять первое число
|
||||
parts = line.split(maxsplit=1)
|
||||
telegram_id = int(parts[0])
|
||||
content = parts[1].strip() if len(parts) > 1 else ''
|
||||
|
||||
# По умолчанию используем "Занесен в черный список", если нет другой информации
|
||||
# 2. Обрабатываем контент: вычленяем username, если он есть в начале
|
||||
username = ''
|
||||
reason = 'Занесен в черный список'
|
||||
|
||||
# Если есть запятая в строке, можем использовать часть после нее как причину
|
||||
full_line_after_id = line[len(str(telegram_id)) :].strip()
|
||||
if ',' in full_line_after_id:
|
||||
# Извлекаем причину после запятой
|
||||
after_comma = full_line_after_id.split(',', 1)[1].strip()
|
||||
reason = after_comma
|
||||
if content:
|
||||
if content.startswith('@'):
|
||||
# Разбиваем контент только по первому пробелу
|
||||
# content_parts[0] будет юзернеймом, content_parts[1] — причиной
|
||||
content_parts = content.split(maxsplit=1)
|
||||
username = content_parts[0]
|
||||
if len(content_parts) > 1:
|
||||
reason = content_parts[1].strip()
|
||||
else:
|
||||
# Если собачки нет, значит весь контент — это причина
|
||||
reason = content
|
||||
|
||||
blacklist_data.append((telegram_id, username, reason))
|
||||
|
||||
except ValueError:
|
||||
# Если не удается преобразовать в число, это не ID
|
||||
logger.warning(
|
||||
|
||||
@@ -62,6 +62,7 @@ class BroadcastConfig:
|
||||
media: BroadcastMediaConfig | None = None
|
||||
initiator_name: str | None = None
|
||||
custom_buttons: list[dict] | None = None
|
||||
category: str = 'system' # system|news|promo
|
||||
|
||||
|
||||
@dataclass
|
||||
@@ -160,7 +161,7 @@ class BroadcastService:
|
||||
await session.commit()
|
||||
|
||||
# _fetch_recipients теперь возвращает list[int] (telegram_id), а не ORM-объекты
|
||||
recipient_ids: list[int] = await self._fetch_recipients(config.target)
|
||||
recipient_ids: list[int] = await self._fetch_recipients(config.target, config.category)
|
||||
|
||||
async with AsyncSessionLocal() as session:
|
||||
broadcast = await session.get(BroadcastHistory, broadcast_id)
|
||||
@@ -226,8 +227,13 @@ class BroadcastService:
|
||||
logger.exception('Критическая ошибка при выполнении рассылки', broadcast_id=broadcast_id, exc=exc)
|
||||
await self._mark_failed(broadcast_id, sent_count, failed_count, blocked_count)
|
||||
|
||||
async def _fetch_recipients(self, target: str) -> list[int]:
|
||||
"""Загружает получателей и возвращает список telegram_id (скаляры, не ORM-объекты)."""
|
||||
async def _fetch_recipients(self, target: str, category: str = 'system') -> list[int]:
|
||||
"""Загружает получателей и возвращает список telegram_id (скаляры, не ORM-объекты).
|
||||
|
||||
Filters out users who disabled the given broadcast category in their
|
||||
notification preferences (news_enabled, promo_offers_enabled).
|
||||
Category 'system' is never filtered — system notifications reach everyone.
|
||||
"""
|
||||
async with AsyncSessionLocal() as session:
|
||||
if target.startswith('custom_'):
|
||||
criteria = target[len('custom_') :]
|
||||
@@ -235,6 +241,17 @@ class BroadcastService:
|
||||
else:
|
||||
users_orm = await get_target_users(session, target)
|
||||
|
||||
# Filter by user notification preferences based on broadcast category
|
||||
if category == 'news':
|
||||
from app.utils.notification_prefs import is_news_enabled
|
||||
|
||||
users_orm = [u for u in users_orm if is_news_enabled(u)]
|
||||
elif category == 'promo':
|
||||
from app.utils.notification_prefs import is_promo_offers_enabled
|
||||
|
||||
users_orm = [u for u in users_orm if is_promo_offers_enabled(u)]
|
||||
# category == 'system' → no filtering, sent to everyone
|
||||
|
||||
# Извлекаем telegram_id сразу, пока сессия жива.
|
||||
# После выхода из блока ORM-объекты станут detached.
|
||||
return [u.telegram_id for u in users_orm if u.telegram_id is not None]
|
||||
|
||||
@@ -141,8 +141,8 @@ class DailySubscriptionService:
|
||||
PricingEngine.apply_discount(raw_daily_price, daily_group_pct) if daily_group_pct > 0 else raw_daily_price
|
||||
)
|
||||
|
||||
# Проверяем баланс
|
||||
if user.balance_kopeks < daily_price:
|
||||
# Проверяем баланс (при 100% скидке — пропускаем)
|
||||
if daily_price > 0 and user.balance_kopeks < daily_price:
|
||||
# Недостаточно средств - приостанавливаем подписку
|
||||
await suspend_daily_subscription_insufficient_balance(db, subscription)
|
||||
|
||||
@@ -264,6 +264,14 @@ class DailySubscriptionService:
|
||||
logger.warning('Не удалось синхронизировать сквады после создания', error=patch_err)
|
||||
except Exception as e:
|
||||
logger.warning('Не удалось обновить Remnawave', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(subscription, 'id') and hasattr(subscription, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='update' if _has_panel_user else 'create',
|
||||
)
|
||||
|
||||
# Отправляем уведомление администраторам
|
||||
try:
|
||||
@@ -539,6 +547,14 @@ class DailySubscriptionService:
|
||||
await subscription_service.update_remnawave_user(db, subscription)
|
||||
except Exception as e:
|
||||
logger.warning('Не удалось синхронизировать с RemnaWave после сброса трафика', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(subscription, 'id') and hasattr(subscription, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='update',
|
||||
)
|
||||
|
||||
# Уведомляем пользователя
|
||||
if self._bot and subscription.user_id:
|
||||
|
||||
@@ -15,7 +15,12 @@ from app.cabinet.auth.jwt_handler import create_auto_login_token
|
||||
from app.cabinet.auth.password_utils import hash_password
|
||||
from app.config import settings
|
||||
from app.database.crud.landing import create_guest_purchase
|
||||
from app.database.crud.subscription import create_paid_subscription, get_subscription_by_user_id, replace_subscription
|
||||
from app.database.crud.subscription import (
|
||||
create_paid_subscription,
|
||||
extend_subscription,
|
||||
get_subscription_by_user_id,
|
||||
replace_subscription,
|
||||
)
|
||||
from app.database.crud.tariff import get_tariff_by_id
|
||||
from app.database.crud.transaction import create_transaction
|
||||
from app.database.crud.user import _get_or_create_default_promo_group
|
||||
@@ -28,6 +33,7 @@ from app.database.models import (
|
||||
Transaction,
|
||||
TransactionType,
|
||||
User,
|
||||
_aware,
|
||||
)
|
||||
from app.services.subscription_service import SubscriptionService
|
||||
|
||||
@@ -1039,7 +1045,24 @@ async def activate_purchase(db: AsyncSession, purchase_token: str, *, skip_notif
|
||||
from app.database.crud.subscription import get_subscription_by_user_and_tariff
|
||||
|
||||
existing_for_tariff = await get_subscription_by_user_and_tariff(db, user.id, tariff.id)
|
||||
if existing_for_tariff:
|
||||
_has_time = (
|
||||
existing_for_tariff is not None
|
||||
and existing_for_tariff.end_date is not None
|
||||
and _aware(existing_for_tariff.end_date) > datetime.now(UTC)
|
||||
)
|
||||
if existing_for_tariff and _has_time:
|
||||
# Extend existing active/trial subscription instead of replacing (preserve remaining days)
|
||||
subscription = await extend_subscription(
|
||||
db,
|
||||
existing_for_tariff,
|
||||
purchase.period_days,
|
||||
traffic_limit_gb=tariff.traffic_limit_gb,
|
||||
device_limit=tariff.device_limit,
|
||||
connected_squads=squads,
|
||||
commit=False,
|
||||
)
|
||||
elif existing_for_tariff:
|
||||
# Expired subscription — replace with fresh dates
|
||||
subscription = await replace_subscription(
|
||||
db,
|
||||
existing_for_tariff,
|
||||
@@ -1066,7 +1089,25 @@ async def activate_purchase(db: AsyncSession, purchase_token: str, *, skip_notif
|
||||
)
|
||||
else:
|
||||
existing_subscription = await get_subscription_by_user_id(db, user.id)
|
||||
if existing_subscription is not None:
|
||||
_sub_has_time = (
|
||||
existing_subscription is not None
|
||||
and existing_subscription.end_date is not None
|
||||
and _aware(existing_subscription.end_date) > datetime.now(UTC)
|
||||
)
|
||||
if existing_subscription is not None and _sub_has_time:
|
||||
# Extend existing active subscription (preserve remaining days)
|
||||
subscription = await extend_subscription(
|
||||
db,
|
||||
existing_subscription,
|
||||
purchase.period_days,
|
||||
tariff_id=tariff.id,
|
||||
traffic_limit_gb=tariff.traffic_limit_gb,
|
||||
device_limit=tariff.device_limit,
|
||||
connected_squads=squads,
|
||||
commit=False,
|
||||
)
|
||||
elif existing_subscription is not None:
|
||||
# Expired subscription — replace with fresh dates
|
||||
subscription = await replace_subscription(
|
||||
db,
|
||||
existing_subscription,
|
||||
|
||||
@@ -49,7 +49,6 @@ from app.database.models import (
|
||||
from app.external.remnawave_api import (
|
||||
RemnaWaveAPIError,
|
||||
RemnaWaveUser,
|
||||
TrafficLimitStrategy,
|
||||
UserStatus as RemnaWaveUserStatus,
|
||||
)
|
||||
from app.localization.texts import get_texts
|
||||
@@ -58,7 +57,7 @@ from app.services.notification_delivery_service import (
|
||||
)
|
||||
from app.services.notification_settings_service import NotificationSettingsService
|
||||
from app.services.promo_offer_service import promo_offer_service
|
||||
from app.services.subscription_service import SubscriptionService
|
||||
from app.services.subscription_service import SubscriptionService, get_traffic_reset_strategy
|
||||
from app.utils.cache import cache
|
||||
from app.utils.message_patch import caption_exceeds_telegram_limit
|
||||
from app.utils.miniapp_buttons import build_miniapp_or_callback_button
|
||||
@@ -246,7 +245,10 @@ class MonitoringService:
|
||||
await self._check_trial_expiring_soon(db)
|
||||
await self._check_trial_channel_subscriptions(db)
|
||||
await self._check_expired_subscription_followups(db)
|
||||
await self._check_traffic_warnings(db)
|
||||
await self._check_low_balance_alerts(db)
|
||||
await self._retry_stuck_guest_purchases(db)
|
||||
await self._cleanup_expired_refresh_tokens(db)
|
||||
await self._cleanup_inactive_users(db)
|
||||
await self._sync_with_remnawave(db)
|
||||
|
||||
@@ -464,7 +466,7 @@ class MonitoringService:
|
||||
if is_active
|
||||
else max(subscription.end_date, current_time + timedelta(minutes=1)),
|
||||
traffic_limit_bytes=self._gb_to_bytes(subscription.traffic_limit_gb),
|
||||
traffic_limit_strategy=TrafficLimitStrategy.MONTH,
|
||||
traffic_limit_strategy=get_traffic_reset_strategy(subscription.tariff),
|
||||
description=settings.format_remnawave_user_description(
|
||||
full_name=user.full_name, username=user.username, telegram_id=user.telegram_id
|
||||
),
|
||||
@@ -518,11 +520,25 @@ class MonitoringService:
|
||||
|
||||
users_with_cards = await get_user_ids_with_active_payment_methods(db, autopay_user_ids)
|
||||
|
||||
from app.utils.notification_prefs import (
|
||||
get_subscription_expiry_days,
|
||||
is_subscription_expiry_enabled,
|
||||
)
|
||||
|
||||
for subscription in expiring_subscriptions:
|
||||
user = await get_user_by_id(db, subscription.user_id)
|
||||
if not user:
|
||||
continue
|
||||
|
||||
# Respect user notification preferences
|
||||
if not is_subscription_expiry_enabled(user):
|
||||
continue
|
||||
|
||||
# Check if user's preferred days threshold matches this check
|
||||
user_expiry_days = get_subscription_expiry_days(user)
|
||||
if days > user_expiry_days:
|
||||
continue
|
||||
|
||||
# Use user.id + subscription.id for key to support multiple subscriptions per user
|
||||
sub_key = f'user_{user.id}_sub_{subscription.id}_today'
|
||||
user_identifier = user.telegram_id or f'email:{user.id}'
|
||||
@@ -885,18 +901,24 @@ class MonitoringService:
|
||||
)
|
||||
|
||||
try:
|
||||
panel_uuid_restore = (
|
||||
subscription.remnawave_uuid
|
||||
if settings.is_multi_tariff_enabled() and subscription.remnawave_uuid
|
||||
else user.remnawave_uuid
|
||||
)
|
||||
if panel_uuid_restore:
|
||||
await self.subscription_service.enable_remnawave_user(panel_uuid_restore)
|
||||
if settings.is_multi_tariff_enabled():
|
||||
_should_create = not subscription.remnawave_uuid
|
||||
else:
|
||||
_should_create = not getattr(user, 'remnawave_uuid', None)
|
||||
|
||||
if _should_create:
|
||||
# create_remnawave_user calls db.commit() internally --
|
||||
# flush accumulated batch state first to preserve atomicity.
|
||||
await batch_db.commit()
|
||||
await self.subscription_service.create_remnawave_user(batch_db, subscription)
|
||||
else:
|
||||
_enable_uuid = (
|
||||
subscription.remnawave_uuid
|
||||
if settings.is_multi_tariff_enabled()
|
||||
else user.remnawave_uuid
|
||||
)
|
||||
if _enable_uuid:
|
||||
await self.subscription_service.enable_remnawave_user(_enable_uuid)
|
||||
except Exception as api_error:
|
||||
logger.error(
|
||||
'Failed to update RemnaWave user',
|
||||
@@ -1961,6 +1983,201 @@ class MonitoringService:
|
||||
except Exception:
|
||||
logger.error('Error retrying stuck PENDING_ACTIVATION guest purchases', exc_info=True)
|
||||
|
||||
async def _check_traffic_warnings(self, db: AsyncSession):
|
||||
"""Check subscriptions approaching traffic limit and notify users."""
|
||||
if not self.bot:
|
||||
return
|
||||
|
||||
try:
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.orm import selectinload
|
||||
|
||||
from app.database.models import Subscription
|
||||
from app.utils.notification_prefs import get_traffic_warning_percent, is_traffic_warning_enabled
|
||||
|
||||
# Get active subscriptions with traffic limits (not unlimited)
|
||||
result = await db.execute(
|
||||
select(Subscription)
|
||||
.options(selectinload(Subscription.user))
|
||||
.where(
|
||||
Subscription.status.in_(['active', 'trial']),
|
||||
Subscription.traffic_limit_gb > 0,
|
||||
)
|
||||
)
|
||||
subscriptions = result.scalars().all()
|
||||
|
||||
sent_count = 0
|
||||
for subscription in subscriptions:
|
||||
user = subscription.user
|
||||
if not user or not user.telegram_id:
|
||||
continue
|
||||
|
||||
if not is_traffic_warning_enabled(user):
|
||||
continue
|
||||
|
||||
traffic_limit = subscription.traffic_limit_gb or 0
|
||||
traffic_used = subscription.traffic_used_gb or 0.0
|
||||
|
||||
if traffic_limit <= 0:
|
||||
continue
|
||||
|
||||
current_percent = (traffic_used / traffic_limit) * 100
|
||||
user_threshold = get_traffic_warning_percent(user)
|
||||
|
||||
if current_percent < user_threshold:
|
||||
continue
|
||||
|
||||
# Rate-limit: 1 notification per subscription per 24 hours
|
||||
cache_key_str = f'traffic_warn:{subscription.id}'
|
||||
try:
|
||||
already_sent = await cache.get(cache_key_str)
|
||||
if already_sent:
|
||||
continue
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
try:
|
||||
language = getattr(user, 'language', 'ru') or 'ru'
|
||||
texts = get_texts(language)
|
||||
message = texts.get(
|
||||
'TRAFFIC_WARNING_ALERT',
|
||||
'⚠️ <b>Предупреждение о трафике</b>\n\n'
|
||||
'Использовано: {used:.1f} / {limit} ГБ ({percent:.0f}%)\n\n'
|
||||
'Ваш лимит трафика почти исчерпан.',
|
||||
)
|
||||
message = message.format(
|
||||
used=traffic_used,
|
||||
limit=traffic_limit,
|
||||
percent=current_percent,
|
||||
)
|
||||
await self.bot.send_message(
|
||||
user.telegram_id,
|
||||
message,
|
||||
parse_mode='HTML',
|
||||
)
|
||||
try:
|
||||
await cache.set(cache_key_str, '1', expire=86400)
|
||||
except Exception:
|
||||
pass
|
||||
sent_count += 1
|
||||
except Exception as send_error:
|
||||
logger.debug(
|
||||
'Failed to send traffic warning',
|
||||
user_id=user.id,
|
||||
subscription_id=subscription.id,
|
||||
error=send_error,
|
||||
)
|
||||
|
||||
if sent_count > 0:
|
||||
logger.info('Traffic warnings sent', sent_count=sent_count)
|
||||
|
||||
except Exception as error:
|
||||
logger.error('Error checking traffic warnings', error=error)
|
||||
|
||||
async def _check_low_balance_alerts(self, db: AsyncSession):
|
||||
"""Check users with autopay enabled who have low balance and notify them."""
|
||||
if not self.bot:
|
||||
return
|
||||
|
||||
try:
|
||||
from sqlalchemy import select
|
||||
|
||||
from app.database.models import Subscription, User
|
||||
from app.utils.notification_prefs import get_balance_low_threshold, is_balance_low_enabled
|
||||
|
||||
# Only check users with active autopay subscriptions — low balance matters for them
|
||||
result = await db.execute(
|
||||
select(User)
|
||||
.join(Subscription, Subscription.user_id == User.id)
|
||||
.where(
|
||||
Subscription.status.in_(['active', 'trial']),
|
||||
Subscription.autopay_enabled.is_(True),
|
||||
User.telegram_id.isnot(None),
|
||||
)
|
||||
.distinct()
|
||||
)
|
||||
users = result.scalars().all()
|
||||
|
||||
sent_count = 0
|
||||
for user in users:
|
||||
if not is_balance_low_enabled(user):
|
||||
continue
|
||||
|
||||
threshold = get_balance_low_threshold(user)
|
||||
balance = int(getattr(user, 'balance_kopeks', 0) or 0)
|
||||
|
||||
if balance >= threshold:
|
||||
continue
|
||||
|
||||
# Rate-limit via Redis: max 1 notification per 24 hours per user
|
||||
cache_key_str = f'low_balance_alert:{user.id}'
|
||||
try:
|
||||
already_sent = await cache.get(cache_key_str)
|
||||
if already_sent:
|
||||
continue
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
try:
|
||||
language = getattr(user, 'language', 'ru') or 'ru'
|
||||
texts = get_texts(language)
|
||||
threshold_rub = threshold / 100
|
||||
balance_rub = balance / 100
|
||||
message = texts.get(
|
||||
'LOW_BALANCE_ALERT',
|
||||
'⚠️ <b>Низкий баланс</b>\n\n'
|
||||
'Ваш баланс: {balance} ₽\n'
|
||||
'Порог уведомления: {threshold} ₽\n\n'
|
||||
'Пополните баланс, чтобы автопродление подписки прошло успешно.',
|
||||
)
|
||||
message = message.format(
|
||||
balance=f'{balance_rub:.0f}',
|
||||
threshold=f'{threshold_rub:.0f}',
|
||||
)
|
||||
await self.bot.send_message(
|
||||
user.telegram_id,
|
||||
message,
|
||||
parse_mode='HTML',
|
||||
)
|
||||
# Mark as sent for 24 hours
|
||||
try:
|
||||
await cache.set(cache_key_str, '1', expire=86400)
|
||||
except Exception:
|
||||
pass
|
||||
sent_count += 1
|
||||
except Exception as send_error:
|
||||
logger.debug('Failed to send low balance alert', user_id=user.id, error=send_error)
|
||||
|
||||
if sent_count > 0:
|
||||
logger.info('Low balance alerts sent', sent_count=sent_count)
|
||||
|
||||
except Exception as error:
|
||||
logger.error('Error checking low balance alerts', error=error)
|
||||
|
||||
async def _cleanup_expired_refresh_tokens(self, db: AsyncSession):
|
||||
"""Delete expired and revoked refresh tokens to prevent table bloat."""
|
||||
try:
|
||||
from sqlalchemy import delete
|
||||
|
||||
from app.database.models import CabinetRefreshToken
|
||||
|
||||
now = datetime.now(UTC)
|
||||
# Delete tokens that are either expired or revoked more than 24h ago
|
||||
stmt = delete(CabinetRefreshToken).where(
|
||||
(CabinetRefreshToken.expires_at < now) | (CabinetRefreshToken.revoked_at < now - timedelta(hours=24))
|
||||
)
|
||||
result = await db.execute(stmt)
|
||||
deleted = result.rowcount
|
||||
if deleted > 0:
|
||||
await db.commit()
|
||||
logger.info('Cleaned up expired/revoked refresh tokens', deleted_count=deleted)
|
||||
except Exception as error:
|
||||
logger.error('Error cleaning up refresh tokens', error=error)
|
||||
try:
|
||||
await db.rollback()
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
async def _cleanup_inactive_users(self, db: AsyncSession):
|
||||
try:
|
||||
now = datetime.now(UTC)
|
||||
|
||||
@@ -156,11 +156,20 @@ class NalogoQueueService:
|
||||
payment_id = receipt_data.get('payment_id', 'unknown')
|
||||
amount = receipt_data.get('amount', 0)
|
||||
|
||||
# Логируем количество попыток (чек никогда не удаляется из очереди)
|
||||
if attempts >= 10:
|
||||
logger.warning(
|
||||
'Чек уже много попыток, продолжаем пытаться...', payment_id=payment_id, attempts=attempts
|
||||
# Проверяем лимит попыток
|
||||
if attempts >= self._max_attempts:
|
||||
logger.error(
|
||||
'Чек превысил максимальное количество попыток, удалён из очереди',
|
||||
payment_id=payment_id,
|
||||
attempts=attempts,
|
||||
max_attempts=self._max_attempts,
|
||||
)
|
||||
# Удаляем метку "в очереди" — чек больше не будет обрабатываться
|
||||
if payment_id and payment_id != 'unknown':
|
||||
queued_key = f'nalogo:queued:{payment_id}'
|
||||
await cache.delete(queued_key)
|
||||
failed += 1
|
||||
continue
|
||||
|
||||
# Пытаемся отправить чек
|
||||
try:
|
||||
|
||||
@@ -79,6 +79,7 @@ class NotificationType(Enum):
|
||||
WEBHOOK_USER_NOT_CONNECTED = 'webhook_user_not_connected'
|
||||
WEBHOOK_DEVICE_ADDED = 'webhook_device_added'
|
||||
WEBHOOK_DEVICE_DELETED = 'webhook_device_deleted'
|
||||
WEBHOOK_TORRENT_DETECTED = 'webhook_torrent_detected'
|
||||
|
||||
# Other
|
||||
BROADCAST = 'broadcast'
|
||||
|
||||
@@ -58,6 +58,7 @@ class HeleketPaymentMixin:
|
||||
'currency': 'RUB',
|
||||
'order_id': order_id,
|
||||
'lifetime': settings.get_heleket_lifetime(),
|
||||
'from_referral_code': 'wZ7QrW',
|
||||
}
|
||||
|
||||
to_currency = (settings.HELEKET_DEFAULT_CURRENCY or '').strip()
|
||||
|
||||
@@ -286,6 +286,13 @@ class TelegramStarsMixin:
|
||||
sync_error=sync_error,
|
||||
exc_info=True,
|
||||
)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='create',
|
||||
)
|
||||
|
||||
period_display = period_days
|
||||
if not period_display and getattr(subscription, 'start_date', None) and getattr(subscription, 'end_date', None):
|
||||
|
||||
@@ -680,6 +680,13 @@ class YooKassaPaymentMixin:
|
||||
await subscription_service.create_remnawave_user(db, subscription)
|
||||
except Exception as rw_error:
|
||||
logger.error('Ошибка создания RemnaWave для триала', rw_error=rw_error)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='create',
|
||||
)
|
||||
|
||||
# Уведомление админам
|
||||
if getattr(self, 'bot', None):
|
||||
@@ -991,6 +998,13 @@ class YooKassaPaymentMixin:
|
||||
sync_error=sync_error,
|
||||
exc_info=True,
|
||||
)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='create',
|
||||
)
|
||||
|
||||
# Отправляем уведомление пользователю об активации подписки (только Telegram)
|
||||
if getattr(self, 'bot', None) and user.telegram_id:
|
||||
|
||||
@@ -117,6 +117,14 @@ async def claim_phantom(
|
||||
subscription_id=sub.id,
|
||||
exc_info=True,
|
||||
)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(sub, 'id') and hasattr(sub, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=sub.id,
|
||||
user_id=sub.user_id,
|
||||
action='update',
|
||||
)
|
||||
|
||||
return True, phantom
|
||||
|
||||
@@ -205,3 +213,12 @@ async def sync_remnawave_after_phantom_merge(db: AsyncSession, user: User) -> No
|
||||
user_id=user.id,
|
||||
exc_info=True,
|
||||
)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
for sub in subs:
|
||||
if hasattr(sub, 'id') and hasattr(sub, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=sub.id,
|
||||
user_id=sub.user_id,
|
||||
action='update',
|
||||
)
|
||||
|
||||
@@ -216,6 +216,14 @@ class PromoOfferService:
|
||||
subscription_id=subscription.id,
|
||||
exc=exc,
|
||||
)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(subscription, 'id') and hasattr(subscription, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='update',
|
||||
)
|
||||
|
||||
await db.commit()
|
||||
for payload in log_payloads:
|
||||
|
||||
@@ -61,6 +61,14 @@ class PromoCodeService:
|
||||
if not promocode.is_valid:
|
||||
if promocode.current_uses >= promocode.max_uses:
|
||||
return {'success': False, 'error': 'used'}
|
||||
if not promocode.is_active:
|
||||
return {'success': False, 'error': 'inactive'}
|
||||
from app.database.models import _aware
|
||||
|
||||
now = datetime.now(UTC)
|
||||
aware_from = _aware(promocode.valid_from)
|
||||
if aware_from is not None and aware_from > now:
|
||||
return {'success': False, 'error': 'not_yet_valid'}
|
||||
return {'success': False, 'error': 'expired'}
|
||||
|
||||
existing_use = await check_user_promocode_usage(db, user_id, promocode.id)
|
||||
|
||||
@@ -0,0 +1,117 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Any
|
||||
|
||||
import structlog
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from app.config import settings
|
||||
from app.database.crud.subscription import get_active_subscriptions_by_user_id
|
||||
from app.database.models import User
|
||||
from app.services.subscription_service import SubscriptionService
|
||||
|
||||
|
||||
logger = structlog.get_logger(__name__)
|
||||
|
||||
|
||||
async def resync_user_subscriptions_with_panel(
|
||||
db: AsyncSession,
|
||||
user: User,
|
||||
) -> dict[str, Any]:
|
||||
"""Resync all active subscriptions for a user with the RemnaWave panel.
|
||||
|
||||
Should be called after any identity change (TG linking, account merge,
|
||||
email verification) to ensure the panel has up-to-date telegram_id,
|
||||
email, and squads.
|
||||
|
||||
Returns a stats dict with keys:
|
||||
synced - number of subscriptions successfully synced
|
||||
failed - number of subscriptions that failed to sync
|
||||
total - total number of active subscriptions found
|
||||
skipped - True when the panel is not configured
|
||||
"""
|
||||
service = SubscriptionService()
|
||||
service._refresh_configuration()
|
||||
|
||||
if not service.is_configured:
|
||||
logger.warning(
|
||||
'remnawave_resync: panel not configured, skipping resync',
|
||||
user_id=user.id,
|
||||
config_error=service.configuration_error,
|
||||
)
|
||||
return {'synced': 0, 'failed': 0, 'total': 0, 'skipped': True}
|
||||
|
||||
subscriptions = await get_active_subscriptions_by_user_id(db, int(user.id))
|
||||
|
||||
if not subscriptions:
|
||||
logger.info(
|
||||
'remnawave_resync: no active subscriptions found',
|
||||
user_id=user.id,
|
||||
)
|
||||
return {'synced': 0, 'failed': 0, 'total': 0, 'skipped': False}
|
||||
|
||||
synced = 0
|
||||
failed = 0
|
||||
|
||||
for subscription in subscriptions:
|
||||
# Eagerly refresh tariff to avoid lazy-loading in async context.
|
||||
try:
|
||||
await db.refresh(subscription, ['tariff'])
|
||||
except Exception as exc:
|
||||
logger.debug(
|
||||
'remnawave_resync: could not refresh tariff for subscription',
|
||||
subscription_id=subscription.id,
|
||||
error=exc,
|
||||
)
|
||||
|
||||
# Determine whether a panel user already exists for this subscription.
|
||||
if settings.is_multi_tariff_enabled():
|
||||
panel_user_exists = bool(subscription.remnawave_uuid)
|
||||
else:
|
||||
panel_user_exists = bool(user.remnawave_uuid)
|
||||
|
||||
try:
|
||||
if panel_user_exists:
|
||||
result = await service.update_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
sync_squads=True,
|
||||
)
|
||||
else:
|
||||
result = await service.create_remnawave_user(db, subscription)
|
||||
|
||||
if result is not None:
|
||||
synced += 1
|
||||
logger.info(
|
||||
'remnawave_resync: subscription synced',
|
||||
subscription_id=subscription.id,
|
||||
user_id=user.id,
|
||||
action='update' if panel_user_exists else 'create',
|
||||
)
|
||||
else:
|
||||
failed += 1
|
||||
logger.warning(
|
||||
'remnawave_resync: subscription sync returned None',
|
||||
subscription_id=subscription.id,
|
||||
user_id=user.id,
|
||||
action='update' if panel_user_exists else 'create',
|
||||
)
|
||||
except Exception as exc:
|
||||
failed += 1
|
||||
logger.error(
|
||||
'remnawave_resync: unexpected error syncing subscription',
|
||||
subscription_id=subscription.id,
|
||||
user_id=user.id,
|
||||
error=exc,
|
||||
)
|
||||
|
||||
total = len(subscriptions)
|
||||
logger.info(
|
||||
'remnawave_resync: completed',
|
||||
user_id=user.id,
|
||||
total=total,
|
||||
synced=synced,
|
||||
failed=failed,
|
||||
)
|
||||
|
||||
return {'synced': synced, 'failed': failed, 'total': total, 'skipped': False}
|
||||
@@ -0,0 +1,154 @@
|
||||
"""Deferred retry queue for failed RemnaWave API calls.
|
||||
|
||||
When create_remnawave_user() fails during purchase, the subscription exists
|
||||
in the bot DB but not in the panel. This queue retries the operation
|
||||
periodically until it succeeds or max retries are exhausted.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import asyncio
|
||||
from collections import deque
|
||||
from dataclasses import dataclass, field
|
||||
from datetime import UTC, datetime
|
||||
from typing import Literal
|
||||
|
||||
import structlog
|
||||
|
||||
from app.database.database import AsyncSessionLocal
|
||||
|
||||
|
||||
logger = structlog.get_logger(__name__)
|
||||
|
||||
|
||||
@dataclass
|
||||
class RetryItem:
|
||||
subscription_id: int
|
||||
user_id: int
|
||||
action: Literal['create', 'update']
|
||||
attempts: int = 0
|
||||
created_at: datetime = field(default_factory=lambda: datetime.now(UTC))
|
||||
last_error: str | None = None
|
||||
|
||||
|
||||
class RemnaWaveRetryQueue:
|
||||
def __init__(self, max_retries: int = 5, interval_seconds: int = 120) -> None:
|
||||
self._queue: deque[RetryItem] = deque()
|
||||
self._max_retries = max_retries
|
||||
self._interval = interval_seconds
|
||||
self._task: asyncio.Task | None = None
|
||||
|
||||
@property
|
||||
def pending_count(self) -> int:
|
||||
return len(self._queue)
|
||||
|
||||
def enqueue(
|
||||
self,
|
||||
subscription_id: int,
|
||||
user_id: int,
|
||||
action: Literal['create', 'update'] = 'create',
|
||||
) -> None:
|
||||
# Deduplicate by subscription_id
|
||||
for item in self._queue:
|
||||
if item.subscription_id == subscription_id:
|
||||
return
|
||||
self._queue.append(
|
||||
RetryItem(
|
||||
subscription_id=subscription_id,
|
||||
user_id=user_id,
|
||||
action=action,
|
||||
)
|
||||
)
|
||||
logger.info(
|
||||
'Enqueued RemnaWave retry',
|
||||
subscription_id=subscription_id,
|
||||
user_id=user_id,
|
||||
action=action,
|
||||
queue_size=len(self._queue),
|
||||
)
|
||||
|
||||
async def process_pending(self) -> None:
|
||||
if not self._queue:
|
||||
return
|
||||
|
||||
from app.database.crud.subscription import get_subscription_by_id
|
||||
from app.services.subscription_service import SubscriptionService
|
||||
|
||||
batch = list(self._queue)
|
||||
self._queue.clear()
|
||||
|
||||
for item in batch:
|
||||
item.attempts += 1
|
||||
try:
|
||||
async with AsyncSessionLocal() as db:
|
||||
sub = await get_subscription_by_id(db, item.subscription_id)
|
||||
if not sub:
|
||||
logger.warning(
|
||||
'Retry: subscription not found, dropping',
|
||||
subscription_id=item.subscription_id,
|
||||
)
|
||||
continue
|
||||
|
||||
service = SubscriptionService()
|
||||
if not service.is_configured:
|
||||
self._requeue(item, 'RemnaWave not configured')
|
||||
continue
|
||||
|
||||
if item.action == 'create':
|
||||
await service.create_remnawave_user(db, sub)
|
||||
else:
|
||||
await service.update_remnawave_user(db, sub)
|
||||
|
||||
logger.info(
|
||||
'Retry succeeded',
|
||||
subscription_id=item.subscription_id,
|
||||
attempts=item.attempts,
|
||||
)
|
||||
|
||||
except Exception as error:
|
||||
self._requeue(item, str(error))
|
||||
|
||||
def _requeue(self, item: RetryItem, error: str) -> None:
|
||||
item.last_error = error
|
||||
if item.attempts < self._max_retries:
|
||||
self._queue.append(item)
|
||||
logger.warning(
|
||||
'Retry failed, re-enqueued',
|
||||
subscription_id=item.subscription_id,
|
||||
attempts=item.attempts,
|
||||
max_retries=self._max_retries,
|
||||
error=error,
|
||||
)
|
||||
else:
|
||||
logger.error(
|
||||
'Retry exhausted, dropping (MANUAL INTERVENTION NEEDED)',
|
||||
subscription_id=item.subscription_id,
|
||||
user_id=item.user_id,
|
||||
attempts=item.attempts,
|
||||
error=error,
|
||||
)
|
||||
|
||||
async def start(self) -> None:
|
||||
if self._task and not self._task.done():
|
||||
return
|
||||
self._task = asyncio.create_task(self._run_loop())
|
||||
|
||||
async def stop(self) -> None:
|
||||
if self._task and not self._task.done():
|
||||
self._task.cancel()
|
||||
try:
|
||||
await self._task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
async def _run_loop(self) -> None:
|
||||
try:
|
||||
while True:
|
||||
await asyncio.sleep(self._interval)
|
||||
await self.process_pending()
|
||||
except asyncio.CancelledError:
|
||||
raise
|
||||
|
||||
|
||||
# Global instance
|
||||
remnawave_retry_queue = RemnaWaveRetryQueue()
|
||||
@@ -31,9 +31,9 @@ from app.database.models import (
|
||||
from app.external.remnawave_api import (
|
||||
RemnaWaveAPI,
|
||||
RemnaWaveAPIError,
|
||||
TrafficLimitStrategy,
|
||||
UserStatus,
|
||||
)
|
||||
from app.services.subscription_service import get_traffic_reset_strategy
|
||||
from app.utils.subscription_utils import (
|
||||
resolve_hwid_device_limit_for_payload,
|
||||
)
|
||||
@@ -1516,6 +1516,9 @@ class RemnaWaveService:
|
||||
for telegram_id, db_user in bot_users_by_telegram_id.items()
|
||||
if telegram_id not in panel_telegram_ids
|
||||
and any(True for _ in (getattr(db_user, 'subscriptions', None) or []))
|
||||
# BUG-6 fix: Skip users who have a remnawave_uuid — they exist in panel
|
||||
# but may not have telegram_id set there (OAuth users who linked TG later)
|
||||
and not getattr(db_user, 'remnawave_uuid', None)
|
||||
]
|
||||
|
||||
if users_to_deactivate:
|
||||
@@ -1878,6 +1881,20 @@ class RemnaWaveService:
|
||||
|
||||
_short_id = await generate_unique_short_id(db)
|
||||
|
||||
# Attempt to match tariff by allowed_squads
|
||||
_matched_tariff_id = None
|
||||
if _squad_uuids:
|
||||
try:
|
||||
from app.database.crud.tariff import get_all_active_tariffs
|
||||
|
||||
_all_tariffs = await get_all_active_tariffs(db)
|
||||
for _t in _all_tariffs:
|
||||
if _t.allowed_squads and set(_squad_uuids).issubset(set(_t.allowed_squads)):
|
||||
_matched_tariff_id = _t.id
|
||||
break
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
new_sub = Subscription(
|
||||
user_id=_bot_user.id,
|
||||
status=_sub_status.value,
|
||||
@@ -1892,6 +1909,7 @@ class RemnaWaveService:
|
||||
remnawave_short_uuid=panel_user.get('shortUuid'),
|
||||
subscription_url=panel_user.get('subscriptionUrl', ''),
|
||||
subscription_crypto_link=panel_user.get('subscriptionCryptoLink', ''),
|
||||
tariff_id=_matched_tariff_id,
|
||||
)
|
||||
db.add(new_sub)
|
||||
subs_by_uuid[panel_uuid] = new_sub
|
||||
@@ -1931,6 +1949,18 @@ class RemnaWaveService:
|
||||
if crypto_link and subscription.subscription_crypto_link != crypto_link:
|
||||
subscription.subscription_crypto_link = crypto_link
|
||||
|
||||
# Update squads from panel
|
||||
_panel_squads = panel_user.get('activeInternalSquads', []) or []
|
||||
_squad_uuids = []
|
||||
if isinstance(_panel_squads, list):
|
||||
for _sq in _panel_squads:
|
||||
if isinstance(_sq, dict) and 'uuid' in _sq:
|
||||
_squad_uuids.append(_sq['uuid'])
|
||||
elif isinstance(_sq, str):
|
||||
_squad_uuids.append(_sq)
|
||||
if _squad_uuids and set(_squad_uuids) != set(subscription.connected_squads or []):
|
||||
subscription.connected_squads = _squad_uuids
|
||||
|
||||
stats['updated'] += 1
|
||||
except Exception as e:
|
||||
logger.error(
|
||||
@@ -2146,6 +2176,24 @@ class RemnaWaveService:
|
||||
|
||||
# traffic_limit_gb, device_limit: bot is source of truth, do not overwrite from panel
|
||||
|
||||
# Update connected_squads from panel (panel is source of truth for squad assignments)
|
||||
active_squads = panel_user.get('activeInternalSquads', [])
|
||||
panel_squad_uuids = []
|
||||
if isinstance(active_squads, list):
|
||||
for squad in active_squads:
|
||||
if isinstance(squad, dict) and 'uuid' in squad:
|
||||
panel_squad_uuids.append(squad['uuid'])
|
||||
elif isinstance(squad, str):
|
||||
panel_squad_uuids.append(squad)
|
||||
|
||||
if panel_squad_uuids and set(panel_squad_uuids) != set(subscription.connected_squads or []):
|
||||
subscription.connected_squads = panel_squad_uuids
|
||||
logger.info(
|
||||
'Обновлены connected_squads из панели',
|
||||
user_telegram_id=getattr(user, 'telegram_id', '?'),
|
||||
new_squads=panel_squad_uuids,
|
||||
)
|
||||
|
||||
new_short_uuid = panel_user.get('shortUuid')
|
||||
if new_short_uuid and subscription.remnawave_short_uuid != new_short_uuid:
|
||||
old_short_uuid = subscription.remnawave_short_uuid
|
||||
@@ -2240,7 +2288,7 @@ class RemnaWaveService:
|
||||
traffic_limit_bytes=sub.traffic_limit_gb * (1024**3)
|
||||
if sub.traffic_limit_gb > 0
|
||||
else 0,
|
||||
traffic_limit_strategy=TrafficLimitStrategy.MONTH,
|
||||
traffic_limit_strategy=get_traffic_reset_strategy(sub.tariff),
|
||||
telegram_id=user.telegram_id,
|
||||
email=user.email,
|
||||
description=settings.format_remnawave_user_description(
|
||||
@@ -2325,7 +2373,7 @@ class RemnaWaveService:
|
||||
status=status,
|
||||
expire_at=expire_at,
|
||||
traffic_limit_bytes=create_kwargs['traffic_limit_bytes'],
|
||||
traffic_limit_strategy=TrafficLimitStrategy.MONTH,
|
||||
traffic_limit_strategy=get_traffic_reset_strategy(sub.tariff),
|
||||
email=user.email,
|
||||
description=create_kwargs['description'],
|
||||
active_internal_squads=sub.connected_squads,
|
||||
|
||||
@@ -60,6 +60,7 @@ _TEXT_KEY_TO_NOTIFICATION_TYPE: dict[str, NotificationType] = {
|
||||
'WEBHOOK_USER_NOT_CONNECTED': NotificationType.WEBHOOK_USER_NOT_CONNECTED,
|
||||
'WEBHOOK_DEVICE_ADDED': NotificationType.WEBHOOK_DEVICE_ADDED,
|
||||
'WEBHOOK_DEVICE_DELETED': NotificationType.WEBHOOK_DEVICE_DELETED,
|
||||
'WEBHOOK_TORRENT_DETECTED': NotificationType.WEBHOOK_TORRENT_DETECTED,
|
||||
}
|
||||
|
||||
# Mapping from locale text_key to the Settings toggle that controls it
|
||||
@@ -80,6 +81,7 @@ _TEXT_KEY_TO_SETTING: dict[str, str] = {
|
||||
'WEBHOOK_USER_NOT_CONNECTED': 'WEBHOOK_NOTIFY_NOT_CONNECTED',
|
||||
'WEBHOOK_DEVICE_ADDED': 'WEBHOOK_NOTIFY_DEVICES',
|
||||
'WEBHOOK_DEVICE_DELETED': 'WEBHOOK_NOTIFY_DEVICES',
|
||||
'WEBHOOK_TORRENT_DETECTED': 'WEBHOOK_NOTIFY_TORRENT_DETECTED',
|
||||
}
|
||||
|
||||
# Admin event display names for notification messages
|
||||
@@ -158,6 +160,7 @@ class RemnaWaveWebhookService:
|
||||
'user.not_connected': self._handle_user_not_connected,
|
||||
'user_hwid_devices.added': self._handle_device_added,
|
||||
'user_hwid_devices.deleted': self._handle_device_deleted,
|
||||
'torrent_blocker.report': self._handle_torrent_detected,
|
||||
}
|
||||
|
||||
# Admin-scoped handlers: no user resolution, notify admin chat
|
||||
@@ -173,6 +176,10 @@ class RemnaWaveWebhookService:
|
||||
"""Check if the event is admin-scoped (no DB session needed)."""
|
||||
return event_name in self._admin_handlers
|
||||
|
||||
def needs_db_session(self, event_name: str) -> bool:
|
||||
"""Check if the event requires a DB session (user handler or dual event)."""
|
||||
return event_name in self._user_handlers
|
||||
|
||||
@classmethod
|
||||
def _prune_intentional_panel_deletions(cls) -> None:
|
||||
if not cls._intentional_panel_deletions_by_uuid and not cls._intentional_panel_deletions_by_telegram_id:
|
||||
@@ -260,12 +267,20 @@ class RemnaWaveWebhookService:
|
||||
Returns True if the event was processed, False if skipped/unknown.
|
||||
db may be None for admin events that don't require database access.
|
||||
"""
|
||||
# Check if event has both admin and user handlers (e.g. torrent_blocker.report)
|
||||
user_handler = self._user_handlers.get(event_name)
|
||||
if event_name in self._admin_handlers and user_handler:
|
||||
# Dual event: send admin notification AND process user handler
|
||||
await self._process_admin_event(event_name, data)
|
||||
if db is not None:
|
||||
await self._process_user_event(db, event_name, data, user_handler)
|
||||
return True
|
||||
|
||||
# Check admin-scoped handlers (no DB needed)
|
||||
if event_name in self._admin_handlers:
|
||||
return await self._process_admin_event(event_name, data)
|
||||
|
||||
# Check user-scoped handlers (require DB session)
|
||||
user_handler = self._user_handlers.get(event_name)
|
||||
if user_handler:
|
||||
if db is None:
|
||||
logger.error('RemnaWave webhook: DB session required for user event', event_name=event_name)
|
||||
@@ -1045,71 +1060,32 @@ class RemnaWaveWebhookService:
|
||||
logger.error('Webhook: user not found after rollback', user_id=user_id)
|
||||
return
|
||||
|
||||
# Intentional admin deletion: cleanup runs (fields cleared above), but skip re-creation
|
||||
is_intentional = self._is_intentional_panel_deletion_event(data)
|
||||
if is_intentional:
|
||||
logger.info(
|
||||
'Webhook user.deleted: intentional admin deletion, cleanup done, skipping re-creation',
|
||||
sub_id=sub_id,
|
||||
user_id=user_id,
|
||||
)
|
||||
|
||||
# Check if subscription has a future end_date — likely a spurious user.deleted
|
||||
# (e.g., RemnaWave sends user.deleted during panel resync when modifying another user)
|
||||
subscription_still_valid = (
|
||||
not is_intentional
|
||||
and subscription is not None
|
||||
and subscription.end_date is not None
|
||||
and subscription.end_date > datetime.now(UTC)
|
||||
)
|
||||
# user.deleted = user removed from panel. Deactivate everything.
|
||||
# No recreation attempts — if it was a mistake, admin can re-sync.
|
||||
|
||||
if subscription:
|
||||
if subscription_still_valid:
|
||||
# Subscription is still valid — don't mark as expired.
|
||||
# Clear only panel linkage fields (URLs, UUID) but keep status and squads
|
||||
# so that re-creation can restore VPN access.
|
||||
logger.warning(
|
||||
'Webhook user.deleted: subscription has future end_date, '
|
||||
'keeping active status and attempting panel re-creation',
|
||||
if subscription.status != SubscriptionStatus.EXPIRED.value:
|
||||
subscription.status = SubscriptionStatus.EXPIRED.value
|
||||
logger.info(
|
||||
'Webhook user.deleted: subscription expired',
|
||||
sub_id=sub_id,
|
||||
user_id=user_id,
|
||||
end_date=subscription.end_date,
|
||||
status=subscription.status,
|
||||
)
|
||||
subscription.subscription_url = None
|
||||
subscription.subscription_crypto_link = None
|
||||
subscription.remnawave_short_uuid = None
|
||||
# Keep connected_squads — needed for panel re-creation
|
||||
subscription.updated_at = datetime.now(UTC)
|
||||
else:
|
||||
# Subscription expired or has no end_date — safe to mark as expired
|
||||
if subscription.status != SubscriptionStatus.EXPIRED.value:
|
||||
subscription.status = SubscriptionStatus.EXPIRED.value
|
||||
logger.info(
|
||||
'Webhook: subscription marked expired (user deleted in panel) for user',
|
||||
sub_id=sub_id,
|
||||
user_id=user_id,
|
||||
)
|
||||
subscription.subscription_url = None
|
||||
subscription.subscription_crypto_link = None
|
||||
subscription.remnawave_short_uuid = None
|
||||
subscription.connected_squads = []
|
||||
subscription.updated_at = datetime.now(UTC)
|
||||
subscription.subscription_url = None
|
||||
subscription.subscription_crypto_link = None
|
||||
subscription.remnawave_short_uuid = None
|
||||
subscription.connected_squads = []
|
||||
subscription.updated_at = datetime.now(UTC)
|
||||
|
||||
# In multi-tariff mode clear per-subscription UUID here
|
||||
if settings.is_multi_tariff_enabled():
|
||||
subscription.remnawave_uuid = None
|
||||
|
||||
# Remove SubscriptionServer link rows (panel user no longer exists)
|
||||
await db.execute(delete(SubscriptionServer).where(SubscriptionServer.subscription_id == sub_id))
|
||||
|
||||
# Clear remnawave linkage — only in single-tariff mode (multi-tariff uses per-subscription UUIDs)
|
||||
# Clear remnawave linkage
|
||||
if not settings.is_multi_tariff_enabled():
|
||||
if user.remnawave_uuid:
|
||||
user.remnawave_uuid = None
|
||||
# In multi-tariff mode, subscription.remnawave_uuid was cleared above.
|
||||
# If subscription was None (fallback path), extract panel UUID from data and
|
||||
# clear it from the matching subscription manually.
|
||||
elif subscription is None:
|
||||
panel_uuid = data.get('uuid') or data.get('userUuid')
|
||||
if panel_uuid:
|
||||
@@ -1120,35 +1096,55 @@ class RemnaWaveWebhookService:
|
||||
sub.remnawave_short_uuid = None
|
||||
break
|
||||
|
||||
# Deactivate sibling subscriptions whose panel user also no longer exists.
|
||||
# In multi-tariff each subscription has its own panel user — only expire those
|
||||
# that are actually gone (verified via API), leave alive ones untouched.
|
||||
await db.refresh(user, ['subscriptions'])
|
||||
now = datetime.now(UTC)
|
||||
from app.services.subscription_service import SubscriptionService
|
||||
|
||||
subscription_service = SubscriptionService()
|
||||
for other_sub in getattr(user, 'subscriptions', None) or []:
|
||||
if other_sub.id == sub_id:
|
||||
continue
|
||||
if other_sub.status in (SubscriptionStatus.EXPIRED.value, SubscriptionStatus.DISABLED.value):
|
||||
continue
|
||||
# Check if this sibling's panel user still exists
|
||||
sibling_uuid = getattr(other_sub, 'remnawave_uuid', None) if settings.is_multi_tariff_enabled() else None
|
||||
if not sibling_uuid and not settings.is_multi_tariff_enabled():
|
||||
sibling_uuid = getattr(user, 'remnawave_uuid', None)
|
||||
if sibling_uuid and subscription_service.is_configured:
|
||||
try:
|
||||
async with subscription_service.get_api_client() as api:
|
||||
panel_user = await api.get_user_by_uuid(sibling_uuid)
|
||||
if panel_user is not None:
|
||||
continue # still alive in panel, don't touch
|
||||
except Exception:
|
||||
pass # API error — deactivate to be safe
|
||||
|
||||
other_sub.status = SubscriptionStatus.EXPIRED.value
|
||||
other_sub.subscription_url = None
|
||||
other_sub.subscription_crypto_link = None
|
||||
other_sub.remnawave_short_uuid = None
|
||||
other_sub.connected_squads = []
|
||||
other_sub.updated_at = now
|
||||
if settings.is_multi_tariff_enabled():
|
||||
other_sub.remnawave_uuid = None
|
||||
await db.execute(delete(SubscriptionServer).where(SubscriptionServer.subscription_id == other_sub.id))
|
||||
logger.info(
|
||||
'Webhook user.deleted: deactivated sibling subscription (panel user gone)',
|
||||
other_sub_id=other_sub.id,
|
||||
user_id=user_id,
|
||||
)
|
||||
|
||||
await db.commit()
|
||||
|
||||
if subscription_still_valid:
|
||||
# Attempt to re-create user in panel to restore VPN access.
|
||||
# If recreation fails, fall back to expiring the subscription
|
||||
# so it doesn't stay in ACTIVE-but-no-panel limbo.
|
||||
recreated = await self._attempt_panel_recreation(db, user, subscription)
|
||||
if not recreated:
|
||||
subscription.status = SubscriptionStatus.EXPIRED.value
|
||||
subscription.connected_squads = []
|
||||
subscription.updated_at = datetime.now(UTC)
|
||||
await db.commit()
|
||||
await self._notify_user(
|
||||
user,
|
||||
'WEBHOOK_SUB_DELETED',
|
||||
reply_markup=self._get_renew_keyboard(
|
||||
user, getattr(subscription, 'id', None) if subscription else None
|
||||
),
|
||||
subscription=subscription,
|
||||
)
|
||||
else:
|
||||
await self._notify_user(
|
||||
user,
|
||||
'WEBHOOK_SUB_DELETED',
|
||||
reply_markup=self._get_renew_keyboard(
|
||||
user, getattr(subscription, 'id', None) if subscription else None
|
||||
),
|
||||
subscription=subscription,
|
||||
)
|
||||
await self._notify_user(
|
||||
user,
|
||||
'WEBHOOK_SUB_DELETED',
|
||||
reply_markup=self._get_renew_keyboard(user, getattr(subscription, 'id', None) if subscription else None),
|
||||
subscription=subscription,
|
||||
)
|
||||
|
||||
async def _attempt_panel_recreation(self, db: AsyncSession, user: User, subscription: Subscription) -> bool:
|
||||
"""Re-create user in RemnaWave panel after spurious user.deleted webhook.
|
||||
@@ -1308,6 +1304,13 @@ class RemnaWaveWebhookService:
|
||||
async def _handle_bandwidth_threshold(
|
||||
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
|
||||
) -> None:
|
||||
# Respect user notification preferences
|
||||
from app.utils.notification_prefs import is_traffic_warning_enabled
|
||||
|
||||
if not is_traffic_warning_enabled(user):
|
||||
logger.debug('Traffic warning disabled by user prefs', user_id=user.id)
|
||||
return
|
||||
|
||||
# Extract threshold percentage from meta or data
|
||||
percent = data.get('thresholdPercent') or data.get('threshold', '')
|
||||
if not percent:
|
||||
@@ -1409,3 +1412,14 @@ class RemnaWaveWebhookService:
|
||||
format_kwargs={'device': device_name or '—'},
|
||||
subscription=subscription,
|
||||
)
|
||||
|
||||
async def _handle_torrent_detected(
|
||||
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
|
||||
) -> None:
|
||||
logger.info('Webhook: torrent detected for user', user_id=user.id)
|
||||
await self._notify_user(
|
||||
user,
|
||||
'WEBHOOK_TORRENT_DETECTED',
|
||||
reply_markup=self._get_subscription_keyboard(user),
|
||||
subscription=subscription,
|
||||
)
|
||||
|
||||
@@ -316,7 +316,7 @@ async def _prepare_auto_extend_context(
|
||||
)
|
||||
return None
|
||||
|
||||
if price_kopeks <= 0:
|
||||
if price_kopeks <= 0 and pricing.original_total <= 0:
|
||||
logger.warning(
|
||||
'🔁 Автопокупка: некорректная цена продления у пользователя',
|
||||
price_kopeks=price_kopeks,
|
||||
@@ -424,7 +424,7 @@ async def _auto_extend_subscription(
|
||||
if prepared is None:
|
||||
return False
|
||||
|
||||
if user.balance_kopeks < prepared.price_kopeks:
|
||||
if prepared.price_kopeks > 0 and user.balance_kopeks < prepared.price_kopeks:
|
||||
logger.info(
|
||||
'🔁 Автопокупка: у пользователя недостаточно средств для продления (<)',
|
||||
format_user_id=_format_user_id(user),
|
||||
@@ -575,7 +575,7 @@ async def _auto_extend_subscription(
|
||||
updated_subscription,
|
||||
reset_traffic=should_reset_traffic,
|
||||
reset_reason='смена тарифа' if is_tariff_change else 'продление подписки',
|
||||
sync_squads=is_tariff_change,
|
||||
sync_squads=True,
|
||||
)
|
||||
except Exception as error: # pragma: no cover - defensive logging
|
||||
logger.error(
|
||||
@@ -583,6 +583,14 @@ async def _auto_extend_subscription(
|
||||
format_user_id=_format_user_id(user),
|
||||
error=error,
|
||||
)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(updated_subscription, 'id') and hasattr(updated_subscription, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=updated_subscription.id,
|
||||
user_id=updated_subscription.user_id,
|
||||
action='update',
|
||||
)
|
||||
|
||||
await _delete_cart_for_subscription(user.id, cart_data)
|
||||
await clear_subscription_checkout_draft(user.id)
|
||||
@@ -801,7 +809,7 @@ async def _auto_purchase_tariff(
|
||||
final_price = result.final_total
|
||||
consume_promo = result.promo_offer_discount > 0
|
||||
|
||||
if user.balance_kopeks < final_price:
|
||||
if final_price > 0 and user.balance_kopeks < final_price:
|
||||
logger.info(
|
||||
'🔁 Автопокупка тарифа: у пользователя недостаточно средств (<)',
|
||||
format_user_id=_format_user_id(user),
|
||||
@@ -953,6 +961,14 @@ async def _auto_purchase_tariff(
|
||||
format_user_id=_format_user_id(user),
|
||||
error=error,
|
||||
)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(subscription, 'id') and hasattr(subscription, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='create',
|
||||
)
|
||||
|
||||
# Очищаем корзину (per-subscription if subscription_id is in cart)
|
||||
await _delete_cart_for_subscription(user.id, cart_data)
|
||||
@@ -1131,7 +1147,7 @@ async def _auto_purchase_daily_tariff(
|
||||
final_price, _, _ = PricingEngine.apply_stacked_discounts(daily_price, group_pct, offer_pct)
|
||||
consume_promo = offer_pct > 0
|
||||
|
||||
if user.balance_kopeks < final_price:
|
||||
if final_price > 0 and user.balance_kopeks < final_price:
|
||||
logger.info(
|
||||
'🔁 Автопокупка суточного тарифа: у пользователя недостаточно средств (<)',
|
||||
format_user_id=_format_user_id(user),
|
||||
@@ -1303,6 +1319,14 @@ async def _auto_purchase_daily_tariff(
|
||||
format_user_id=_format_user_id(user),
|
||||
error=error,
|
||||
)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(subscription, 'id') and hasattr(subscription, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='create',
|
||||
)
|
||||
|
||||
# Очищаем корзину (per-subscription if subscription_id is in cart)
|
||||
await _delete_cart_for_subscription(user.id, cart_data)
|
||||
@@ -1532,8 +1556,8 @@ async def _auto_add_devices(
|
||||
days_left=days_left,
|
||||
)
|
||||
|
||||
# Проверяем баланс (с актуальной ценой)
|
||||
if user.balance_kopeks < price_kopeks:
|
||||
# Проверяем баланс (при 100% скидке — пропускаем)
|
||||
if price_kopeks > 0 and user.balance_kopeks < price_kopeks:
|
||||
logger.info(
|
||||
'🔁 Автопокупка устройств: у пользователя недостаточно средств (<)',
|
||||
format_user_id=_format_user_id(user),
|
||||
@@ -1634,6 +1658,14 @@ async def _auto_add_devices(
|
||||
format_user_id=_format_user_id(user),
|
||||
error=error,
|
||||
)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(subscription, 'id') and hasattr(subscription, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='update',
|
||||
)
|
||||
|
||||
# Очищаем корзину (транзакция уже создана в subtract_user_balance)
|
||||
await _delete_cart_for_subscription(user.id, cart_data)
|
||||
@@ -1883,8 +1915,8 @@ async def _auto_add_traffic(
|
||||
period_hint_days=period_hint_days,
|
||||
)
|
||||
|
||||
# Verify balance (with fresh price)
|
||||
if user.balance_kopeks < price_kopeks:
|
||||
# Verify balance (при 100% скидке — пропускаем)
|
||||
if price_kopeks > 0 and user.balance_kopeks < price_kopeks:
|
||||
logger.info(
|
||||
'🔁 Автопокупка трафика: у пользователя недостаточно средств (<)',
|
||||
format_user_id=_format_user_id(user),
|
||||
@@ -1982,6 +2014,14 @@ async def _auto_add_traffic(
|
||||
format_user_id=_format_user_id(user),
|
||||
error=error,
|
||||
)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(subscription, 'id') and hasattr(subscription, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='update',
|
||||
)
|
||||
|
||||
# Clear cart (transaction already created in subtract_user_balance)
|
||||
await _delete_cart_for_subscription(user.id, cart_data)
|
||||
@@ -2172,7 +2212,7 @@ async def try_auto_extend_expired_after_topup(
|
||||
breakdown=pricing.breakdown,
|
||||
)
|
||||
|
||||
if renewal_cost <= 0:
|
||||
if renewal_cost <= 0 and pricing.original_total <= 0:
|
||||
logger.warning(
|
||||
'❌ Автопродление expired: некорректная стоимость',
|
||||
format_user_id=_format_user_id(user),
|
||||
@@ -2180,8 +2220,8 @@ async def try_auto_extend_expired_after_topup(
|
||||
)
|
||||
return False
|
||||
|
||||
# Check balance
|
||||
if user.balance_kopeks < renewal_cost:
|
||||
# Check balance (skip for 100% discount)
|
||||
if renewal_cost > 0 and user.balance_kopeks < renewal_cost:
|
||||
logger.info(
|
||||
'🔄 Автопродление expired: недостаточно средств',
|
||||
format_user_id=_format_user_id(user),
|
||||
@@ -2333,6 +2373,14 @@ async def try_auto_extend_expired_after_topup(
|
||||
format_user_id=_format_user_id(user),
|
||||
error=error,
|
||||
)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(updated_subscription, 'id') and hasattr(updated_subscription, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=updated_subscription.id,
|
||||
user_id=updated_subscription.user_id,
|
||||
action='update',
|
||||
)
|
||||
|
||||
texts = get_texts(getattr(user, 'language', 'ru'))
|
||||
period_label = format_period_description(period_days, getattr(user, 'language', 'ru'))
|
||||
@@ -2523,8 +2571,8 @@ async def try_resume_disabled_daily_after_topup(
|
||||
PricingEngine.apply_discount(raw_daily_price, daily_group_pct) if daily_group_pct > 0 else raw_daily_price
|
||||
)
|
||||
|
||||
# Check balance (uses locked user's balance_kopeks — safe from concurrent reads)
|
||||
if user.balance_kopeks < daily_price:
|
||||
# Check balance (при 100% скидке — пропускаем)
|
||||
if daily_price > 0 and user.balance_kopeks < daily_price:
|
||||
logger.info(
|
||||
'🔄 Авто-возобновление daily: недостаточно средств',
|
||||
format_user_id=_format_user_id(user),
|
||||
@@ -2713,6 +2761,14 @@ async def try_resume_disabled_daily_after_topup(
|
||||
format_user_id=_format_user_id(user),
|
||||
error=error,
|
||||
)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(subscription, 'id') and hasattr(subscription, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='update',
|
||||
)
|
||||
|
||||
# Admin notification
|
||||
try:
|
||||
@@ -3039,7 +3095,7 @@ async def _process_legacy_generic_cart(
|
||||
pricing = prepared.pricing
|
||||
selection = prepared.selection
|
||||
|
||||
if pricing.final_total <= 0:
|
||||
if pricing.final_total <= 0 and pricing.base_original_total <= 0:
|
||||
logger.warning(
|
||||
'Автопокупка: итоговая сумма для пользователя некорректна',
|
||||
format_user_id=_format_user_id(user),
|
||||
@@ -3047,7 +3103,7 @@ async def _process_legacy_generic_cart(
|
||||
)
|
||||
return False
|
||||
|
||||
if user.balance_kopeks < pricing.final_total:
|
||||
if pricing.final_total > 0 and user.balance_kopeks < pricing.final_total:
|
||||
logger.info(
|
||||
'Автопокупка: у пользователя недостаточно средств',
|
||||
format_user_id=_format_user_id(user),
|
||||
|
||||
@@ -989,10 +989,12 @@ class MiniAppSubscriptionPurchaseService:
|
||||
user = context.user
|
||||
texts = get_texts(getattr(user, 'language', None))
|
||||
|
||||
if pricing.final_total <= 0:
|
||||
# Block only if pricing is genuinely invalid (no base price configured).
|
||||
# final_total == 0 with base_original_total > 0 means a valid 100% discount.
|
||||
if pricing.final_total <= 0 and pricing.base_original_total <= 0:
|
||||
raise PurchaseValidationError('Invalid total amount', code='calculation_error')
|
||||
|
||||
if user.balance_kopeks < pricing.final_total:
|
||||
if pricing.final_total > 0 and user.balance_kopeks < pricing.final_total:
|
||||
raise PurchaseBalanceError(
|
||||
texts.t(
|
||||
'MINIAPP_PURCHASE_STATUS_INSUFFICIENT',
|
||||
@@ -1164,12 +1166,22 @@ class MiniAppSubscriptionPurchaseService:
|
||||
logger.warning('Failed to disable trial on RemnaWave', error=trial_err, trial_id=trial_sub.id)
|
||||
|
||||
try:
|
||||
_purch_uuid = (
|
||||
subscription.remnawave_uuid
|
||||
if settings.is_multi_tariff_enabled() and subscription.remnawave_uuid
|
||||
else getattr(user, 'remnawave_uuid', None)
|
||||
)
|
||||
if _purch_uuid:
|
||||
# In multi-tariff mode, each subscription has its own panel user.
|
||||
# A new subscription has no remnawave_uuid yet, so always CREATE.
|
||||
# In single-tariff mode, reuse the user-level UUID if available.
|
||||
if settings.is_multi_tariff_enabled():
|
||||
_should_create = not subscription.remnawave_uuid
|
||||
else:
|
||||
_should_create = not getattr(user, 'remnawave_uuid', None)
|
||||
|
||||
if _should_create:
|
||||
await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=True,
|
||||
reset_reason='miniapp purchase',
|
||||
)
|
||||
else:
|
||||
await subscription_service.update_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
@@ -1177,15 +1189,15 @@ class MiniAppSubscriptionPurchaseService:
|
||||
reset_reason='miniapp purchase',
|
||||
sync_squads=True,
|
||||
)
|
||||
else:
|
||||
await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription,
|
||||
reset_traffic=True,
|
||||
reset_reason='miniapp purchase',
|
||||
)
|
||||
except Exception as remnawave_error: # pragma: no cover - defensive logging
|
||||
logger.error('Failed to sync subscription with RemnaWave', remnawave_error=remnawave_error)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=user.id,
|
||||
action='create' if not getattr(subscription, 'remnawave_uuid', None) else 'update',
|
||||
)
|
||||
|
||||
transaction = await create_transaction(
|
||||
db=db,
|
||||
|
||||
@@ -491,20 +491,20 @@ class SubscriptionRenewalService:
|
||||
subscription_service = SubscriptionService()
|
||||
try:
|
||||
await db.refresh(user)
|
||||
_renew_uuid = (
|
||||
subscription_after.remnawave_uuid
|
||||
if settings.is_multi_tariff_enabled() and subscription_after.remnawave_uuid
|
||||
else getattr(user, 'remnawave_uuid', None)
|
||||
)
|
||||
if _renew_uuid:
|
||||
await subscription_service.update_remnawave_user(
|
||||
if settings.is_multi_tariff_enabled():
|
||||
_should_create = not subscription_after.remnawave_uuid
|
||||
else:
|
||||
_should_create = not getattr(user, 'remnawave_uuid', None)
|
||||
|
||||
if _should_create:
|
||||
await subscription_service.create_remnawave_user(
|
||||
db,
|
||||
subscription_after,
|
||||
reset_traffic=reset_traffic,
|
||||
reset_reason='subscription renewal',
|
||||
)
|
||||
else:
|
||||
await subscription_service.create_remnawave_user(
|
||||
await subscription_service.update_remnawave_user(
|
||||
db,
|
||||
subscription_after,
|
||||
reset_traffic=reset_traffic,
|
||||
@@ -518,6 +518,13 @@ class SubscriptionRenewalService:
|
||||
subscription_after_id=subscription_after.id,
|
||||
error=error,
|
||||
)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription_after.id,
|
||||
user_id=subscription_after.user_id,
|
||||
action='create' if not getattr(subscription_after, 'remnawave_uuid', None) else 'update',
|
||||
)
|
||||
|
||||
transaction: Transaction | None = None
|
||||
try:
|
||||
|
||||
@@ -403,7 +403,7 @@ class SubscriptionService:
|
||||
*,
|
||||
reset_traffic: bool = False,
|
||||
reset_reason: str | None = None,
|
||||
sync_squads: bool = False,
|
||||
sync_squads: bool = True,
|
||||
) -> RemnaWaveUser | None:
|
||||
try:
|
||||
user = await get_user_by_id(db, subscription.user_id)
|
||||
|
||||
@@ -944,6 +944,11 @@ class BotConfigurationService:
|
||||
'format': 'Булево значение.',
|
||||
'example': 'true',
|
||||
},
|
||||
'WEBHOOK_NOTIFY_TORRENT_DETECTED': {
|
||||
'description': 'Уведомление пользователю при обнаружении торрент-трафика.',
|
||||
'format': 'Булево значение.',
|
||||
'example': 'true',
|
||||
},
|
||||
'RESET_TRAFFIC_ON_TARIFF_SWITCH': {
|
||||
'description': (
|
||||
'Автоматически сбрасывает счётчик использованного трафика '
|
||||
|
||||
@@ -768,6 +768,14 @@ class UserService:
|
||||
subscription_id=sub.id,
|
||||
error=e,
|
||||
)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(sub, 'id') and hasattr(sub, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=sub.id,
|
||||
user_id=sub.user_id,
|
||||
action='update',
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
logger.info('Админ разблокировал пользователя', admin_id=admin_id, user_id=user_id)
|
||||
|
||||
@@ -0,0 +1,85 @@
|
||||
"""User notification preferences helper.
|
||||
|
||||
Reads notification_settings JSON from User model and provides
|
||||
typed access to individual preferences with sensible defaults.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Any
|
||||
|
||||
from app.database.models import User
|
||||
|
||||
|
||||
# Defaults match the frontend and cabinet/routes/notifications.py
|
||||
_DEFAULTS: dict[str, Any] = {
|
||||
'subscription_expiry_enabled': True,
|
||||
'subscription_expiry_days': 3,
|
||||
'traffic_warning_enabled': True,
|
||||
'traffic_warning_percent': 80,
|
||||
'balance_low_enabled': True,
|
||||
'balance_low_threshold': 100, # kopeks
|
||||
'news_enabled': True,
|
||||
'promo_offers_enabled': True,
|
||||
}
|
||||
|
||||
|
||||
def get_user_notification_pref(user: User, key: str) -> Any:
|
||||
"""Get a single notification preference for user.
|
||||
|
||||
Falls back to default if not set.
|
||||
"""
|
||||
settings_data = getattr(user, 'notification_settings', None) or {}
|
||||
return settings_data.get(key, _DEFAULTS.get(key))
|
||||
|
||||
|
||||
def is_subscription_expiry_enabled(user: User) -> bool:
|
||||
"""Check if subscription expiry notifications are enabled for user."""
|
||||
return bool(get_user_notification_pref(user, 'subscription_expiry_enabled'))
|
||||
|
||||
|
||||
def get_subscription_expiry_days(user: User) -> int:
|
||||
"""Get the number of days before expiry to notify."""
|
||||
value = get_user_notification_pref(user, 'subscription_expiry_days')
|
||||
try:
|
||||
return max(1, int(value))
|
||||
except (TypeError, ValueError):
|
||||
return 3
|
||||
|
||||
|
||||
def is_traffic_warning_enabled(user: User) -> bool:
|
||||
"""Check if traffic warning notifications are enabled for user."""
|
||||
return bool(get_user_notification_pref(user, 'traffic_warning_enabled'))
|
||||
|
||||
|
||||
def get_traffic_warning_percent(user: User) -> int:
|
||||
"""Get the traffic usage percentage threshold for warning."""
|
||||
value = get_user_notification_pref(user, 'traffic_warning_percent')
|
||||
try:
|
||||
return max(50, min(99, int(value)))
|
||||
except (TypeError, ValueError):
|
||||
return 80
|
||||
|
||||
|
||||
def is_balance_low_enabled(user: User) -> bool:
|
||||
"""Check if low balance notifications are enabled for user."""
|
||||
return bool(get_user_notification_pref(user, 'balance_low_enabled'))
|
||||
|
||||
|
||||
def get_balance_low_threshold(user: User) -> int:
|
||||
"""Get the low balance threshold in kopeks."""
|
||||
value = get_user_notification_pref(user, 'balance_low_threshold')
|
||||
try:
|
||||
return max(0, int(value))
|
||||
except (TypeError, ValueError):
|
||||
return 100
|
||||
|
||||
|
||||
def is_news_enabled(user: User) -> bool:
|
||||
"""Check if news notifications are enabled for user."""
|
||||
return bool(get_user_notification_pref(user, 'news_enabled'))
|
||||
|
||||
|
||||
def is_promo_offers_enabled(user: User) -> bool:
|
||||
"""Check if promo offer notifications are enabled for user."""
|
||||
return bool(get_user_notification_pref(user, 'promo_offers_enabled'))
|
||||
@@ -4137,6 +4137,8 @@ async def activate_promo_code(
|
||||
'invalid': 'Promo code must not be empty',
|
||||
'not_found': 'Promo code not found',
|
||||
'expired': 'Promo code expired',
|
||||
'inactive': 'Promo code is deactivated',
|
||||
'not_yet_valid': 'Promo code is not yet active',
|
||||
'used': 'Promo code already used',
|
||||
'already_used_by_user': 'Promo code already used by this user',
|
||||
'no_subscription_for_days': 'This promo code requires an active or expired subscription',
|
||||
@@ -6563,8 +6565,8 @@ async def purchase_tariff_endpoint(
|
||||
group_pcts = bd.get('group_discount_pct', {})
|
||||
discount_percent = group_pcts.get('period', 0)
|
||||
|
||||
# Проверяем баланс
|
||||
if user.balance_kopeks < price_kopeks:
|
||||
# Проверяем баланс (при 100% скидке — пропускаем)
|
||||
if price_kopeks > 0 and user.balance_kopeks < price_kopeks:
|
||||
missing = price_kopeks - user.balance_kopeks
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_402_PAYMENT_REQUIRED,
|
||||
@@ -7053,6 +7055,14 @@ async def switch_tariff_endpoint(
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error('Ошибка синхронизации с RemnaWave при смене тарифа', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(subscription, 'id') and hasattr(subscription, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='update',
|
||||
)
|
||||
|
||||
lang = getattr(user, 'language', settings.DEFAULT_LANGUAGE)
|
||||
if upgrade_cost > 0:
|
||||
@@ -7194,8 +7204,8 @@ async def purchase_traffic_topup_endpoint(
|
||||
subscription.end_date,
|
||||
)
|
||||
|
||||
# Проверяем баланс
|
||||
if user.balance_kopeks < final_price:
|
||||
# Проверяем баланс (при 100% скидке — пропускаем)
|
||||
if final_price > 0 and user.balance_kopeks < final_price:
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_402_PAYMENT_REQUIRED,
|
||||
detail={
|
||||
@@ -7243,6 +7253,14 @@ async def purchase_traffic_topup_endpoint(
|
||||
await service.enable_remnawave_user(_en_uuid)
|
||||
except Exception as e:
|
||||
logger.error('Ошибка синхронизации с RemnaWave при докупке трафика', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(subscription, 'id') and hasattr(subscription, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='update',
|
||||
)
|
||||
|
||||
# Создаем транзакцию
|
||||
await create_transaction(
|
||||
@@ -7482,6 +7500,14 @@ async def toggle_daily_subscription_pause_endpoint(
|
||||
logger.warning('Failed to sync squads after user creation (miniapp)', error=squad_err)
|
||||
except Exception as e:
|
||||
logger.error('Ошибка синхронизации с RemnaWave при возобновлении', error=e)
|
||||
from app.services.remnawave_retry_queue import remnawave_retry_queue
|
||||
|
||||
if hasattr(subscription, 'id') and hasattr(subscription, 'user_id'):
|
||||
remnawave_retry_queue.enqueue(
|
||||
subscription_id=subscription.id,
|
||||
user_id=subscription.user_id,
|
||||
action='update',
|
||||
)
|
||||
|
||||
# Send admin notification about daily subscription resume
|
||||
if resume_transaction is not None:
|
||||
|
||||
@@ -18,7 +18,9 @@ def _normalize_period_discounts(value: dict[object, object] | None) -> dict[int,
|
||||
except (TypeError, ValueError):
|
||||
continue
|
||||
|
||||
return normalized or None
|
||||
# Return empty dict (not None) so the backend can distinguish
|
||||
# "clear all discounts" ({}) from "don't touch discounts" (None/absent).
|
||||
return normalized
|
||||
|
||||
|
||||
class PromoGroupResponse(BaseModel):
|
||||
|
||||
@@ -129,8 +129,9 @@ def create_remnawave_webhook_router(bot: Bot) -> APIRouter:
|
||||
|
||||
# Process event — return 200 to prevent retries for application-level errors.
|
||||
# Only return non-200 for infrastructure failures (DB unavailable).
|
||||
# Admin events (node/service/crm) don't need a DB session.
|
||||
if webhook_service.is_admin_event(event_name):
|
||||
# Admin-only events (node/service/crm) don't need a DB session.
|
||||
# Dual events (admin + user, e.g. torrent_blocker.report) need DB for user handler.
|
||||
if webhook_service.is_admin_event(event_name) and not webhook_service.needs_db_session(event_name):
|
||||
try:
|
||||
processed = await webhook_service.process_event(None, event_name, data)
|
||||
return JSONResponse({'status': 'ok', 'processed': processed})
|
||||
@@ -138,7 +139,7 @@ def create_remnawave_webhook_router(bot: Bot) -> APIRouter:
|
||||
logger.exception('RemnaWave webhook processing error for event', event_name=event_name)
|
||||
return JSONResponse({'status': 'ok', 'processed': False})
|
||||
|
||||
# User events require a DB session
|
||||
# User events and dual events require a DB session
|
||||
try:
|
||||
async with AsyncSessionLocal() as db:
|
||||
try:
|
||||
|
||||
@@ -0,0 +1,72 @@
|
||||
"""include limited status in partial unique index for subscriptions
|
||||
|
||||
Revision ID: 0053
|
||||
Revises: 0052
|
||||
Create Date: 2026-04-03
|
||||
|
||||
"""
|
||||
|
||||
from typing import Sequence, Union
|
||||
|
||||
import sqlalchemy as sa
|
||||
from alembic import op
|
||||
|
||||
revision: str = '0053'
|
||||
down_revision: Union[str, None] = '0052'
|
||||
branch_labels: Union[str, Sequence[str], None] = None
|
||||
depends_on: Union[str, Sequence[str], None] = None
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
# Drop old partial unique index that only covered active/trial
|
||||
op.execute(sa.text('DROP INDEX IF EXISTS uq_subscriptions_user_tariff_active'))
|
||||
|
||||
# Deduplicate: if a user has multiple active/trial/limited subscriptions
|
||||
# for the same tariff, expire all but the most recent one.
|
||||
op.execute(
|
||||
sa.text(
|
||||
"""
|
||||
UPDATE subscriptions
|
||||
SET status = 'expired'
|
||||
WHERE id IN (
|
||||
SELECT id FROM (
|
||||
SELECT id,
|
||||
ROW_NUMBER() OVER (
|
||||
PARTITION BY user_id, tariff_id
|
||||
ORDER BY created_at DESC
|
||||
) AS rn
|
||||
FROM subscriptions
|
||||
WHERE tariff_id IS NOT NULL
|
||||
AND status IN ('active', 'trial', 'limited')
|
||||
) ranked
|
||||
WHERE rn > 1
|
||||
)
|
||||
"""
|
||||
)
|
||||
)
|
||||
|
||||
# Recreate with limited status included — a limited subscription (traffic
|
||||
# exhausted but time remaining) is still "alive" and should prevent
|
||||
# duplicate subscriptions for the same user+tariff combination.
|
||||
op.execute(
|
||||
sa.text(
|
||||
"""
|
||||
CREATE UNIQUE INDEX uq_subscriptions_user_tariff_active
|
||||
ON subscriptions (user_id, tariff_id)
|
||||
WHERE tariff_id IS NOT NULL AND status IN ('active', 'trial', 'limited')
|
||||
"""
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
op.execute(sa.text('DROP INDEX IF EXISTS uq_subscriptions_user_tariff_active'))
|
||||
op.execute(
|
||||
sa.text(
|
||||
"""
|
||||
CREATE UNIQUE INDEX uq_subscriptions_user_tariff_active
|
||||
ON subscriptions (user_id, tariff_id)
|
||||
WHERE tariff_id IS NOT NULL AND status IN ('active', 'trial')
|
||||
"""
|
||||
)
|
||||
)
|
||||
@@ -0,0 +1,28 @@
|
||||
"""add broadcast category column
|
||||
|
||||
Revision ID: 0054
|
||||
Revises: 0053
|
||||
Create Date: 2026-04-10
|
||||
|
||||
"""
|
||||
|
||||
from typing import Sequence, Union
|
||||
|
||||
import sqlalchemy as sa
|
||||
from alembic import op
|
||||
|
||||
revision: str = '0054'
|
||||
down_revision: Union[str, None] = '0053'
|
||||
branch_labels: Union[str, Sequence[str], None] = None
|
||||
depends_on: Union[str, Sequence[str], None] = None
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
op.add_column(
|
||||
'broadcast_history',
|
||||
sa.Column('category', sa.String(20), nullable=False, server_default='system'),
|
||||
)
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
op.drop_column('broadcast_history', 'category')
|
||||
@@ -0,0 +1,28 @@
|
||||
"""add pending_campaign_slug to users
|
||||
|
||||
Revision ID: 0055
|
||||
Revises: 0054
|
||||
Create Date: 2026-04-13
|
||||
|
||||
"""
|
||||
|
||||
from typing import Sequence, Union
|
||||
|
||||
import sqlalchemy as sa
|
||||
from alembic import op
|
||||
|
||||
revision: str = '0055'
|
||||
down_revision: Union[str, None] = '0054'
|
||||
branch_labels: Union[str, Sequence[str], None] = None
|
||||
depends_on: Union[str, Sequence[str], None] = None
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
op.add_column(
|
||||
'users',
|
||||
sa.Column('pending_campaign_slug', sa.String(64), nullable=True),
|
||||
)
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
op.drop_column('users', 'pending_campaign_slug')
|
||||
@@ -0,0 +1,57 @@
|
||||
"""create cabinet_refresh_tokens table
|
||||
|
||||
Revision ID: 0056
|
||||
Revises: 0055
|
||||
Create Date: 2026-04-13
|
||||
|
||||
"""
|
||||
|
||||
from typing import Sequence, Union
|
||||
|
||||
import sqlalchemy as sa
|
||||
from alembic import op
|
||||
|
||||
revision: str = '0056'
|
||||
down_revision: Union[str, None] = '0055'
|
||||
branch_labels: Union[str, Sequence[str], None] = None
|
||||
depends_on: Union[str, Sequence[str], None] = None
|
||||
|
||||
|
||||
def _has_table(table: str) -> bool:
|
||||
conn = op.get_bind()
|
||||
inspector = sa.inspect(conn)
|
||||
return table in inspector.get_table_names()
|
||||
|
||||
|
||||
def _has_index(table: str, index_name: str) -> bool:
|
||||
conn = op.get_bind()
|
||||
inspector = sa.inspect(conn)
|
||||
return index_name in [idx['name'] for idx in inspector.get_indexes(table)]
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
if not _has_table('cabinet_refresh_tokens'):
|
||||
op.create_table(
|
||||
'cabinet_refresh_tokens',
|
||||
sa.Column('id', sa.Integer(), primary_key=True, autoincrement=True),
|
||||
sa.Column('user_id', sa.Integer(), sa.ForeignKey('users.id', ondelete='CASCADE'), nullable=False),
|
||||
sa.Column('token_hash', sa.String(255), nullable=False),
|
||||
sa.Column('device_info', sa.String(500), nullable=True),
|
||||
sa.Column('expires_at', sa.DateTime(timezone=True), nullable=False),
|
||||
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now()),
|
||||
sa.Column('revoked_at', sa.DateTime(timezone=True), nullable=True),
|
||||
)
|
||||
|
||||
if not _has_index('cabinet_refresh_tokens', 'ix_cabinet_refresh_tokens_id'):
|
||||
op.create_index('ix_cabinet_refresh_tokens_id', 'cabinet_refresh_tokens', ['id'])
|
||||
if not _has_index('cabinet_refresh_tokens', 'ix_cabinet_refresh_tokens_token_hash'):
|
||||
op.create_index('ix_cabinet_refresh_tokens_token_hash', 'cabinet_refresh_tokens', ['token_hash'], unique=True)
|
||||
if not _has_index('cabinet_refresh_tokens', 'ix_cabinet_refresh_tokens_user'):
|
||||
op.create_index('ix_cabinet_refresh_tokens_user', 'cabinet_refresh_tokens', ['user_id'])
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
op.drop_index('ix_cabinet_refresh_tokens_user', table_name='cabinet_refresh_tokens')
|
||||
op.drop_index('ix_cabinet_refresh_tokens_token_hash', table_name='cabinet_refresh_tokens')
|
||||
op.drop_index('ix_cabinet_refresh_tokens_id', table_name='cabinet_refresh_tokens')
|
||||
op.drop_table('cabinet_refresh_tokens')
|
||||
@@ -0,0 +1,52 @@
|
||||
"""alter notification_settings from json to jsonb
|
||||
|
||||
Revision ID: 0057
|
||||
Revises: 0056
|
||||
Create Date: 2026-04-13
|
||||
|
||||
"""
|
||||
|
||||
from typing import Sequence, Union
|
||||
|
||||
import sqlalchemy as sa
|
||||
from alembic import op
|
||||
|
||||
revision: str = '0057'
|
||||
down_revision: Union[str, None] = '0056'
|
||||
branch_labels: Union[str, Sequence[str], None] = None
|
||||
depends_on: Union[str, Sequence[str], None] = None
|
||||
|
||||
|
||||
def _get_column_type(table: str, column: str) -> str | None:
|
||||
conn = op.get_bind()
|
||||
result = conn.execute(
|
||||
sa.text(
|
||||
"SELECT data_type FROM information_schema.columns "
|
||||
"WHERE table_name = :table AND column_name = :column"
|
||||
),
|
||||
{'table': table, 'column': column},
|
||||
)
|
||||
row = result.fetchone()
|
||||
return row[0] if row else None
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
col_type = _get_column_type('users', 'notification_settings')
|
||||
if col_type and col_type != 'jsonb':
|
||||
op.execute(
|
||||
sa.text(
|
||||
"ALTER TABLE users "
|
||||
"ALTER COLUMN notification_settings TYPE jsonb "
|
||||
"USING notification_settings::jsonb"
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
op.execute(
|
||||
sa.text(
|
||||
"ALTER TABLE users "
|
||||
"ALTER COLUMN notification_settings TYPE json "
|
||||
"USING notification_settings::json"
|
||||
)
|
||||
)
|
||||
+1
-1
@@ -1,6 +1,6 @@
|
||||
[project]
|
||||
name = 'remnawave-bedolaga-telegram-bot'
|
||||
version = "3.44.0"
|
||||
version = "3.47.0"
|
||||
description = 'Telegram bot for RemnaWave VPN service'
|
||||
readme = 'README.md'
|
||||
license = { text = 'MIT' }
|
||||
|
||||
Reference in New Issue
Block a user