diff --git a/flowspec/admin.py b/flowspec/admin.py index 548994dab29361f7a4cfe6d8a080bd6c83c97a78..03f612763edcaf6aafe263b8deb65afd37563beb 100644 --- a/flowspec/admin.py +++ b/flowspec/admin.py @@ -10,21 +10,6 @@ from flowspy.peers.models import * from flowspy.flowspec.forms import * import datetime from django.conf import settings -from django.forms import ModelForm -from django.contrib.admin.widgets import FilteredSelectMultiple - -class PeerAdminForm(ModelForm): - networks=forms.ModelMultipleChoiceField(PeerRange.objects.all(),widget= - FilteredSelectMultiple("PeerRange",True), required=False) - - class Meta: - model= Peer - -class PeerAdmin(admin.ModelAdmin): - form = PeerAdminForm - -class PeerTechcAdmin(admin.ModelAdmin): - list_display = ('get_peer_name', 'emails') class RouteAdmin(admin.ModelAdmin): form = RouteForm @@ -66,7 +51,7 @@ class UserProfileInline(admin.StackedInline): class UserProfileAdmin(UserAdmin): actions = ['deactivate', 'activate'] - list_display = ('username', 'email', 'first_name' , 'last_name', 'is_staff', 'is_active','get_userprofile_peer') + list_display = ('username', 'email', 'first_name' , 'last_name', 'is_staff', 'is_active', 'is_superuser', 'get_userprofile_peer') inlines = [UserProfileInline] def deactivate(self, request, queryset): @@ -88,22 +73,11 @@ class UserProfileAdmin(UserAdmin): # kwargs['widget'] = PasswordInput # return db_field.formfield(**kwargs) -#admin.site.register(MatchAddress) admin.site.unregister(User) admin.site.register(MatchPort) admin.site.register(MatchDscp) admin.site.register(UserProfile) -#admin.site.register(MatchFragmentType) -#admin.site.register(MatchIcmpCode) -#admin.site.register(MatchIcmpType) -#admin.site.register(MatchPacketLength) -#admin.site.register(MatchProtocol) -#admin.site.register(MatchTcpFlag) admin.site.register(ThenAction) -#admin.site.register(ThenStatement) -#admin.site.register(MatchStatement) -admin.site.register(Peer, PeerAdmin) -admin.site.register(PeerTechc, PeerTechcAdmin) admin.site.register(Route, RouteAdmin) admin.site.register(User, UserProfileAdmin) admin.site.disable_action('delete_selected') diff --git a/peers/admin.py b/peers/admin.py new file mode 100644 index 0000000000000000000000000000000000000000..2fe8069f2558c69b363d6dd355208cd880a0b317 --- /dev/null +++ b/peers/admin.py @@ -0,0 +1,24 @@ +from django.contrib import admin + +from flowspy.peers.models import * +from flowspy.flowspec.forms import * +from django.conf import settings +from django.forms import ModelForm +from django.contrib.admin.widgets import FilteredSelectMultiple + +class PeerAdminForm(ModelForm): + networks=forms.ModelMultipleChoiceField(PeerRange.objects.all(),widget= + FilteredSelectMultiple("PeerRange",True), required=False) + + class Meta: + model= Peer + +class PeerAdmin(admin.ModelAdmin): + form = PeerAdminForm + +class PeerTechcAdmin(admin.ModelAdmin): + list_display = ('get_peer_name', 'emails') + +admin.site.register(Peer, PeerAdmin) +admin.site.register(PeerTechc, PeerTechcAdmin) +admin.site.register(PeerRange) \ No newline at end of file diff --git a/peers/models.py b/peers/models.py index 311c408d4d8dff50eae167d0cb258a14ce72fa6b..405a8038d856164bf6741ed6c6f37c705f102ed8 100644 --- a/peers/models.py +++ b/peers/models.py @@ -9,6 +9,7 @@ class PeerRange(models.Model): return self.network class Meta: db_table = u'peer_range' + ordering = ['network'] # Create your models here. class Peer(models.Model):