|
|
from django.contrib import admin |
|
|
from django.http import HttpResponseRedirect |
|
|
from django.shortcuts import render |
|
|
|
|
|
from dtb.settings import DEBUG |
|
|
|
|
|
from users.models import Location |
|
|
from users.models import User |
|
|
from users.forms import BroadcastForm |
|
|
|
|
|
from users.tasks import broadcast_message |
|
|
from tgbot.handlers.broadcast_message.utils import send_one_message |
|
|
|
|
|
|
|
|
@admin.register(User) |
|
|
class UserAdmin(admin.ModelAdmin): |
|
|
list_display = [ |
|
|
'user_id', 'username', 'first_name', 'last_name', |
|
|
'language_code', 'deep_link', |
|
|
'created_at', 'updated_at', "is_blocked_bot", |
|
|
] |
|
|
list_filter = ["is_blocked_bot", ] |
|
|
search_fields = ('username', 'user_id') |
|
|
|
|
|
actions = ['broadcast'] |
|
|
|
|
|
def broadcast(self, request, queryset): |
|
|
""" Select users via check mark in django-admin panel, then select "Broadcast" to send message""" |
|
|
user_ids = queryset.values_list('user_id', flat=True).distinct().iterator() |
|
|
if 'apply' in request.POST: |
|
|
broadcast_message_text = request.POST["broadcast_text"] |
|
|
|
|
|
if DEBUG: |
|
|
for user_id in user_ids: |
|
|
send_one_message( |
|
|
user_id=user_id, |
|
|
text=broadcast_message_text, |
|
|
) |
|
|
self.message_user(request, f"Just broadcasted to {len(queryset)} users") |
|
|
else: |
|
|
broadcast_message.delay(text=broadcast_message_text, user_ids=list(user_ids)) |
|
|
self.message_user(request, f"Broadcasting of {len(queryset)} messages has been started") |
|
|
|
|
|
return HttpResponseRedirect(request.get_full_path()) |
|
|
else: |
|
|
form = BroadcastForm(initial={'_selected_action': user_ids}) |
|
|
return render( |
|
|
request, "admin/broadcast_message.html", {'form': form, 'title': u'Broadcast message'} |
|
|
) |
|
|
|
|
|
|
|
|
@admin.register(Location) |
|
|
class LocationAdmin(admin.ModelAdmin): |
|
|
list_display = ['id', 'user_id', 'created_at'] |
|
|
|
