24 lines
693 B
Python
24 lines
693 B
Python
|
from django.contrib import admin
|
||
|
|
|
||
|
|
from orchestra.core import services
|
||
|
|
|
||
|
|
from .models import Pack, Price, Rate
|
||
|
|
|
||
|
|
|
||
|
|
class RateInline(admin.TabularInline):
|
||
|
|
model = Rate
|
||
|
|
|
||
|
|
|
||
|
|
class PriceAdmin(admin.ModelAdmin):
|
||
|
|
inlines = [RateInline]
|
||
|
|
|
||
|
|
def formfield_for_dbfield(self, db_field, **kwargs):
|
||
|
|
""" Improve performance of account field and filter by account """
|
||
|
|
if db_field.name == 'service':
|
||
|
|
models = [model._meta.model_name for model in services.get().keys()]
|
||
|
|
kwargs['queryset'] = db_field.rel.to.objects.filter(model__in=models)
|
||
|
|
return super(PriceAdmin, self).formfield_for_dbfield(db_field, **kwargs)
|
||
|
|
|
||
|
|
|
||
|
|
admin.site.register(Price, PriceAdmin)
|