12345678910111213141516171819 |
- import os
- import yaml
- class YAMLLoader:
- def __init__(self, yaml_file):
- self.file = yaml_file
- def get_config(self):
- """
- Load settings file
- :return: cfg : the configuration file
- """
- # Load settings.
- __location__ = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__)))
- with open(os.path.join(__location__, self.file)) as ymlfile:
- cfg = yaml.load(ymlfile)
- return cfg
|