Improve codec script

Author: davxy

codec/validate.py

@@ -1,28 +1,22 @@
 #!/usr/bin/env python
 
-from pathlib import Path
 import asn1tools
+import glob
 import os
 import sys
 
 script_dir = os.path.dirname(os.path.abspath(__file__))
 sys.path.append(os.path.abspath(os.path.join(script_dir, '../jam-types-asn')))
-print(sys.path)
+
+from utils import get_schema_files, validate  # noqa: E402
 
 os.chdir(script_dir)
 
-from utils import get_schema_files, validate
+def validate_spec(spec_name, is_full):
+    print(f"[Validating '{spec_name}' spec]")
+    schema = asn1tools.compile_files(get_schema_files(is_full), codec="jer")
+    for json_file in glob.glob(f"{spec_name}/*.json"):
+        validate(schema, json_file)
 
-# Validate tiny
-print("[Validating Tiny Spec]")
-schema = asn1tools.compile_files(get_schema_files(False), codec="jer")
-for path in Path("tiny").iterdir():
-    if path.is_file() and path.suffix == ".json":
-        validate(schema, path)
- 
-# Validate full
-print("[Validating Full Spec]")
-schema = asn1tools.compile_files(get_schema_files(True), codec="jer")
-for path in Path("full").iterdir():
-    if path.is_file() and path.suffix == ".json":
-        validate(schema, path)
+validate_spec("tiny", False)
+validate_spec("full", True)