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 from flowspec.tasks import *
   6 from django.contrib.auth.models import User
   7 from django.contrib.auth.admin import UserAdmin
   8 from accounts.models import UserProfile
   9 from flowspy.flowspec.forms import *
  10
  11
  12 class RouteAdmin(admin.ModelAdmin):
  13     form = RouteForm
  14     actions = ['deactivate']
  15
  16     def deactivate(self, request, queryset):
  17         queryset = queryset.filter(status='ACTIVE')
  18         response = batch_delete.delay(queryset, reason="ADMININACTIVE")
  19         self.message_user(request, "Added request %s to job que. Check in a while for result" % response)
  20     deactivate.short_description = "Remove selected routes from network"
  21
  22     def save_model(self, request, obj, form, change):
  23         obj.status = "PENDING"
  24         obj.save()
  25         obj.commit_add()
  26
  27     def has_delete_permission(self, request, obj=None):
  28         return False
  29
  30     list_display = ('name', 'status', 'applier' , 'applier_peer', 'get_match', 'get_then', 'response', "expires", "comments")
  31
  32     fieldsets = [
  33         (None,               {'fields': ['name','applier']}),
  34         ("Match",               {'fields': ['source', 'sourceport', 'destination', 'destinationport', 'port']}),
  35         ('Advanced Match Statements', {'fields': ['dscp', 'fragmenttype', 'icmpcode', 'icmptype', 'packetlength', 'protocol', 'tcpflag'], 'classes': ['collapse']}),
  36         ("Then",               {'fields': ['then' ]}),
  37         (None,               {'fields': ['comments',]}),
  38
  39     ]
  40
  41
  42
  43 class UserProfileInline(admin.StackedInline):
  44     model = UserProfile
  45
  46 class UserProfileAdmin(UserAdmin):
  47     inlines = [UserProfileInline]
  48 #    fields = ('name', 'applier', 'expires')
  49
  50     #def formfield_for_dbfield(self, db_field, **kwargs):
  51     #    if db_field.name == 'password':
  52     #        kwargs['widget'] = PasswordInput
  53     #    return db_field.formfield(**kwargs)
  54
  55 #admin.site.register(MatchAddress)
  56 admin.site.unregister(User)
  57 admin.site.register(MatchPort)
  58 admin.site.register(MatchDscp)
  59 admin.site.register(UserProfile)
  60 #admin.site.register(MatchFragmentType)
  61 #admin.site.register(MatchIcmpCode)
  62 #admin.site.register(MatchIcmpType)
  63 #admin.site.register(MatchPacketLength)
  64 #admin.site.register(MatchProtocol)
  65 #admin.site.register(MatchTcpFlag)
  66 admin.site.register(ThenAction)
  67 #admin.site.register(ThenStatement)
  68 #admin.site.register(MatchStatement)
  69 admin.site.register(Route, RouteAdmin)
  70 admin.site.register(User, UserProfileAdmin)
  71 admin.site.disable_action('delete_selected')
  72
  73