Extracts method

Author: pedrocamargo

gmnspy/schema.py

@@ -37,6 +37,27 @@
 }
 
 
+def read_schema_for_resource(resource_df: pd.DataFrame, table_name: str, raise_error: bool) -> dict:
+    """
+    Read in schema from schema json file and returns as dictionary.
+
+    ##TODO validate schema itself
+
+    Args:
+        schema_file: File location of the schema json file.
+
+    Returns: The schema as a dictionary
+    """
+    matching_schema_paths = resource_df.loc[resource_df["name"] == table_name, "fullpath_schema"]
+    if matching_schema_paths.empty:
+        msg = f"FAIL. Could not find schema path for table {table_name}"
+        logger.error(msg)
+        if raise_error:
+            raise Exception(msg)
+        return dict()
+    return read_schema(schema_file=matching_schema_paths.iloc[0])
+
+
 def read_schema(schema_file: str) -> dict:
     """
     Read in schema from schema json file and returns as dictionary.

gmnspy/validation/foreign_keys.py

@@ -4,7 +4,7 @@
 
 import pandas as pd
 
-from gmnspy.schema import read_schema
+from gmnspy.schema import read_schema_for_resource
 from gmnspy.utils import logger
 
 
@@ -28,14 +28,9 @@ def validate_foreign_keys(gmns_net_d: Dict[str, pd.DataFrame], resource_df: pd.D
 
     fkey_errors = []
     for table_name, df in gmns_net_d.items():
-        matching_schema_paths = resource_df.loc[resource_df["name"] == table_name, "fullpath_schema"]
-        if matching_schema_paths.empty:
-            msg = f"FAIL. Could not find schema path for table {table_name}"
-            logger.error(msg)
-            if raise_error:
-                raise Exception(msg)
+        schema = read_schema_for_resource(resource_df, table_name, raise_error)
+        if len(schema) == 0:
             continue
-        schema = read_schema(schema_file=matching_schema_paths.iloc[0])
 
         foreign_keys = [
             (f["name"], f["foreign_key"])