Share function that loads AutoPkg recipes

Author: homebysix

pre_commit_hooks/check_autopkg_recipes.py

@@ -4,17 +4,13 @@
 requirements."""
 
 import argparse
-import json
 import os
-import plistlib
 import sys
 from contextlib import contextmanager
 from distutils.version import LooseVersion
-from xml.parsers.expat import ExpatError
-
-from ruamel import yaml
 
 from pre_commit_hooks.util import (
+    load_autopkg_recipe,
     validate_pkginfo_key_types,
     validate_required_keys,
     validate_restart_action_key,
@@ -486,35 +482,10 @@ def main(argv=None):
     retval = 0
     for filename in args.filenames:
 
-        if filename.endswith(".yaml"):
-            try:
-                # try to read it as yaml
-                with open(filename, "rb") as f:
-                    recipe = yaml.safe_load(f)
-            except Exception as err:
-                print("{}: yaml parsing error: {}".format(filename, err))
-                retval = 1
-                break  # No need to continue checking this file
-
-        elif filename.endswith(".json"):
-            try:
-                # try to read it as json
-                with open(filename, "rb") as f:
-                    recipe = json.load(f)
-            except Exception as err:
-                print("{}: json parsing error: {}".format(filename, err))
-                retval = 1
-                break  # No need to continue checking this file
-
-        else:
-            try:
-                # try to read it as a plist
-                with open(filename, "rb") as f:
-                    recipe = plistlib.load(f)
-            except Exception as err:
-                print("{}: plist parsing error: {}".format(filename, err))
-                retval = 1
-                break  # No need to continue checking this file
+        recipe = load_autopkg_recipe(filename)
+        if not recipe:
+            retval = 1
+            break  # No need to continue checking this file
 
         # For future implementation of validate_unused_input_vars()
         # with open(filename, "r") as openfile:

pre_commit_hooks/forbid_autopkg_overrides.py

@@ -3,8 +3,8 @@
 """This hook prevents AutoPkg overrides from being added to the repo."""
 
 import argparse
-import plistlib
-from xml.parsers.expat import ExpatError
+
+from pre_commit_hooks.util import load_autopkg_recipe
 
 
 def build_argument_parser():
@@ -29,18 +29,14 @@ def main(argv=None):
 
     retval = 0
     for filename in args.filenames:
-        try:
-            with open(filename, "rb") as openfile:
-                recipe = plistlib.load(openfile)
-            for req_key in required_keys:
-                if req_key not in recipe:
-                    print("{}: possible AutoPkg recipe override".format(filename))
-                    retval = 1
-                    break  # No need to continue checking this file.
-
-        except (ExpatError, ValueError) as err:
-            print("{}: plist parsing error: {}".format(filename, err))
+        recipe = load_autopkg_recipe(filename)
+        if not recipe:
             retval = 1
+            break  # No need to continue checking this file.
+        for req_key in required_keys:
+            if req_key not in recipe:
+                print("{}: possible AutoPkg recipe override".format(filename))
+                retval = 1
 
     return retval
 

pre_commit_hooks/forbid_autopkg_trust_info.py

@@ -4,8 +4,7 @@
 repo."""
 
 import argparse
-import plistlib
-from xml.parsers.expat import ExpatError
+from pre_commit_hooks.util import load_autopkg_recipe
 
 
 def build_argument_parser():
@@ -27,16 +26,11 @@ def main(argv=None):
 
     retval = 0
     for filename in args.filenames:
-        try:
-            with open(filename, "rb") as openfile:
-                recipe = plistlib.load(openfile)
-            if "ParentRecipeTrustInfo" in recipe:
-                print("{}: trust info in recipe".format(filename))
-                retval = 1
-                break  # No need to continue checking this file.
-
-        except (ExpatError, ValueError) as err:
-            print("{}: plist parsing error: {}".format(filename, err))
+        recipe = load_autopkg_recipe(filename)
+        if not recipe:
+            retval = 1
+        elif "ParentRecipeTrustInfo" in recipe:
+            print("{}: trust info in recipe".format(filename))
             retval = 1
 
     return retval

pre_commit_hooks/util.py

@@ -1,9 +1,42 @@
 #!/usr/bin/python
 # -*- coding: utf-8 -*-
 
+import json
+import plistlib
 import sys
 from datetime import datetime
 
+from ruamel import yaml
+
+
+def load_autopkg_recipe(path):
+    """Loads an AutoPkg recipe in plist, yaml, or json format."""
+    recipe = None
+
+    if path.endswith(".yaml"):
+        try:
+            # try to read it as yaml
+            with open(path, "rb") as f:
+                recipe = yaml.safe_load(f)
+        except Exception as err:
+            print("{}: yaml parsing error: {}".format(path, err))
+    elif path.endswith(".json"):
+        try:
+            # try to read it as json
+            with open(path, "rb") as f:
+                recipe = json.load(f)
+        except Exception as err:
+            print("{}: json parsing error: {}".format(path, err))
+    else:
+        try:
+            # try to read it as a plist
+            with open(path, "rb") as f:
+                recipe = plistlib.load(f)
+        except Exception as err:
+            print("{}: plist parsing error: {}".format(path, err))
+
+    return recipe
+
 
 def validate_required_keys(plist, filename, required_keys):
     """Verifies that required_keys are present in dictionary plist."""