source: django_apps/wndb/admin.py@ 422

from wndb.models import *

from django.contrib import admin

from django.db.models import get_models, get_app

from django.contrib.admin.sites import AlreadyRegistered

from django import forms

from django.db.models import Q



class InterfaceInline(admin.TabularInline):

  extra = 3

  model = Interface



class NodeAdmin(admin.ModelAdmin):

  inlines = [

    InterfaceInline,

  ]



admin.site.register(Node,NodeAdmin)



class LinkForm(forms.ModelForm):

  def __init__(self, *args, **kwargs):

    super(LinkForm, self).__init__(*args, **kwargs)

    if 'initial' in kwargs:

      self.fields['ip_block'].queryset = IPBlock.objects.filter(Q(link__isnull=True))

    else:

      self.fields['ip_block'].queryset = IPBlock.objects.filter(Q(link__isnull=True)|Q(pk=self.instance.ip_block.id))



class LinkAdmin(admin.ModelAdmin):

  form = LinkForm



admin.site.register(Link,LinkAdmin)





for model in get_models(get_app('wndb')):

  try:

    admin.site.register(model)

  except AlreadyRegistered:

    pass