mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-08-03 18:38:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Imported specific models in a few tests that didn't.

Browse source
This commit is contained in:
Tim Graham 2016-12-08 11:00:14 -05:00
parent 9b79281e31
commit e744c7e459
4 changed files with 181 additions and 166 deletions
Download patch file Download diff file Expand all files Collapse all files

31
tests/admin_widgets/widgetadmin.py
View file

@ -1,6 +1,9 @@
from django.contrib import admin
from . import models
from .models import (
Advisor, Album, Band, Bee, Car, CarTire, Event, Inventory, Member, Profile,
School, User,
)
class WidgetAdmin(admin.AdminSite):
@ -15,7 +18,7 @@ class CarAdmin(admin.ModelAdmin):
class CarTireAdmin(admin.ModelAdmin):
def formfield_for_foreignkey(self, db_field, request, **kwargs):
if db_field.name == "car":
kwargs["queryset"] = models.Car.objects.filter(owner=request.user)
kwargs["queryset"] = Car.objects.filter(owner=request.user)
return db_field.formfield(**kwargs)
return super(CarTireAdmin, self).formfield_for_foreignkey(db_field, request, **kwargs)
@ -36,21 +39,21 @@ class SchoolAdmin(admin.ModelAdmin):
site = WidgetAdmin(name='widget-admin')
site.register(models.User)
site.register(models.Car, CarAdmin)
site.register(models.CarTire, CarTireAdmin)
site.register(User)
site.register(Car, CarAdmin)
site.register(CarTire, CarTireAdmin)
site.register(models.Member)
site.register(models.Band)
site.register(models.Event, EventAdmin)
site.register(models.Album, AlbumAdmin)
site.register(Member)
site.register(Band)
site.register(Event, EventAdmin)
site.register(Album, AlbumAdmin)
site.register(models.Inventory)
site.register(Inventory)
site.register(models.Bee)
site.register(Bee)
site.register(models.Advisor)
site.register(Advisor)
site.register(models.School, SchoolAdmin)
site.register(School, SchoolAdmin)
site.register(models.Profile)
site.register(Profile)