diff --git a/src/bin/pegleg/pegleg/cli.py b/src/bin/pegleg/pegleg/cli.py index c09ceab4..4d5b02f1 100644 --- a/src/bin/pegleg/pegleg/cli.py +++ b/src/bin/pegleg/pegleg/cli.py @@ -69,7 +69,6 @@ def site(primary_repo, aux_repo): 'save_location', type=click.Path( file_okay=False, dir_okay=True, writable=True, resolve_path=True), - default=sys.stdout, help='Where to output') @click.argument('site_name') def collect(*, save_location, site_name): diff --git a/src/bin/pegleg/pegleg/engine/site.py b/src/bin/pegleg/pegleg/engine/site.py index d96dbdae..29c032d4 100644 --- a/src/bin/pegleg/pegleg/engine/site.py +++ b/src/bin/pegleg/pegleg/engine/site.py @@ -29,6 +29,10 @@ LOG = logging.getLogger(__name__) def collect(site_name, save_location): try: save_files = dict() + if save_location is None: + raise ValueError('Missing param: save-location') + elif not os.path.exists(save_location): + raise FileNotFoundError('Invalid save-location path') for (repo_base, filename) in util.definition.site_files_by_repo(site_name): repo_name = os.path.normpath(repo_base).split(os.sep)[-1]