summaryrefslogtreecommitdiffstats
path: root/vio/vio/api_v2/api_definition/utils.py
blob: a8da2a7a623dd0571bc1449bf178bb9eb6e8f3b4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
# Copyright (c) 2017-2018 VMware, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at:
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

import pkg_resources
import yaml


def get_definition_list():
    """ Get API Definition from YAML files. """

    api_def = []
    definition_dir = __name__[:__name__.rfind(".")]
    for f in pkg_resources.resource_listdir(definition_dir, '.'):
        if not f.endswith(".yaml"):
            continue

        with pkg_resources.resource_stream(definition_dir, f) as fd:
            # TODO(xiaohhui): Should add exception handler to inform user of
            # potential error.
            api_def.append(yaml.safe_load(fd))

    return api_def