X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=api%2Fdatabase%2Fv2%2Fhandlers.py;h=1bc32bf0e26606baaa1624af5a59c43c9dd3510a;hb=deccb02a8adabe609c2e903ce9f60e8b39db6c29;hp=095ad724c359e5d52394d88de6beb088f346fb97;hpb=48c0cb1b5f3ccd023df1c4ae708236d2f459bfa8;p=yardstick.git

diff --git a/api/database/v2/handlers.py b/api/database/v2/handlers.py
index 095ad724c..1bc32bf0e 100644
--- 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)
@@ -130,6 +136,9 @@ class V2ContainerHandler(object):
 
 class V2ProjectHandler(object):
 
+    def list_all(self):
+        return V2Project.query.all()
+
     def insert(self, kwargs):
         project = V2Project(**kwargs)
         db_session.add(project)
@@ -164,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)