34 lines
998 B
Python
34 lines
998 B
Python
from django.contrib import admin
|
|
from django.urls.resolvers import get_resolver
|
|
from django import forms
|
|
|
|
from .models import Tip
|
|
|
|
def url_choices():
|
|
choices = [('', '-- All Pages --')]
|
|
for entry in get_resolver(None).url_patterns:
|
|
if entry.pattern.name:
|
|
choices.append((entry.pattern.name, entry.pattern.name))
|
|
return choices
|
|
|
|
# Register your models here.
|
|
class TipForm(forms.ModelForm):
|
|
class Meta:
|
|
model = Tip
|
|
fields = '__all__'
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.fields['view'].widget = forms.Select(choices=url_choices())
|
|
|
|
class TipAdmin(admin.ModelAdmin):
|
|
list_filter =('level', 'view')
|
|
list_display = ('name', 'level', 'view', 'run_start', 'run_end', 'seen_count')
|
|
search_fields = ('name', 'view')
|
|
form = TipForm
|
|
|
|
def seen_count(self, tip):
|
|
return tip.seen_by.count()
|
|
seen_count.short_description = 'Seen by'
|
|
|
|
admin.site.register(Tip, TipAdmin)
|