Validation functions modified to check for the presence of required columns in the input files without enforcing a specific order

school_center.py

@@ -142,6 +142,58 @@ def is_allocated(scode1: str, scode2:str) -> bool:
         return allocations[scode1].get(scode2) != None
     else:
         return False
+
+# validating the format of the school input file
+def validate_school_file(file_path: str) -> bool:
+
+    if not os.path.exists(file_path):
+        print(f"School file '{file_path}' not found.")
+        return False
+
+    # open the file and read the header
+    with open(file_path, 'r', newline='', encoding='utf-8') as file:
+        header = file.readline().strip().split('\t')
+        required_columns = ['scode', 'count', 'name-address', 'lat', 'long']
+        if not all(col in header for col in required_columns):
+            print(f"Invalid school file format. Required columns: {required_columns}")
+            return False
+
+    return True
+
+# validating the format of the center input file
+def validate_center_file(file_path: str) -> bool:
+    # check if file exists
+    if not os.path.exists(file_path):
+        print(f"Center file '{file_path}' not found.")
+        return False
+    # open files and read header
+    with open(file_path, 'r', newline='', encoding='utf-8') as file:
+        header = file.readline().strip().split('\t')
+        required_columns = ['cscode', 'capacity', 'name', 'address', 'lat', 'long']
+        # compare the header with required columns
+        if not all(col in header for col in required_columns):
+            print(f"Invalid center file format. Required columns: {required_columns}")
+            return False
+
+    return True
+
+# validating the format of the preference input file
+def validate_preference_file(file_path: str) -> bool:
+    # check if the file exists
+    if not os.path.exists(file_path):
+        print(f"Preference file '{file_path}' not found.")
+        return False
+
+    # open the file and read the header
+    with open(file_path, 'r', newline='', encoding='utf-8') as file:
+        header = file.readline().strip().split('\t')
+        required_columns = ['scode', 'cscode', 'pref', 'reason']
+        # compare the header with required columns
+        if not all(col in header for col in required_columns):
+            print(f"Invalid preference file format. Required columns: {required_columns}")
+            return False
+
+    return True
 
 parser = argparse.ArgumentParser(
                     prog='center randomizer',
@@ -214,4 +266,4 @@ def is_allocated(scode1: str, scode2:str) -> bool:
     logger.info("Remaining capacity at each center (remaining_capacity cscode):")
     logger.info(sorted([(v,k) for k, v in centers_remaining_cap.items() if v != 0]))
     logger.info(f"Total remaining capacity across all centers: {sum({k:v for k, v in centers_remaining_cap.items() if v != 0}.values())}")
-    logger.info(f"Students not assigned: {remaining}")
+    logger.info(f"Students not assigned: {remaining}")