Merge pull request #12 from Danya-Djan/issue/4

This commit is contained in:
Danya 2024-12-16 19:51:21 +03:00 committed by GitHub
commit 7f5d87b031
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 35 additions and 36 deletions

View File

@ -44,26 +44,26 @@ class AuctionAdmin(admin.ModelAdmin):
def view_product_link(self, obj): def view_product_link(self, obj):
url = reverse("admin:auction_product_change", args=[obj.product_id]) url = reverse("admin:auction_product_change", args=[obj.product_id])
return format_html(f'<a href="{url}">{obj.product.name} ({obj.product_id})</a>') return format_html('<a href="{}">{} ({})</a>', url, obj.product.name, obj.product_id)
view_product_link.short_description = 'Товар' view_product_link.short_description = 'Товар'
def view_betters_link(self, obj): def view_betters_link(self, obj):
count = obj.betters.distinct().count() count = obj.betters.distinct().count()
url = reverse('admin:users_tguser_changelist') + '?' + urlencode({'betters__pk': f'{obj.pk}'}) url = reverse('admin:users_tguser_changelist') + '?' + urlencode({'betters__pk': f'{obj.pk}'})
return format_html(f'<a href="{url}"> {count} users </a>') return format_html('<a href="{}"> {} users </a>', url, count)
view_betters_link.short_description = 'Пользователи, сделавшие ставки' view_betters_link.short_description = 'Пользователи, сделавшие ставки'
def view_bets_link(self, obj): def view_bets_link(self, obj):
count = obj.bets.count() count = obj.bets.count()
url = reverse('admin:auction_bet_changelist') + '?' + urlencode({'auction_id': f'{obj.pk}'}) url = reverse('admin:auction_bet_changelist') + '?' + urlencode({'auction_id': f'{obj.pk}'})
return format_html(f'<a href="{url}"> {count} bets </a>') return format_html('<a href="{}"> {} bets </a>', url, count)
view_bets_link.short_description = 'Ставки' view_bets_link.short_description = 'Ставки'
def view_winners_link(self, obj): def view_winners_link(self, obj):
winning_user_ids = obj.bets.filter(_is_winning=True).values_list('user_id', flat=True) winning_user_ids = obj.bets.filter(_is_winning=True).values_list('user_id', flat=True)
count = winning_user_ids.count() count = winning_user_ids.count()
url = reverse('admin:users_tguser_changelist') + '?' + urlencode({'pk__in': ','.join(map(str, winning_user_ids))}) url = reverse('admin:users_tguser_changelist') + '?' + urlencode({'pk__in': ','.join(map(str, winning_user_ids))})
return format_html(f'<a href="{url}"> {count} победителей </a>') return format_html('<a href="{}"> {} победителей </a>', url, count)
view_winners_link.short_description = 'Победители' view_winners_link.short_description = 'Победители'
@ -86,7 +86,7 @@ class ProductAdmin(admin.ModelAdmin):
def view_auctions_link(self, obj): def view_auctions_link(self, obj):
count = obj.auctions.count() count = obj.auctions.count()
url = reverse('admin:auction_auction_changelist') + '?' + urlencode({'product_id': f'{obj.pk}'}) url = reverse('admin:auction_auction_changelist') + '?' + urlencode({'product_id': f'{obj.pk}'})
return format_html(f'<a href="{url}"> {count} auctions </a>') return format_html('<a href="{}"> {} auctions </a>', url, count)
view_auctions_link.short_description = 'Аукционы' view_auctions_link.short_description = 'Аукционы'
@ -132,16 +132,16 @@ class BetAdmin(admin.ModelAdmin):
def view_user_link(self, obj): def view_user_link(self, obj):
url = reverse("admin:users_tguser_change", args=[obj.user_id]) url = reverse("admin:users_tguser_change", args=[obj.user_id])
return format_html(f'<a href="{url}">{obj.user}</a>') return format_html('<a href="{}">{}</a>', url, obj.user)
view_user_link.short_description = 'Пользователь' view_user_link.short_description = 'Пользователь'
def view_auction_link(self, obj): def view_auction_link(self, obj):
url = reverse("admin:auction_auction_change", args=[obj.auction_id]) url = reverse("admin:auction_auction_change", args=[obj.auction_id])
return format_html(f'<a href="{url}">{obj.auction}</a>') return format_html('<a href="{}">{}</a>', url, obj.auction)
view_auction_link.short_description = 'Аукцион' view_auction_link.short_description = 'Аукцион'
def view_transactions_link(self, obj): def view_transactions_link(self, obj):
count = obj.transactions.count() count = obj.transactions.count()
url = reverse('admin:users_bettransaction_changelist') + '?' + urlencode({'bet_id': f'{obj.pk}'}) url = reverse('admin:users_bettransaction_changelist') + '?' + urlencode({'bet_id': f'{obj.pk}'})
return format_html(f'<a href="{url}"> {count} transactions </a>') return format_html('<a href="{}"> {} transactions </a>', url, count)
view_transactions_link.short_description = 'Транзакции' view_transactions_link.short_description = 'Транзакции'

View File

@ -28,10 +28,10 @@ class ClickAdmin(admin.ModelAdmin):
def view_user_link(self, obj): def view_user_link(self, obj):
url = reverse("admin:users_tguser_change", args=[obj.user_id]) url = reverse("admin:users_tguser_change", args=[obj.user_id])
return format_html(f'<a href="{url}">{obj.user}</a>') return format_html('<a href="{}">{}</a>', url, obj.user)
view_user_link.short_description = 'Пользователь' view_user_link.short_description = 'Пользователь'
def view_transaction_link(self, obj): def view_transaction_link(self, obj):
url = reverse("admin:users_clicktransaction_change", args=[obj.transaction.id]) url = reverse("admin:users_clicktransaction_change", args=[obj.transaction.id])
return format_html(f'<a href="{url}">{obj.transaction}</a>') return format_html('<a href="{}">{}</a>', url, obj.transaction)
view_transaction_link.short_description = 'Транзакция' view_transaction_link.short_description = 'Транзакция'

View File

@ -58,20 +58,20 @@ class TGUserAdmin(admin.ModelAdmin):
def view_user_link(self, obj): def view_user_link(self, obj):
url = reverse("admin:auth_user_change", args=[obj.user_id]) url = reverse("admin:auth_user_change", args=[obj.user_id])
return format_html(f'<a href="{url}">{obj.user.username} ({obj.user_id})</a>') return format_html('<a href="{}">{} ({})</a>', url, obj.user.username, obj.user_id)
view_user_link.short_description = 'Системный пользователь' view_user_link.short_description = 'Системный пользователь'
def view_referred_by_link(self, obj): def view_referred_by_link(self, obj):
if not obj.referred_by: if not obj.referred_by:
return return
url = reverse("admin:users_tguser_change", args=[obj.referred_by.tg_id]) url = reverse("admin:users_tguser_change", args=[obj.referred_by.tg_id])
return format_html(f'<a href="{url}">{obj.referred_by.username} ({obj.referred_by.tg_id})</a>') return format_html('<a href="{}">{} ({})</a>', url, obj.referred_by.username, obj.referred_by.tg_id)
view_referred_by_link.short_description = 'Кем был приглашен' view_referred_by_link.short_description = 'Кем был приглашен'
def view_referred_users_link(self, obj): def view_referred_users_link(self, obj):
count = obj.referrees.count() count = obj.referrees.count()
url = reverse('admin:users_tguser_changelist') + '?' + urlencode({'referred_by__tg_id': f'{obj.tg_id}'}) url = reverse('admin:users_tguser_changelist') + '?' + urlencode({'referred_by__tg_id': f'{obj.tg_id}'})
return format_html(f'<a href="{url}"> {count} users </a>') return format_html('<a href="{}"> {} users </a>', url, count)
view_referred_users_link.short_description = 'Приглашенные пользователи' view_referred_users_link.short_description = 'Приглашенные пользователи'
def view_transactions(self, obj): def view_transactions(self, obj):
@ -82,26 +82,25 @@ class TGUserAdmin(admin.ModelAdmin):
referral_url = reverse('admin:users_referraltransaction_changelist') + '?' + urlencode({'user_id': f'{obj.tg_id}'}) referral_url = reverse('admin:users_referraltransaction_changelist') + '?' + urlencode({'user_id': f'{obj.tg_id}'})
return format_html( return format_html(
f'<a href="{all_url}"> все </a> // ' '<a href="{}"> все </a> // '
f'<a href="{click_url}"> клики </a> // ' '<a href="{}"> клики </a> // '
f'<a href="{bet_url}"> ставки </a> // ' '<a href="{}"> ставки </a> // '
f'<a href="{commission_url}"> комиссии </a> // ' '<a href="{}"> комиссии </a> // '
f'<a href="{referral_url}"> реферальная программа </a>' '<a href="{}"> реферальная программа </a>',
all_url, click_url, bet_url, commission_url, referral_url
) )
view_transactions.short_description = 'Транзакции' view_transactions.short_description = 'Транзакции'
def view_clicks_link(self, obj): def view_clicks_link(self, obj):
count = obj.clicks.count() count = obj.clicks.count()
url = reverse('admin:clicks_click_changelist') + '?' + urlencode({'user_id': f'{obj.tg_id}'}) url = reverse('admin:clicks_click_changelist') + '?' + urlencode({'user_id': f'{obj.tg_id}'})
return format_html(f'<a href="{url}"> {count} clicks </a>') return format_html('<a href="{}"> {} clicks </a>', url, count)
view_clicks_link.short_description = 'Клики' view_clicks_link.short_description = 'Клики'
@admin.action(description="Создать рассылку для выбранных пользователей") @admin.action(description="Создать рассылку для выбранных пользователей")
def create_mailing_list(self, request, queryset): def create_mailing_list(self, request, queryset):
request.session['user_ids'] = list(queryset.values_list('pk', flat=True)) request.session['user_ids'] = list(queryset.values_list('pk', flat=True))
return HttpResponseRedirect( return HttpResponseRedirect(reverse('admin:users_mailinglist_add'))
f'/admin/users/mailinglist/add/'
)
class MailingListAdminForm(forms.ModelForm): class MailingListAdminForm(forms.ModelForm):
@ -162,14 +161,14 @@ class MailingListReceiverInfoAdmin(admin.ModelAdmin):
if not obj.user: if not obj.user:
return None return None
link = reverse("admin:users_tguser_change", args=[obj.user.tg_id]) link = reverse("admin:users_tguser_change", args=[obj.user.tg_id])
return format_html(f'<a href="{link}">{obj.user}</a>') return format_html('<a href="{}">{}</a>', link, obj.user)
view_user_link.short_description = 'Пользователь' view_user_link.short_description = 'Пользователь'
def view_mailing_list_link(self, obj): def view_mailing_list_link(self, obj):
if not obj.mailing_list: if not obj.mailing_list:
return None return None
link = reverse("admin:users_mailinglist_change", args=[obj.mailing_list.pk]) link = reverse("admin:users_mailinglist_change", args=[obj.mailing_list.pk])
return format_html(f'<a href="{link}">{obj.mailing_list}</a>') return format_html('<a href="{}">{}</a>', link, obj.mailing_list)
view_mailing_list_link.short_description = 'Рассылка' view_mailing_list_link.short_description = 'Рассылка'
@ -205,28 +204,28 @@ class MailingListAdmin(admin.ModelAdmin):
count = obj.users.count() count = obj.users.count()
url = reverse('admin:users_tguser_changelist') + '?' + urlencode( url = reverse('admin:users_tguser_changelist') + '?' + urlencode(
{'mailing_lists__id': f'{obj.id}'}) {'mailing_lists__id': f'{obj.id}'})
return format_html(f'<a href="{url}"> {count} пользователей </a>') return format_html('<a href="{}"> {} пользователей </a>', url, count)
view_users_link.short_description = 'Пользователи' view_users_link.short_description = 'Пользователи'
def view_mailing_list_receiver_infos_link(self, obj): def view_mailing_list_receiver_infos_link(self, obj):
count = obj.mailing_list_receiver_infos.count() count = obj.mailing_list_receiver_infos.count()
url = reverse('admin:users_mailinglistreceiverinfo_changelist') + '?' + urlencode( url = reverse('admin:users_mailinglistreceiverinfo_changelist') + '?' + urlencode(
{'mailing_list_id': f'{obj.id}'}) {'mailing_list_id': f'{obj.id}'})
return format_html(f'<a href="{url}"> {count} получателей </a>') return format_html('<a href="{}"> {} получателей </a>', url, count)
view_mailing_list_receiver_infos_link.short_description = 'Информация о получателях' view_mailing_list_receiver_infos_link.short_description = 'Информация о получателях'
def view_main_button_link(self, obj): def view_main_button_link(self, obj):
if not obj.main_button: if not obj.main_button:
return return
url = reverse("admin:misc_button_change", args=[obj.main_button_id]) url = reverse("admin:misc_button_change", args=[obj.main_button_id])
return format_html(f'<a href="{url}"> Кнопка №{obj.main_button_id}</a>') return format_html('<a href="{}"> Кнопка №{}</a>', url, obj.main_button_id)
view_main_button_link.short_description = 'Основная кнопка' view_main_button_link.short_description = 'Основная кнопка'
def view_webapp_button_link(self, obj): def view_webapp_button_link(self, obj):
if not obj.webapp_button: if not obj.webapp_button:
return return
url = reverse("admin:misc_button_change", args=[obj.webapp_button_id]) url = reverse("admin:misc_button_change", args=[obj.webapp_button_id])
return format_html(f'<a href="{url}"> Кнопка №{obj.webapp_button_id}</a>') return format_html('<a href="{}"> Кнопка №{}</a>', url, obj.webapp_button_id)
view_webapp_button_link.short_description = 'Кнопка, открывающая вебапп' view_webapp_button_link.short_description = 'Кнопка, открывающая вебапп'
@ -242,7 +241,7 @@ class TransactionChildAdmin(PolymorphicChildModelAdmin):
if not obj.user: if not obj.user:
return return
url = reverse("admin:users_tguser_change", args=[obj.user.tg_id]) url = reverse("admin:users_tguser_change", args=[obj.user.tg_id])
return format_html(f'<a href="{url}">{obj.user}</a>') return format_html('<a href="{}">{}</a>', url, obj.user)
view_user_link.short_description = 'Пользователь' view_user_link.short_description = 'Пользователь'
def has_delete_permission(self, request, obj=None): def has_delete_permission(self, request, obj=None):
@ -269,7 +268,7 @@ class ClickTransactionAdmin(TransactionChildAdmin):
def view_click_link(self, obj): def view_click_link(self, obj):
link = reverse("admin:clicks_click_change", args=[obj.click_id]) link = reverse("admin:clicks_click_change", args=[obj.click_id])
return format_html(f'<a href="{link}"> {obj.click}</a>') return format_html('<a href="{}"> {}</a>', link, obj.click)
view_click_link.short_description = 'Клик' view_click_link.short_description = 'Клик'
@ -293,7 +292,7 @@ class BetTransactionAdmin(TransactionChildAdmin):
def view_bet_link(self, obj): def view_bet_link(self, obj):
link = reverse("admin:auction_bet_change", args=[obj.bet_id]) link = reverse("admin:auction_bet_change", args=[obj.bet_id])
return format_html(f'<a href="{link}">{obj.bet}</a>') return format_html('<a href="{}">{}</a>', link, obj.bet)
view_bet_link.short_description = 'Ставка' view_bet_link.short_description = 'Ставка'
def view_commission_link(self, obj): def view_commission_link(self, obj):
@ -302,7 +301,7 @@ class BetTransactionAdmin(TransactionChildAdmin):
except ObjectDoesNotExist: except ObjectDoesNotExist:
return None return None
link = reverse("admin:users_commissiontransaction_change", args=[obj.commission.id]) link = reverse("admin:users_commissiontransaction_change", args=[obj.commission.id])
return format_html(f'<a href="{link}"> {obj.commission} </a>') return format_html('<a href="{}"> {} </a>', link, obj.commission)
view_commission_link.short_description = 'Комиссия' view_commission_link.short_description = 'Комиссия'
def view_refunded_by_link(self, obj): def view_refunded_by_link(self, obj):
@ -311,14 +310,14 @@ class BetTransactionAdmin(TransactionChildAdmin):
except ObjectDoesNotExist: except ObjectDoesNotExist:
return None return None
link = reverse("admin:users_bettransaction_change", args=[obj.refunded_by.id]) link = reverse("admin:users_bettransaction_change", args=[obj.refunded_by.id])
return format_html(f'<a href="{link}"> {obj.refunded_by} </a>') return format_html('<a href="{}"> {} </a>', link, obj.refunded_by)
view_refunded_by_link.short_description = 'Чем компенсирована' view_refunded_by_link.short_description = 'Чем компенсирована'
def view_refund_to_link(self, obj): def view_refund_to_link(self, obj):
if not obj.refund_to: if not obj.refund_to:
return None return None
link = reverse("admin:users_bettransaction_change", args=[obj.refund_to_id]) link = reverse("admin:users_bettransaction_change", args=[obj.refund_to_id])
return format_html(f'<a href="{link}"> {obj.refund_to} </a>') return format_html('<a href="{}"> {} </a>', link, obj.refund_to)
view_refund_to_link.short_description = 'Что компенсирует' view_refund_to_link.short_description = 'Что компенсирует'
@ -339,7 +338,7 @@ class CommissionTransactionAdmin(TransactionChildAdmin):
def view_bet_transaction_link(self, obj): def view_bet_transaction_link(self, obj):
link = reverse("admin:users_bettransaction_change", args=[obj.parent_transaction]) link = reverse("admin:users_bettransaction_change", args=[obj.parent_transaction])
return format_html(f'<a href="{link}"> {obj.parent_transaction} </a>') return format_html('<a href="{}"> {} </a>', link, obj.parent_transaction)
view_bet_transaction_link.short_description = 'Родительская транзакция' view_bet_transaction_link.short_description = 'Родительская транзакция'
@ -389,7 +388,7 @@ class TransactionParentAdmin(PolymorphicParentModelAdmin):
if not obj.user: if not obj.user:
return return
url = reverse("admin:users_tguser_change", args=[obj.user.tg_id]) url = reverse("admin:users_tguser_change", args=[obj.user.tg_id])
return format_html(f'<a href="{url}">{obj.user}</a>') return format_html('<a href="{}">{}</a>', url, obj.user)
view_user_link.short_description = 'Пользователь' view_user_link.short_description = 'Пользователь'
def view_type(self, obj): def view_type(self, obj):