fix: add client cleanup, put error handling, and cumulative validation to migration script

Author: sspickle

ide/shell_cmds/buildUserHistory.py

@@ -39,6 +39,7 @@ def build_history(client):
                 skipped += 1
 
         print(f"Done: {count} users, {skipped} skipped (no joinDate)")
+        expected = count - skipped
 
         sorted_months = sorted(monthly_new.keys())
         cumulative = 0
@@ -47,6 +48,9 @@ def build_history(client):
             cumulative += monthly_new[month]
             points.append({'month': month, 'count': cumulative})
 
+        if points and points[-1]['count'] != expected:
+            print(f"WARNING: cumulative total {points[-1]['count']} != expected {expected}")
+
         history = {
             'updated': datetime.now(timezone.utc).strftime('%Y-%m-%d'),
             'points': points,
@@ -56,16 +60,25 @@ def build_history(client):
         if not existing:
             existing = Setting(id='user_count_history')
         existing.value = json.dumps(history)
-        existing.put()
+        try:
+            existing.put()
+        except Exception as e:
+            import sys
+            print(f"ERROR: Failed to store user history: {e}", file=sys.stderr)
+            sys.exit(1)
 
         print(f"Stored {len(points)} monthly data points")
         if points:
             print(f"Latest: {points[-1]}")
 
 
 if __name__ == '__main__':
+    import sys
     project = os.environ.get('GOOGLE_CLOUD_PROJECT', 'glowscript')
     emulator = os.environ.get('DATASTORE_EMULATOR_HOST')
-    client = ndb.Client(project='glowscript-dev' if emulator else project)
+    client = ndb.Client(project=project)
     print(f"Connecting to {'emulator at ' + emulator if emulator else 'production Datastore'}...")
-    build_history(client)
+    try:
+        build_history(client)
+    finally:
+        client.close()