admin.py 34 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160
  1. import datetime
  2. import os
  3. import tempfile
  4. from io import StringIO
  5. from wsgiref.util import FileWrapper
  6. from django import forms
  7. from django.contrib import admin
  8. from django.contrib.admin import BooleanFieldListFilter
  9. from django.contrib.admin.views.main import ChangeList
  10. from django.contrib.auth.admin import GroupAdmin, UserAdmin
  11. from django.contrib.auth.models import Group, User
  12. from django.core.exceptions import ValidationError
  13. from django.core.files.storage import FileSystemStorage
  14. from django.core.mail import EmailMessage
  15. from django.db import models
  16. from django.forms.models import BaseModelFormSet
  17. from django.http import HttpResponse, StreamingHttpResponse
  18. from django.urls import path
  19. from django.utils.html import format_html
  20. from django.utils.safestring import mark_safe
  21. from .forms import MediaActionForm
  22. from .models import (
  23. Actor, AdminOrderedAdminMethod, AdminOrderedCallable, AdminOrderedField,
  24. AdminOrderedModelMethod, Album, Answer, Answer2, Article, BarAccount, Book,
  25. Bookmark, Category, Chapter, ChapterXtra1, Child, ChildOfReferer, Choice,
  26. City, Collector, Color, Color2, ComplexSortedPerson, CoverLetter,
  27. CustomArticle, CyclicOne, CyclicTwo, DependentChild, DooHickey, EmptyModel,
  28. EmptyModelHidden, EmptyModelMixin, EmptyModelVisible, ExplicitlyProvidedPK,
  29. ExternalSubscriber, Fabric, FancyDoodad, FieldOverridePost,
  30. FilteredManager, FooAccount, FoodDelivery, FunkyTag, Gadget, Gallery,
  31. GenRelReference, Grommet, ImplicitlyGeneratedPK, Ingredient,
  32. InlineReference, InlineReferer, Inquisition, Language, Link,
  33. MainPrepopulated, ModelWithStringPrimaryKey, NotReferenced, OldSubscriber,
  34. OtherStory, Paper, Parent, ParentWithDependentChildren, ParentWithUUIDPK,
  35. Person, Persona, Picture, Pizza, Plot, PlotDetails, PlotProxy,
  36. PluggableSearchPerson, Podcast, Post, PrePopulatedPost,
  37. PrePopulatedPostLargeSlug, PrePopulatedSubPost, Promo, Question,
  38. ReadablePizza, ReadOnlyPizza, Recipe, Recommendation, Recommender,
  39. ReferencedByGenRel, ReferencedByInline, ReferencedByParent,
  40. RelatedPrepopulated, RelatedWithUUIDPKModel, Report, Reservation,
  41. Restaurant, RowLevelChangePermissionModel, Section, ShortMessage, Simple,
  42. Sketch, State, Story, StumpJoke, Subscriber, SuperVillain, Telegram, Thing,
  43. Topping, UnchangeableObject, UndeletableObject, UnorderedObject,
  44. UserMessenger, UserProxy, Villain, Vodcast, Whatsit, Widget, Worker,
  45. WorkHour,
  46. )
  47. def callable_year(dt_value):
  48. try:
  49. return dt_value.year
  50. except AttributeError:
  51. return None
  52. callable_year.admin_order_field = 'date'
  53. class ArticleInline(admin.TabularInline):
  54. model = Article
  55. fk_name = 'section'
  56. prepopulated_fields = {
  57. 'title': ('content',)
  58. }
  59. fieldsets = (
  60. ('Some fields', {
  61. 'classes': ('collapse',),
  62. 'fields': ('title', 'content')
  63. }),
  64. ('Some other fields', {
  65. 'classes': ('wide',),
  66. 'fields': ('date', 'section')
  67. })
  68. )
  69. class ChapterInline(admin.TabularInline):
  70. model = Chapter
  71. class ChapterXtra1Admin(admin.ModelAdmin):
  72. list_filter = (
  73. 'chap',
  74. 'chap__title',
  75. 'chap__book',
  76. 'chap__book__name',
  77. 'chap__book__promo',
  78. 'chap__book__promo__name',
  79. 'guest_author__promo__book',
  80. )
  81. class ArticleForm(forms.ModelForm):
  82. extra_form_field = forms.BooleanField(required=False)
  83. class Meta:
  84. fields = '__all__'
  85. model = Article
  86. class ArticleAdmin(admin.ModelAdmin):
  87. list_display = (
  88. 'content', 'date', callable_year, 'model_year', 'modeladmin_year',
  89. 'model_year_reversed', 'section', lambda obj: obj.title,
  90. 'order_by_expression',
  91. )
  92. list_editable = ('section',)
  93. list_filter = ('date', 'section')
  94. autocomplete_fields = ('section',)
  95. view_on_site = False
  96. form = ArticleForm
  97. fieldsets = (
  98. ('Some fields', {
  99. 'classes': ('collapse',),
  100. 'fields': ('title', 'content', 'extra_form_field'),
  101. }),
  102. ('Some other fields', {
  103. 'classes': ('wide',),
  104. 'fields': ('date', 'section', 'sub_section')
  105. })
  106. )
  107. def order_by_expression(self, obj):
  108. return obj.model_year
  109. # This ordering isn't particularly useful but shows that expressions can
  110. # be used for admin_order_field.
  111. order_by_expression.admin_order_field = models.F('date') + datetime.timedelta(days=3)
  112. def changelist_view(self, request):
  113. return super().changelist_view(request, extra_context={'extra_var': 'Hello!'})
  114. def modeladmin_year(self, obj):
  115. return obj.date.year
  116. modeladmin_year.admin_order_field = 'date'
  117. modeladmin_year.short_description = None
  118. def delete_model(self, request, obj):
  119. EmailMessage(
  120. 'Greetings from a deleted object',
  121. 'I hereby inform you that some user deleted me',
  122. 'from@example.com',
  123. ['to@example.com']
  124. ).send()
  125. return super().delete_model(request, obj)
  126. def save_model(self, request, obj, form, change=True):
  127. EmailMessage(
  128. 'Greetings from a created object',
  129. 'I hereby inform you that some user created me',
  130. 'from@example.com',
  131. ['to@example.com']
  132. ).send()
  133. return super().save_model(request, obj, form, change)
  134. class ArticleAdmin2(admin.ModelAdmin):
  135. def has_module_permission(self, request):
  136. return False
  137. class RowLevelChangePermissionModelAdmin(admin.ModelAdmin):
  138. def has_change_permission(self, request, obj=None):
  139. """ Only allow changing objects with even id number """
  140. return request.user.is_staff and (obj is not None) and (obj.id % 2 == 0)
  141. def has_view_permission(self, request, obj=None):
  142. """Only allow viewing objects if id is a multiple of 3."""
  143. return request.user.is_staff and obj is not None and obj.id % 3 == 0
  144. class CustomArticleAdmin(admin.ModelAdmin):
  145. """
  146. Tests various hooks for using custom templates and contexts.
  147. """
  148. change_list_template = 'custom_admin/change_list.html'
  149. change_form_template = 'custom_admin/change_form.html'
  150. add_form_template = 'custom_admin/add_form.html'
  151. object_history_template = 'custom_admin/object_history.html'
  152. delete_confirmation_template = 'custom_admin/delete_confirmation.html'
  153. delete_selected_confirmation_template = 'custom_admin/delete_selected_confirmation.html'
  154. popup_response_template = 'custom_admin/popup_response.html'
  155. def changelist_view(self, request):
  156. return super().changelist_view(request, extra_context={'extra_var': 'Hello!'})
  157. class ThingAdmin(admin.ModelAdmin):
  158. list_filter = ('color', 'color__warm', 'color__value', 'pub_date')
  159. class InquisitionAdmin(admin.ModelAdmin):
  160. list_display = ('leader', 'country', 'expected', 'sketch')
  161. def sketch(self, obj):
  162. # A method with the same name as a reverse accessor.
  163. return 'list-display-sketch'
  164. class SketchAdmin(admin.ModelAdmin):
  165. raw_id_fields = ('inquisition', 'defendant0', 'defendant1')
  166. class FabricAdmin(admin.ModelAdmin):
  167. list_display = ('surface',)
  168. list_filter = ('surface',)
  169. class BasePersonModelFormSet(BaseModelFormSet):
  170. def clean(self):
  171. for person_dict in self.cleaned_data:
  172. person = person_dict.get('id')
  173. alive = person_dict.get('alive')
  174. if person and alive and person.name == "Grace Hopper":
  175. raise forms.ValidationError("Grace is not a Zombie")
  176. class PersonAdmin(admin.ModelAdmin):
  177. list_display = ('name', 'gender', 'alive')
  178. list_editable = ('gender', 'alive')
  179. list_filter = ('gender',)
  180. search_fields = ('^name',)
  181. save_as = True
  182. def get_changelist_formset(self, request, **kwargs):
  183. return super().get_changelist_formset(request, formset=BasePersonModelFormSet, **kwargs)
  184. def get_queryset(self, request):
  185. # Order by a field that isn't in list display, to be able to test
  186. # whether ordering is preserved.
  187. return super().get_queryset(request).order_by('age')
  188. class FooAccountAdmin(admin.StackedInline):
  189. model = FooAccount
  190. extra = 1
  191. class BarAccountAdmin(admin.StackedInline):
  192. model = BarAccount
  193. extra = 1
  194. class PersonaAdmin(admin.ModelAdmin):
  195. inlines = (
  196. FooAccountAdmin,
  197. BarAccountAdmin
  198. )
  199. class SubscriberAdmin(admin.ModelAdmin):
  200. actions = ['mail_admin']
  201. action_form = MediaActionForm
  202. def delete_queryset(self, request, queryset):
  203. SubscriberAdmin.overridden = True
  204. super().delete_queryset(request, queryset)
  205. def mail_admin(self, request, selected):
  206. EmailMessage(
  207. 'Greetings from a ModelAdmin action',
  208. 'This is the test email from an admin action',
  209. 'from@example.com',
  210. ['to@example.com']
  211. ).send()
  212. def external_mail(modeladmin, request, selected):
  213. EmailMessage(
  214. 'Greetings from a function action',
  215. 'This is the test email from a function action',
  216. 'from@example.com',
  217. ['to@example.com']
  218. ).send()
  219. external_mail.short_description = 'External mail (Another awesome action)'
  220. def redirect_to(modeladmin, request, selected):
  221. from django.http import HttpResponseRedirect
  222. return HttpResponseRedirect('/some-where-else/')
  223. redirect_to.short_description = 'Redirect to (Awesome action)'
  224. def download(modeladmin, request, selected):
  225. buf = StringIO('This is the content of the file')
  226. return StreamingHttpResponse(FileWrapper(buf))
  227. download.short_description = 'Download subscription'
  228. def no_perm(modeladmin, request, selected):
  229. return HttpResponse(content='No permission to perform this action', status=403)
  230. no_perm.short_description = 'No permission to run'
  231. class ExternalSubscriberAdmin(admin.ModelAdmin):
  232. actions = [redirect_to, external_mail, download, no_perm]
  233. class PodcastAdmin(admin.ModelAdmin):
  234. list_display = ('name', 'release_date')
  235. list_editable = ('release_date',)
  236. date_hierarchy = 'release_date'
  237. ordering = ('name',)
  238. class VodcastAdmin(admin.ModelAdmin):
  239. list_display = ('name', 'released')
  240. list_editable = ('released',)
  241. ordering = ('name',)
  242. class ChildInline(admin.StackedInline):
  243. model = Child
  244. class ParentAdmin(admin.ModelAdmin):
  245. model = Parent
  246. inlines = [ChildInline]
  247. save_as = True
  248. list_display = ('id', 'name',)
  249. list_display_links = ('id',)
  250. list_editable = ('name',)
  251. def save_related(self, request, form, formsets, change):
  252. super().save_related(request, form, formsets, change)
  253. first_name, last_name = form.instance.name.split()
  254. for child in form.instance.child_set.all():
  255. if len(child.name.split()) < 2:
  256. child.name = child.name + ' ' + last_name
  257. child.save()
  258. class EmptyModelAdmin(admin.ModelAdmin):
  259. def get_queryset(self, request):
  260. return super().get_queryset(request).filter(pk__gt=1)
  261. class OldSubscriberAdmin(admin.ModelAdmin):
  262. actions = None
  263. temp_storage = FileSystemStorage(tempfile.mkdtemp())
  264. UPLOAD_TO = os.path.join(temp_storage.location, 'test_upload')
  265. class PictureInline(admin.TabularInline):
  266. model = Picture
  267. extra = 1
  268. class GalleryAdmin(admin.ModelAdmin):
  269. inlines = [PictureInline]
  270. class PictureAdmin(admin.ModelAdmin):
  271. pass
  272. class LanguageAdmin(admin.ModelAdmin):
  273. list_display = ['iso', 'shortlist', 'english_name', 'name']
  274. list_editable = ['shortlist']
  275. class RecommendationAdmin(admin.ModelAdmin):
  276. show_full_result_count = False
  277. search_fields = ('=titletranslation__text', '=the_recommender__titletranslation__text',)
  278. class WidgetInline(admin.StackedInline):
  279. model = Widget
  280. class DooHickeyInline(admin.StackedInline):
  281. model = DooHickey
  282. class GrommetInline(admin.StackedInline):
  283. model = Grommet
  284. class WhatsitInline(admin.StackedInline):
  285. model = Whatsit
  286. class FancyDoodadInline(admin.StackedInline):
  287. model = FancyDoodad
  288. class CategoryAdmin(admin.ModelAdmin):
  289. list_display = ('id', 'collector', 'order')
  290. list_editable = ('order',)
  291. class CategoryInline(admin.StackedInline):
  292. model = Category
  293. class CollectorAdmin(admin.ModelAdmin):
  294. inlines = [
  295. WidgetInline, DooHickeyInline, GrommetInline, WhatsitInline,
  296. FancyDoodadInline, CategoryInline
  297. ]
  298. class LinkInline(admin.TabularInline):
  299. model = Link
  300. extra = 1
  301. readonly_fields = ("posted", "multiline", "readonly_link_content")
  302. def multiline(self, instance):
  303. return "InlineMultiline\ntest\nstring"
  304. class SubPostInline(admin.TabularInline):
  305. model = PrePopulatedSubPost
  306. prepopulated_fields = {
  307. 'subslug': ('subtitle',)
  308. }
  309. def get_readonly_fields(self, request, obj=None):
  310. if obj and obj.published:
  311. return ('subslug',)
  312. return self.readonly_fields
  313. def get_prepopulated_fields(self, request, obj=None):
  314. if obj and obj.published:
  315. return {}
  316. return self.prepopulated_fields
  317. class PrePopulatedPostAdmin(admin.ModelAdmin):
  318. list_display = ['title', 'slug']
  319. prepopulated_fields = {
  320. 'slug': ('title',)
  321. }
  322. inlines = [SubPostInline]
  323. def get_readonly_fields(self, request, obj=None):
  324. if obj and obj.published:
  325. return ('slug',)
  326. return self.readonly_fields
  327. def get_prepopulated_fields(self, request, obj=None):
  328. if obj and obj.published:
  329. return {}
  330. return self.prepopulated_fields
  331. class PrePopulatedPostReadOnlyAdmin(admin.ModelAdmin):
  332. prepopulated_fields = {'slug': ('title',)}
  333. def has_change_permission(self, *args, **kwargs):
  334. return False
  335. class PostAdmin(admin.ModelAdmin):
  336. list_display = ['title', 'public']
  337. readonly_fields = (
  338. 'posted', 'awesomeness_level', 'coolness', 'value',
  339. 'multiline', 'multiline_html', lambda obj: "foo",
  340. 'readonly_content',
  341. )
  342. inlines = [
  343. LinkInline
  344. ]
  345. def coolness(self, instance):
  346. if instance.pk:
  347. return "%d amount of cool." % instance.pk
  348. else:
  349. return "Unknown coolness."
  350. def value(self, instance):
  351. return 1000
  352. value.short_description = 'Value in $US'
  353. def multiline(self, instance):
  354. return "Multiline\ntest\nstring"
  355. def multiline_html(self, instance):
  356. return mark_safe("Multiline<br>\nhtml<br>\ncontent")
  357. class FieldOverridePostForm(forms.ModelForm):
  358. model = FieldOverridePost
  359. class Meta:
  360. help_texts = {
  361. 'posted': 'Overridden help text for the date',
  362. }
  363. labels = {
  364. 'public': 'Overridden public label',
  365. }
  366. class FieldOverridePostAdmin(PostAdmin):
  367. form = FieldOverridePostForm
  368. class CustomChangeList(ChangeList):
  369. def get_queryset(self, request):
  370. return self.root_queryset.order_by('pk').filter(pk=9999) # Doesn't exist
  371. class GadgetAdmin(admin.ModelAdmin):
  372. def get_changelist(self, request, **kwargs):
  373. return CustomChangeList
  374. class ToppingAdmin(admin.ModelAdmin):
  375. readonly_fields = ('pizzas',)
  376. class PizzaAdmin(admin.ModelAdmin):
  377. readonly_fields = ('toppings',)
  378. class StudentAdmin(admin.ModelAdmin):
  379. search_fields = ('name',)
  380. class ReadOnlyPizzaAdmin(admin.ModelAdmin):
  381. readonly_fields = ('name', 'toppings')
  382. def has_add_permission(self, request):
  383. return False
  384. def has_change_permission(self, request, obj=None):
  385. return True
  386. def has_delete_permission(self, request, obj=None):
  387. return True
  388. class WorkHourAdmin(admin.ModelAdmin):
  389. list_display = ('datum', 'employee')
  390. list_filter = ('employee',)
  391. class FoodDeliveryAdmin(admin.ModelAdmin):
  392. list_display = ('reference', 'driver', 'restaurant')
  393. list_editable = ('driver', 'restaurant')
  394. class CoverLetterAdmin(admin.ModelAdmin):
  395. """
  396. A ModelAdmin with a custom get_queryset() method that uses defer(), to test
  397. verbose_name display in messages shown after adding/editing CoverLetter
  398. instances. Note that the CoverLetter model defines a __str__ method.
  399. For testing fix for ticket #14529.
  400. """
  401. def get_queryset(self, request):
  402. return super().get_queryset(request).defer('date_written')
  403. class PaperAdmin(admin.ModelAdmin):
  404. """
  405. A ModelAdmin with a custom get_queryset() method that uses only(), to test
  406. verbose_name display in messages shown after adding/editing Paper
  407. instances.
  408. For testing fix for ticket #14529.
  409. """
  410. def get_queryset(self, request):
  411. return super().get_queryset(request).only('title')
  412. class ShortMessageAdmin(admin.ModelAdmin):
  413. """
  414. A ModelAdmin with a custom get_queryset() method that uses defer(), to test
  415. verbose_name display in messages shown after adding/editing ShortMessage
  416. instances.
  417. For testing fix for ticket #14529.
  418. """
  419. def get_queryset(self, request):
  420. return super().get_queryset(request).defer('timestamp')
  421. class TelegramAdmin(admin.ModelAdmin):
  422. """
  423. A ModelAdmin with a custom get_queryset() method that uses only(), to test
  424. verbose_name display in messages shown after adding/editing Telegram
  425. instances. Note that the Telegram model defines a __str__ method.
  426. For testing fix for ticket #14529.
  427. """
  428. def get_queryset(self, request):
  429. return super().get_queryset(request).only('title')
  430. class StoryForm(forms.ModelForm):
  431. class Meta:
  432. widgets = {'title': forms.HiddenInput}
  433. class StoryAdmin(admin.ModelAdmin):
  434. list_display = ('id', 'title', 'content')
  435. list_display_links = ('title',) # 'id' not in list_display_links
  436. list_editable = ('content',)
  437. form = StoryForm
  438. ordering = ['-id']
  439. class OtherStoryAdmin(admin.ModelAdmin):
  440. list_display = ('id', 'title', 'content')
  441. list_display_links = ('title', 'id') # 'id' in list_display_links
  442. list_editable = ('content',)
  443. ordering = ['-id']
  444. class ComplexSortedPersonAdmin(admin.ModelAdmin):
  445. list_display = ('name', 'age', 'is_employee', 'colored_name')
  446. ordering = ('name',)
  447. def colored_name(self, obj):
  448. return format_html('<span style="color: #ff00ff;">{}</span>', obj.name)
  449. colored_name.admin_order_field = 'name'
  450. class PluggableSearchPersonAdmin(admin.ModelAdmin):
  451. list_display = ('name', 'age')
  452. search_fields = ('name',)
  453. def get_search_results(self, request, queryset, search_term):
  454. queryset, use_distinct = super().get_search_results(request, queryset, search_term)
  455. try:
  456. search_term_as_int = int(search_term)
  457. except ValueError:
  458. pass
  459. else:
  460. queryset |= self.model.objects.filter(age=search_term_as_int)
  461. return queryset, use_distinct
  462. class AlbumAdmin(admin.ModelAdmin):
  463. list_filter = ['title']
  464. class QuestionAdmin(admin.ModelAdmin):
  465. ordering = ['-posted']
  466. search_fields = ['question']
  467. autocomplete_fields = ['related_questions']
  468. class AnswerAdmin(admin.ModelAdmin):
  469. autocomplete_fields = ['question']
  470. class PrePopulatedPostLargeSlugAdmin(admin.ModelAdmin):
  471. prepopulated_fields = {
  472. 'slug': ('title',)
  473. }
  474. class AdminOrderedFieldAdmin(admin.ModelAdmin):
  475. ordering = ('order',)
  476. list_display = ('stuff', 'order')
  477. class AdminOrderedModelMethodAdmin(admin.ModelAdmin):
  478. ordering = ('order',)
  479. list_display = ('stuff', 'some_order')
  480. class AdminOrderedAdminMethodAdmin(admin.ModelAdmin):
  481. def some_admin_order(self, obj):
  482. return obj.order
  483. some_admin_order.admin_order_field = 'order'
  484. ordering = ('order',)
  485. list_display = ('stuff', 'some_admin_order')
  486. def admin_ordered_callable(obj):
  487. return obj.order
  488. admin_ordered_callable.admin_order_field = 'order'
  489. class AdminOrderedCallableAdmin(admin.ModelAdmin):
  490. ordering = ('order',)
  491. list_display = ('stuff', admin_ordered_callable)
  492. class ReportAdmin(admin.ModelAdmin):
  493. def extra(self, request):
  494. return HttpResponse()
  495. def get_urls(self):
  496. # Corner case: Don't call parent implementation
  497. return [path('extra/', self.extra, name='cable_extra')]
  498. class CustomTemplateBooleanFieldListFilter(BooleanFieldListFilter):
  499. template = 'custom_filter_template.html'
  500. class CustomTemplateFilterColorAdmin(admin.ModelAdmin):
  501. list_filter = (('warm', CustomTemplateBooleanFieldListFilter),)
  502. # For Selenium Prepopulated tests -------------------------------------
  503. class RelatedPrepopulatedInline1(admin.StackedInline):
  504. fieldsets = (
  505. (None, {
  506. 'fields': (
  507. ('fk', 'm2m'),
  508. ('pubdate', 'status'),
  509. ('name', 'slug1', 'slug2',),
  510. ),
  511. }),
  512. )
  513. formfield_overrides = {models.CharField: {'strip': False}}
  514. model = RelatedPrepopulated
  515. extra = 1
  516. autocomplete_fields = ['fk', 'm2m']
  517. prepopulated_fields = {
  518. 'slug1': ['name', 'pubdate'],
  519. 'slug2': ['status', 'name'],
  520. }
  521. class RelatedPrepopulatedInline2(admin.TabularInline):
  522. model = RelatedPrepopulated
  523. extra = 1
  524. autocomplete_fields = ['fk', 'm2m']
  525. prepopulated_fields = {
  526. 'slug1': ['name', 'pubdate'],
  527. 'slug2': ['status', 'name'],
  528. }
  529. class RelatedPrepopulatedInline3(admin.TabularInline):
  530. model = RelatedPrepopulated
  531. extra = 0
  532. autocomplete_fields = ['fk', 'm2m']
  533. class MainPrepopulatedAdmin(admin.ModelAdmin):
  534. inlines = [RelatedPrepopulatedInline1, RelatedPrepopulatedInline2, RelatedPrepopulatedInline3]
  535. fieldsets = (
  536. (None, {
  537. 'fields': (('pubdate', 'status'), ('name', 'slug1', 'slug2', 'slug3'))
  538. }),
  539. )
  540. formfield_overrides = {models.CharField: {'strip': False}}
  541. prepopulated_fields = {
  542. 'slug1': ['name', 'pubdate'],
  543. 'slug2': ['status', 'name'],
  544. 'slug3': ['name'],
  545. }
  546. class UnorderedObjectAdmin(admin.ModelAdmin):
  547. list_display = ['id', 'name']
  548. list_display_links = ['id']
  549. list_editable = ['name']
  550. list_per_page = 2
  551. class UndeletableObjectAdmin(admin.ModelAdmin):
  552. def change_view(self, *args, **kwargs):
  553. kwargs['extra_context'] = {'show_delete': False}
  554. return super().change_view(*args, **kwargs)
  555. class UnchangeableObjectAdmin(admin.ModelAdmin):
  556. def get_urls(self):
  557. # Disable change_view, but leave other urls untouched
  558. urlpatterns = super().get_urls()
  559. return [p for p in urlpatterns if p.name and not p.name.endswith("_change")]
  560. def callable_on_unknown(obj):
  561. return obj.unknown
  562. class AttributeErrorRaisingAdmin(admin.ModelAdmin):
  563. list_display = [callable_on_unknown]
  564. class CustomManagerAdmin(admin.ModelAdmin):
  565. def get_queryset(self, request):
  566. return FilteredManager.objects
  567. class MessageTestingAdmin(admin.ModelAdmin):
  568. actions = ["message_debug", "message_info", "message_success",
  569. "message_warning", "message_error", "message_extra_tags"]
  570. def message_debug(self, request, selected):
  571. self.message_user(request, "Test debug", level="debug")
  572. def message_info(self, request, selected):
  573. self.message_user(request, "Test info", level="info")
  574. def message_success(self, request, selected):
  575. self.message_user(request, "Test success", level="success")
  576. def message_warning(self, request, selected):
  577. self.message_user(request, "Test warning", level="warning")
  578. def message_error(self, request, selected):
  579. self.message_user(request, "Test error", level="error")
  580. def message_extra_tags(self, request, selected):
  581. self.message_user(request, "Test tags", extra_tags="extra_tag")
  582. class ChoiceList(admin.ModelAdmin):
  583. list_display = ['choice']
  584. readonly_fields = ['choice']
  585. fields = ['choice']
  586. class DependentChildAdminForm(forms.ModelForm):
  587. """
  588. Issue #20522
  589. Form to test child dependency on parent object's validation
  590. """
  591. def clean(self):
  592. parent = self.cleaned_data.get('parent')
  593. if parent.family_name and parent.family_name != self.cleaned_data.get('family_name'):
  594. raise ValidationError("Children must share a family name with their parents " +
  595. "in this contrived test case")
  596. return super().clean()
  597. class DependentChildInline(admin.TabularInline):
  598. model = DependentChild
  599. form = DependentChildAdminForm
  600. class ParentWithDependentChildrenAdmin(admin.ModelAdmin):
  601. inlines = [DependentChildInline]
  602. # Tests for ticket 11277 ----------------------------------
  603. class FormWithoutHiddenField(forms.ModelForm):
  604. first = forms.CharField()
  605. second = forms.CharField()
  606. class FormWithoutVisibleField(forms.ModelForm):
  607. first = forms.CharField(widget=forms.HiddenInput)
  608. second = forms.CharField(widget=forms.HiddenInput)
  609. class FormWithVisibleAndHiddenField(forms.ModelForm):
  610. first = forms.CharField(widget=forms.HiddenInput)
  611. second = forms.CharField()
  612. class EmptyModelVisibleAdmin(admin.ModelAdmin):
  613. form = FormWithoutHiddenField
  614. fieldsets = (
  615. (None, {
  616. 'fields': (('first', 'second'),),
  617. }),
  618. )
  619. class EmptyModelHiddenAdmin(admin.ModelAdmin):
  620. form = FormWithoutVisibleField
  621. fieldsets = EmptyModelVisibleAdmin.fieldsets
  622. class EmptyModelMixinAdmin(admin.ModelAdmin):
  623. form = FormWithVisibleAndHiddenField
  624. fieldsets = EmptyModelVisibleAdmin.fieldsets
  625. class CityInlineAdmin(admin.TabularInline):
  626. model = City
  627. view_on_site = False
  628. class StateAdmin(admin.ModelAdmin):
  629. inlines = [CityInlineAdmin]
  630. class RestaurantInlineAdmin(admin.TabularInline):
  631. model = Restaurant
  632. view_on_site = True
  633. class CityAdmin(admin.ModelAdmin):
  634. inlines = [RestaurantInlineAdmin]
  635. view_on_site = True
  636. class WorkerAdmin(admin.ModelAdmin):
  637. def view_on_site(self, obj):
  638. return '/worker/%s/%s/' % (obj.surname, obj.name)
  639. class WorkerInlineAdmin(admin.TabularInline):
  640. model = Worker
  641. def view_on_site(self, obj):
  642. return '/worker_inline/%s/%s/' % (obj.surname, obj.name)
  643. class RestaurantAdmin(admin.ModelAdmin):
  644. inlines = [WorkerInlineAdmin]
  645. view_on_site = False
  646. def get_changeform_initial_data(self, request):
  647. return {'name': 'overridden_value'}
  648. class FunkyTagAdmin(admin.ModelAdmin):
  649. list_display = ('name', 'content_object')
  650. class InlineReferenceInline(admin.TabularInline):
  651. model = InlineReference
  652. class InlineRefererAdmin(admin.ModelAdmin):
  653. inlines = [InlineReferenceInline]
  654. class PlotReadonlyAdmin(admin.ModelAdmin):
  655. readonly_fields = ('plotdetails',)
  656. class GetFormsetsArgumentCheckingAdmin(admin.ModelAdmin):
  657. fields = ['name']
  658. def add_view(self, request, *args, **kwargs):
  659. request.is_add_view = True
  660. return super().add_view(request, *args, **kwargs)
  661. def change_view(self, request, *args, **kwargs):
  662. request.is_add_view = False
  663. return super().change_view(request, *args, **kwargs)
  664. def get_formsets_with_inlines(self, request, obj=None):
  665. if request.is_add_view and obj is not None:
  666. raise Exception("'obj' passed to get_formsets_with_inlines wasn't None during add_view")
  667. if not request.is_add_view and obj is None:
  668. raise Exception("'obj' passed to get_formsets_with_inlines was None during change_view")
  669. return super().get_formsets_with_inlines(request, obj)
  670. site = admin.AdminSite(name="admin")
  671. site.site_url = '/my-site-url/'
  672. site.register(Article, ArticleAdmin)
  673. site.register(CustomArticle, CustomArticleAdmin)
  674. site.register(
  675. Section, save_as=True, inlines=[ArticleInline],
  676. readonly_fields=['name_property'], search_fields=['name'],
  677. )
  678. site.register(ModelWithStringPrimaryKey)
  679. site.register(Color)
  680. site.register(Thing, ThingAdmin)
  681. site.register(Actor)
  682. site.register(Inquisition, InquisitionAdmin)
  683. site.register(Sketch, SketchAdmin)
  684. site.register(Person, PersonAdmin)
  685. site.register(Persona, PersonaAdmin)
  686. site.register(Subscriber, SubscriberAdmin)
  687. site.register(ExternalSubscriber, ExternalSubscriberAdmin)
  688. site.register(OldSubscriber, OldSubscriberAdmin)
  689. site.register(Podcast, PodcastAdmin)
  690. site.register(Vodcast, VodcastAdmin)
  691. site.register(Parent, ParentAdmin)
  692. site.register(EmptyModel, EmptyModelAdmin)
  693. site.register(Fabric, FabricAdmin)
  694. site.register(Gallery, GalleryAdmin)
  695. site.register(Picture, PictureAdmin)
  696. site.register(Language, LanguageAdmin)
  697. site.register(Recommendation, RecommendationAdmin)
  698. site.register(Recommender)
  699. site.register(Collector, CollectorAdmin)
  700. site.register(Category, CategoryAdmin)
  701. site.register(Post, PostAdmin)
  702. site.register(FieldOverridePost, FieldOverridePostAdmin)
  703. site.register(Gadget, GadgetAdmin)
  704. site.register(Villain)
  705. site.register(SuperVillain)
  706. site.register(Plot)
  707. site.register(PlotDetails)
  708. site.register(PlotProxy, PlotReadonlyAdmin)
  709. site.register(Bookmark)
  710. site.register(CyclicOne)
  711. site.register(CyclicTwo)
  712. site.register(WorkHour, WorkHourAdmin)
  713. site.register(Reservation)
  714. site.register(FoodDelivery, FoodDeliveryAdmin)
  715. site.register(RowLevelChangePermissionModel, RowLevelChangePermissionModelAdmin)
  716. site.register(Paper, PaperAdmin)
  717. site.register(CoverLetter, CoverLetterAdmin)
  718. site.register(ShortMessage, ShortMessageAdmin)
  719. site.register(Telegram, TelegramAdmin)
  720. site.register(Story, StoryAdmin)
  721. site.register(OtherStory, OtherStoryAdmin)
  722. site.register(Report, ReportAdmin)
  723. site.register(MainPrepopulated, MainPrepopulatedAdmin)
  724. site.register(UnorderedObject, UnorderedObjectAdmin)
  725. site.register(UndeletableObject, UndeletableObjectAdmin)
  726. site.register(UnchangeableObject, UnchangeableObjectAdmin)
  727. site.register(State, StateAdmin)
  728. site.register(City, CityAdmin)
  729. site.register(Restaurant, RestaurantAdmin)
  730. site.register(Worker, WorkerAdmin)
  731. site.register(FunkyTag, FunkyTagAdmin)
  732. site.register(ReferencedByParent)
  733. site.register(ChildOfReferer)
  734. site.register(ReferencedByInline)
  735. site.register(InlineReferer, InlineRefererAdmin)
  736. site.register(ReferencedByGenRel)
  737. site.register(GenRelReference)
  738. site.register(ParentWithUUIDPK)
  739. site.register(RelatedPrepopulated, search_fields=['name'])
  740. site.register(RelatedWithUUIDPKModel)
  741. # We intentionally register Promo and ChapterXtra1 but not Chapter nor ChapterXtra2.
  742. # That way we cover all four cases:
  743. # related ForeignKey object registered in admin
  744. # related ForeignKey object not registered in admin
  745. # related OneToOne object registered in admin
  746. # related OneToOne object not registered in admin
  747. # when deleting Book so as exercise all four paths through
  748. # contrib.admin.utils's get_deleted_objects function.
  749. site.register(Book, inlines=[ChapterInline])
  750. site.register(Promo)
  751. site.register(ChapterXtra1, ChapterXtra1Admin)
  752. site.register(Pizza, PizzaAdmin)
  753. site.register(ReadOnlyPizza, ReadOnlyPizzaAdmin)
  754. site.register(ReadablePizza)
  755. site.register(Topping, ToppingAdmin)
  756. site.register(Album, AlbumAdmin)
  757. site.register(Question, QuestionAdmin)
  758. site.register(Answer, AnswerAdmin, date_hierarchy='question__posted')
  759. site.register(Answer2, date_hierarchy='question__expires')
  760. site.register(PrePopulatedPost, PrePopulatedPostAdmin)
  761. site.register(ComplexSortedPerson, ComplexSortedPersonAdmin)
  762. site.register(FilteredManager, CustomManagerAdmin)
  763. site.register(PluggableSearchPerson, PluggableSearchPersonAdmin)
  764. site.register(PrePopulatedPostLargeSlug, PrePopulatedPostLargeSlugAdmin)
  765. site.register(AdminOrderedField, AdminOrderedFieldAdmin)
  766. site.register(AdminOrderedModelMethod, AdminOrderedModelMethodAdmin)
  767. site.register(AdminOrderedAdminMethod, AdminOrderedAdminMethodAdmin)
  768. site.register(AdminOrderedCallable, AdminOrderedCallableAdmin)
  769. site.register(Color2, CustomTemplateFilterColorAdmin)
  770. site.register(Simple, AttributeErrorRaisingAdmin)
  771. site.register(UserMessenger, MessageTestingAdmin)
  772. site.register(Choice, ChoiceList)
  773. site.register(ParentWithDependentChildren, ParentWithDependentChildrenAdmin)
  774. site.register(EmptyModelHidden, EmptyModelHiddenAdmin)
  775. site.register(EmptyModelVisible, EmptyModelVisibleAdmin)
  776. site.register(EmptyModelMixin, EmptyModelMixinAdmin)
  777. site.register(StumpJoke)
  778. site.register(Recipe)
  779. site.register(Ingredient)
  780. site.register(NotReferenced)
  781. site.register(ExplicitlyProvidedPK, GetFormsetsArgumentCheckingAdmin)
  782. site.register(ImplicitlyGeneratedPK, GetFormsetsArgumentCheckingAdmin)
  783. site.register(UserProxy)
  784. # Register core models we need in our tests
  785. site.register(User, UserAdmin)
  786. site.register(Group, GroupAdmin)
  787. # Used to test URL namespaces
  788. site2 = admin.AdminSite(name="namespaced_admin")
  789. site2.register(User, UserAdmin)
  790. site2.register(Group, GroupAdmin)
  791. site2.register(ParentWithUUIDPK)
  792. site2.register(
  793. RelatedWithUUIDPKModel,
  794. list_display=['pk', 'parent'],
  795. list_editable=['parent'],
  796. raw_id_fields=['parent'],
  797. )
  798. site2.register(Person, save_as_continue=False)
  799. site7 = admin.AdminSite(name="admin7")
  800. site7.register(Article, ArticleAdmin2)
  801. site7.register(Section)
  802. site7.register(PrePopulatedPost, PrePopulatedPostReadOnlyAdmin)
  803. # Used to test ModelAdmin.sortable_by and get_sortable_by().
  804. class ArticleAdmin6(admin.ModelAdmin):
  805. list_display = (
  806. 'content', 'date', callable_year, 'model_year', 'modeladmin_year',
  807. 'model_year_reversed', 'section',
  808. )
  809. sortable_by = ('date', callable_year)
  810. def modeladmin_year(self, obj):
  811. return obj.date.year
  812. modeladmin_year.admin_order_field = 'date'
  813. class ActorAdmin6(admin.ModelAdmin):
  814. list_display = ('name', 'age')
  815. sortable_by = ('name',)
  816. def get_sortable_by(self, request):
  817. return ('age',)
  818. class ChapterAdmin6(admin.ModelAdmin):
  819. list_display = ('title', 'book')
  820. sortable_by = ()
  821. class ColorAdmin6(admin.ModelAdmin):
  822. list_display = ('value',)
  823. def get_sortable_by(self, request):
  824. return ()
  825. site6 = admin.AdminSite(name='admin6')
  826. site6.register(Article, ArticleAdmin6)
  827. site6.register(Actor, ActorAdmin6)
  828. site6.register(Chapter, ChapterAdmin6)
  829. site6.register(Color, ColorAdmin6)
  830. class ArticleAdmin9(admin.ModelAdmin):
  831. def has_change_permission(self, request, obj=None):
  832. # Simulate that the user can't change a specific object.
  833. return obj is None
  834. site9 = admin.AdminSite(name='admin9')
  835. site9.register(Article, ArticleAdmin9)
  836. class ArticleAdmin10(admin.ModelAdmin):
  837. def has_change_permission(self, request, obj=None):
  838. return False
  839. site10 = admin.AdminSite(name='admin10')
  840. site10.register(Article, ArticleAdmin10)