Add DB refresh/expire on specific parameters

Previously we could only refresh or expire the entire object.

Change-Id: I2207349d6ffff435538a577943316cd13e7f1355
Signed-off-by: Zane Bitter <zbitter@redhat.com>

diff --git a/heat/db/sqlalchemy/models.py b/heat/db/sqlalchemy/models.py
index 25f63b0891b733845a0483642ad9baf8859c32b2..625ee7fd3e763ea609fb9f1ed011235ec7d44be9 100644 (file)
--- a/heat/db/sqlalchemy/models.py
+++ b/heat/db/sqlalchemy/models.py
@@ -61,21 +61,21 @@ class HeatBase(object):
             else:
                 raise
 
-    def expire(self, session=None):
+    def expire(self, session=None, attrs=None):
         """Expire this object ()."""
         if not session:
             session = Session.object_session(self)
             if not session:
                 session = get_session()
-        session.expire(self)
+        session.expire(self, attrs)
 
-    def refresh(self, session=None):
+    def refresh(self, session=None, attrs=None):
         """Refresh this object."""
         if not session:
             session = Session.object_session(self)
             if not session:
                 session = get_session()
-        session.refresh(self)
+        session.refresh(self, attrs)
 
     def delete(self, session=None):
         """Delete this object."""