Add rate limit to push return (#143)

Rate limit information added to _push return
richard-better · Oct 20, 2020 · ad8b878 · ad8b878
1 parent 3071dc2
commit ad8b878
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pushbullet/pushbullet.py b/pushbullet/pushbullet.py
@@ -275,7 +275,11 @@ def push_link(self, title, url, body=None, device=None, chat=None, email=None, c
     def _push(self, data):
         r = self._session.post(self.PUSH_URL, data=json.dumps(data))
         if r.status_code == requests.codes.ok:
-            return r.json()
+            js = r.json()
+            js['Ratelimit-Reset'] = r.headers['X-Ratelimit-Reset']
+            js['Ratelimit-Limit'] = r.headers['X-Ratelimit-Limit']
+            js['Ratelimit-Remaining'] = r.headers['X-Ratelimit-Remaining']
+            return js
         else:
             raise PushError(r.text)