notifications.py refactored to use `repr()` on `Connection()`

diff --git a/diaspy/notifications.py b/diaspy/notifications.py
index 418c93944800c13eb913ffd073bd73c75ee50d3d..bfcfadac21784a69002bfa12cf0aae80896dab18 100644 (file)
--- a/diaspy/notifications.py
+++ b/diaspy/notifications.py
@@ -27,7 +27,7 @@ class Notifications():
         """Returns list of most recent notifications.
         """
         params = {'per_page': 5, '_': int(round(time.time(), 3)*1000)}
-        headers = {'x-csrf-token': self._connection.get_token()}
+        headers = {'x-csrf-token': repr(self._connection)}
 
         request = self._connection.get('notifications.json', headers=headers, params=params)
 
@@ -39,7 +39,7 @@ class Notifications():
         """Returns list of notifications.
         """
         params = {'per_page': per_page, 'page': page}
-        headers = {'x-csrf-token': self._connection.get_token()}
+        headers = {'x-csrf-token': repr(self._connection)}
 
         request = self._connection.get('notifications.json', headers=headers, params=params)