lcm/workflows/graphflow/flow/load.py

   1 # Copyright 2018 ZTE Corporation.
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #         http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14
  15 import importlib
  16 import logging
  17
  18
  19 logger = logging.getLogger(__name__)
  20
  21
  22 def load_module(imp_module):
  23     try:
  24         imp_module = importlib.import_module(imp_module)
  25     except Exception:
  26         logger.debug("load_module error: %s", imp_module)
  27         imp_module = None
  28     return imp_module
  29
  30
  31 def load_class(imp_module, imp_class):
  32     try:
  33         cls = getattr(imp_module, imp_class)
  34     except Exception:
  35         logger.debug("load_class error: %s", imp_class)
  36         cls = None
  37     return cls
  38
  39
  40 def load_class_from_config(config):
  41     class_set = {}
  42     for k, v in list(config.items()):
  43         imp_module = load_module(v["module"])
  44         cls = load_class(imp_module, v["class"])
  45         class_set[k] = cls
  46     return class_set