ingest/registry.py

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"
]