diff --git a/core/management/commands/load_by_doab.py b/core/management/commands/load_by_doab.py index 611f18ed..beb32483 100644 --- a/core/management/commands/load_by_doab.py +++ b/core/management/commands/load_by_doab.py @@ -4,7 +4,10 @@ from regluit.core.loaders import doab class Command(BaseCommand): help = "load doab books by doab_id via oai" - args = "" + + def add_arguments(self, parser): + parser.add_argument('doab_ids', nargs='+', type=int, default=1, help="doab ids to add") - def handle(self, doab_id, **options): - doab.add_by_doab(doab_id) + def handle(self, doab_ids, **options): + for doab_id in doab_ids: + doab.add_by_doab(doab_id)