Create ingest/registry.py

ingest/registry.py

@@ -0,0 +1,32 @@
+from typing import Dict, List
+from ingest.cia_reading_room import CIAAdapter
+from ingest.fbi_vault import FBIAdapter
+from ingest.cia_extended import CIAExtendedAdapter
+from ingest.nsa_extended import NSAExtendedAdapter
+from ingest.nro_extended import NROExtendedAdapter
+from ingest.dod_special_extended import (
+    AATIPExtendedAdapter,
+    SAPExtendedAdapter,
+    TENCAPExtendedAdapter,
+    SpecialActivitiesExtendedAdapter
+)
+
+def get_all_adapters():
+    return {
+        "CIA": CIAAdapter(),
+        "FBI": FBIAdapter(),
+
+        # Extended
+        "CIA_EXT": CIAExtendedAdapter(),
+        "NSA": NSAExtendedAdapter(),
+        "NRO": NROExtendedAdapter(),
+        "AATIP": AATIPExtendedAdapter(),
+        "SAP": SAPExtendedAdapter(),
+        "TENCAP": TENCAPExtendedAdapter(),
+        "SPECIAL_ACTIVITIES": SpecialActivitiesExtendedAdapter(),
+    }
+
+BASE_AGENCIES = ["CIA", "FBI"]
+EXTENDED_AGENCIES = [
+    "CIA_EXT", "NSA", "NRO", "AATIP", "SAP", "TENCAP", "SPECIAL_ACTIVITIES"
+]