Remove unnecessary aspy.yaml

asottile · asottile · commit 3bc1c31ce770 · 2020-01-31T20:48:15.000-08:00
diff --git a/make_all_hooks.py b/make_all_hooks.py
@@ -1,4 +1,4 @@
-import collections
+import functools
 import json
 import multiprocessing
 import os.path
@@ -8,9 +8,12 @@
 from typing import Dict
 from typing import Tuple
 
-import aspy.yaml
+import yaml
 from pre_commit.clientlib import load_manifest
 
+Loader = getattr(yaml, 'CSafeLoader', yaml.SafeLoader)
+fast_load = functools.partial(yaml.load, Loader=Loader)
+
 
 def get_manifest(repo_path: str) -> Tuple[str, Dict[str, Any]]:
     print(f'*** {repo_path}')
@@ -21,13 +24,16 @@ def get_manifest(repo_path: str) -> Tuple[str, Dict[str, Any]]:
         manifest_path = os.path.join(repo_dir, '.pre-commit-hooks.yaml')
         # Validate the manifest just to make sure it's ok.
         load_manifest(manifest_path)
-        return (repo_path, aspy.yaml.ordered_load(open(manifest_path)))
+        with open(manifest_path) as f:
+            return repo_path, fast_load(f)
 
 
 def main() -> int:
-    repos = aspy.yaml.ordered_load(open('all-repos.yaml'))
+    with open('all-repos.yaml') as f:
+        repos = fast_load(f)
+
     pool = multiprocessing.Pool(4)
-    hooks_json = collections.OrderedDict(pool.map(get_manifest, repos))
+    hooks_json = dict(pool.map(get_manifest, repos))
 
     with open('all-hooks.json', 'w') as hooks_json_file:
         json.dump(hooks_json, hooks_json_file, indent=4)
diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -1,4 +1,3 @@
-aspy.yaml
 mako
 libsass
 markdown-code-blocks
