More EAFP, less LBYL

1 files changed, 10 insertions, 9 deletions

diff --git a/push.py b/push.py
index 1d3f5ab..c96e0e6 100755
--- a/push.py
+++ b/push.py
@@ -115,16 +115,17 @@ msg = payload.strip()
 
 # Send notification to each user key
 for user_token in user_tokens:
-  if type(user_token) is dict:
-    try:
-      k = user_token.keys()[0]
-      device = user_token[k]
-    except (IndexError, KeyError):
-      next
-    else:
-      user_token = k
+  device = None
+
+  try:
+    k = user_token.keys()[0]
+    device = user_token[k]
+  except AttributeError:
+    pass
+  except (IndexError, KeyError):
+    next
   else:
-    device = None
+    user_token = k
 
   p = pushover.Client(user_token, api_token=api_token)
   p.send_message(msg, title=title, device=device)