Merge "base: remove redudant ip_to_convert"

[yardstick.git] / api / database / v2 / handlers.py

diff --git a/api/database/v2/handlers.py b/api/database/v2/handlers.py
index 7aa02a0..1bc32bf 100644 (file)
--- a/api/database/v2/handlers.py
+++ b/api/database/v2/handlers.py
@@ -122,6 +122,12 @@ class V2ContainerHandler(object):
             raise ValueError
         return container
 
+    def update_attr(self, uuid, attr):
+        container = self.get_by_uuid(uuid)
+        for k, v in attr.items():
+            setattr(container, k, v)
+        db_session.commit()
+
     def delete_by_uuid(self, uuid):
         container = self.get_by_uuid(uuid)
         db_session.delete(container)
@@ -167,6 +173,9 @@ class V2ProjectHandler(object):
 
 class V2TaskHandler(object):
 
+    def list_all(self):
+        return V2Task.query.all()
+
     def insert(self, kwargs):
         task = V2Task(**kwargs)
         db_session.add(task)