feat: split metadata into smal pieces

Reviewed-by: Hasko, Vladimir <vladimir.hasko@t-systems.com>
Co-authored-by: gtema <artem.goncharov@gmail.com>
Co-committed-by: gtema <artem.goncharov@gmail.com>
This commit is contained in:
2023-06-02 07:20:02 +00:00
committed by zuul
parent 177ba3a3fa
commit bac06c250f
108 changed files with 1133 additions and 538 deletions

View File

@ -12,30 +12,45 @@
# limitations under the License.
import os
import pathlib
import yaml
__all__ = ['read_data']
__all__ = ["read_data"]
DATA_DIR = os.path.dirname(__file__)
def read_data(filename):
"""Return data that is shipped inside the Python package.
"""
"""Return data that is shipped inside the Python package."""
filepath = os.path.join(DATA_DIR, filename)
with open(filepath, 'r') as fd:
return yaml.safe_load(fd)
with open(filepath, "r") as fd:
data = yaml.safe_load(fd)
# Merge data found in individual element files
data.setdefault("documents", list())
data.setdefault("services", list())
data.setdefault("service_categories", list())
for item in pathlib.Path(DATA_DIR, "documents").glob("*.yaml"):
with open(item, "r") as fp:
data["documents"].append(yaml.safe_load(fp))
for item in pathlib.Path(DATA_DIR, "services").glob("*.yaml"):
with open(item, "r") as fp:
data["services"].append(yaml.safe_load(fp))
for item in pathlib.Path(DATA_DIR, "service_categories").glob(
"*.yaml"
):
with open(item, "r") as fp:
data["service_categories"].append(yaml.safe_load(fp))
return data
def rewrite_data(filename, data):
"""Rewrites data formatting it
"""
"""Rewrites data formatting it"""
from ruamel.yaml import YAML
_yaml = YAML()
_yaml.indent(mapping=2, sequence=4, offset=2)
filepath = os.path.join(DATA_DIR, filename)
with open(filepath, 'w') as fd:
with open(filepath, "w") as fd:
_yaml.dump(data, fd)