obi import: progress bar fixed when using --only option

2019-03-30 15:16:57 +01:00
parent c293cfabbb
commit ce686e9569
1 changed files with 5 additions and 1 deletions
--- a/python/obitools3/commands/import.pyx
+++ b/python/obitools3/commands/import.pyx
@ -105,7 +105,11 @@ def run(config):
        if input is None:  # TODO check for bytes instead now?
            raise Exception("Could not open input URI")
        
+        if config['obi']['only'] is not None:
+            entry_count = min(input[4], config['obi']['only'])
+        else:
            entry_count = input[4]
+        
        logger("info", "Importing %d entries", entry_count)
        
        # TODO a bit dirty?