Revision 1e02e391 flowspec/admin.py

b/flowspec/admin.py
5 5
from flowspec.tasks import *
6 6
from django.contrib.auth.models import User
7 7
from django.contrib.auth.admin import UserAdmin
8
from accounts.models import UserProfile
8
from flowspy.accounts.models import UserProfile
9
from flowspy.peers.models import *
9 10
from flowspy.flowspec.forms import *
10 11
import datetime
11 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')
12 28

  
13 29
class RouteAdmin(admin.ModelAdmin):
14 30
    form = RouteForm
......
71 87
admin.site.register(ThenAction)
72 88
#admin.site.register(ThenStatement)
73 89
#admin.site.register(MatchStatement)
90
admin.site.register(Peer, PeerAdmin)
91
admin.site.register(PeerTechc, PeerTechcAdmin)
74 92
admin.site.register(Route, RouteAdmin)
75 93
admin.site.register(User, UserProfileAdmin)
76 94
admin.site.disable_action('delete_selected')

Also available in: Unified diff