Revision dc9b0d59 flowspec/admin.py

b/flowspec/admin.py
10 10
from flowspy.flowspec.forms import *
11 11
import datetime
12 12
from django.conf import settings
13
from django.forms import ModelForm
14
from django.contrib.admin.widgets import FilteredSelectMultiple
15

  
16
class PeerAdminForm(ModelForm):
17
    networks=forms.ModelMultipleChoiceField(PeerRange.objects.all(),widget=
18
            FilteredSelectMultiple("PeerRange",True), required=False)
19

  
20
    class Meta:
21
        model= Peer
22

  
23
class PeerAdmin(admin.ModelAdmin):
24
    form = PeerAdminForm
25

  
26
class PeerTechcAdmin(admin.ModelAdmin):
27
    list_display = ('get_peer_name', 'emails')
28 13

  
29 14
class RouteAdmin(admin.ModelAdmin):
30 15
    form = RouteForm
......
66 51
    
67 52
class UserProfileAdmin(UserAdmin):
68 53
    actions = ['deactivate', 'activate']
69
    list_display = ('username', 'email', 'first_name' , 'last_name', 'is_staff', 'is_active','get_userprofile_peer')
54
    list_display = ('username', 'email', 'first_name' , 'last_name', 'is_staff', 'is_active', 'is_superuser', 'get_userprofile_peer')
70 55
    inlines = [UserProfileInline]
71 56

  
72 57
    def deactivate(self, request, queryset):
......
88 73
    #        kwargs['widget'] = PasswordInput
89 74
    #    return db_field.formfield(**kwargs)
90 75

  
91
#admin.site.register(MatchAddress)
92 76
admin.site.unregister(User)
93 77
admin.site.register(MatchPort)
94 78
admin.site.register(MatchDscp)
95 79
admin.site.register(UserProfile)
96
#admin.site.register(MatchFragmentType)
97
#admin.site.register(MatchIcmpCode)
98
#admin.site.register(MatchIcmpType)
99
#admin.site.register(MatchPacketLength)
100
#admin.site.register(MatchProtocol)
101
#admin.site.register(MatchTcpFlag)
102 80
admin.site.register(ThenAction)
103
#admin.site.register(ThenStatement)
104
#admin.site.register(MatchStatement)
105
admin.site.register(Peer, PeerAdmin)
106
admin.site.register(PeerTechc, PeerTechcAdmin)
107 81
admin.site.register(Route, RouteAdmin)
108 82
admin.site.register(User, UserProfileAdmin)
109 83
admin.site.disable_action('delete_selected')

Also available in: Unified diff