Changeset 4852 for waeup

Timestamp:

18 Jan 2010, 09:06:22 (15 years ago)

Author:

uli

Message:

When uploading files, insert username before filename extension.

File:

• waeup/branches/ulif-importers/src/waeup/browser/pages.py (modified) (3 diffs)

waeup/branches/ulif-importers/src/waeup/browser/pages.py

@@ -6 +6 @@
 import grok
 import os
+import re
 import sys
 from hurry import yui
@@ -270 +271 @@
         try:
             filename = uploadfile.filename
-            target = os.path.join(self.context.storage, filename)
+            target = os.path.join(self.context.storage,
+                                  self.getNormalizedFileName(filename))
             open(target, 'wb').write(uploadfile.read())
         except IOError:
@@ -278 +280 @@
         self.redirect(self.url(self.context))
 
+    def getNormalizedFileName(self, filename):
+        """Build sane filename.
+
+        An uploaded file foo.csv will be stored as foo_USERNAME.csv
+        where username is the principal id of the currently logged in
+        user.
+
+        Spaces in filename are replaced by underscore.
+        """
+        username = self.request.principal.id
+        filename = filename.replace(' ', '_')
+        # Only accept typical filname chars...
+        filtered_username = ''.join(re.findall('[a-zA-Z0-9_\.\-]', username))
+        base, ext = os.path.splitext(filename)
+        return '%s_%s%s' % (base, filtered_username, ext.lower())
 
 class DataCenterImportCSVPage(WAeUPPage):