From 7208cdd8dcd45d89adbeafffbd342346ace949aa Mon Sep 17 00:00:00 2001
From: Reiko Asakura
Date: Sun, 14 Mar 2021 17:08:10 -0400
Subject: Use ruamel instead of PyYAML

---
 scripts/nids.py | 13 ++-----------
 1 file changed, 2 insertions(+), 11 deletions(-)

(limited to 'scripts/nids.py')

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):
 
-- 
cgit v1.2.3