def test_modelform_factory_exclude_name(self):
self.form = autocomplete_light.modelform_factory(self.model_class,
exclude=['name'])()
self.assertNotInForm('name')
self.assertExpectedFormField()
self.assertIsAutocomplete('noise')
def test_modelform_factory_fields_name(self):
self.form = autocomplete_light.modelform_factory(self.model_class,
fields=['name'])()
self.assertInForm('name')
self.assertNotInForm('relation')
self.assertNotInForm('noise')
def test_modelform_factory_autocomplete_exclude_relation(self):
self.form = autocomplete_light.modelform_factory(self.model_class,
autocomplete_exclude=['relation'])()
self.assertNotInForm('relation')
self.assertInForm('name')
self.assertIsAutocomplete('noise')
def test_modelform_factory_fields_relation(self):
self.form = autocomplete_light.modelform_factory(self.model_class,
fields=['relation'])()
self.assertExpectedFormField()
self.assertNotInForm('name')
self.assertNotInForm('noise')
def get_changelist_form(self, request, **kwargs):
defaults = {
"form": self.form,
"formfield_callback": curry(self.formfield_for_dbfield,
request=request),
}
defaults.update(kwargs)
return autocomplete_light.modelform_factory(self.model, **defaults)
def test_modelform_factory_does_not_warn(self):
# fix for #257
with mock.patch('warnings.warn') as warn:
self.form = autocomplete_light.modelform_factory(self.model_class,
fields='__all__')()
self.assertEqual(warn.call_count, 0)
self.assertExpectedFormField()
def test_modelform_factory_autocomplete_fields_relation(self):
if VERSION < (1, 7):
fields = None
else:
fields = '__all__'
self.form = autocomplete_light.modelform_factory(self.model_class,
autocomplete_fields=['relation'], fields=fields)()
self.assertExpectedFormField()
self.assertNotIsAutocomplete('noise')
self.assertInForm('name')
def test_modelform_factory_autocomplete_names(self):
SpecialAutocomplete = self.get_new_autocomplete_class()
autocomplete_light.registry.register(SpecialAutocomplete)
ModelForm = autocomplete_light.modelform_factory(self.model_class,
autocomplete_names={'relation': 'SpecialAutocomplete'})
self.form = ModelForm()
self.assertInForm('name')
self.assertIsAutocomplete('relation')
self.assertIsAutocomplete('noise')
self.assertTrue(issubclass(self.form.fields['relation'].autocomplete,
SpecialAutocomplete))
def inline_instances(self):
inline_instances = []
for name, value in self.model_class.__dict__.items():
if not isinstance(value, ForeignRelatedObjectsDescriptor):
continue
model = getattr(self.model_class, name).related.model
cls = inlineformset_factory(self.model_class, model, form=modelform_factory(model), extra=25)
instance = cls(prefix=name, *self.inline_args, **self.inline_kwargs)
instance.name = instance.model._meta.verbose_name_plural
inline_instances.append(instance)
return inline_instances
def form_class(self):
attributes = self.form_model.enabled_field_set
form = modelform_factory(self.form_model.contenttype.model_class(), fields=[a.name for a in attributes])
for attribute in attributes:
name = attribute.name
if attribute.verbose_name:
form.base_fields[name].label = attribute.verbose_name
if attribute.help_text:
form.base_fields[name].help_text = attribute.help_text
form.helper = FormHelper()
form.helper.layout = self.form_model.to_crispy()
form.helper.form_tag = False
return form
class UserContactAdmin(admin.ModelAdmin):
form = autocomplete_light.modelform_factory(UserContact)
class ProfileAdmin(admin.ModelAdmin):
list_display = ('id', 'username', 'organization',)
search_fields = ('username', 'organization', 'profile', )
autocomplete_light.modelform_factory(Profile, fields='__all__')
add_form_template = 'admin/auth/user/add_form.html'
change_user_password_template = None
fieldsets = (
(None, {'fields': ('username', 'password')}),
(_('Personal info'), {'fields': ('first_name', 'last_name', 'email')}),
(_('Permissions'), {'fields': ('is_active', 'is_staff', 'is_superuser',
'groups')}),
(_('Important dates'), {'fields': ('last_login', 'date_joined')}),
(_('Extended profile'), {'fields': ('organization', 'profile',
'position', 'voice', 'fax',
'delivery', 'city', 'area',
'zipcode', 'country',
'keywords')}),
)
add_fieldsets = (
(None, {
'classes': ('wide',),
'fields': ('username', 'password1', 'password2')}
),
)
form = ProfileChangeForm
add_form = ProfileCreationForm
change_password_form = AdminPasswordChangeForm
list_display = ('username', 'email', 'first_name', 'last_name', 'is_staff')
list_filter = ('is_staff', 'is_superuser', 'is_active', 'groups')
search_fields = ('username', 'first_name', 'last_name', 'email')
ordering = ('username',)
filter_horizontal = ('groups', 'user_permissions',)
def get_fieldsets(self, request, obj=None):
if not obj:
return self.add_fieldsets
return super(ProfileAdmin, self).get_fieldsets(request, obj)
def get_form(self, request, obj=None, **kwargs):
"""
Use special form during user creation
"""
defaults = {}
if obj is None:
defaults.update({
'form': self.add_form,
'fields': admin.utils.flatten_fieldsets(self.add_fieldsets),
})
defaults.update(kwargs)
return super(ProfileAdmin, self).get_form(request, obj, **defaults)
def get_urls(self):
from django.conf.urls import patterns
return patterns('',
(r'^(\d+)/password/$',
self.admin_site.admin_view(self.user_change_password))
) + super(ProfileAdmin, self).get_urls()
def lookup_allowed(self, lookup, value):
# See #20078: we don't want to allow any lookups involving passwords.
if lookup.startswith('password'):
return False
return super(ProfileAdmin, self).lookup_allowed(lookup, value)
@sensitive_post_parameters_m
@csrf_protect_m
@transaction.atomic
def add_view(self, request, form_url='', extra_context=None):
# It's an error for a user to have add permission but NOT change
# permission for users. If we allowed such users to add users, they
# could create superusers, which would mean they would essentially have
# the permission to change users. To avoid the problem entirely, we
# disallow users from adding users if they don't have change
# permission.
if not self.has_change_permission(request):
if self.has_add_permission(request) and settings.DEBUG:
# Raise Http404 in debug mode so that the user gets a helpful
# error message.
raise Http404(
'Your user does not have the "Change user" permission. In '
'order to add users, Django requires that your user '
'account have both the "Add user" and "Change user" '
'permissions set.')
raise PermissionDenied
if extra_context is None:
extra_context = {}
username_field = self.model._meta.get_field(self.model.USERNAME_FIELD)
defaults = {
'auto_populated_fields': (),
'username_help_text': username_field.help_text,
}
extra_context.update(defaults)
return super(ProfileAdmin, self).add_view(request, form_url,
extra_context)
@sensitive_post_parameters_m
def user_change_password(self, request, id, form_url=''):
if not self.has_change_permission(request):
raise PermissionDenied
user = get_object_or_404(self.get_queryset(request), pk=id)
if request.method == 'POST':
form = self.change_password_form(user, request.POST)
if form.is_valid():
form.save()
change_message = self.construct_change_message(
request,
form,
None)
self.log_change(request, user, change_message)
msg = ugettext('Password changed successfully.')
messages.success(request, msg)
return HttpResponseRedirect('..')
else:
form = self.change_password_form(user)
fieldsets = [(None, {'fields': list(form.base_fields)})]
adminForm = admin.helpers.AdminForm(form, fieldsets, {})
context = {
'title': _('Change password: %s') % escape(user.get_username()),
'adminForm': adminForm,
'form_url': form_url,
'form': form,
'is_popup': IS_POPUP_VAR in request.REQUEST,
'add': True,
'change': False,
'has_delete_permission': False,
'has_change_permission': True,
'has_absolute_url': False,
'opts': self.model._meta,
'original': user,
'save_as': False,
'show_save': True,
}
return TemplateResponse(request,
self.change_user_password_template or
'admin/auth/user/change_password.html',
context, current_app=self.admin_site.name)
def response_add(self, request, obj, post_url_continue=None):
"""
Determines the HttpResponse for the add_view stage. It mostly defers to
its superclass implementation but is customized because the User model
has a slightly different workflow.
"""
# We should allow further modification of the user just added i.e. the
# 'Save' button should behave like the 'Save and continue editing'
# button except in two scenarios:
# * The user has pressed the 'Save and add another' button
# * We are adding a user in a popup
if '_addanother' not in request.POST and IS_POPUP_VAR not in request.POST:
request.POST['_continue'] = 1
return super(ProfileAdmin, self).response_add(request, obj,
post_url_continue)
class PledgedItemInline(admin.TabularInline):
form = autocomplete_light.modelform_factory(PledgedItem)
model = PledgedItem
exclude = ['pledge_particulars']
extra = 3
formset = PledgedItemInlineFormset
def get_form_class(self,*args,**kwargs):
form = autocomplete_light.modelform_factory(self.model)
#raise Exception(dir(form))
if 'summary' in form.base_fields:
form.base_fields['summary'].widget = CKEditorWidget()
return form
class UserProfileInline(AdminImageMixin, admin.StackedInline):
"""UserProfile Inline model for UserAdmin."""
model = UserProfile
readonly_fields = ['date_vouched', 'vouched_by', 'basket_token']
form = autocomplete_light.modelform_factory(UserProfile)
class IssueReportAdmin(BaseAdmin):
form = autocomplete_light.modelform_factory(IssueReport, fields="__all__")
fields = [('name', 'status'), 'type', 'report', 'report_duplicate', 'description',
('created_by', 'created_at'), ('reviewed_by', 'reviewed_at'), 'resolve_reason']
readonly_fields = ['name', 'status', 'created_by', 'created_at', 'reviewed_by', 'reviewed_at', 'resolve_reason']
list_display = ['name', 'type', 'created_by', 'created_at', 'status',]
search_fields = ['name', 'created_by__name']
list_filter = ['type', 'status']
date_hierarchy = 'created_at'
def get_fields(self, request, obj=None):
""" Override to hide the 'usecase_duplicate' if type is not 'duplicate' """
fields = super(IssueReportAdmin, self).get_fields(request, obj)
if obj and obj.type != 'duplicate' and 'report_duplicate' in fields:
fields.remove('report_duplicate')
return fields
def response_change(self, request, obj, *args, **kwargs):
'''
Override response_change method of admin/options.py to handle the click of
newly added buttons
'''
# Get the metadata about self (it tells you app and current model)
opts = self.model._meta
# Get the primary key of the model object i.e. Issue Report
pk_value = obj._get_pk_val()
preserved_filters = self.get_preserved_filters(request)
redirect_url = reverse('admin:%s_%s_change' %
(opts.app_label, opts.model_name),
args=(pk_value,))
redirect_url = add_preserved_filters({'preserved_filters': preserved_filters, 'opts': opts}, redirect_url)
# Check which button is clicked, handle accordingly.
try:
if "_investigate" in request.POST:
obj.action_investigate(request.user)
msg = "The issue is now being investigated."
elif "_resolve" in request.POST:
resolve_reason = request.POST.get('resolve_reason_text', '')
obj.action_resolve(resolve_reason,request.user)
msg = "The issue is now resolved because " + resolve_reason
elif "_reopen" in request.POST:
obj.action_reopen(request.user)
msg = "The issue has been re-opened."
elif "_open" in request.POST:
obj.action_open(request.user)
msg = "The issue is now opened."
except ValueError as e:
# In case the state of the object is not suitable for the corresponding action,
# model will raise the value exception with the appropriate message. Catch the
# exception and show the error message to the user
msg = e.message
self.message_user(request, msg, messages.ERROR)
self.message_user(request, msg, messages.SUCCESS)
return HttpResponseRedirect(redirect_url)
class ResourceBaseAdmin(admin.ModelAdmin):
list_display = ('id', 'title', 'date', 'category')
list_display_links = ('id', )
form = autocomplete_light.modelform_factory(ResourceBase)
class MyUserAdmin(admin.ModelAdmin):
form = autocomplete_light.modelform_factory(User, fields='__all__')
def get_form_class(self, **kwargs):
return autocomplete_light.modelform_factory(self.model)
class DeviceAdmin(admin.ModelAdmin):
form = autocomplete_light.modelform_factory(Device, fields='__all__')
import autocomplete_light
from .models import NonAdminAddAnotherModel
NonAdminAddAnotherModelForm = autocomplete_light.modelform_factory(
NonAdminAddAnotherModel)
class PhotoAdmin(ModelAdmin):
@staticmethod
def _distance_between_two_points_on_sphere(lon_1, lat_1, lon_2, lat_2):
import math
rad = math.pi / 180.0
equatorial_radius_meters = 6378137
lon_1_rad = lon_1 * rad
lat_1_rad = lat_1 * rad
lon_2_rad = lon_2 * rad
lat_2_rad = lat_2 * rad
cos_angle = math.sin(lat_1_rad) * math.sin(lat_2_rad) + math.cos(
lat_1_rad) * math.cos(lat_2_rad) * math.cos(lon_2_rad - lon_1_rad)
if cos_angle >= 1:
return 0
angle = math.acos(cos_angle)
return angle * equatorial_radius_meters
def save_model(self, request, obj, form, change):
if obj.lat and obj.lon and obj.bounding_circle_radius:
# If an administrator sets a bounding circle, invalidate GeoTags outside of it
all_photo_geo_tags = GeoTag.objects.filter(photo_id=obj.id)
for geo_tag in all_photo_geo_tags:
d = self._distance_between_two_points_on_sphere(
obj.lon, obj.lat, geo_tag.lon, geo_tag.lat)
if d > obj.bounding_circle_radius:
geo_tag.is_correct = False
else:
geo_tag.is_correct = True
geo_tag.save()
obj.save()
def _invertcolors(self, id):
from django.http.response import HttpResponse
photo = Photo.objects.filter(pk=id).first()
if photo:
photo_path = settings.MEDIA_ROOT + "/" + str(photo.image)
img = Image.open(photo_path)
inverted_grayscale_image = ImageOps.invert(img).convert('L')
inverted_grayscale_image.save(photo_path)
photo.invert = not photo.invert
sorl_delete(photo.image, delete_file=False)
photo.light_save()
return HttpResponse(u'Photo inverted!')
return HttpResponse(u'Failed to invert photo!')
extra_buttons = [
{
'url': '_invertcolors',
'textname': _('Invert colors'),
'func': _invertcolors
},
]
def change_view(self, request, object_id, form_url='', extra_context={}):
extra_context['extra_buttons'] = self.extra_buttons
return super(PhotoAdmin, self).change_view(request,
object_id,
form_url,
extra_context=extra_context)
def get_urls(self):
from django.conf.urls import url
urls = super(PhotoAdmin, self).get_urls()
my_urls = list((url(r'^(.+)/%(url)s/$' % b,
self.admin_site.admin_view(b['func']))
for b in self.extra_buttons))
return my_urls + urls
inlines = (AlbumPhotoInline, )
form = autocomplete_light.modelform_factory(Photo, fields='__all__')
class DatingConfirmationAdmin(ModelAdmin):
form = autocomplete_light.modelform_factory(DatingConfirmation,
fields='__all__')
class MapLayerAdmin(admin.ModelAdmin):
list_display = ('id', 'map', 'name')
list_filter = ('map',)
search_fields = ('map__title', 'name',)
form = autocomplete_light.modelform_factory(MapLayer)
def test_modelform_factory(self):
self.form = autocomplete_light.modelform_factory(self.model_class)()
self.assertExpectedFormField()
class VisitedCitiesAdmin(admin.ModelAdmin):
form = autocomplete_light.modelform_factory(VisitedCities)
class PcaGrantInlineAdmin(admin.TabularInline):
form = autocomplete_light.modelform_factory(Grant)
model = PCAGrant
verbose_name = 'Grant'
verbose_name_plural = 'Grants'
extra = 0
class NonAdminAddAnotherModelAdmin(admin.ModelAdmin):
form = autocomplete_light.modelform_factory(NonAdminAddAnotherModel)
class VolunteerSiteAdmin(admin.ModelAdmin):
form = autocomplete_light.modelform_factory(VolunteerSite)
list_display = ('volunteer', 'supervisor', 'site_approval', 'contract',
'hours_confirmed', 'inactive')
actions = [approve_site, reject_site, time_fulfilled]
inlines = [HoursInline]
class GroupMembershipInline(admin.TabularInline):
model = GroupMembership
extra = 1
form = autocomplete_light.modelform_factory(GroupMembership)
def test_modelform_factory(self):
self.form = autocomplete_light.modelform_factory(self.model_class)()
self.assertExpectedFormField()
class CityAdmin(admin.ModelAdmin):
form = autocomplete_light.modelform_factory(City)
list_display = ('name', 'pincode', 'post')
list_per_page = 25
ordering = ['name']
search_fields = ['name', 'pincode']
class AreaAdmin(admin.ModelAdmin):
form = autocomplete_light.modelform_factory(Area, fields='__all__')
from django.views import generic
from django.forms.models import modelformset_factory
import autocomplete_light
from models import OutsideAdmin
OutsideAdminModelForm = autocomplete_light.modelform_factory(OutsideAdmin, widgets={
'tags': autocomplete_light.TextWidget('TagAutocomplete'),})
OutsideAdminModelFormset = modelformset_factory(OutsideAdmin, form=OutsideAdminModelForm)
class FormsetView(generic.FormView):
form_class = OutsideAdminModelFormset
template_name = 'inlines_outside_admin/formset.html'
success_url = '/inlines_outside_admin/'
def form_valid(self, form):
form.save()
return super(FormsetView, self).form_valid(form)
class AlbumPhotoAdmin(admin.ModelAdmin):
form = autocomplete_light.modelform_factory(AlbumPhoto, fields='__all__')
def get_form_class(self,**kwargs):
return autocomplete_light.modelform_factory(self.model)
class AlbumAdmin(ModelAdmin):
form = autocomplete_light.modelform_factory(Album, fields='__all__')
class TransactionCommentPlaceInline(admin.TabularInline):
model = Transaction.comment_places.through
form = autocomplete_light.modelform_factory(Place, exclude=[])
extra = 10
class PointsAdmin(ModelAdmin):
form = autocomplete_light.modelform_factory(Points, fields='__all__')
import autocomplete_light
from .models import NonAdminAddAnotherModel
NonAdminAddAnotherModelForm = autocomplete_light.modelform_factory(
NonAdminAddAnotherModel)
class ProfileAdmin(ModelAdmin):
form = autocomplete_light.modelform_factory(Profile, fields='__all__')
class ContactRoleAdmin(admin.ModelAdmin):
model = ContactRole
list_display_links = ('id', )
list_display = ('id', 'contact', 'resource', 'role')
list_editable = ('contact', 'resource', 'role')
form = autocomplete_light.modelform_factory(ContactRole)
class GeoTagAdmin(ModelAdmin):
form = autocomplete_light.modelform_factory(GeoTag, fields='__all__')
class ItemModelAdmin(admin.ModelAdmin):
form = autocomplete_light.modelform_factory(ItemModel)
class VideoAdmin(admin.ModelAdmin):
form = autocomplete_light.modelform_factory(Video, fields='__all__')
