Compare commits

5 Commits

Author SHA1 Message Date
842710fe5c minor bugfix, UI fix 2025-08-03 14:16:52 +09:00
1d659ea5ee bugfix 2025-08-03 10:10:10 +09:00
7acdcc7465 fixed draw and draw-results notification logics 2025-07-22 05:09:28 +09:00
37b0507c07 db_data removed 2025-07-21 17:22:39 +09:00
421bebb770 Bot container restart from admin-panel 2025-07-21 17:16:07 +09:00
26 changed files with 552 additions and 365 deletions

1
lottery/.gitignore vendored
View File

@@ -12,3 +12,4 @@ var/mysql/
.db
.png
db_data/

2
lottery/__init__.py Normal file
View File

@@ -0,0 +1,2 @@
from .lottery.celery import app as celery_app
__all__ = ("celery_app",)

View File

@@ -1,37 +1,101 @@
# import requests
# import json
# from dotenv import load_dotenv
# import os
# load_dotenv()
# API_URL='http://106.245.250.243:8000/api/clients'
# API_KEY='g1XqG9lir4RneLuX01VAob+F0MpVCZTpG2is8UBWLZ0='
# def fetch_clients(api_url, api_key):
# """
# Подключаемся к API с использованием заданного X-API-Key
# """
# headers = {
# "X-API-Key": api_key
# }
# try:
# response = requests.get(api_url, headers=headers)
# response.raise_for_status()
# return response.json()
# except requests.exceptions.RequestException as e:
# print(f"Ошибка при запросе к API: {e}")
# return None
# if __name__ == '__main__':
# api_endpoint = f"{os.getenv('API_URL')}"
# api_key = os.getenv("API_KEY")
# print(api_endpoint, api_key)
# if not api_endpoint or not api_key:
# print("Необходимо задать API_URL и API_KEY в .env файле.")
# exit(1)
# clients = fetch_clients(api_endpoint, api_key)
# if clients is not None:
# print(json.dumps(clients, indent=4, ensure_ascii=False))
# else:
# print("Не удалось получить данные с API.")
import requests
import json
from dotenv import load_dotenv
import os
from dotenv import load_dotenv
load_dotenv('.env')
load_dotenv()
def fetch_clients(api_url, api_key):
"""
Подключаемся к API с использованием заданного X-API-Key
"""
headers = {
"X-API-Key": api_key
}
try:
response = requests.get(api_url, headers=headers)
response.raise_for_status()
return response.json()
except requests.exceptions.RequestException as e:
print(f"Ошибка при запросе к API: {e}")
return None
class ApiClient:
def __init__(self, base_url=None, api_key=None):
self.base_url = base_url or os.getenv('API_URL', 'http://106.245.250.243:8000/api')
self.api_key = api_key or os.getenv('API_KEY')
if not self.api_key:
raise ValueError("API_KEY не задан ни в .env, ни через параметры конструктора")
self.headers = {
"X-API-Key": self.api_key
}
def fetch_clients(self):
"""Получение списка клиентов"""
try:
url = f"{self.base_url}/clients"
response = requests.get(url, headers=self.headers)
response.raise_for_status()
return response.json()
except requests.exceptions.RequestException as e:
print(f"Ошибка при получении клиентов: {e}")
return None
def fetch_invoices(self, filters=None):
"""Получение списка счетов с возможными фильтрами"""
try:
url = f"{self.base_url}/invoices"
response = requests.get(url, headers=self.headers, params=filters)
response.raise_for_status()
return response.json()
except requests.exceptions.RequestException as e:
print(f"Ошибка при получении счетов: {e}")
return None
if __name__ == '__main__':
api_endpoint = os.getenv("API_URL")
api_key = os.getenv("API_KEY")
print(api_endpoint, api_key)
if not api_endpoint or not api_key:
print("Необходимо задать API_URL и API_KEY в .env файле.")
exit(1)
clients = fetch_clients(api_endpoint, api_key)
if clients is not None:
client = ApiClient()
print("=== Клиенты ===")
clients = client.fetch_clients()
if clients:
print(json.dumps(clients, indent=4, ensure_ascii=False))
else:
print("Не удалось получить данные с API.")
print("Не удалось получить клиентов")
print("\n=== Счета ===")
filters = {
"page": 1,
# "created_at[after]": "2024-12-01T00:00:00",
# "created_at[before]": "2025-01-01T00:00:00"
}
invoices = client.fetch_invoices(filters)
if invoices:
print(json.dumps(invoices, indent=4, ensure_ascii=False))
else:
print("Не удалось получить счета")

View File

@@ -1,11 +1,36 @@
from django.contrib import admin
from django.urls import path
from django.shortcuts import redirect
from django.contrib import messages
from .models import BotConfig
from .models import BotConfig, WelcomeMessage
from .tasks import restart_bot_container
@admin.register(BotConfig)
class BotConfigAdmin(admin.ModelAdmin):
list_display = ("bot_name", "channel_id", "bot_token")
search_fields = ("bot_name", "channel_id", "bot_token")
change_form_template = "admin/botconfig_change_form.html"
def get_urls(self):
urls = super().get_urls()
custom_urls = [
path(
"<int:botconfig_id>/restart/",
self.admin_site.admin_view(self.restart_bot),
name="botconfig-restart"
),
]
return custom_urls + urls
def restart_bot(self, request, botconfig_id):
try:
BotConfig.objects.get(id=botconfig_id)
except BotConfig.DoesNotExist:
self.message_user(request, "Настройка бота с таким ID не найдена.", level=messages.ERROR)
return redirect("..")
restart_bot_container.delay()
self.message_user(request, "Бот будет перезапущен в фоновом режиме.", messages.SUCCESS)
return redirect("..")
@admin.register(WelcomeMessage)
class WWelcomeMessageAdmin(admin.ModelAdmin):

View File

@@ -1,11 +1,12 @@
# bot/handlers.py
import re
from telegram import Update, InlineKeyboardMarkup, InlineKeyboardButton
from telegram.ext import ContextTypes
from asgiref.sync import sync_to_async
from webapp.models import BindingRequest, Client, Invoice
from draw.models import Lottery, DrawResult, LotteryParticipant
from draw.models import DrawResult
from bot.state import BINDING_PENDING
from bot.models import BotConfig
@sync_to_async
def get_client_by_telegram_id(telegram_id: int):
@@ -32,9 +33,11 @@ def create_binding_request(telegram_chat_id, client_card):
client_card=client_card
)
async def whoami(update: Update, context: ContextTypes.DEFAULT_TYPE):
if not update.message:
return
chat_id = update.message.chat_id
client = await get_client_by_telegram_id(chat_id)
if not client:
@@ -45,7 +48,7 @@ async def whoami(update: Update, context: ContextTypes.DEFAULT_TYPE):
invoices = await get_invoices_for_client(client)
draws = await get_draws_for_client(client)
message_text = f"👤 *Профиль клиента:*\n" \
f"• *Имя:* {client.name}\n" \
f"• *Клубная карта:* {client.club_card_number}\n\n"
@@ -55,8 +58,7 @@ async def whoami(update: Update, context: ContextTypes.DEFAULT_TYPE):
show_all = len(invoices) > 5
displayed_invoices = invoices[:5] if show_all else invoices
for inv in displayed_invoices:
invoice_str = str(inv).split('/')[0].strip()
message_text += f"{invoice_str} (*{inv.sum}*)\n"
message_text += f" • ID {inv.id} (*{inv.sum}*)\n"
else:
message_text += " _Нет счетов_\n"
@@ -64,12 +66,8 @@ async def whoami(update: Update, context: ContextTypes.DEFAULT_TYPE):
if draws:
for draw in draws:
lottery_name = draw.lottery.name if draw.lottery else "неизвестно"
invoice_info = (draw.participant.invoice
if draw.participant and hasattr(draw.participant, 'invoice')
else "неизвестно")
prize_info = (draw.prize.reward
if draw.prize and hasattr(draw.prize, 'reward')
else "неизвестно")
invoice_info = getattr(draw.participant.invoice, 'id', 'неизвестно') if draw.participant else 'неизвестно'
prize_info = getattr(draw.prize, 'reward', 'неизвестно')
message_text += f" • Лотерея: *{lottery_name}*\n Счет: _{invoice_info}_\n Выигрыш: *{prize_info}*\n\n"
else:
message_text += " _Нет результатов розыгрышей_\n"
@@ -82,43 +80,37 @@ async def whoami(update: Update, context: ContextTypes.DEFAULT_TYPE):
await update.message.reply_text(message_text, parse_mode="Markdown", reply_markup=reply_markup)
async def handle_client_card(update: Update, context: ContextTypes.DEFAULT_TYPE):
"""
Обработчик команды /bind.
После ввода команды бот запрашивает ввод номера КК и ожидает сообщение с кодом.
Здесь мы просто добавляем ID пользователя в pending.
"""
if not update.message:
return
telegram_chat_id = update.message.chat_id
# Если пользователь уже привязан, сообщаем об этом.
client = await get_client_by_telegram_id(telegram_chat_id)
if client:
await update.message.reply_text("✅ Вы уже привязаны!")
return
# Добавляем пользователя в список pending и просим ввести номер КК.
BINDING_PENDING.add(telegram_chat_id)
await update.message.reply_text("Введите номер вашей клубной карты (КК):")
async def process_binding_input(update: Update, context: ContextTypes.DEFAULT_TYPE):
"""
Обрабатывает ввод номера КК от пользователя, который находится в состоянии ожидания.
"""
if not update.message or not update.message.text:
return
client_card = update.message.text.strip()
telegram_chat_id = update.message.chat_id
if not re.fullmatch(r'\d{3,}', client_card):
await update.message.reply_text("❌ Неверный формат клиентской карты. Пожалуйста, введите корректный код (минимум 3 цифры).")
await update.message.reply_text("❌ Неверный формат клиентской карты. Пожалуйста, введите минимум 3 цифры.")
return
await create_binding_request(telegram_chat_id, client_card)
if telegram_chat_id in BINDING_PENDING:
BINDING_PENDING.remove(telegram_chat_id)
BINDING_PENDING.discard(telegram_chat_id)
await update.message.reply_text("✅ Заявка отправлена. После подтверждения вы сможете участвовать в чате.")
async def show_all_invoices_callback(update: Update, context: ContextTypes.DEFAULT_TYPE):
query = update.callback_query
await query.answer()
@@ -132,9 +124,18 @@ async def show_all_invoices_callback(update: Update, context: ContextTypes.DEFAU
message_text = "💳 *Все счета:*\n"
if invoices:
for inv in invoices:
invoice_str = str(inv).split('/')[0].strip()
message_text += f"{invoice_str} (*{inv.sum}*)\n"
message_text += f" • ID {inv.id} (*{inv.sum}*)\n"
else:
message_text += " _Нет счетов_\n"
await query.edit_message_text(message_text, parse_mode="Markdown")
async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
if not update.message:
return
bot_config = await sync_to_async(BotConfig.objects.filter(is_active=True).first)()
if bot_config and bot_config.active_welcome:
await update.message.reply_text(bot_config.active_welcome.welcome_message)
else:
await update.message.reply_text("Привет! Бот работает.")

View File

@@ -0,0 +1,23 @@
# Generated by Django 5.1.6 on 2025-07-20 09:49
from django.db import migrations, models
class Migration(migrations.Migration):
dependencies = [
('bot', '0005_alter_welcomemessage_welcome_image'),
]
operations = [
migrations.AlterField(
model_name='botconfig',
name='global_block_message',
field=models.TextField(blank=True, help_text='Сообщение, которое показывается пользователям, если чат отключён.', null=True, verbose_name='Сообщение о глобальном блоке'),
),
migrations.AlterField(
model_name='welcomemessage',
name='welcome_image',
field=models.ImageField(blank=True, help_text='Загрузите изображение для приветствия', null=True, upload_to='static/upload_image/', verbose_name='Приветственное изображение'),
),
]

View File

@@ -0,0 +1,29 @@
# Generated by Django 5.1.6 on 2025-07-20 23:33
import django.db.models.deletion
from django.db import migrations, models
class Migration(migrations.Migration):
dependencies = [
('bot', '0006_alter_botconfig_global_block_message_and_more'),
]
operations = [
migrations.AddField(
model_name='botconfig',
name='active_welcome',
field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, related_name='used_by', to='bot.welcomemessage', verbose_name='Активное приветствие'),
),
migrations.AddField(
model_name='botconfig',
name='is_active',
field=models.BooleanField(default=False, verbose_name='Включен'),
),
migrations.AddField(
model_name='botconfig',
name='welcome_messages',
field=models.ManyToManyField(blank=True, to='bot.welcomemessage', verbose_name='Варианты приветствия'),
),
]

View File

@@ -1,5 +1,6 @@
from django.db import models
class BotConfig(models.Model):
bot_token = models.CharField(max_length=255, help_text="Токен для подключения к Telegram API")
channel_id = models.CharField(max_length=100, help_text="ID канала/чата, куда бот будет отправлять сообщения")
@@ -15,67 +16,79 @@ class BotConfig(models.Model):
verbose_name="Сообщение о глобальном блоке",
help_text="Сообщение, которое показывается пользователям, если чат отключён."
)
welcome_messages = models.ManyToManyField('WelcomeMessage', blank=True, verbose_name="Варианты приветствия")
active_welcome = models.ForeignKey(
'WelcomeMessage',
null=True,
blank=True,
on_delete=models.SET_NULL,
related_name="used_by",
verbose_name="Активное приветствие"
)
is_active = models.BooleanField(default=False, verbose_name="Включен")
def __str__(self):
return self.bot_name
class Meta:
verbose_name = "Настройка бота"
verbose_name_plural = "Настройки Ботов"
class WelcomeMessage(models.Model):
bot = models.ForeignKey(BotConfig, on_delete=models.CASCADE, verbose_name="Бот")
welcome_message = models.TextField(
verbose_name="Приветственное сообщение",
blank=True,
verbose_name="Приветственное сообщение",
blank=True,
null=True,
help_text="Текст, который будет отправлен при запуске команды /start"
)
welcome_image = models.ImageField(
upload_to='static/upload_image/',
verbose_name="Приветственное изображение",
blank=True,
verbose_name="Приветственное изображение",
blank=True,
null=True,
help_text="Загрузите изображение для приветствия"
)
admin_contact = models.CharField(
max_length=255,
verbose_name="Контакт администратора",
blank=True,
max_length=255,
verbose_name="Контакт администратора",
blank=True,
null=True,
help_text="Ссылка или контакт для связи с администратором"
)
channel_link = models.URLField(
verbose_name="Ссылка на канал",
blank=True,
verbose_name="Ссылка на канал",
blank=True,
null=True,
help_text="URL канала бота"
)
group_link = models.URLField(
verbose_name="Ссылка на группу",
blank=True,
verbose_name="Ссылка на группу",
blank=True,
null=True,
help_text="URL группы бота"
)
custom_link1_name = models.CharField(
max_length=100,
verbose_name="Название кастомной ссылки 1",
blank=True,
max_length=100,
verbose_name="Название кастомной ссылки 1",
blank=True,
null=True
)
custom_link1_url = models.URLField(
verbose_name="URL кастомной ссылки 1",
blank=True,
verbose_name="URL кастомной ссылки 1",
blank=True,
null=True
)
custom_link2_name = models.CharField(
max_length=100,
verbose_name="Название кастомной ссылки 2",
blank=True,
max_length=100,
verbose_name="Название кастомной ссылки 2",
blank=True,
null=True
)
custom_link2_url = models.URLField(
verbose_name="URL кастомной ссылки 2",
blank=True,
verbose_name="URL кастомной ссылки 2",
blank=True,
null=True
)
@@ -84,4 +97,4 @@ class WelcomeMessage(models.Model):
verbose_name_plural = "Приветственные сообщения"
def __str__(self):
return f"Приветствие для {self.bot}"
return f"Приветствие для {self.bot}"

21
lottery/bot/tasks.py Normal file
View File

@@ -0,0 +1,21 @@
from celery import shared_task
import subprocess
@shared_task
def restart_bot_container(container_name="bot"):
try:
result = subprocess.run(
["docker", "restart", container_name],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
text=True,
check=True
)
return f"Success: {result.stdout.strip()}"
except subprocess.CalledProcessError as e:
return f"Error restarting container: {e.stderr.strip()}"
@shared_task
def ping():
return "pong"

View File

@@ -20,6 +20,7 @@ services:
image: lottery_py
build: .
restart: unless-stopped
container_name: bot
entrypoint: ["./wait-for-db.sh"]
command: ["python3", "manage.py", "runbot"]
depends_on:
@@ -31,7 +32,26 @@ services:
- ./var:/app/var
networks:
default:
celery:
build:
context: .
dockerfile: docker/celery.Dockerfile
container_name: celery_worker
volumes:
- .:/app
- /var/run/docker.sock:/var/run/docker.sock
depends_on:
- db
- bot
- redis
environment:
- DJANGO_SETTINGS_MODULE=lottery.settings
networks:
- default
redis:
image: redis:7
container_name: redis
restart: unless-stopped
web:
image: lottery_py
build: .
@@ -42,13 +62,36 @@ services:
- "${DOCKER_HTTP_BIND:-8000}:8000"
depends_on:
- db
- redis
env_file:
- .env
environment:
- DJANGO_SETTINGS_MODULE=lottery.settings
- CELERY_BROKER_URL=redis://redis:6379/0
volumes:
- .:/app
networks:
default:
flower:
build:
context: .
dockerfile: docker/flower.Dockerfile
container_name: flower
restart: unless-stopped
working_dir: /app
volumes:
- .:/app
ports:
- "5555:5555"
command: ["python", "-m", "celery", "-A", "lottery", "-b", "redis://redis:6379/0", "flower", "--port=5555", "--basic_auth=admin:secret"]
depends_on:
- redis
- celery
networks:
- default
volumes:
db_data:

View File

@@ -0,0 +1,23 @@
FROM python:3.12-slim
# Установим необходимые системные зависимости
RUN apt-get update && apt-get install -y \
build-essential \
libmariadb-dev \
default-libmysqlclient-dev \
pkg-config \
docker.io \
&& rm -rf /var/lib/apt/lists/*
# Создание рабочей директории
WORKDIR /app
# Копирование зависимостей и установка Python-библиотек
COPY requirements.txt ./
RUN pip install --no-cache-dir -r requirements.txt
# Копируем весь проект
COPY . .
# Запуск Celery worker
CMD ["celery", "-A", "lottery", "worker", "--loglevel=info"]

View File

@@ -0,0 +1,16 @@
FROM python:3.12-slim
# Установим системные зависимости, нужные для mysqlclient и других пакетов
RUN apt-get update && apt-get install -y \
build-essential \
default-libmysqlclient-dev \
libmariadb-dev \
pkg-config \
&& rm -rf /var/lib/apt/lists/*
WORKDIR /app
COPY docker/flower.requirements.txt ./
RUN pip install --no-cache-dir -r flower.requirements.txt
CMD ["python", "-m", "celery", "-A", "lottery", "flower", "--port=5555", "--basic_auth=admin:secret", "-b", "redis://redis:6379/0"]

View File

@@ -0,0 +1,14 @@
Django
django-jazzmin
django-celery-beat
django-celery-results
django-redis
pillow
python-dotenv
python-telegram-bot
requests
mysqlclient
celery
redis
docker
flower

View File

@@ -1,220 +1,3 @@
# import random
# import logging
# from asgiref.sync import async_to_sync
# from django.contrib import admin, messages
# from django.urls import path, reverse
# from django.shortcuts import render, redirect, get_object_or_404
# from django.utils import timezone
# from django.http import HttpResponseRedirect, HttpResponse
# from django.utils.html import format_html
# from .models import Lottery, Prize, LotteryParticipant, DrawResult
# from .forms import AddParticipantsForm
# from webapp.models import Invoice, Client, BindingRequest
# from bot.notifications import NotificationService
# from bot.utils import create_bot_instance
# # Настройка логгера
# logger = logging.getLogger(__name__)
# logger.setLevel(logging.DEBUG)
# if not logger.handlers:
# console_handler = logging.StreamHandler()
# console_handler.setLevel(logging.DEBUG)
# formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
# console_handler.setFormatter(formatter)
# logger.addHandler(console_handler)
# def add_participants_view(request):
# lottery_id = request.GET.get("lottery_id")
# if not lottery_id:
# return HttpResponse("Не указан параметр lottery_id", status=400)
# lottery = get_object_or_404(Lottery, id=lottery_id)
# used_invoice_ids = LotteryParticipant.objects.filter(lottery=lottery).values_list("invoice_id", flat=True)
# qs = Invoice.objects.exclude(id__in=used_invoice_ids)
# deposit_min = request.GET.get("deposit_min")
# deposit_max = request.GET.get("deposit_max")
# if deposit_min:
# qs = qs.filter(deposit_sum__gte=deposit_min)
# if deposit_max:
# qs = qs.filter(deposit_sum__lte=deposit_max)
# if request.method == "POST":
# form = AddParticipantsForm(request.POST)
# form.fields["invoices"].queryset = qs
# if form.is_valid():
# selected_invoices = form.cleaned_data["invoices"]
# for invoice in selected_invoices:
# LotteryParticipant.objects.create(lottery=lottery, invoice=invoice)
# messages.success(request, "Участники успешно добавлены.")
# return redirect("admin:draw_lotteryparticipant_changelist")
# else:
# form = AddParticipantsForm()
# form.fields["invoices"].queryset = qs
# context = {"form": form, "lottery": lottery}
# return render(request, "admin/add_participants.html", context)
# def get_client_by_invoice(invoice):
# """
# Возвращает клиента, используя значение поля client_club_card_number у счета.
# Предполагается, что у модели Invoice есть поле client_club_card_number.
# """
# try:
# return Client.objects.get(club_card_number=invoice.client_club_card_number)
# except Client.DoesNotExist:
# return None
# def start_draw(request, lottery_id):
# lottery = get_object_or_404(Lottery, id=lottery_id)
# logger.info("Запуск розыгрыша для лотереи: %s", lottery.name)
# notifier = NotificationService(bot=create_bot_instance())
# # Используем async_to_sync для уведомления о запуске розыгрыша
# async_to_sync(notifier.notify_draw_start)(lottery)
# for prize in lottery.prizes.all():
# logger.info("Обработка приза: %s", prize.prize_place)
# # Если для приза уже назначен победитель вручную, сохраняем его в таблице результатов
# if prize.winner:
# logger.info("Приз '%s' имеет установленного вручную победителя. Сохраняем в таблице результатов.", prize.prize_place)
# try:
# draw_result = lottery.draw_results.get(prize=prize)
# draw_result.participant = prize.winner
# draw_result.drawn_at = timezone.now()
# draw_result.confirmed = False
# draw_result.save()
# except DrawResult.DoesNotExist:
# DrawResult.objects.create(
# lottery=lottery,
# prize=prize,
# participant=prize.winner,
# confirmed=False,
# drawn_at=timezone.now()
# )
# continue
# try:
# draw_result = lottery.draw_results.get(prize=prize)
# if draw_result.confirmed:
# logger.info("Приз '%s' уже подтвержден.", prize.prize_place)
# continue
# except DrawResult.DoesNotExist:
# draw_result = None
# participants = list(lottery.participants.filter(used=False))
# logger.info("Найдено свободных участников для приза '%s': %d", prize.prize_place, len(participants))
# if not participants:
# logger.warning("Нет свободных участников для приза '%s'.", prize.prize_place)
# continue
# winner_participant = random.choice(participants)
# winner_participant.used = True
# winner_participant.save()
# logger.info("Выбран участник с счетом '%s' для приза '%s'.", winner_participant.invoice.api_id, prize.prize_place)
# if draw_result:
# draw_result.participant = winner_participant
# draw_result.drawn_at = timezone.now()
# draw_result.confirmed = False
# draw_result.save()
# logger.info("Обновлен результат розыгрыша для приза '%s'.", prize.prize_place)
# else:
# DrawResult.objects.create(
# lottery=lottery,
# prize=prize,
# participant=winner_participant,
# confirmed=False,
# drawn_at=timezone.now()
# )
# logger.info("Создан результат розыгрыша для приза '%s'.", prize.prize_place)
# draw_results = lottery.draw_results.all()
# async_to_sync(notifier.notify_draw_results)(lottery, draw_results)
# return render(request, "admin/draw_result.html", {"lottery": lottery, "draw_results": draw_results})
# def confirm_draw_result(request, result_id):
# from django.http import HttpResponseRedirect
# result = get_object_or_404(DrawResult, id=result_id)
# result.confirmed = True
# result.save()
# prize = result.prize
# prize.winner = result.participant
# prize.save()
# logger.info("Подтвержден результат розыгрыша для приза '%s'.", prize.prize_place)
# messages.success(request, f"Результат для приза '{prize.prize_place}' подтвержден.")
# # Получаем клиента по счету участника
# client = get_client_by_invoice(result.participant.invoice)
# notifier = NotificationService(bot=create_bot_instance())
# if client:
# async_to_sync(notifier.notify_prize_status_update)(client, result)
# return HttpResponseRedirect(reverse("admin:start_draw", args=[result.lottery.id]))
# @admin.register(Lottery)
# class LotteryAdmin(admin.ModelAdmin):
# list_display = ("name", "description", "created_at", "start_draw_button")
# search_fields = ("name", "description")
# def get_urls(self):
# urls = super().get_urls()
# custom_urls = [
# path('<int:lottery_id>/start-draw/', self.admin_site.admin_view(start_draw), name='start_draw'),
# path('confirm-draw-result/<int:result_id>/', self.admin_site.admin_view(confirm_draw_result), name='confirm_draw_result'),
# ]
# return custom_urls + urls
# def start_draw_button(self, obj):
# url = reverse("admin:start_draw", args=[obj.pk])
# return format_html('<a class="btn btn-primary" href="{}">Запуск розыгрыша</a>', url)
# start_draw_button.short_description = "Розыгрыш"
# @admin.register(Prize)
# class PrizeAdmin(admin.ModelAdmin):
# list_display = ("lottery", "prize_place", "reward", "winner")
# list_filter = ("lottery",)
# search_fields = ("prize_place", "description")
# @admin.register(LotteryParticipant)
# class LotteryParticipantAdmin(admin.ModelAdmin):
# list_display = ("lottery", "invoice", "added_at", "used")
# list_filter = ("lottery", "used")
# search_fields = ("invoice__api_id", "lottery__name")
# change_list_template = "admin/draw/lotteryparticipant/change_list.html"
# def changelist_view(self, request, extra_context=None):
# extra_context = extra_context or {}
# from draw.models import Lottery
# active_lotteries = Lottery.objects.filter(prizes__winner__isnull=True).distinct()
# extra_context['active_lotteries'] = active_lotteries
# return super().changelist_view(request, extra_context=extra_context)
# def get_urls(self):
# from django.urls import path
# urls = super().get_urls()
# custom_urls = [
# path('add-participants/', self.admin_site.admin_view(add_participants_view), name="add_participants"),
# ]
# return custom_urls + urls
# @admin.register(DrawResult)
# class DrawResultAdmin(admin.ModelAdmin):
# list_display = ("lottery", "prize", "participant", "confirmed", "drawn_at")
# list_filter = ("lottery", "confirmed")
# search_fields = ("prize__prize_place", "participant__invoice__api_id")
import random
import logging
from asgiref.sync import async_to_sync
@@ -229,6 +12,10 @@ from .forms import AddParticipantsForm
from webapp.models import Invoice, Client, BindingRequest
from bot.notifications import NotificationService
from bot.utils import create_bot_instance
from .views import view_draw_results
from django import forms
from django.utils.dateparse import parse_datetime, parse_date
from datetime import datetime
# Настройка логгера
logger = logging.getLogger(__name__)
@@ -253,11 +40,30 @@ def add_participants_view(request):
deposit_min = request.GET.get("deposit_min")
deposit_max = request.GET.get("deposit_max")
created_after = request.GET.get("created_after")
created_before = request.GET.get("created_before")
if deposit_min:
qs = qs.filter(deposit_sum__gte=deposit_min)
if deposit_max:
qs = qs.filter(deposit_sum__lte=deposit_max)
if created_after:
try:
dt = parse_date(created_after)
if dt:
qs = qs.filter(created_at__gte=dt)
except ValueError:
pass
if created_before:
try:
dt = parse_date(created_before)
if dt:
qs = qs.filter(created_at__lte=dt)
except ValueError:
pass
if request.method == "POST":
form = AddParticipantsForm(request.POST)
form.fields["invoices"].queryset = qs
@@ -293,7 +99,6 @@ def start_draw(request, lottery_id):
lottery = get_object_or_404(Lottery, id=lottery_id)
logger.info("Запуск розыгрыша для лотереи: %s", lottery.name)
# Если лотерея уже завершена, кнопку запускать не показываем
if lottery.finished:
messages.warning(request, "Розыгрыш уже завершён.")
return redirect("..")
@@ -301,10 +106,19 @@ def start_draw(request, lottery_id):
notifier = NotificationService(bot=create_bot_instance())
async_to_sync(notifier.notify_draw_start)(lottery)
# Собираем ID счетов вручную назначенных победителей
manually_assigned_invoice_ids = set()
for prize in lottery.prizes.all():
if prize.winner and prize.winner.invoice:
manually_assigned_invoice_ids.add(prize.winner.invoice_id)
# Помечаем вручную выбранного участника как использованного
prize.winner.used = True
prize.winner.save()
for prize in lottery.prizes.all():
logger.info("Обработка приза: %s", prize.prize_place)
# Если для приза уже назначен победитель вручную, сохраняем его в таблице результатов
# Если у приза уже установлен победитель вручную сохраняем его в таблицу результатов
if prize.winner:
logger.info("Приз '%s' имеет установленного вручную победителя. Сохраняем в таблице результатов.", prize.prize_place)
try:
@@ -331,7 +145,10 @@ def start_draw(request, lottery_id):
except DrawResult.DoesNotExist:
draw_result = None
participants = list(lottery.participants.filter(used=False))
# Получаем всех неиспользованных участников, исключая вручную выбранных
participants = list(
lottery.participants.filter(used=False).exclude(invoice_id__in=manually_assigned_invoice_ids)
)
logger.info("Найдено свободных участников для приза '%s': %d", prize.prize_place, len(participants))
if not participants:
logger.warning("Нет свободных участников для приза '%s'.", prize.prize_place)
@@ -361,7 +178,6 @@ def start_draw(request, lottery_id):
draw_results = lottery.draw_results.all()
async_to_sync(notifier.notify_draw_results)(lottery, draw_results)
# Если все призы розыгрыша подтверждены, устанавливаем флаг завершения лотереи
if not lottery.prizes.filter(winner__isnull=True).exists():
lottery.finished = True
lottery.save()
@@ -390,8 +206,6 @@ def confirm_draw_result(request, result_id):
# Получаем клиента по счету участника
client = get_client_by_invoice(result.participant.invoice)
# Если уведомление вызывает ошибки, можно временно его отключить,
# чтобы проверить базовую функциональность подтверждения.
if client:
try:
# Можно попробовать запуск уведомления в отдельном потоке или отключить его временно:
@@ -399,7 +213,7 @@ def confirm_draw_result(request, result_id):
except Exception as e:
logger.error(f"Ошибка отправки уведомления о статусе приза пользователю {client.telegram_id}: {e}")
return HttpResponseRedirect(reverse("admin:start_draw", args=[result.lottery.id]))
return HttpResponseRedirect(reverse("admin:view_draw_results", args=[result.lottery.id]))
@admin.register(Lottery)
@@ -412,6 +226,8 @@ class LotteryAdmin(admin.ModelAdmin):
custom_urls = [
path('<int:lottery_id>/start-draw/', self.admin_site.admin_view(start_draw), name='start_draw'),
path('confirm-draw-result/<int:result_id>/', self.admin_site.admin_view(confirm_draw_result), name='confirm_draw_result'),
path('<int:lottery_id>/view-draw-results/', self.admin_site.admin_view(view_draw_results), name='view_draw_results'),
]
return custom_urls + urls

View File

@@ -1,3 +1,12 @@
from django.shortcuts import render
from django.shortcuts import get_object_or_404
from .models import Lottery, DrawResult
# Create your views here.
def view_draw_results(request, lottery_id):
lottery = get_object_or_404(Lottery, id=lottery_id)
draw_results = lottery.draw_results.all()
return render(request, "admin/draw_result.html", {
"lottery": lottery,
"draw_results": draw_results
})

View File

@@ -0,0 +1,8 @@
import os
from celery import Celery
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "lottery.settings")
app = Celery("lottery")
app.config_from_object("django.conf:settings", namespace="CELERY")
app.autodiscover_tasks()

View File

@@ -108,6 +108,11 @@ else: # по умолчанию SQLite
}
}
CELERY_BROKER_URL = 'redis://redis:6379/0'
CELERY_RESULT_BACKEND = 'redis://redis:6379/0'
CELERY_ACCEPT_CONTENT = ['json']
CELERY_TASK_SERIALIZER = 'json'
# Password validation
# https://docs.djangoproject.com/en/5.1/ref/settings/#auth-password-validators

View File

@@ -17,4 +17,10 @@ sniffio==1.3.1
sqlparse==0.5.3
typing_extensions==4.12.2
urllib3==2.3.0
mysqlclient
celery
redis
docker
django-celery-beat
django-celery-results
django-redis
flower

Binary file not shown.

After

Width:  |  Height:  |  Size: 120 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 80 KiB

View File

@@ -14,7 +14,7 @@
const selectAllCheckbox = document.getElementById("select-all");
if (selectAllCheckbox) {
selectAllCheckbox.addEventListener("click", function(){
const checkboxes = document.querySelectorAll("input[name='invoices']");
const checkboxes = document.querySelectorAll("input[name='invoices[]']");
checkboxes.forEach(chk => chk.checked = selectAllCheckbox.checked);
});
}
@@ -23,64 +23,92 @@
{% endblock extrahead %}
{% block content %}
<div class="container-fluid">
<div class="container-fluid custom-container">
<h1>Добавление участников лотереи: {{ lottery.name }}</h1>
<p>{{ lottery.description }}</p>
<!-- Форма фильтрации -->
<form method="get" class="form-inline mb-3">
<form method="get" class="mb-4">
<input type="hidden" name="lottery_id" value="{{ lottery.id }}">
<div class="form-group mr-3">
<label for="id_deposit_min" class="mr-2">Минимальный депозит:</label>
<input type="number" step="0.01" name="deposit_min" id="id_deposit_min" class="form-control">
<div class="row">
<div class="col-md-3">
<label>Минимальная сумма депозита:</label>
<input type="number" step="0.01" name="deposit_min" value="{{ request.GET.deposit_min }}" class="form-control">
</div>
<div class="col-md-3">
<label>Максимальная сумма депозита:</label>
<input type="number" step="0.01" name="deposit_max" value="{{ request.GET.deposit_max }}" class="form-control">
</div>
<div class="col-md-3">
<label>Дата создания от:</label>
<input type="date" name="created_after" value="{{ request.GET.created_after }}" class="form-control">
</div>
<div class="col-md-3">
<label>Дата создания до:</label>
<input type="date" name="created_before" value="{{ request.GET.created_before }}" class="form-control">
</div>
</div>
<div class="form-group mr-3">
<label for="id_deposit_max" class="mr-2">Максимальный депозит:</label>
<input type="number" step="0.01" name="deposit_max" id="id_deposit_max" class="form-control">
<div class="mt-3">
<button type="submit" class="btn btn-primary">Применить фильтр</button>
</div>
<button type="submit" class="btn btn-info">Фильтровать</button>
</form>
<!-- Форма добавления участников -->
<form method="post">
{% csrf_token %}
<div class="table-responsive">
<table class="table table-striped table-bordered table-sm">
<thead>
<p><strong>Найдено подходящих счетов: {{ invoice_count }}</strong></p>
<!-- Форма добавления участников -->
<form method="post">
{% csrf_token %}
<div class="table-responsive">
<table class="table table-striped table-bordered table-sm">
<thead>
<tr>
<th><input type="checkbox" id="select-all" /></th>
<th>Создан</th>
<th>Закрыт</th>
<th>Счет</th>
<th>Клиент</th>
<th>Номер клиента</th>
<th>Сумма счета</th>
<th>Бонус</th>
<th>ФД</th>
<th>Депозит</th>
<th>Примечание</th>
</tr>
</thead>
<tbody>
{% for invoice in form.fields.invoices.queryset %}
<tr>
<th><input type="checkbox" id="select-all" /></th>
<th>Счет</th>
<th>Владелец счета</th>
<th>Депозит</th>
<input type="checkbox" name="invoices" value="{{ invoice.id }}">
<td>{{ invoice.created_at|date:"d.m.Y H:i" }}</td>
<td>
{% if invoice.closed_at %}
{{ invoice.closed_at|date:"d.m.Y H:i" }}
{% else %}
{% endif %}
</td>
<td>{{ invoice.ext_id|default:"—" }}</td>
<td>{{ invoice.client.name|default:"Не указан" }}</td>
<td>{{ invoice.client.club_card_number|default:"—" }}</td>
<td>{{ invoice.sum|default:"—" }}</td>
<td>{{ invoice.bonus|default:"—" }}</td>
<td>{{ invoice.start_bonus|default:"—" }}</td>
<td>{{ invoice.deposit_sum|default:"—" }}</td>
<td>{{ invoice.notes|default:"—" }}</td>
</tr>
</thead>
<tbody>
{% for invoice in form.fields.invoices.queryset %}
<tr>
<td>
<input type="checkbox" name="invoices" value="{{ invoice.id }}" />
</td>
<td>{{ invoice.ext_id }}</td>
<td>
{% if invoice.client_name %}
{{ invoice.client_name }}
{% else %}
Не указан
{% endif %}
</td>
<td>{{ invoice.deposit_sum }}</td>
</tr>
{% empty %}
<tr>
<td colspan="5" class="text-center">Нет доступных счетов</td>
</tr>
{% endfor %}
</tbody>
</table>
</div>
<button type="submit" class="btn btn-primary">Добавить выбранные счета</button>
</form>
{% empty %}
<tr>
<td colspan="11" class="text-center">Нет доступных счетов</td>
</tr>
{% endfor %}
</tbody>
</table>
</div>
<button type="submit" class="btn btn-primary">Добавить выбранные счета</button>
</form>
<a href="{% url 'admin:draw_lotteryparticipant_changelist' %}" class="btn btn-secondary mt-3">Вернуться к списку участников</a>
</div>
{% endblock content %}

View File

@@ -0,0 +1,14 @@
{% extends "admin/change_form.html" %}
{% load static %}
{% block object-tools %}
{{ block.super }}
{% if original %}
<li>
<a class="button"
href="{% url 'admin:botconfig-restart' original.pk %}">
🔁 Перезапустить бота
</a>
</li>
{% endif %}
{% endblock %}

View File

@@ -4,4 +4,4 @@ from django.apps import AppConfig
class WebappConfig(AppConfig):
default_auto_field = 'django.db.models.BigAutoField'
name = 'webapp'
verbose_name='Основноая информация'
verbose_name='Основная информация'

View File

@@ -0,0 +1,17 @@
# Generated by Django 5.1.6 on 2025-08-03 05:05
from django.db import migrations
class Migration(migrations.Migration):
dependencies = [
('webapp', '0008_invoice_used'),
]
operations = [
migrations.AlterModelOptions(
name='apisettings',
options={'verbose_name': 'Настройка API', 'verbose_name_plural': 'Настройки API'},
),
]

View File

@@ -57,10 +57,13 @@ class Invoice(models.Model):
class APISettings(models.Model):
api_url = models.URLField("API URL")
api_key = models.CharField("API KEY", max_length=255)
class Meta:
verbose_name = "Настройка API"
verbose_name_plural = "Настройки API"
def __str__(self):
return f"Настройки API: {self.api_url}"
class BindingRequest(models.Model):
STATUS_CHOICES = (
('pending', 'Ожидает проверки'),

View File

@@ -95,14 +95,19 @@ class API_SYNC:
self.logger.warning("Запись клиента %s пропущена: отсутствует club_card_num. Запись: %s", index, item)
continue
# Используем update_or_create для обновления существующей записи
defaults = {
'name': item.get("name"),
}
telegram_id = item.get("telegram_id")
if telegram_id:
defaults['telegram_id'] = telegram_id # Обновим только если значение есть
obj, created = Client.objects.update_or_create(
club_card_number=club_card_number,
defaults={
'name': item.get("name"),
'telegram_id': item.get("telegram_id"),
}
defaults=defaults
)
new_or_updated += 1
if created:
self.logger.info("Запись клиента %s создана: club_card_num %s.", index, club_card_number)
@@ -112,6 +117,7 @@ class API_SYNC:
self.logger.error("Запись клиента %s имеет неожиданный тип: %s. Значение: %s", index, type(item), item)
return new_or_updated
def sync_invoices(self):
api_client = APIClient()
data = api_client.get_invoices()