artemis/artemis.py
changeset 69 88c088722a30
parent 67 30bd40ef9165
child 70 5796e44b0e17
--- a/artemis/artemis.py	Thu Apr 28 08:32:55 2011 -0700
+++ b/artemis/artemis.py	Thu Apr 28 08:50:31 2011 -0700
@@ -255,6 +255,8 @@
                 if not filename:
                     ext = mimetypes.guess_extension(part.get_content_type()) or ''
                     filename = 'attachment-%03d%s' % (counter, ext)
+                else:
+                    filename = os.path.basename(filename)
                 fp = open(filename, 'wb')
                 fp.write(part.get_payload(decode = True))
                 fp.close()
@@ -419,7 +421,7 @@
             # Encode the payload using Base64
             encoders.encode_base64(attachment)
         # Set the filename parameter
-        attachment.add_header('Content-Disposition', 'attachment', filename=filename)
+        attachment.add_header('Content-Disposition', 'attachment', filename=os.path.basename(filename))
         outer.attach(attachment)
     return outer