瀏覽代碼

startup: define functions yaml_dump & yaml_load

Fabian Peter Hammerle 3 年之前
父節點
當前提交
6e3ce76321
共有 1 個文件被更改,包括 11 次插入0 次删除
  1. 11 0
      profile_default/startup/init.py

+ 11 - 0
profile_default/startup/init.py

@@ -6,6 +6,7 @@ import numpy
 import pandas
 import scipy.io.wavfile
 import sympy
+import yaml
 from matplotlib import pyplot  # pylint: disable=unused-import; frequently used in shell
 
 # https://pandas.pydata.org/pandas-docs/stable/user_guide/options.html
@@ -74,3 +75,13 @@ def wavfile_read_mono(
     for channel_index in range(1, data.shape[1]):
         assert (data_first_channel == data[:, channel_index]).all()
     return rate, data_first_channel
+
+
+def yaml_dump(path: typing.Union[pathlib.Path, str], data: typing.Any) -> None:
+    with pathlib.Path(path).open("w") as stream:
+        yaml.safe_dump(data, stream)
+
+
+def yaml_load(path: typing.Union[pathlib.Path, str]) -> typing.Any:
+    with pathlib.Path(path).open("r") as stream:
+        return yaml.safe_load(stream)