code.grnet.gr Git - flowspy/blob - flowspec/admin.py

   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
   6 class RouteAdmin(admin.ModelAdmin):
   7
   8     actions = ['deactivate']
   9
  10     def deactivate(self, request, queryset):
  11         applier = PR.Applier(route_objects=queryset)
  12         commit, response = applier.apply(configuration=applier.delete_routes())
  13         if commit:
  14             rows = queryset.update(is_online=False, is_active=False)
  15             queryset.update(response="Successfully removed route from network")
  16             self.message_user(request, "Successfully removed %s routes from network" % rows)
  17         else:
  18             self.message_user(request, "Could not remove routes from network")
  19     deactivate.short_description = "Deactivate selected routes from network"
  20
  21     list_display = ('name', 'is_online', 'applier', 'get_match', 'get_then', 'response')
  22     fieldsets = [
  23         (None,               {'fields': ['name','applier']}),
  24         ("Match",               {'fields': ['source', 'sourceport', 'destination', 'destinationport', 'port']}),
  25         ('Advanced Match Statements', {'fields': ['dscp', 'fragmenttype', 'icmpcode', 'icmptype', 'packetlength', 'protocol', 'tcpflag'], 'classes': ['collapse']}),
  26         ("Then",               {'fields': ['then' ]}),
  27         (None,               {'fields': ['comments',]}),
  28
  29     ]
  30 #    fields = ('name', 'applier', 'expires')
  31
  32     #def formfield_for_dbfield(self, db_field, **kwargs):
  33     #    if db_field.name == 'password':
  34     #        kwargs['widget'] = PasswordInput
  35     #    return db_field.formfield(**kwargs)
  36
  37 #admin.site.register(MatchAddress)
  38 admin.site.register(MatchPort)
  39 admin.site.register(MatchDscp)
  40 admin.site.register(UserProfile)
  41 #admin.site.register(MatchFragmentType)
  42 #admin.site.register(MatchIcmpCode)
  43 #admin.site.register(MatchIcmpType)
  44 #admin.site.register(MatchPacketLength)
  45 #admin.site.register(MatchProtocol)
  46 #admin.site.register(MatchTcpFlag)
  47 admin.site.register(ThenAction)
  48 #admin.site.register(ThenStatement)
  49 #admin.site.register(MatchStatement)
  50 admin.site.register(Route, RouteAdmin)
  51
  52 admin.site.disable_action('delete_selected')
  53
  54