from flowspec.tasks import *
from django.contrib.auth.models import User
from django.contrib.auth.admin import UserAdmin
-from flowspy.accounts.models import UserProfile
from flowspy.peers.models import *
from flowspy.flowspec.forms import *
import datetime
from django.conf import settings
-from django.forms import ModelForm
-from django.contrib.admin.widgets import FilteredSelectMultiple
-class PeerAdminForm(ModelForm):
- networks=forms.ModelMultipleChoiceField(PeerRange.objects.all(),widget=
- FilteredSelectMultiple("PeerRange",True), required=False)
-
- class Meta:
- model= Peer
-
-class PeerAdmin(admin.ModelAdmin):
- form = PeerAdminForm
-
-class PeerTechcAdmin(admin.ModelAdmin):
- list_display = ('get_peer_name', 'emails')
+from flowspy.monkey_patch.forms import UserCreationForm, UserChangeForm
class RouteAdmin(admin.ModelAdmin):
form = RouteForm
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(Peer, PeerAdmin)
-admin.site.register(PeerTechc, PeerTechcAdmin)
admin.site.register(Route, RouteAdmin)
admin.site.register(User, UserProfileAdmin)
admin.site.disable_action('delete_selected')