Merge pull request #54 from prio-data/status_message_pagination

fixed pagination of queryset-fetch status messages
prio-data · Mar 15, 2024 · 17bfbd6 · 17bfbd6
2 parents c47ec7d + d028e29
commit 17bfbd6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "viewser"
-version = "6.0.3"
+version = "6.0.4"
 description = "The Views 3 CLI tool"
 authors = ["peder2911 <pglandsverk@gmail.com>"]
 readme = "README.md"

diff --git a/viewser/commands/queryset/operations.py b/viewser/commands/queryset/operations.py
@@ -170,13 +170,16 @@ def _fetch(
 
             try:
                 data = pd.read_parquet(io.BytesIO(data.value.content))
-                clear_output(wait=True)
-                print(f'{retries+1}: Queryset data read successfully', end="\r")
+#                clear_output(wait=True)
+                print(f'{retries+1}: Queryset {name} read successfully')
                 succeeded = True
             except:
                 message = data.value.content.decode()
-                clear_output(wait=True)
-                print(f'{retries+1}: {message}', end="\r")
+                if retries == 0:
+                    print(f'{retries + 1}: {message}')
+                else:
+                    clear_output(wait=True)
+                    print(f'{retries+1}: {message}', end="\r")
                 if 'failed' in message:
                     failed = True
                     data = message