Cosmétique (début).

[Portfolio.git] / skins / portfolio_presentation_form.py

diff --git a/skins/portfolio_presentation_form.py b/skins/portfolio_presentation_form.py
index ab77d5d..e84a0bc 100755 (executable)
--- a/skins/portfolio_presentation_form.py
+++ b/skins/portfolio_presentation_form.py
@@ -1,36 +1,38 @@
 ##parameters=createPresentationPage=None, savePresentation=None, deletePresentation=None, defineSample=None, deleteSample=None, ajax=''
-
+from Products.CMFCore.utils import getToolByName
+atool = getToolByName(context, 'portal_attachment')
+attachments = atool.getAttachmentsFor(context)
 form = context.REQUEST.form
 
 if createPresentationPage and \
-       context.createPresentationPage() and \
-       context.setStatus(True, 'Object created.') and \
-       context.setRedirect(context, 'object/presentation', ajax=ajax):
-       return
+    context.createPresentationPage() and \
+    context.setStatus(True, 'Object created.') and \
+    context.setRedirect(context, 'object/presentation', ajax=ajax):
+    return
 
 if savePresentation and \
-       context.editPresentationPage(form.get('presentation_page', '')) and \
-       context.setStatus(True, 'Saved changes.') and \
-       context.setRedirect(context, 'object/view', ajax=ajax) :
-       return
+    context.editPresentationPage(form.get('presentation_page', '')) and \
+    context.setStatus(True, 'Saved changes.') :
+    attachments.removeUnusedAttachments(context.presentation_page)
+    return context.setRedirect(context, 'object/view', ajax=ajax)
 
 if deletePresentation and \
-       context.deletePresentationPage() and \
-       context.setStatus(True, 'Object deleted.') and \
-       context.setRedirect(context, 'object/view', ajax=ajax) :
-       return
+    context.deletePresentationPage() and \
+    context.setStatus(True, 'Object deleted.') :
+    attachments.removeUnusedAttachments('')
+    return context.setRedirect(context, 'object/view', ajax=ajax)
 
 if defineSample and \
-       context.setSamplePhoto(form.get('path')) and \
-       context.setStatus(True, 'Saved changes.') and \
-       context.setRedirect(context, 'object/view', ajax=ajax) :
-       return
+    context.setSamplePhoto(form.get('path')) and \
+    context.setStatus(True, 'Saved changes.') and \
+    context.setRedirect(context, 'object/view', ajax=ajax) :
+    return
 
 if deleteSample and \
-       context.setSamplePhoto(None) and \
-       context.setStatus(True, 'Saved changes.') and \
-       context.setRedirect(context, 'object/view', ajax=ajax) :
-       return
+    context.setSamplePhoto(None) and \
+    context.setStatus(True, 'Saved changes.') and \
+    context.setRedirect(context, 'object/view', ajax=ajax) :
+    return
 
 options = {}
 return context.portfolio_presentation_template(**options)
\ No newline at end of file