From f58f110436a65a37a8cfacadf8bc4bafa3b18817 Mon Sep 17 00:00:00 2001 From: Patrick Robertson Date: Wed, 26 Feb 2025 17:59:13 +0000 Subject: [PATCH] Check at least 1 URL provided for new cli_feeder module rewrite --- src/auto_archiver/modules/cli_feeder/cli_feeder.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/auto_archiver/modules/cli_feeder/cli_feeder.py b/src/auto_archiver/modules/cli_feeder/cli_feeder.py index 1f1fe26..20ca6ae 100644 --- a/src/auto_archiver/modules/cli_feeder/cli_feeder.py +++ b/src/auto_archiver/modules/cli_feeder/cli_feeder.py @@ -5,6 +5,11 @@ from auto_archiver.core.metadata import Metadata class CLIFeeder(Feeder): + def setup(self) -> None: + self.urls = self.config['urls'] + if not self.urls: + raise ValueError("No URLs provided. Please provide at least one URL via the command line, or set up an alternative feeder. Use --help for more information.") + def __iter__(self) -> Metadata: urls = self.config['urls'] for url in urls: