Move set_created_field() to shared utils class

Signed-off-by: Dan McGee <dan@archlinux.org>

1 files changed, 10 insertions, 0 deletions

diff --git a/main/utils.py b/main/utils.py
index 12d1250..2ca69bc 100644
--- a/main/utils.py
+++ b/main/utils.py
@@ -2,6 +2,9 @@ try:
     import cPickle as pickle
 except ImportError:
     import pickle
+
+from datetime import datetime
+
 from django.core.cache import cache
 from django.utils.hashcompat import md5_constructor
 
@@ -81,4 +84,11 @@ def retrieve_latest(sender):
         pass
     return None
 
+def set_created_field(sender, **kwargs):
+    '''This will set the 'created' field on any object to datetime.utcnow() if
+    it is unset. For use as a pre_save signal handler.'''
+    obj = kwargs['instance']
+    if hasattr(obj, 'created') and not obj.created:
+        obj.created = datetime.utcnow()
+
 # vim: set ts=4 sw=4 et: