123456789101112131415161718192021222324252627 |
- import os
- import yaml
- class YAMLFileNotFound(Exception):
- pass
- class YAMLLoader:
- def __init__(self):
- pass
- @classmethod
- def get_config(cls, yaml_file):
- """
- Load settings file
- :return: cfg : the configuration file
- """
- # Load settings.
- __location__ = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__)))
- try:
- with open(os.path.join(__location__, yaml_file)) as ymlfile:
- cfg = yaml.load(ymlfile)
- return cfg
- except IOError:
- raise YAMLFileNotFound("The file path %s does not exist" % yaml_file)
|