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 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     actions = ['deactivate', 'activate']
  69     list_display = ('username', 'email', 'first_name' , 'last_name', 'is_staff', 'is_active','get_userprofile_peer')
  70     inlines = [UserProfileInline]
  71
  72     def deactivate(self, request, queryset):
  73         queryset = queryset.update(is_active=False)
  74     deactivate.short_description = "Deactivate Selected Users"
  75
  76     def activate(self, request, queryset):
  77         queryset = queryset.update(is_active=True)
  78     activate.short_description = "Activate Selected Users"
  79
  80     def get_userprofile_peer(self, instance):
  81         # instance is User instance
  82         return instance.get_profile().peer
  83     get_userprofile_peer.short_description = "User Peer"
  84 #    fields = ('name', 'applier', 'expires')
  85
  86     #def formfield_for_dbfield(self, db_field, **kwargs):
  87     #    if db_field.name == 'password':
  88     #        kwargs['widget'] = PasswordInput
  89     #    return db_field.formfield(**kwargs)
  90
  91 #admin.site.register(MatchAddress)
  92 admin.site.unregister(User)
  93 admin.site.register(MatchPort)
  94 admin.site.register(MatchDscp)
  95 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 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 admin.site.register(Route, RouteAdmin)
 108 admin.site.register(User, UserProfileAdmin)
 109 admin.site.disable_action('delete_selected')
 110
 111