expose geography in generate_schemas.py entrypoint

Author: thekaveman

schema/README.md

@@ -7,10 +7,10 @@ This directory contains the templates and code that _generate_ the official JSON
 At a command prompt within this `schema` directory run:
 
 ```bash
-python generate_schemas.py [--agency] [--policy] [--provider]
+python generate_schemas.py [--agency] [--geography] [--policy] [--provider]
 ```
 
-The optional flags `--agency`, `--policy`, and `--provider` can be used to specify which
+The optional flags `--agency`, `--geography`, `--policy`, and `--provider` can be used to specify which
 set of schemas to generate. The default is to generate all schemas.
 
 ## Updating Schemas

schema/generate_schemas.py

@@ -2,25 +2,30 @@
 Generate the JSON Schema documents for MDS endpoints.
 
 USAGE:
-    python generate_schemas.py [--agency] [--policy] [--provider]
+    python generate_schemas.py [--agency] [--geography] [--policy] [--provider]
 """
 
 import sys
 
 import agency
+import geography
 import policy
 import provider
 
 
 if __name__ == "__main__":
     if len(sys.argv) == 1:
         agency.write_schema_files()
+        geography.write_schema_files()
         policy.write_schema_files()
         provider.write_schema_files()
     else:
         if "--agency" in sys.argv:
             agency.write_schema_files()
             sys.argv.remove("--agency")
+        if "--geography" in sys.argv:
+            geography.write_schema_files()
+            sys.argv.remove("--geography")
         if "--policy" in sys.argv:
             policy.write_schema_files()
             sys.argv.remove("--policy")