Move the body of set_last_modified to main/utils

Instead of having multiple methods, move this into our single 'created'
setter method. If the 'last_modified' property is present, we now update
it accordingly when saving any model with this signal attached.

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

1 files changed, 8 insertions, 2 deletions

diff --git a/main/utils.py b/main/utils.py
index d12e5e1a..17ca386e 100644
--- a/main/utils.py
+++ b/main/utils.py
@@ -106,10 +106,16 @@ def retrieve_latest(sender, latest_by=None):
 
 def set_created_field(sender, **kwargs):
     '''This will set the 'created' field on any object to the current UTC time
-    if it is unset. For use as a pre_save signal handler.'''
+    if it is unset.
+    Additionally, this will set the 'last_modified' field on any object to the
+    current UTC time on any save of the object.
+    For use as a pre_save signal handler.'''
     obj = kwargs['instance']
+    time = now()
     if hasattr(obj, 'created') and not obj.created:
-        obj.created = now()
+        obj.created = time
+    if hasattr(obj, 'last_modified'):
+        obj.last_modified = time
 
 
 def database_vendor(model, mode='read'):