Excpetion handling when loading json file from cli

Author: andrewhabib

jsonsubschema/_utils.py

@@ -10,6 +10,7 @@
 import numbers
 import re
 import sys
+import json
 
 import jsonschema
 import intervals as I
@@ -332,3 +333,16 @@ def are_intervals_mergable(i1, i2):
     return i1.overlaps(i2) \
         or (is_num(i1.lower) and is_num(i2.upper) and i1.lower - i2.upper == 1) \
         or (is_num(i2.lower) and is_num(i1.upper) and i2.lower - i1.upper == 1)
+
+
+def load_json_file(path, msg=None):
+    with open(path, "r") as fh:
+        try:
+            return json.load(fh)
+        except Exception as e:
+            if msg:
+                print(msg, e)
+                sys.exit(1)
+            else:
+                print(e)
+                sys.exit(1)

jsonsubschema/cli.py

@@ -3,10 +3,9 @@
 @author: Andrew Habib
 '''
 
-import json
-import jsonref
 import sys
 
+from jsonsubschema._utils import load_json_file
 from jsonsubschema.api import isSubschema
 
 
@@ -18,12 +17,8 @@ def main():
     s1_file = sys.argv[1]
     s2_file = sys.argv[2]
 
-    with open(s1_file, 'r') as f1:
-        s1 = json.load(f1)
-        # s1 = jsonref.load(f1)
-    with open(s2_file, 'r') as f2:
-        s2 = json.load(f2)
-        # s2 = jsonref.load(f2)
+    s1 = load_json_file(s1_file, "LHS file:")
+    s2 = load_json_file(s2_file, "RHS file:")
 
     print("LHS <: RHS", isSubschema(s1, s2))
     print("RHS <: LHS", isSubschema(s2, s1))