importpkg.py + readyaml.py: prefer the C libyaml implementation

2 files changed, 2 insertions, 2 deletions

diff --git a/importpkg.py b/importpkg.py
index 7bca70b..4f00407 100755
--- a/importpkg.py
+++ b/importpkg.py
@@ -79,7 +79,7 @@ def main():
                         type=urllib.request.urlopen,
                         help="read from this location instead of stdin")
     args = parser.parse_args()
-    dumper = yaml.SafeDumper(sys.stdout)
+    dumper = yaml.CSafeDumper(sys.stdout)
     dumper.open()
     if args.hash:
         args.input = HashedStream(args.input, hashlib.sha256())
diff --git a/readyaml.py b/readyaml.py
index a4837cf..b88d1e0 100755
--- a/readyaml.py
+++ b/readyaml.py
@@ -12,7 +12,7 @@ import yaml
 def readyaml(db, stream):
     cur = db.cursor()
     cur.execute("PRAGMA foreign_keys = ON;")
-    gen = yaml.safe_load_all(stream)
+    gen = yaml.load_all(stream, yaml.CSafeLoader)
     metadata = next(gen)
     package = metadata["package"]
     cur.execute("SELECT id, version FROM package WHERE name = ?;",