1 files changed, 69 insertions, 0 deletions
diff --git a/apps/pci/optimizers/configdb.py b/apps/pci/optimizers/configdb.py
new file mode 100644
index 0000000..8f003c2
--- /dev/null
+++ b/apps/pci/optimizers/configdb.py
@@ -0,0 +1,69 @@
+# -------------------------------------------------------------------------
+#   Copyright (c) 2018 AT&T Intellectual Property
+#
+#   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 datetime import datetime as dt
+
+from osdf.logging.osdf_logging import debug_log
+from osdf.utils.interfaces import RestClient
+
+
+def request(req_object, osdf_config, flat_policies):
+    """
+    Process a configdb request from a Client (build Conductor API call, make the call, return result)
+    :param req_object: Request parameters from the client
+    :param osdf_config: Configuration specific to OSDF application (core + deployment)
+    :param flat_policies: policies related to PCI Opt (fetched based on request)
+    :return: response from ConfigDB (accounting for redirects from Conductor service
+    """
+    cell_list_response = {}
+    config = osdf_config.deployment
+    local_config = osdf_config.core
+    uid, passwd = config['configDbUserName'], config['configDbPassword']
+    req_id = req_object['requestInfo']['requestId']
+    transaction_id = req_object['requestInfo']['transactionId']
+    headers = dict(transaction_id=transaction_id)
+
+    network_id = req_object['cellInfo']['networkId']
+
+    cell_list_response['network_id'] = network_id
+
+    ts = dt.strftime(dt.now(), '%Y-%m-%d %H:%M:%S%z')
+
+    rc = RestClient(userid=uid, passwd=passwd, method="GET", log_func=debug_log.debug, headers=headers)
+
+    cell_list_url = '{}/{}/{}/{}'.format(config['configDbUrl'], config['configDbGetCellListUrl'], network_id, ts)
+
+    cell_list_resp = rc.request(raw_response=True, url=cell_list_url)
+    cell_resp = cell_list_resp.json()
+
+    cell_list = []
+    count = 0
+    for cell_id in cell_resp:
+        cell_info = {'cell_id': cell_id, 'id': count}
+        nbr_list_url = '{}/{}/{}/{}'.format(config['configDbUrl'], config['configDbGetNbrListUrl'], cell_id, ts)
+        nbr_list_raw = rc.request(url=nbr_list_url, raw_response=True)
+        cell_info['nbr_list'] = get_neighbor_list(nbr_list_raw.json())
+        cell_list.append(cell_info)
+        count += 1
+
+    cell_list_response['cell_list'] = cell_list
+    return cell_resp, cell_list_response
+
+
+def get_neighbor_list(nbr_list_response):
+    return nbr_list_response.get('nbrList', [])