from flowspy.flowspec.models import *
from flowspy.accounts.models import *
from utils import proxy as PR
+from flowspec.tasks import *
+from django.contrib.auth.models import User
+from django.contrib.auth.admin import UserAdmin
+from flowspy.peers.models import *
+from flowspy.flowspec.forms import *
+import datetime
+from django.conf import settings
-#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','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',]}),
-#
-# ]
+from flowspy.monkey_patch.forms import UserCreationForm, UserChangeForm
+
+class RouteAdmin(admin.ModelAdmin):
+ form = RouteForm
+ actions = ['deactivate']
+
+ def deactivate(self, request, queryset):
+ queryset = queryset.filter(status='ACTIVE')
+ 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"
+
+ def save_model(self, request, obj, form, change):
+ obj.status = "PENDING"
+ obj.save()
+ if change:
+ obj.commit_edit()
+ else:
+ obj.commit_add()
+
+ def has_delete_permission(self, request, obj=None):
+ return False
+
+ list_display = ('name', 'status', 'applier' , 'applier_peer', 'get_match', 'get_then', 'response', "expires", "comments")
+
+ 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' ]}),
+ ("Expires", {'fields': ['expires' ]}),
+ (None, {'fields': ['comments',]}),
+
+ ]
+
+
+
+class UserProfileInline(admin.StackedInline):
+ model = UserProfile
+
+class UserProfileAdmin(UserAdmin):
+ add_form = UserCreationForm
+ form = UserChangeForm
+ actions = ['deactivate', 'activate']
+ list_display = ('username', 'email', 'first_name' , 'last_name', 'is_staff', 'is_active', 'is_superuser', 'get_userprofile_peer')
+ inlines = [UserProfileInline]
+
+ def deactivate(self, request, queryset):
+ queryset = queryset.update(is_active=False)
+ deactivate.short_description = "Deactivate Selected Users"
+
+ def activate(self, request, queryset):
+ queryset = queryset.update(is_active=True)
+ activate.short_description = "Activate Selected Users"
+
+ def get_userprofile_peer(self, instance):
+ # instance is User instance
+ return instance.get_profile().peer
+ get_userprofile_peer.short_description = "User Peer"
# fields = ('name', 'applier', 'expires')
#def formfield_for_dbfield(self, db_field, **kwargs):
# kwargs['widget'] = PasswordInput
# return db_field.formfield(**kwargs)
-#admin.site.register(MatchAddress)
+admin.site.unregister(User)
admin.site.register(MatchPort)
+admin.site.register(MatchProtocol)
admin.site.register(MatchDscp)
-admin.site.register(UserProfile)
-#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)
-
+admin.site.register(Route, RouteAdmin)
+admin.site.register(User, UserProfileAdmin)
admin.site.disable_action('delete_selected')