python - 如何在 django admin 中添加自定义按钮以运行 django 管理命令
问题描述
我必须在我的 django-admin 页面的一个模型中添加一个更新按钮,match_action 是 django-admin 上的更新按钮,match_status 是我的管理命令,这是我的代码,实际上我想要,当我单击更新按钮时想要使用管理命令运行名为 goalserveService 的服务
管理员.py`
@admin.register(Match)
class MatchAdmin(admin.ModelAdmin):
list_display = ['id', 'start_time', "home_club", 'away_club', 'created_on', 'status', 'lookup_id', 'timer', 'match_actions']
search_fields = ['id', 'home_club__name', 'away_club__name']
list_filter = ['status', 'season', 'start_time']
inlines = [ MatchLookupInline, MatchEventInline, ]
def lookup_id(self, obj):
if obj.matchlookup_set.all().exists():
return obj.matchlookup_set.get().source_match_id
else:
return None
def get_urls(self):
urls = super().get_urls()
custom_urls = [
url(
r'^(?P<match_id>.+)/update/$',
self.admin_site.admin_view(self.match_update),
name='match-update',
),
]
return custom_urls + urls
def match_actions(self, obj):
print (obj.id)
return format_html(
'<a class="button" href="{}">Update</a>',
reverse('admin:match-update', args=(obj.id,)),
)
def match_update(self, request, match_id, *args):
from django.core.management import call_command
call_command("match_status", match_id)
核心/管理/命令/match_status.py
from django.utils import timezone
from django.core.management.base import BaseCommand, CommandError
from core.models import UserTeam
from ourapp.models import Match
class Command(BaseCommand):
help = 'Update Match Status'
def add_arguments(self, parser):
parser.add_argument('match_id', nargs='+', type=int)
def handle(self, *args, **options):
# NOT_STARTED = 0
# HF = 1
# FT = 2
# IN_PLAY = 3
# FINISH = 4
match = Match.objects.filter(id=options.get('match_id'))
self.stdout.write(self.style.NOTICE("Match competition Name %s Match Found" % match.competition.name))
current_time = timezone.now()
try:
import time
self.stdout.write(self.style.NOTICE("Match competition Name %s Match Found" % match.competition.name if match.competition else "N/A"))
match = Match.objects.filter(id=options.get('match_id'))
time.sleep(5)
from ourapp.services import goalserveService
try:
goalserveService.run_match_events(match.pk, 2, update_df=False, update_round=False)
time.sleep(5)
except:
pass
from django.core.management import call_command
call_command("update_dfc_rank", 1)
except Match.DoesNotExist:
raise CommandError('League "%s" does not exist' % match)
self.stdout.write(self.style.SUCCESS('Successfully closed league "%s"' % match))
但它显示一个错误
解决方案
每当您.objects.filter
在代码中看到时,您都会返回一个QuerySet
. QuerySet
s 可以解释为“列表”。对于初学者,替换.filter
为.get
以解决当前错误(但我不能保证您之后不会遇到新错误)。