root / flowspec / admin.py @ 1e02e391
History | View | Annotate | Download (3.3 kB)
1 |
from django.contrib import admin |
---|---|
2 |
from flowspy.flowspec.models import * |
3 |
from flowspy.accounts.models import * |
4 |
from utils import proxy as PR |
5 |
from flowspec.tasks import * |
6 |
from django.contrib.auth.models import User |
7 |
from django.contrib.auth.admin import UserAdmin |
8 |
from flowspy.accounts.models import UserProfile |
9 |
from flowspy.peers.models import * |
10 |
from flowspy.flowspec.forms import * |
11 |
import datetime |
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 |
|
29 |
class RouteAdmin(admin.ModelAdmin): |
30 |
form = RouteForm |
31 |
actions = ['deactivate']
|
32 |
|
33 |
def deactivate(self, request, queryset): |
34 |
queryset = queryset.filter(status='ACTIVE')
|
35 |
response = batch_delete.delay(queryset, reason="ADMININACTIVE")
|
36 |
self.message_user(request, "Added request %s to job que. Check in a while for result" % response) |
37 |
deactivate.short_description = "Remove selected routes from network"
|
38 |
|
39 |
def save_model(self, request, obj, form, change): |
40 |
obj.status = "PENDING"
|
41 |
obj.save() |
42 |
if change:
|
43 |
obj.commit_edit() |
44 |
else:
|
45 |
obj.commit_add() |
46 |
|
47 |
def has_delete_permission(self, request, obj=None): |
48 |
return False |
49 |
|
50 |
list_display = ('name', 'status', 'applier' , 'applier_peer', 'get_match', 'get_then', 'response', "expires", "comments") |
51 |
|
52 |
fieldsets = [ |
53 |
(None, {'fields': ['name','applier']}), |
54 |
("Match", {'fields': ['source', 'sourceport', 'destination', 'destinationport', 'port']}), |
55 |
('Advanced Match Statements', {'fields': ['dscp', 'fragmenttype', 'icmpcode', 'icmptype', 'packetlength', 'protocol', 'tcpflag'], 'classes': ['collapse']}), |
56 |
("Then", {'fields': ['then' ]}), |
57 |
("Expires", {'fields': ['expires' ]}), |
58 |
(None, {'fields': ['comments',]}), |
59 |
|
60 |
] |
61 |
|
62 |
|
63 |
|
64 |
class UserProfileInline(admin.StackedInline): |
65 |
model = UserProfile |
66 |
|
67 |
class UserProfileAdmin(UserAdmin): |
68 |
inlines = [UserProfileInline] |
69 |
# fields = ('name', 'applier', 'expires')
|
70 |
|
71 |
#def formfield_for_dbfield(self, db_field, **kwargs):
|
72 |
# if db_field.name == 'password':
|
73 |
# kwargs['widget'] = PasswordInput
|
74 |
# return db_field.formfield(**kwargs)
|
75 |
|
76 |
#admin.site.register(MatchAddress)
|
77 |
admin.site.unregister(User) |
78 |
admin.site.register(MatchPort) |
79 |
admin.site.register(MatchDscp) |
80 |
admin.site.register(UserProfile) |
81 |
#admin.site.register(MatchFragmentType)
|
82 |
#admin.site.register(MatchIcmpCode)
|
83 |
#admin.site.register(MatchIcmpType)
|
84 |
#admin.site.register(MatchPacketLength)
|
85 |
#admin.site.register(MatchProtocol)
|
86 |
#admin.site.register(MatchTcpFlag)
|
87 |
admin.site.register(ThenAction) |
88 |
#admin.site.register(ThenStatement)
|
89 |
#admin.site.register(MatchStatement)
|
90 |
admin.site.register(Peer, PeerAdmin) |
91 |
admin.site.register(PeerTechc, PeerTechcAdmin) |
92 |
admin.site.register(Route, RouteAdmin) |
93 |
admin.site.register(User, UserProfileAdmin) |
94 |
admin.site.disable_action('delete_selected')
|
95 |
|
96 |
|