Some error handling

1 files changed, 7 insertions, 3 deletions

diff --git a/push.py b/push.py
index c6dc613..1d3f5ab 100755
--- a/push.py
+++ b/push.py
@@ -116,9 +116,13 @@ msg = payload.strip()
 # Send notification to each user key
 for user_token in user_tokens:
   if type(user_token) is dict:
-    k = user_token.keys()[0]
-    device = user_token.get(k, None)
-    user_token = k
+    try:
+      k = user_token.keys()[0]
+      device = user_token[k]
+    except (IndexError, KeyError):
+      next
+    else:
+      user_token = k
   else:
     device = None