CSV Loader

loader_2_header.py

@@ -0,0 +1,53 @@
+import csv
+from enum import IntEnum
+from lib.csv_loader import CSVLoader, CSVRow
+from lib.csv_loader.csv_loader import BoolValuePair, CSVRowDefaultConfig
+from lib.csv_loader.mapping_strategies import MappingStrategyByHeader
+
+
+class ActionEnum(IntEnum):
+    INSERT = 0
+    UPDATE = 1
+    DELETE = 2
+
+
+class AssetRow(CSVRow):
+    asset_action: ActionEnum
+    asset_id: str
+    brand_id: int
+    model_id: int
+    serial_number: str
+    is_active: bool
+
+    class Config(CSVRowDefaultConfig):
+        bool_value_pair = BoolValuePair(true="YES", false="NO")
+
+
+def main():
+    with open("csv/assets.csv", "r") as f:
+        reader = csv.reader(f, delimiter=";")
+
+        mapping_strategy = MappingStrategyByHeader(model_cls=AssetRow)
+
+        csv_loader = CSVLoader[AssetRow](
+            reader=reader,
+            output_model_cls=AssetRow,
+            has_header=True,
+            aggregate_errors=True,
+            mapping_strategy=mapping_strategy,
+        )
+
+        result = csv_loader.read_rows()
+
+        "Results:"
+        for row in result.rows:
+            print(row)
+
+        if result.has_errors():
+            print("Errors:")
+            for error in result.errors:
+                print(f"Line: {error.line_number}: {error.original_error}")
+
+
+if __name__ == "__main__":
+    main()