@@ -266,9 +266,15 @@ def deletefolder(request):
266266@staff_member_required (redirect_field_name = "referrer" )
267267def deletefile (request ):
268268 if request .POST .get ("id" ) and request .POST .get ("classname" ):
269- file = get_object_or_404 (
270- eval (request .POST .get ("classname" )), id = request .POST .get ("id" )
271- )
269+ classname = request .POST .get ("classname" )
270+ if classname == "CustomImageModel" :
271+ file = get_object_or_404 (
272+ CustomImageModel , id = request .POST .get ("id" )
273+ )
274+ else :
275+ file = get_object_or_404 (
276+ CustomFileModel , id = request .POST .get ("id" )
277+ )
272278 folder = file .folder
273279 if request .user != file .created_by and not (
274280 request .user .is_superuser
@@ -398,9 +404,16 @@ def changefile(request):
398404 )
399405 raise PermissionDenied
400406
401- file = get_object_or_404 (
402- eval (request .POST .get ("file_type" )), id = request .POST .get ("file_id" )
403- )
407+ file_type = request .POST .get ("file_type" )
408+ if file_type == "CustomImageModel" :
409+ file = get_object_or_404 (
410+ CustomImageModel , id = request .POST .get ("file_id" )
411+ )
412+ else :
413+ file = get_object_or_404 (
414+ CustomFileModel , id = request .POST .get ("file_id" )
415+ )
416+
404417 if request .user != file .created_by and not (
405418 request .user .is_superuser
406419 or request .user .has_perm ("podfile.change_customfilemodel" )
@@ -410,9 +423,14 @@ def changefile(request):
410423 messages .add_message (request , messages .ERROR , _ ("You cannot edit this file." ))
411424 raise PermissionDenied
412425
413- form_file = eval ("%sForm" % request .POST .get ("file_type" ))(
414- request .POST , request .FILES , instance = file
415- )
426+ if file_type == "CustomImageModel" :
427+ form_file = CustomImageModelForm (
428+ request .POST , request .FILES , instance = file
429+ )
430+ else :
431+ form_file = CustomFileModelForm (
432+ request .POST , request .FILES , instance = file
433+ )
416434
417435 if form_file .is_valid ():
418436 if form_file .cleaned_data ["folder" ] != folder :
0 commit comments