1 files changed, 4 insertions, 5 deletions
diff --git a/at.py b/at.py
index 9e1dfa0..05d78a0 100644
--- a/at.py
+++ b/at.py
@@ -240,7 +240,7 @@ def main_view():
 
 @app.route('/api')
 def list_all():
-    result = now_at()
+    data = now_at()
 
     def prettify_user((user, atime)):
         return {
@@ -248,10 +248,9 @@ def list_all():
             'timestamp': atime,
             'pretty_time': strfts(atime),
         }
-    result['users'] = map(prettify_user, result['users'])
-    result['unknown'] = len(result['unknown'])
-    result['kektops'] = len(result['kektops'])
-    result['esps'] = len(result['esps'])
+    result = {}
+    result['users'] = map(prettify_user, data.pop('users'))
+    result.update((k, len(v)) for k, v in data.items())
     res = make_response(json.dumps(result), 200)
     res.headers['Access-Control-Allow-Origin'] = '*'
     return res