mirror of
https://github.com/django/django.git
synced 2025-08-03 18:38:50 +00:00
Refs #33476 -- Reformatted code with Black.
This commit is contained in:
parent
f68fa8b45d
commit
9c19aff7c7
1992 changed files with 139577 additions and 96284 deletions
|
@ -3,12 +3,17 @@ from datetime import date
|
|||
from django import forms
|
||||
from django.contrib.admin.models import ADDITION, CHANGE, DELETION, LogEntry
|
||||
from django.contrib.admin.options import (
|
||||
HORIZONTAL, VERTICAL, ModelAdmin, TabularInline,
|
||||
HORIZONTAL,
|
||||
VERTICAL,
|
||||
ModelAdmin,
|
||||
TabularInline,
|
||||
get_content_type_for_model,
|
||||
)
|
||||
from django.contrib.admin.sites import AdminSite
|
||||
from django.contrib.admin.widgets import (
|
||||
AdminDateWidget, AdminRadioSelect, AutocompleteSelect,
|
||||
AdminDateWidget,
|
||||
AdminRadioSelect,
|
||||
AutocompleteSelect,
|
||||
AutocompleteSelectMultiple,
|
||||
)
|
||||
from django.contrib.auth.models import User
|
||||
|
@ -34,12 +39,11 @@ request.user = MockSuperUser()
|
|||
|
||||
|
||||
class ModelAdminTests(TestCase):
|
||||
|
||||
@classmethod
|
||||
def setUpTestData(cls):
|
||||
cls.band = Band.objects.create(
|
||||
name='The Doors',
|
||||
bio='',
|
||||
name="The Doors",
|
||||
bio="",
|
||||
sign_date=date(1965, 1, 1),
|
||||
)
|
||||
|
||||
|
@ -48,7 +52,7 @@ class ModelAdminTests(TestCase):
|
|||
|
||||
def test_modeladmin_str(self):
|
||||
ma = ModelAdmin(Band, self.site)
|
||||
self.assertEqual(str(ma), 'modeladmin.ModelAdmin')
|
||||
self.assertEqual(str(ma), "modeladmin.ModelAdmin")
|
||||
|
||||
def test_default_attributes(self):
|
||||
ma = ModelAdmin(Band, self.site)
|
||||
|
@ -59,9 +63,13 @@ class ModelAdminTests(TestCase):
|
|||
|
||||
def test_default_fields(self):
|
||||
ma = ModelAdmin(Band, self.site)
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ['name', 'bio', 'sign_date'])
|
||||
self.assertEqual(list(ma.get_fields(request)), ['name', 'bio', 'sign_date'])
|
||||
self.assertEqual(list(ma.get_fields(request, self.band)), ['name', 'bio', 'sign_date'])
|
||||
self.assertEqual(
|
||||
list(ma.get_form(request).base_fields), ["name", "bio", "sign_date"]
|
||||
)
|
||||
self.assertEqual(list(ma.get_fields(request)), ["name", "bio", "sign_date"])
|
||||
self.assertEqual(
|
||||
list(ma.get_fields(request, self.band)), ["name", "bio", "sign_date"]
|
||||
)
|
||||
self.assertIsNone(ma.get_exclude(request, self.band))
|
||||
|
||||
def test_default_fieldsets(self):
|
||||
|
@ -72,43 +80,50 @@ class ModelAdminTests(TestCase):
|
|||
# Here's the default case. There are no custom form_add/form_change methods,
|
||||
# no fields argument, and no fieldsets argument.
|
||||
ma = ModelAdmin(Band, self.site)
|
||||
self.assertEqual(ma.get_fieldsets(request), [(None, {'fields': ['name', 'bio', 'sign_date']})])
|
||||
self.assertEqual(ma.get_fieldsets(request, self.band), [(None, {'fields': ['name', 'bio', 'sign_date']})])
|
||||
self.assertEqual(
|
||||
ma.get_fieldsets(request),
|
||||
[(None, {"fields": ["name", "bio", "sign_date"]})],
|
||||
)
|
||||
self.assertEqual(
|
||||
ma.get_fieldsets(request, self.band),
|
||||
[(None, {"fields": ["name", "bio", "sign_date"]})],
|
||||
)
|
||||
|
||||
def test_get_fieldsets(self):
|
||||
# get_fieldsets() is called when figuring out form fields (#18681).
|
||||
class BandAdmin(ModelAdmin):
|
||||
def get_fieldsets(self, request, obj=None):
|
||||
return [(None, {'fields': ['name', 'bio']})]
|
||||
return [(None, {"fields": ["name", "bio"]})]
|
||||
|
||||
ma = BandAdmin(Band, self.site)
|
||||
form = ma.get_form(None)
|
||||
self.assertEqual(form._meta.fields, ['name', 'bio'])
|
||||
self.assertEqual(form._meta.fields, ["name", "bio"])
|
||||
|
||||
class InlineBandAdmin(TabularInline):
|
||||
model = Concert
|
||||
fk_name = 'main_band'
|
||||
fk_name = "main_band"
|
||||
can_delete = False
|
||||
|
||||
def get_fieldsets(self, request, obj=None):
|
||||
return [(None, {'fields': ['day', 'transport']})]
|
||||
return [(None, {"fields": ["day", "transport"]})]
|
||||
|
||||
ma = InlineBandAdmin(Band, self.site)
|
||||
form = ma.get_formset(None).form
|
||||
self.assertEqual(form._meta.fields, ['day', 'transport'])
|
||||
self.assertEqual(form._meta.fields, ["day", "transport"])
|
||||
|
||||
def test_lookup_allowed_allows_nonexistent_lookup(self):
|
||||
"""
|
||||
A lookup_allowed allows a parameter whose field lookup doesn't exist.
|
||||
(#21129).
|
||||
"""
|
||||
|
||||
class BandAdmin(ModelAdmin):
|
||||
fields = ['name']
|
||||
fields = ["name"]
|
||||
|
||||
ma = BandAdmin(Band, self.site)
|
||||
self.assertTrue(ma.lookup_allowed('name__nonexistent', 'test_value'))
|
||||
self.assertTrue(ma.lookup_allowed("name__nonexistent", "test_value"))
|
||||
|
||||
@isolate_apps('modeladmin')
|
||||
@isolate_apps("modeladmin")
|
||||
def test_lookup_allowed_onetoone(self):
|
||||
class Department(models.Model):
|
||||
code = models.CharField(max_length=4, unique=True)
|
||||
|
@ -125,28 +140,34 @@ class ModelAdminTests(TestCase):
|
|||
|
||||
class EmployeeProfileAdmin(ModelAdmin):
|
||||
list_filter = [
|
||||
'employee__employeeinfo__description',
|
||||
'employee__department__code',
|
||||
"employee__employeeinfo__description",
|
||||
"employee__department__code",
|
||||
]
|
||||
|
||||
ma = EmployeeProfileAdmin(EmployeeProfile, self.site)
|
||||
# Reverse OneToOneField
|
||||
self.assertIs(ma.lookup_allowed('employee__employeeinfo__description', 'test_value'), True)
|
||||
self.assertIs(
|
||||
ma.lookup_allowed("employee__employeeinfo__description", "test_value"), True
|
||||
)
|
||||
# OneToOneField and ForeignKey
|
||||
self.assertIs(ma.lookup_allowed('employee__department__code', 'test_value'), True)
|
||||
self.assertIs(
|
||||
ma.lookup_allowed("employee__department__code", "test_value"), True
|
||||
)
|
||||
|
||||
def test_field_arguments(self):
|
||||
# If fields is specified, fieldsets_add and fieldsets_change should
|
||||
# just stick the fields into a formsets structure and return it.
|
||||
class BandAdmin(ModelAdmin):
|
||||
fields = ['name']
|
||||
fields = ["name"]
|
||||
|
||||
ma = BandAdmin(Band, self.site)
|
||||
|
||||
self.assertEqual(list(ma.get_fields(request)), ['name'])
|
||||
self.assertEqual(list(ma.get_fields(request, self.band)), ['name'])
|
||||
self.assertEqual(ma.get_fieldsets(request), [(None, {'fields': ['name']})])
|
||||
self.assertEqual(ma.get_fieldsets(request, self.band), [(None, {'fields': ['name']})])
|
||||
self.assertEqual(list(ma.get_fields(request)), ["name"])
|
||||
self.assertEqual(list(ma.get_fields(request, self.band)), ["name"])
|
||||
self.assertEqual(ma.get_fieldsets(request), [(None, {"fields": ["name"]})])
|
||||
self.assertEqual(
|
||||
ma.get_fieldsets(request, self.band), [(None, {"fields": ["name"]})]
|
||||
)
|
||||
|
||||
def test_field_arguments_restricted_on_form(self):
|
||||
# If fields or fieldsets is specified, it should exclude fields on the
|
||||
|
@ -157,41 +178,41 @@ class ModelAdminTests(TestCase):
|
|||
|
||||
# Using `fields`.
|
||||
class BandAdmin(ModelAdmin):
|
||||
fields = ['name']
|
||||
fields = ["name"]
|
||||
|
||||
ma = BandAdmin(Band, self.site)
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ['name'])
|
||||
self.assertEqual(list(ma.get_form(request, self.band).base_fields), ['name'])
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ["name"])
|
||||
self.assertEqual(list(ma.get_form(request, self.band).base_fields), ["name"])
|
||||
|
||||
# Using `fieldsets`.
|
||||
class BandAdmin(ModelAdmin):
|
||||
fieldsets = [(None, {'fields': ['name']})]
|
||||
fieldsets = [(None, {"fields": ["name"]})]
|
||||
|
||||
ma = BandAdmin(Band, self.site)
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ['name'])
|
||||
self.assertEqual(list(ma.get_form(request, self.band).base_fields), ['name'])
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ["name"])
|
||||
self.assertEqual(list(ma.get_form(request, self.band).base_fields), ["name"])
|
||||
|
||||
# Using `exclude`.
|
||||
class BandAdmin(ModelAdmin):
|
||||
exclude = ['bio']
|
||||
exclude = ["bio"]
|
||||
|
||||
ma = BandAdmin(Band, self.site)
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ['name', 'sign_date'])
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ["name", "sign_date"])
|
||||
|
||||
# You can also pass a tuple to `exclude`.
|
||||
class BandAdmin(ModelAdmin):
|
||||
exclude = ('bio',)
|
||||
exclude = ("bio",)
|
||||
|
||||
ma = BandAdmin(Band, self.site)
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ['name', 'sign_date'])
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ["name", "sign_date"])
|
||||
|
||||
# Using `fields` and `exclude`.
|
||||
class BandAdmin(ModelAdmin):
|
||||
fields = ['name', 'bio']
|
||||
exclude = ['bio']
|
||||
fields = ["name", "bio"]
|
||||
exclude = ["bio"]
|
||||
|
||||
ma = BandAdmin(Band, self.site)
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ['name'])
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ["name"])
|
||||
|
||||
def test_custom_form_meta_exclude_with_readonly(self):
|
||||
"""
|
||||
|
@ -203,25 +224,25 @@ class ModelAdminTests(TestCase):
|
|||
class AdminBandForm(forms.ModelForm):
|
||||
class Meta:
|
||||
model = Band
|
||||
exclude = ['bio']
|
||||
exclude = ["bio"]
|
||||
|
||||
class BandAdmin(ModelAdmin):
|
||||
readonly_fields = ['name']
|
||||
readonly_fields = ["name"]
|
||||
form = AdminBandForm
|
||||
|
||||
ma = BandAdmin(Band, self.site)
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ['sign_date'])
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ["sign_date"])
|
||||
|
||||
# With InlineModelAdmin
|
||||
class AdminConcertForm(forms.ModelForm):
|
||||
class Meta:
|
||||
model = Concert
|
||||
exclude = ['day']
|
||||
exclude = ["day"]
|
||||
|
||||
class ConcertInline(TabularInline):
|
||||
readonly_fields = ['transport']
|
||||
readonly_fields = ["transport"]
|
||||
form = AdminConcertForm
|
||||
fk_name = 'main_band'
|
||||
fk_name = "main_band"
|
||||
model = Concert
|
||||
|
||||
class BandAdmin(ModelAdmin):
|
||||
|
@ -230,7 +251,8 @@ class ModelAdminTests(TestCase):
|
|||
ma = BandAdmin(Band, self.site)
|
||||
self.assertEqual(
|
||||
list(list(ma.get_formsets_with_inlines(request))[0][0]().forms[0].fields),
|
||||
['main_band', 'opening_band', 'id', 'DELETE'])
|
||||
["main_band", "opening_band", "id", "DELETE"],
|
||||
)
|
||||
|
||||
def test_custom_formfield_override_readonly(self):
|
||||
class AdminBandForm(forms.ModelForm):
|
||||
|
@ -242,25 +264,19 @@ class ModelAdminTests(TestCase):
|
|||
|
||||
class BandAdmin(ModelAdmin):
|
||||
form = AdminBandForm
|
||||
readonly_fields = ['name']
|
||||
readonly_fields = ["name"]
|
||||
|
||||
ma = BandAdmin(Band, self.site)
|
||||
|
||||
# `name` shouldn't appear in base_fields because it's part of
|
||||
# readonly_fields.
|
||||
self.assertEqual(
|
||||
list(ma.get_form(request).base_fields),
|
||||
['bio', 'sign_date']
|
||||
)
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ["bio", "sign_date"])
|
||||
# But it should appear in get_fields()/fieldsets() so it can be
|
||||
# displayed as read-only.
|
||||
self.assertEqual(
|
||||
list(ma.get_fields(request)),
|
||||
['bio', 'sign_date', 'name']
|
||||
)
|
||||
self.assertEqual(list(ma.get_fields(request)), ["bio", "sign_date", "name"])
|
||||
self.assertEqual(
|
||||
list(ma.get_fieldsets(request)),
|
||||
[(None, {'fields': ['bio', 'sign_date', 'name']})]
|
||||
[(None, {"fields": ["bio", "sign_date", "name"]})],
|
||||
)
|
||||
|
||||
def test_custom_form_meta_exclude(self):
|
||||
|
@ -272,25 +288,25 @@ class ModelAdminTests(TestCase):
|
|||
class AdminBandForm(forms.ModelForm):
|
||||
class Meta:
|
||||
model = Band
|
||||
exclude = ['bio']
|
||||
exclude = ["bio"]
|
||||
|
||||
class BandAdmin(ModelAdmin):
|
||||
exclude = ['name']
|
||||
exclude = ["name"]
|
||||
form = AdminBandForm
|
||||
|
||||
ma = BandAdmin(Band, self.site)
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ['bio', 'sign_date'])
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ["bio", "sign_date"])
|
||||
|
||||
# With InlineModelAdmin
|
||||
class AdminConcertForm(forms.ModelForm):
|
||||
class Meta:
|
||||
model = Concert
|
||||
exclude = ['day']
|
||||
exclude = ["day"]
|
||||
|
||||
class ConcertInline(TabularInline):
|
||||
exclude = ['transport']
|
||||
exclude = ["transport"]
|
||||
form = AdminConcertForm
|
||||
fk_name = 'main_band'
|
||||
fk_name = "main_band"
|
||||
model = Concert
|
||||
|
||||
class BandAdmin(ModelAdmin):
|
||||
|
@ -299,41 +315,41 @@ class ModelAdminTests(TestCase):
|
|||
ma = BandAdmin(Band, self.site)
|
||||
self.assertEqual(
|
||||
list(list(ma.get_formsets_with_inlines(request))[0][0]().forms[0].fields),
|
||||
['main_band', 'opening_band', 'day', 'id', 'DELETE']
|
||||
["main_band", "opening_band", "day", "id", "DELETE"],
|
||||
)
|
||||
|
||||
def test_overriding_get_exclude(self):
|
||||
class BandAdmin(ModelAdmin):
|
||||
def get_exclude(self, request, obj=None):
|
||||
return ['name']
|
||||
return ["name"]
|
||||
|
||||
self.assertEqual(
|
||||
list(BandAdmin(Band, self.site).get_form(request).base_fields),
|
||||
['bio', 'sign_date']
|
||||
["bio", "sign_date"],
|
||||
)
|
||||
|
||||
def test_get_exclude_overrides_exclude(self):
|
||||
class BandAdmin(ModelAdmin):
|
||||
exclude = ['bio']
|
||||
exclude = ["bio"]
|
||||
|
||||
def get_exclude(self, request, obj=None):
|
||||
return ['name']
|
||||
return ["name"]
|
||||
|
||||
self.assertEqual(
|
||||
list(BandAdmin(Band, self.site).get_form(request).base_fields),
|
||||
['bio', 'sign_date']
|
||||
["bio", "sign_date"],
|
||||
)
|
||||
|
||||
def test_get_exclude_takes_obj(self):
|
||||
class BandAdmin(ModelAdmin):
|
||||
def get_exclude(self, request, obj=None):
|
||||
if obj:
|
||||
return ['sign_date']
|
||||
return ['name']
|
||||
return ["sign_date"]
|
||||
return ["name"]
|
||||
|
||||
self.assertEqual(
|
||||
list(BandAdmin(Band, self.site).get_form(request, self.band).base_fields),
|
||||
['name', 'bio']
|
||||
["name", "bio"],
|
||||
)
|
||||
|
||||
def test_custom_form_validation(self):
|
||||
|
@ -346,48 +362,55 @@ class ModelAdminTests(TestCase):
|
|||
form = AdminBandForm
|
||||
|
||||
ma = BandAdmin(Band, self.site)
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ['name', 'bio', 'sign_date', 'delete'])
|
||||
self.assertEqual(type(ma.get_form(request).base_fields['sign_date'].widget), AdminDateWidget)
|
||||
self.assertEqual(
|
||||
list(ma.get_form(request).base_fields),
|
||||
["name", "bio", "sign_date", "delete"],
|
||||
)
|
||||
self.assertEqual(
|
||||
type(ma.get_form(request).base_fields["sign_date"].widget), AdminDateWidget
|
||||
)
|
||||
|
||||
def test_form_exclude_kwarg_override(self):
|
||||
"""
|
||||
The `exclude` kwarg passed to `ModelAdmin.get_form()` overrides all
|
||||
other declarations (#8999).
|
||||
"""
|
||||
|
||||
class AdminBandForm(forms.ModelForm):
|
||||
class Meta:
|
||||
model = Band
|
||||
exclude = ['name']
|
||||
exclude = ["name"]
|
||||
|
||||
class BandAdmin(ModelAdmin):
|
||||
exclude = ['sign_date']
|
||||
exclude = ["sign_date"]
|
||||
form = AdminBandForm
|
||||
|
||||
def get_form(self, request, obj=None, **kwargs):
|
||||
kwargs['exclude'] = ['bio']
|
||||
kwargs["exclude"] = ["bio"]
|
||||
return super().get_form(request, obj, **kwargs)
|
||||
|
||||
ma = BandAdmin(Band, self.site)
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ['name', 'sign_date'])
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ["name", "sign_date"])
|
||||
|
||||
def test_formset_exclude_kwarg_override(self):
|
||||
"""
|
||||
The `exclude` kwarg passed to `InlineModelAdmin.get_formset()`
|
||||
overrides all other declarations (#8999).
|
||||
"""
|
||||
|
||||
class AdminConcertForm(forms.ModelForm):
|
||||
class Meta:
|
||||
model = Concert
|
||||
exclude = ['day']
|
||||
exclude = ["day"]
|
||||
|
||||
class ConcertInline(TabularInline):
|
||||
exclude = ['transport']
|
||||
exclude = ["transport"]
|
||||
form = AdminConcertForm
|
||||
fk_name = 'main_band'
|
||||
fk_name = "main_band"
|
||||
model = Concert
|
||||
|
||||
def get_formset(self, request, obj=None, **kwargs):
|
||||
kwargs['exclude'] = ['opening_band']
|
||||
kwargs["exclude"] = ["opening_band"]
|
||||
return super().get_formset(request, obj, **kwargs)
|
||||
|
||||
class BandAdmin(ModelAdmin):
|
||||
|
@ -396,22 +419,22 @@ class ModelAdminTests(TestCase):
|
|||
ma = BandAdmin(Band, self.site)
|
||||
self.assertEqual(
|
||||
list(list(ma.get_formsets_with_inlines(request))[0][0]().forms[0].fields),
|
||||
['main_band', 'day', 'transport', 'id', 'DELETE']
|
||||
["main_band", "day", "transport", "id", "DELETE"],
|
||||
)
|
||||
|
||||
def test_formset_overriding_get_exclude_with_form_fields(self):
|
||||
class AdminConcertForm(forms.ModelForm):
|
||||
class Meta:
|
||||
model = Concert
|
||||
fields = ['main_band', 'opening_band', 'day', 'transport']
|
||||
fields = ["main_band", "opening_band", "day", "transport"]
|
||||
|
||||
class ConcertInline(TabularInline):
|
||||
form = AdminConcertForm
|
||||
fk_name = 'main_band'
|
||||
fk_name = "main_band"
|
||||
model = Concert
|
||||
|
||||
def get_exclude(self, request, obj=None):
|
||||
return ['opening_band']
|
||||
return ["opening_band"]
|
||||
|
||||
class BandAdmin(ModelAdmin):
|
||||
inlines = [ConcertInline]
|
||||
|
@ -419,22 +442,22 @@ class ModelAdminTests(TestCase):
|
|||
ma = BandAdmin(Band, self.site)
|
||||
self.assertEqual(
|
||||
list(list(ma.get_formsets_with_inlines(request))[0][0]().forms[0].fields),
|
||||
['main_band', 'day', 'transport', 'id', 'DELETE']
|
||||
["main_band", "day", "transport", "id", "DELETE"],
|
||||
)
|
||||
|
||||
def test_formset_overriding_get_exclude_with_form_exclude(self):
|
||||
class AdminConcertForm(forms.ModelForm):
|
||||
class Meta:
|
||||
model = Concert
|
||||
exclude = ['day']
|
||||
exclude = ["day"]
|
||||
|
||||
class ConcertInline(TabularInline):
|
||||
form = AdminConcertForm
|
||||
fk_name = 'main_band'
|
||||
fk_name = "main_band"
|
||||
model = Concert
|
||||
|
||||
def get_exclude(self, request, obj=None):
|
||||
return ['opening_band']
|
||||
return ["opening_band"]
|
||||
|
||||
class BandAdmin(ModelAdmin):
|
||||
inlines = [ConcertInline]
|
||||
|
@ -442,7 +465,7 @@ class ModelAdminTests(TestCase):
|
|||
ma = BandAdmin(Band, self.site)
|
||||
self.assertEqual(
|
||||
list(list(ma.get_formsets_with_inlines(request))[0][0]().forms[0].fields),
|
||||
['main_band', 'day', 'transport', 'id', 'DELETE']
|
||||
["main_band", "day", "transport", "id", "DELETE"],
|
||||
)
|
||||
|
||||
def test_raw_id_fields_widget_override(self):
|
||||
|
@ -450,27 +473,35 @@ class ModelAdminTests(TestCase):
|
|||
The autocomplete_fields, raw_id_fields, and radio_fields widgets may
|
||||
overridden by specifying a widget in get_formset().
|
||||
"""
|
||||
|
||||
class ConcertInline(TabularInline):
|
||||
model = Concert
|
||||
fk_name = 'main_band'
|
||||
raw_id_fields = ('opening_band',)
|
||||
fk_name = "main_band"
|
||||
raw_id_fields = ("opening_band",)
|
||||
|
||||
def get_formset(self, request, obj=None, **kwargs):
|
||||
kwargs['widgets'] = {'opening_band': Select}
|
||||
kwargs["widgets"] = {"opening_band": Select}
|
||||
return super().get_formset(request, obj, **kwargs)
|
||||
|
||||
class BandAdmin(ModelAdmin):
|
||||
inlines = [ConcertInline]
|
||||
|
||||
ma = BandAdmin(Band, self.site)
|
||||
band_widget = list(ma.get_formsets_with_inlines(request))[0][0]().forms[0].fields['opening_band'].widget
|
||||
band_widget = (
|
||||
list(ma.get_formsets_with_inlines(request))[0][0]()
|
||||
.forms[0]
|
||||
.fields["opening_band"]
|
||||
.widget
|
||||
)
|
||||
# Without the override this would be ForeignKeyRawIdWidget.
|
||||
self.assertIsInstance(band_widget, Select)
|
||||
|
||||
def test_queryset_override(self):
|
||||
# If the queryset of a ModelChoiceField in a custom form is overridden,
|
||||
# RelatedFieldWidgetWrapper doesn't mess that up.
|
||||
band2 = Band.objects.create(name='The Beatles', bio='', sign_date=date(1962, 1, 1))
|
||||
band2 = Band.objects.create(
|
||||
name="The Beatles", bio="", sign_date=date(1962, 1, 1)
|
||||
)
|
||||
|
||||
ma = ModelAdmin(Concert, self.site)
|
||||
form = ma.get_form(request)()
|
||||
|
@ -482,13 +513,15 @@ class ModelAdminTests(TestCase):
|
|||
'<option value="" selected>---------</option>'
|
||||
'<option value="%d">The Beatles</option>'
|
||||
'<option value="%d">The Doors</option>'
|
||||
'</select></div>' % (band2.id, self.band.id)
|
||||
"</select></div>" % (band2.id, self.band.id),
|
||||
)
|
||||
|
||||
class AdminConcertForm(forms.ModelForm):
|
||||
def __init__(self, *args, **kwargs):
|
||||
super().__init__(*args, **kwargs)
|
||||
self.fields["main_band"].queryset = Band.objects.filter(name='The Doors')
|
||||
self.fields["main_band"].queryset = Band.objects.filter(
|
||||
name="The Doors"
|
||||
)
|
||||
|
||||
class ConcertAdminWithForm(ModelAdmin):
|
||||
form = AdminConcertForm
|
||||
|
@ -502,7 +535,7 @@ class ModelAdminTests(TestCase):
|
|||
'<select name="main_band" id="id_main_band" required>'
|
||||
'<option value="" selected>---------</option>'
|
||||
'<option value="%d">The Doors</option>'
|
||||
'</select></div>' % self.band.id
|
||||
"</select></div>" % self.band.id,
|
||||
)
|
||||
|
||||
def test_regression_for_ticket_15820(self):
|
||||
|
@ -510,18 +543,19 @@ class ModelAdminTests(TestCase):
|
|||
`obj` is passed from `InlineModelAdmin.get_fieldsets()` to
|
||||
`InlineModelAdmin.get_formset()`.
|
||||
"""
|
||||
|
||||
class CustomConcertForm(forms.ModelForm):
|
||||
class Meta:
|
||||
model = Concert
|
||||
fields = ['day']
|
||||
fields = ["day"]
|
||||
|
||||
class ConcertInline(TabularInline):
|
||||
model = Concert
|
||||
fk_name = 'main_band'
|
||||
fk_name = "main_band"
|
||||
|
||||
def get_formset(self, request, obj=None, **kwargs):
|
||||
if obj:
|
||||
kwargs['form'] = CustomConcertForm
|
||||
kwargs["form"] = CustomConcertForm
|
||||
return super().get_formset(request, obj, **kwargs)
|
||||
|
||||
class BandAdmin(ModelAdmin):
|
||||
|
@ -531,9 +565,13 @@ class ModelAdminTests(TestCase):
|
|||
ma = BandAdmin(Band, self.site)
|
||||
inline_instances = ma.get_inline_instances(request)
|
||||
fieldsets = list(inline_instances[0].get_fieldsets(request))
|
||||
self.assertEqual(fieldsets[0][1]['fields'], ['main_band', 'opening_band', 'day', 'transport'])
|
||||
fieldsets = list(inline_instances[0].get_fieldsets(request, inline_instances[0].model))
|
||||
self.assertEqual(fieldsets[0][1]['fields'], ['day'])
|
||||
self.assertEqual(
|
||||
fieldsets[0][1]["fields"], ["main_band", "opening_band", "day", "transport"]
|
||||
)
|
||||
fieldsets = list(
|
||||
inline_instances[0].get_fieldsets(request, inline_instances[0].model)
|
||||
)
|
||||
self.assertEqual(fieldsets[0][1]["fields"], ["day"])
|
||||
|
||||
# radio_fields behavior ###########################################
|
||||
|
||||
|
@ -546,25 +584,27 @@ class ModelAdminTests(TestCase):
|
|||
cma = ModelAdmin(Concert, self.site)
|
||||
cmafa = cma.get_form(request)
|
||||
|
||||
self.assertEqual(type(cmafa.base_fields['main_band'].widget.widget), Select)
|
||||
self.assertEqual(type(cmafa.base_fields["main_band"].widget.widget), Select)
|
||||
self.assertEqual(
|
||||
list(cmafa.base_fields['main_band'].widget.choices),
|
||||
[('', '---------'), (self.band.id, 'The Doors')])
|
||||
list(cmafa.base_fields["main_band"].widget.choices),
|
||||
[("", "---------"), (self.band.id, "The Doors")],
|
||||
)
|
||||
|
||||
self.assertEqual(type(cmafa.base_fields['opening_band'].widget.widget), Select)
|
||||
self.assertEqual(type(cmafa.base_fields["opening_band"].widget.widget), Select)
|
||||
self.assertEqual(
|
||||
list(cmafa.base_fields['opening_band'].widget.choices),
|
||||
[('', '---------'), (self.band.id, 'The Doors')]
|
||||
list(cmafa.base_fields["opening_band"].widget.choices),
|
||||
[("", "---------"), (self.band.id, "The Doors")],
|
||||
)
|
||||
self.assertEqual(type(cmafa.base_fields['day'].widget), Select)
|
||||
self.assertEqual(type(cmafa.base_fields["day"].widget), Select)
|
||||
self.assertEqual(
|
||||
list(cmafa.base_fields['day'].widget.choices),
|
||||
[('', '---------'), (1, 'Fri'), (2, 'Sat')]
|
||||
list(cmafa.base_fields["day"].widget.choices),
|
||||
[("", "---------"), (1, "Fri"), (2, "Sat")],
|
||||
)
|
||||
self.assertEqual(type(cmafa.base_fields['transport'].widget), Select)
|
||||
self.assertEqual(type(cmafa.base_fields["transport"].widget), Select)
|
||||
self.assertEqual(
|
||||
list(cmafa.base_fields['transport'].widget.choices),
|
||||
[('', '---------'), (1, 'Plane'), (2, 'Train'), (3, 'Bus')])
|
||||
list(cmafa.base_fields["transport"].widget.choices),
|
||||
[("", "---------"), (1, "Plane"), (2, "Train"), (3, "Bus")],
|
||||
)
|
||||
|
||||
def test_foreign_key_as_radio_field(self):
|
||||
# Now specify all the fields as radio_fields. Widgets should now be
|
||||
|
@ -573,67 +613,81 @@ class ModelAdminTests(TestCase):
|
|||
# 'radiolist' attr, and 'inline' as well if the field is specified HORIZONTAL.
|
||||
class ConcertAdmin(ModelAdmin):
|
||||
radio_fields = {
|
||||
'main_band': HORIZONTAL,
|
||||
'opening_band': VERTICAL,
|
||||
'day': VERTICAL,
|
||||
'transport': HORIZONTAL,
|
||||
"main_band": HORIZONTAL,
|
||||
"opening_band": VERTICAL,
|
||||
"day": VERTICAL,
|
||||
"transport": HORIZONTAL,
|
||||
}
|
||||
|
||||
cma = ConcertAdmin(Concert, self.site)
|
||||
cmafa = cma.get_form(request)
|
||||
|
||||
self.assertEqual(type(cmafa.base_fields['main_band'].widget.widget), AdminRadioSelect)
|
||||
self.assertEqual(cmafa.base_fields['main_band'].widget.attrs, {'class': 'radiolist inline'})
|
||||
self.assertEqual(
|
||||
list(cmafa.base_fields['main_band'].widget.choices),
|
||||
[(self.band.id, 'The Doors')]
|
||||
type(cmafa.base_fields["main_band"].widget.widget), AdminRadioSelect
|
||||
)
|
||||
self.assertEqual(
|
||||
cmafa.base_fields["main_band"].widget.attrs, {"class": "radiolist inline"}
|
||||
)
|
||||
self.assertEqual(
|
||||
list(cmafa.base_fields["main_band"].widget.choices),
|
||||
[(self.band.id, "The Doors")],
|
||||
)
|
||||
|
||||
self.assertEqual(type(cmafa.base_fields['opening_band'].widget.widget), AdminRadioSelect)
|
||||
self.assertEqual(cmafa.base_fields['opening_band'].widget.attrs, {'class': 'radiolist'})
|
||||
self.assertEqual(
|
||||
list(cmafa.base_fields['opening_band'].widget.choices),
|
||||
[('', 'None'), (self.band.id, 'The Doors')]
|
||||
type(cmafa.base_fields["opening_band"].widget.widget), AdminRadioSelect
|
||||
)
|
||||
self.assertEqual(type(cmafa.base_fields['day'].widget), AdminRadioSelect)
|
||||
self.assertEqual(cmafa.base_fields['day'].widget.attrs, {'class': 'radiolist'})
|
||||
self.assertEqual(list(cmafa.base_fields['day'].widget.choices), [(1, 'Fri'), (2, 'Sat')])
|
||||
|
||||
self.assertEqual(type(cmafa.base_fields['transport'].widget), AdminRadioSelect)
|
||||
self.assertEqual(cmafa.base_fields['transport'].widget.attrs, {'class': 'radiolist inline'})
|
||||
self.assertEqual(
|
||||
list(cmafa.base_fields['transport'].widget.choices),
|
||||
[('', 'None'), (1, 'Plane'), (2, 'Train'), (3, 'Bus')]
|
||||
cmafa.base_fields["opening_band"].widget.attrs, {"class": "radiolist"}
|
||||
)
|
||||
self.assertEqual(
|
||||
list(cmafa.base_fields["opening_band"].widget.choices),
|
||||
[("", "None"), (self.band.id, "The Doors")],
|
||||
)
|
||||
self.assertEqual(type(cmafa.base_fields["day"].widget), AdminRadioSelect)
|
||||
self.assertEqual(cmafa.base_fields["day"].widget.attrs, {"class": "radiolist"})
|
||||
self.assertEqual(
|
||||
list(cmafa.base_fields["day"].widget.choices), [(1, "Fri"), (2, "Sat")]
|
||||
)
|
||||
|
||||
self.assertEqual(type(cmafa.base_fields["transport"].widget), AdminRadioSelect)
|
||||
self.assertEqual(
|
||||
cmafa.base_fields["transport"].widget.attrs, {"class": "radiolist inline"}
|
||||
)
|
||||
self.assertEqual(
|
||||
list(cmafa.base_fields["transport"].widget.choices),
|
||||
[("", "None"), (1, "Plane"), (2, "Train"), (3, "Bus")],
|
||||
)
|
||||
|
||||
class AdminConcertForm(forms.ModelForm):
|
||||
class Meta:
|
||||
model = Concert
|
||||
exclude = ('transport',)
|
||||
exclude = ("transport",)
|
||||
|
||||
class ConcertAdmin(ModelAdmin):
|
||||
form = AdminConcertForm
|
||||
|
||||
ma = ConcertAdmin(Concert, self.site)
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ['main_band', 'opening_band', 'day'])
|
||||
self.assertEqual(
|
||||
list(ma.get_form(request).base_fields), ["main_band", "opening_band", "day"]
|
||||
)
|
||||
|
||||
class AdminConcertForm(forms.ModelForm):
|
||||
extra = forms.CharField()
|
||||
|
||||
class Meta:
|
||||
model = Concert
|
||||
fields = ['extra', 'transport']
|
||||
fields = ["extra", "transport"]
|
||||
|
||||
class ConcertAdmin(ModelAdmin):
|
||||
form = AdminConcertForm
|
||||
|
||||
ma = ConcertAdmin(Concert, self.site)
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ['extra', 'transport'])
|
||||
self.assertEqual(list(ma.get_form(request).base_fields), ["extra", "transport"])
|
||||
|
||||
class ConcertInline(TabularInline):
|
||||
form = AdminConcertForm
|
||||
model = Concert
|
||||
fk_name = 'main_band'
|
||||
fk_name = "main_band"
|
||||
can_delete = True
|
||||
|
||||
class BandAdmin(ModelAdmin):
|
||||
|
@ -642,30 +696,30 @@ class ModelAdminTests(TestCase):
|
|||
ma = BandAdmin(Band, self.site)
|
||||
self.assertEqual(
|
||||
list(list(ma.get_formsets_with_inlines(request))[0][0]().forms[0].fields),
|
||||
['extra', 'transport', 'id', 'DELETE', 'main_band']
|
||||
["extra", "transport", "id", "DELETE", "main_band"],
|
||||
)
|
||||
|
||||
def test_log_actions(self):
|
||||
ma = ModelAdmin(Band, self.site)
|
||||
mock_request = MockRequest()
|
||||
mock_request.user = User.objects.create(username='bill')
|
||||
mock_request.user = User.objects.create(username="bill")
|
||||
content_type = get_content_type_for_model(self.band)
|
||||
tests = (
|
||||
(ma.log_addition, ADDITION, {'added': {}}),
|
||||
(ma.log_change, CHANGE, {'changed': {'fields': ['name', 'bio']}}),
|
||||
(ma.log_addition, ADDITION, {"added": {}}),
|
||||
(ma.log_change, CHANGE, {"changed": {"fields": ["name", "bio"]}}),
|
||||
(ma.log_deletion, DELETION, str(self.band)),
|
||||
)
|
||||
for method, flag, message in tests:
|
||||
with self.subTest(name=method.__name__):
|
||||
created = method(mock_request, self.band, message)
|
||||
fetched = LogEntry.objects.filter(action_flag=flag).latest('id')
|
||||
fetched = LogEntry.objects.filter(action_flag=flag).latest("id")
|
||||
self.assertEqual(created, fetched)
|
||||
self.assertEqual(fetched.action_flag, flag)
|
||||
self.assertEqual(fetched.content_type, content_type)
|
||||
self.assertEqual(fetched.object_id, str(self.band.pk))
|
||||
self.assertEqual(fetched.user, mock_request.user)
|
||||
if flag == DELETION:
|
||||
self.assertEqual(fetched.change_message, '')
|
||||
self.assertEqual(fetched.change_message, "")
|
||||
self.assertEqual(fetched.object_repr, message)
|
||||
else:
|
||||
self.assertEqual(fetched.change_message, str(message))
|
||||
|
@ -673,37 +727,46 @@ class ModelAdminTests(TestCase):
|
|||
|
||||
def test_get_autocomplete_fields(self):
|
||||
class NameAdmin(ModelAdmin):
|
||||
search_fields = ['name']
|
||||
search_fields = ["name"]
|
||||
|
||||
class SongAdmin(ModelAdmin):
|
||||
autocomplete_fields = ['featuring']
|
||||
fields = ['featuring', 'band']
|
||||
autocomplete_fields = ["featuring"]
|
||||
fields = ["featuring", "band"]
|
||||
|
||||
class OtherSongAdmin(SongAdmin):
|
||||
def get_autocomplete_fields(self, request):
|
||||
return ['band']
|
||||
return ["band"]
|
||||
|
||||
self.site.register(Band, NameAdmin)
|
||||
try:
|
||||
# Uses autocomplete_fields if not overridden.
|
||||
model_admin = SongAdmin(Song, self.site)
|
||||
form = model_admin.get_form(request)()
|
||||
self.assertIsInstance(form.fields['featuring'].widget.widget, AutocompleteSelectMultiple)
|
||||
self.assertIsInstance(
|
||||
form.fields["featuring"].widget.widget, AutocompleteSelectMultiple
|
||||
)
|
||||
# Uses overridden get_autocomplete_fields
|
||||
model_admin = OtherSongAdmin(Song, self.site)
|
||||
form = model_admin.get_form(request)()
|
||||
self.assertIsInstance(form.fields['band'].widget.widget, AutocompleteSelect)
|
||||
self.assertIsInstance(form.fields["band"].widget.widget, AutocompleteSelect)
|
||||
finally:
|
||||
self.site.unregister(Band)
|
||||
|
||||
def test_get_deleted_objects(self):
|
||||
mock_request = MockRequest()
|
||||
mock_request.user = User.objects.create_superuser(username='bob', email='bob@test.com', password='test')
|
||||
mock_request.user = User.objects.create_superuser(
|
||||
username="bob", email="bob@test.com", password="test"
|
||||
)
|
||||
self.site.register(Band, ModelAdmin)
|
||||
ma = self.site._registry[Band]
|
||||
deletable_objects, model_count, perms_needed, protected = ma.get_deleted_objects([self.band], request)
|
||||
self.assertEqual(deletable_objects, ['Band: The Doors'])
|
||||
self.assertEqual(model_count, {'bands': 1})
|
||||
(
|
||||
deletable_objects,
|
||||
model_count,
|
||||
perms_needed,
|
||||
protected,
|
||||
) = ma.get_deleted_objects([self.band], request)
|
||||
self.assertEqual(deletable_objects, ["Band: The Doors"])
|
||||
self.assertEqual(model_count, {"bands": 1})
|
||||
self.assertEqual(perms_needed, set())
|
||||
self.assertEqual(protected, [])
|
||||
|
||||
|
@ -713,7 +776,9 @@ class ModelAdminTests(TestCase):
|
|||
for permissions checking.
|
||||
"""
|
||||
mock_request = MockRequest()
|
||||
mock_request.user = User.objects.create_superuser(username='bob', email='bob@test.com', password='test')
|
||||
mock_request.user = User.objects.create_superuser(
|
||||
username="bob", email="bob@test.com", password="test"
|
||||
)
|
||||
|
||||
class TestModelAdmin(ModelAdmin):
|
||||
def has_delete_permission(self, request, obj=None):
|
||||
|
@ -721,10 +786,15 @@ class ModelAdminTests(TestCase):
|
|||
|
||||
self.site.register(Band, TestModelAdmin)
|
||||
ma = self.site._registry[Band]
|
||||
deletable_objects, model_count, perms_needed, protected = ma.get_deleted_objects([self.band], request)
|
||||
self.assertEqual(deletable_objects, ['Band: The Doors'])
|
||||
self.assertEqual(model_count, {'bands': 1})
|
||||
self.assertEqual(perms_needed, {'band'})
|
||||
(
|
||||
deletable_objects,
|
||||
model_count,
|
||||
perms_needed,
|
||||
protected,
|
||||
) = ma.get_deleted_objects([self.band], request)
|
||||
self.assertEqual(deletable_objects, ["Band: The Doors"])
|
||||
self.assertEqual(model_count, {"bands": 1})
|
||||
self.assertEqual(perms_needed, {"band"})
|
||||
self.assertEqual(protected, [])
|
||||
|
||||
def test_modeladmin_repr(self):
|
||||
|
@ -736,26 +806,25 @@ class ModelAdminTests(TestCase):
|
|||
|
||||
|
||||
class ModelAdminPermissionTests(SimpleTestCase):
|
||||
|
||||
class MockUser:
|
||||
def has_module_perms(self, app_label):
|
||||
return app_label == 'modeladmin'
|
||||
return app_label == "modeladmin"
|
||||
|
||||
class MockViewUser(MockUser):
|
||||
def has_perm(self, perm, obj=None):
|
||||
return perm == 'modeladmin.view_band'
|
||||
return perm == "modeladmin.view_band"
|
||||
|
||||
class MockAddUser(MockUser):
|
||||
def has_perm(self, perm, obj=None):
|
||||
return perm == 'modeladmin.add_band'
|
||||
return perm == "modeladmin.add_band"
|
||||
|
||||
class MockChangeUser(MockUser):
|
||||
def has_perm(self, perm, obj=None):
|
||||
return perm == 'modeladmin.change_band'
|
||||
return perm == "modeladmin.change_band"
|
||||
|
||||
class MockDeleteUser(MockUser):
|
||||
def has_perm(self, perm, obj=None):
|
||||
return perm == 'modeladmin.delete_band'
|
||||
return perm == "modeladmin.delete_band"
|
||||
|
||||
def test_has_view_permission(self):
|
||||
"""
|
||||
|
@ -803,7 +872,7 @@ class ModelAdminPermissionTests(SimpleTestCase):
|
|||
request = MockRequest()
|
||||
request.user = self.MockAddUser()
|
||||
self.assertEqual(ma.get_inline_instances(request), [])
|
||||
band = Band(name='The Doors', bio='', sign_date=date(1965, 1, 1))
|
||||
band = Band(name="The Doors", bio="", sign_date=date(1965, 1, 1))
|
||||
inline_instances = ma.get_inline_instances(request, band)
|
||||
self.assertEqual(len(inline_instances), 1)
|
||||
self.assertIsInstance(inline_instances[0], ConcertInline)
|
||||
|
@ -857,7 +926,7 @@ class ModelAdminPermissionTests(SimpleTestCase):
|
|||
self.assertTrue(ma.has_module_permission(request))
|
||||
|
||||
original_app_label = ma.opts.app_label
|
||||
ma.opts.app_label = 'anotherapp'
|
||||
ma.opts.app_label = "anotherapp"
|
||||
try:
|
||||
request.user = self.MockViewUser()
|
||||
self.assertIs(ma.has_module_permission(request), False)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue