From 9f429d11c0f6aaabd56b0417b87e93d367dcffa3 Mon Sep 17 00:00:00 2001 From: David A Minton Date: Tue, 22 Nov 2022 17:19:07 -0500 Subject: [PATCH] Added tqdm progress bar --- python/swiftest/swiftest/simulation_class.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/swiftest/swiftest/simulation_class.py b/python/swiftest/swiftest/simulation_class.py index 9f1555411..436cfb605 100644 --- a/python/swiftest/swiftest/simulation_class.py +++ b/python/swiftest/swiftest/simulation_class.py @@ -26,6 +26,7 @@ import subprocess import shlex import warnings +from tqdm import tqdm from typing import ( Literal, Dict, @@ -376,7 +377,8 @@ def _run_swiftest_driver(self): env=env, universal_newlines=True) as p: process_output = False - for line in p.stdout: + noutput = int((self.param['TSTOP'] - self.param['T0']) / (self.param['DT'] * self.param['ISTEP_OUT'])) + for line in tqdm(p.stdout,total=noutput): if "SWIFTEST STOP" in line: process_output = False