From 48e620475e74bc0de94632e812107edffd55ca59 Mon Sep 17 00:00:00 2001 From: Emmanuel Cazenave Date: Wed, 9 Jun 2021 14:22:26 +0200 Subject: [PATCH] venissieux-technocarte: add --sample-numlines option --- venissieux-technocarte/run.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/venissieux-technocarte/run.py b/venissieux-technocarte/run.py index 0549a0a..9d206b2 100644 --- a/venissieux-technocarte/run.py +++ b/venissieux-technocarte/run.py @@ -81,7 +81,7 @@ def get_rows(args): with open(args.filepath) as csvfile: reader = csv.DictReader(csvfile, delimiter=';', quotechar='"') for i, row in enumerate(reader): - if args.mode != 'full' and i > 100: + if args.mode != 'full' and i > args.sample_numlines: break yield row @@ -283,6 +283,7 @@ parser_import = subparsers.add_parser('import') parser_import.set_defaults(func=import_data) parser_import.add_argument('--filepath') parser_import.add_argument('--mode', default='sample', choices=('sample', 'full')) +parser_import.add_argument('--sample-numlines', default=100, type=int) parser_import.add_argument('--wcs-api-url') parser_import.add_argument('--wcs-email') parser_import.add_argument('--wcs-orig')