netdata/python.d/python_modules/bases/loaders.py at master · jamesblunt/netdata · GitHub

History

66 lines (52 loc) · 1.72 KB

Raw

# -*- coding: utf-8 -*-

# Description:

# Author: Ilya Mashchenko (l2isbad)

import types

from sys import version_info

PY_VERSION = version_info[:2]

if PY_VERSION > (3, 1):

from pyyaml3 import SafeLoader as YamlSafeLoader

from importlib.machinery import SourceFileLoader

DEFAULT_MAPPING_TAG = 'tag:yaml.org,2002:map'

else:

from pyyaml2 import SafeLoader as YamlSafeLoader

from imp import load_source as SourceFileLoader

DEFAULT_MAPPING_TAG = u'tag:yaml.org,2002:map'

try:

from collections import OrderedDict

except ImportError:

from third_party.ordereddict import OrderedDict

def dict_constructor(loader, node):

return OrderedDict(loader.construct_pairs(node))

YamlSafeLoader.add_constructor(DEFAULT_MAPPING_TAG, dict_constructor)

class YamlOrderedLoader:

@staticmethod

def load_config_from_file(file_name):

opened, loaded = False, False

try:

stream = open(file_name, 'r')

opened = True

loader = YamlSafeLoader(stream)

loaded = True

parsed = loader.get_single_data() or dict()

except Exception as error:

return dict(), error

else:

return parsed, None

finally:

if opened:

stream.close()

if loaded:

loader.dispose()

class SourceLoader:

@staticmethod

def load_module_from_file(name, path):

try:

loaded = SourceFileLoader(name, path)

if isinstance(loaded, types.ModuleType):

return loaded, None

return loaded.load_module(), None

except Exception as error:

return None, error

class ModuleAndConfigLoader(YamlOrderedLoader, SourceLoader):

pass

netdata/python.d/python_modules/bases/loaders.py at master · jamesblunt/netdata · GitHub

Expand file tree

Breadcrumbs

loaders.py

Latest commit

History

Breadcrumbs

loaders.py

File metadata and controls