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:
@ -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)
|
||||
|
Reference in New Issue
Block a user