Firewall on Demand

from django.contrib import admin

from flowspy.flowspec.models import *

from utils import proxy as PR

class RouteAdmin(admin.ModelAdmin):

    actions = ['deactivate']

    def deactivate(self, request, queryset):

        applier = PR.Applier(route_objects=queryset)

        commit, response = applier.apply(configuration=applier.delete_routes())

        if commit:

            rows = queryset.update(is_online=False, is_active=False)

            queryset.update(response="Successfully removed route from network")

            self.message_user(request, "Successfully removed %s routes from network" % rows)

        else:

            self.message_user(request, "Could not remove routes from network")

    deactivate.short_description = "Deactivate selected routes from network"

    list_display = ('name', 'is_online', 'applier', 'get_match', 'get_then', 'response')

    fieldsets = [

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

        ("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',]}),

    ]

#    fields = ('name', 'applier', 'expires')

    #def formfield_for_dbfield(self, db_field, **kwargs):

    #    if db_field.name == 'password':

    #        kwargs['widget'] = PasswordInput

    #    return db_field.formfield(**kwargs)

#admin.site.register(MatchAddress)

admin.site.register(MatchPort)

admin.site.register(MatchDscp)

#admin.site.register(MatchFragmentType)

#admin.site.register(MatchIcmpCode)

#admin.site.register(MatchIcmpType)

#admin.site.register(MatchPacketLength)

#admin.site.register(MatchProtocol)

#admin.site.register(MatchTcpFlag)

admin.site.register(ThenAction)

#admin.site.register(ThenStatement)

#admin.site.register(MatchStatement)

admin.site.register(Route, RouteAdmin)

admin.site.disable_action('delete_selected')