Feature-5950: Management of quotas in VIM Account

[osm/MON.git] / osm_mon / core / common_db.py

diff --git a/osm_mon/core/common_db.py b/osm_mon/core/common_db.py
index cf52d85..1e78bc8 100644 (file)
--- a/osm_mon/core/common_db.py
+++ b/osm_mon/core/common_db.py
@@ -144,6 +144,15 @@ class CommonDbClient:
                     )
         return vim_account
 
+    def set_vim_account(self, vim_account_id: str, update_dict: dict) -> bool:
+        try:
+            # Set vim_account resources in mongo
+            self.common_db.set_one('vim_accounts', {"_id": vim_account_id}, update_dict)
+            # self.common_db.set_one('vim_accounts', {"name": "test-vim"}, update_dict)
+            return True
+        except Exception:
+            return False
+
     def get_sdncs(self):
         return self.common_db.get_list("sdns")