Firewall on Demand

from flowspec.tasks import *

from django.contrib.auth.models import User

from django.contrib.auth.admin import UserAdmin

from accounts.models import UserProfile

class RouteAdmin(admin.ModelAdmin):

    actions = ['deactivate']

    def deactivate(self, request, queryset):

        response = batch_delete.delay(queryset, reason="ADMININACTIVE")

        self.message_user(request, "Added request %s to job que. Check in a while for result" % response)

    deactivate.short_description = "Remove selected routes from network"

    list_display = ('name', 'status', 'applier' , 'applier_peer', 'get_match', 'get_then', 'response')

    fieldsets = [

        (None,               {'fields': ['name','applier']}),

        ("Match",               {'fields': ['source', 'sourceport', 'destination', 'destinationport', 'port']}),

        ('Advanced Match Statements', {'fields': ['dscp', 'fragmenttype', 'icmpcode', 'icmptype', 'packetlength', 'protocol', 'tcpflag'], 'classes': ['collapse']}),

        ("Then",               {'fields': ['then' ]}),

        (None,               {'fields': ['comments',]}),

    ]

class UserProfileInline(admin.StackedInline):

    model = UserProfile

class UserProfileAdmin(UserAdmin):

    inlines = [UserProfileInline]

admin.site.unregister(User)

admin.site.register(Route, RouteAdmin)

admin.site.register(User, UserProfileAdmin)

    ("INACTIVE", "INACTIVE"),

    ("ADMININACTIVE", "ADMININACTIVE"),           

    @property

    def applier_peer(self):

        try:

            applier_peer = self.applier.get_profile().peer

        except:

            applier_peer = None

        return applier_peer

            elif "reason" in kwargs and kwargs['reason']!='EXPIRED':

                status = kwargs['reason']

                reason_text = " Reason: %s " %status

        if route.has_expired() and (route.status != 'EXPIRED' or route.status != 'ADMININACTIVE' or route.status != 'INACTIVE'):

        elif route.has_expired() and (route.status == 'ADMININACTIVE' or route.status == 'INACTIVE'):

            route.status = 'EXPIRED'

            route.response = 'Route Expired'

            route.save()

    if route_edit.status == "ADMININACTIVE" :

        messages.add_message(request, messages.WARNING,

                             "Administrator has disabled editing of rule %s" %(route_slug))

        return HttpResponseRedirect(reverse("group-routes"))

    if route_edit.status == "EXPIRED" :

        messages.add_message(request, messages.WARNING,

                             "Cannot edit the expired rule %s. Contact helpdesk to enable it" %(route_slug))

        return HttpResponseRedirect(reverse("group-routes"))