Changeset 1830

Timestamp:

01/17/12 10:01:42 (4 months ago)

Author:

vic

Message:

LockForUpdate?: add native lock call for django >=1.4

Files:

• DjHDGutils/trunk/dbutils.py (modified) (2 diffs)

DjHDGutils/trunk/dbutils.py

@@ -249 +249 @@
 
 class LockForUpate(object):
-
-    def lock_for_update(self):
-        """
-        lock object for update
-        Return locked object
-
-        """
+    def _raw_lock(self):
         cursor = connection.cursor()
         cursor.execute("""select id from \"%s\" where id=%%s for update""" % self._meta.db_table,
@@ -263 +257 @@
         transaction.set_dirty()
         return type(self).objects.get(pk=self.id)
+        
+    def _native_lock(self):
+        return type(self).objects.select_for_update().get(pk=self.id)
+
+    def lock_for_update(self):
+        """
+        lock object for update
+        Return locked object
+
+        """
+        if hasattr(self._base_manager, 'select_for_update'):
+            return self._native_lock()
+
+        return self._raw_lock()