diff options
| author | Reiko Asakura | 2021-03-14 17:08:10 -0400 |
|---|---|---|
| committer | Reiko Asakura | 2021-03-14 17:10:39 -0400 |
| commit | 7208cdd8dcd45d89adbeafffbd342346ace949aa (patch) | |
| tree | ed7645e0babd6a54fc33402155c616bfc0e006be /scripts/nids.py | |
| parent | Add ScePafThread functions (diff) | |
| download | vds-libraries-7208cdd8dcd45d89adbeafffbd342346ace949aa.tar.gz | |
Use ruamel instead of PyYAML
Diffstat (limited to 'scripts/nids.py')
| -rw-r--r-- | scripts/nids.py | 13 |
1 files changed, 2 insertions, 11 deletions
diff --git a/scripts/nids.py b/scripts/nids.py index 6966076..e9a5567 100644 --- a/scripts/nids.py +++ b/scripts/nids.py @@ -9,19 +9,10 @@ import sys from pathlib import Path -import yaml -try: - from yaml import CLoader as YamlLoader -except ImportError: - from yaml import SafeLoader as YamlLoader +from ruamel.yaml import YAML def load(nids): - if isinstance(nids, Path): - nids = yaml.load(nids.read_bytes(), Loader=YamlLoader) - elif not isinstance(nids, dict): - nids = yaml.load(nids, Loader=YamlLoader) - - return nids + return nids if isinstance(nids, dict) else YAML(typ='safe').load(nids) def process(nids, modules_cb=None, module_cb=None, libraries_cb=None, library_cb=None, functions_cb=None, variables_cb=None, export_cb=None, user_data=None): |