Feature 10954 to automatically select WIMs for inter-datacenter networks

[osm/LCM.git] / osm_lcm / data_utils / database / wim_account.py

diff --git a/osm_lcm/data_utils/database/wim_account.py b/osm_lcm/data_utils/database/wim_account.py
new file mode 100644 (file)
index 0000000..8b0b5f6
--- /dev/null
+++ b/osm_lcm/data_utils/database/wim_account.py
@@ -0,0 +1,48 @@
+# -*- coding: utf-8 -*-
+
+# This file is part of OSM Life-Cycle Management module
+#
+# Copyright 2022 ETSI
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License. You may obtain
+# a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+##
+
+from osm_lcm.data_utils.database.database import Database
+
+__author__ = (
+    "Lluis Gifre <lluis.gifre@cttc.es>, Ricard Vilalta <ricard.vilalta@cttc.es>"
+)
+
+
+class WimAccountDB:
+    db = None
+    db_wims = {}
+
+    def initialize_db():
+        WimAccountDB.db = Database().instance.db
+
+    def get_wim_account_with_id(wim_account_id):
+        if not WimAccountDB.db:
+            WimAccountDB.initialize_db()
+        if wim_account_id in WimAccountDB.db_wims:
+            return WimAccountDB.db_wims[wim_account_id]
+        db_wim = WimAccountDB.db.get_one("wim_accounts", {"_id": wim_account_id}) or {}
+        WimAccountDB.db_wims[wim_account_id] = db_wim
+        return db_wim
+
+    def get_all_wim_accounts():
+        if not WimAccountDB.db:
+            WimAccountDB.initialize_db()
+        db_wims_list = WimAccountDB.db.get_list("wim_accounts")
+        WimAccountDB.db_wims.update({db_wim["_id"]: db_wim for db_wim in db_wims_list})
+        return WimAccountDB.db_wims