From 17d75d44c428de036655767bae2932075e544d19 Mon Sep 17 00:00:00 2001
From: nishanth
Date: Fri, 26 Feb 2010 15:15:16 +0530
Subject: modified image storage.

---
 taskapp/models.py | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

(limited to 'taskapp/models.py')

diff --git a/taskapp/models.py b/taskapp/models.py
index db7d6ab..6429336 100644
--- a/taskapp/models.py
+++ b/taskapp/models.py
@@ -26,6 +26,13 @@ STATUS_CHOICES = (
 IMAGES_DIR = "./images"
 UPLOADS_DIR = "./uploads"
 
+def get_key():
+    """ generate a 10 character name with random alphabets and digits.
+    """
+
+    name = ''.join([ random.choice(string.uppercase+string.digits) for i in range(10)])
+    return name
+
 class CustomImageStorage(FileSystemStorage):
 
     def path(self, name):
@@ -39,10 +46,10 @@ class CustomImageStorage(FileSystemStorage):
         """
     
         root, ext = os.path.splitext(name)
-        name = ''.join([ random.choice(string.uppercase+string.digits) for i in range(10)])+ext
-        while self.exists(name):
-            name = ''.join([ random.choice(string.uppercase+string.digits) for i in range(10)])+ext
-        return name
+        file_name = get_key() + ext
+        while self.exists(file_name):
+            file_name = get_key() + ext
+        return file_name
 
 class Profile(models.Model):
     
-- 
cgit