From 6c5d9de2c26f527f36e8af3c60a953c44af6fba1 Mon Sep 17 00:00:00 2001
From: Shing Chan <cshing.m@gmail.com>
Date: Wed, 6 Nov 2024 01:41:07 +0000
Subject: [PATCH] fix: early exit if no data to process

---
 src/stepcount/stepcount.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/stepcount/stepcount.py b/src/stepcount/stepcount.py
index c568fcd..aac913e 100644
--- a/src/stepcount/stepcount.py
+++ b/src/stepcount/stepcount.py
@@ -1,5 +1,6 @@
 import warnings
 import os
+import sys
 import pathlib
 import urllib
 import shutil
@@ -101,6 +102,20 @@ def main():
     # Update wear time stats after exclusions
     info.update(utils.calculate_wear_stats(data))
 
+    # If no data, save Info.json and exit
+    if len(data) == 0 or data.isna().any(axis=1).all():  # TODO: check na only on x,y,z cols?
+        # Save Info.json
+        with open(f"{outdir}/{basename}-Info.json", 'w') as f:
+            json.dump(info, f, indent=4, cls=utils.NpEncoder)
+        # Print
+        print("\nSummary\n-------")
+        print(json.dumps(
+            {k: v for k, v in info.items() if not re.search(r'_Weekend|_Weekday|_Hour\d{2}', k)},
+            indent=4, cls=utils.NpEncoder
+        ))
+        print("No data to process. Exiting early...")
+        sys.exit(0)
+
     # Run model
     if verbose:
         print("Loading model...")