a1/utils.py

   1 # ==================================================================================
   2 #       Copyright (c) 2019 Nokia
   3 #       Copyright (c) 2018-2019 AT&T Intellectual Property.
   4 #
   5 #   Licensed under the Apache License, Version 2.0 (the "License");
   6 #   you may not use this file except in compliance with the License.
   7 #   You may obtain a copy of the License at
   8 #
   9 #          http://www.apache.org/licenses/LICENSE-2.0
  10 #
  11 #   Unless required by applicable law or agreed to in writing, software
  12 #   distributed under the License is distributed on an "AS IS" BASIS,
  13 #   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 #   See the License for the specific language governing permissions and
  15 #   limitations under the License.
  16 # ==================================================================================
  17 import json
  18 from jsonschema import validate
  19 from a1 import get_module_logger
  20 from a1 import exceptions
  21
  22
  23 logger = get_module_logger(__name__)
  24
  25
  26 def _get_rmr_mapping_table(cache={}):
  27     """
  28     Get the rmr mapping file
  29     Broken out for ease of unit testing
  30     """
  31     try:
  32         return open("/opt/rmr_string_int_mapping.txt", "r").readlines()
  33     except FileNotFoundError:
  34         logger.error("Missing RMR Mapping!")
  35         raise exceptions.MissingRmrMapping
  36
  37
  38 # Public
  39
  40
  41 def validate_json(instance, schema):
  42     """
  43     validate incoming policy payloads
  44     """
  45     validate(instance=instance, schema=schema)
  46
  47
  48 def get_ric_manifest():
  49     """
  50     Get the ric level manifest
  51     """
  52     try:
  53         with open("/opt/ricmanifest.json", "r") as f:
  54             content = f.read()
  55             manifest = json.loads(content)
  56             return manifest
  57     except FileNotFoundError:
  58         logger.error("Missing A1 Manifest!")
  59         raise exceptions.MissingManifest
  60
  61
  62 def rmr_string_to_int(rmrs, cache={}):
  63     """
  64     map an rmr string to an int.
  65     TODO: should we allow for dynamic updates to this file? If so, we shouldn't cache
  66     """
  67     if cache == {}:  # fill the cache if not populated
  68         lines = _get_rmr_mapping_table()
  69         for l in lines:
  70             items = l.split(":")
  71             cache[items[0]] = int(items[1])
  72
  73     if rmrs in cache:
  74         return cache[rmrs]
  75     raise exceptions.MissingRmrString(rmrs)