Skip to content

index_reference_data#

Command#

Bases: BaseCommand

Source code in src/apps/refdata/management/commands/index_reference_data.py 
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
class Command(BaseCommand):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.type_choices = [*settings.REFERENCE_DATA_SOURCES.keys()]

    def add_arguments(self, parser):
        parser.add_argument(
            "types",
            nargs="*",
            help=f"List of reference data types to index. If omitted, index all types. Available: {self.type_choices}",
        )

    def handle(self, *args, **options):
        types = options["types"] or self.type_choices

        if len(types) != len(set(types)):
            raise CommandError("Duplicate arguments supplied")

        unknown = set(types) - set(self.type_choices)
        if len(unknown) != 0:
            raise CommandError(f"Unknown types: {sorted(unknown)}, available: {self.type_choices}")

        return indexer.index(types=types)