diaspy.connection.Connection.login() returns Connection() object as well

[diaspy.git] / diaspy / connection.py

diff --git a/diaspy/connection.py b/diaspy/connection.py
index baec218808328592cbdb4724093f5decc7365245..50b1d44cf0577f2c05edbbba93904c315c0d1a76 100644 (file)
--- a/diaspy/connection.py
+++ b/diaspy/connection.py
@@ -147,7 +147,6 @@ class Connection():
                             allow_redirects=False)
         if request.status_code != 302:
             raise errors.LoginError('{0}: login failed'.format(request.status_code))
-        return request.status_code
 
     def login(self, remember_me=1):
         """This function is used to log in to a pod.
@@ -158,7 +157,7 @@ class Connection():
         self._login_data['user[remember_me]'] = remember_me
         status = self._login()
         self._login_data = {}
-        return status
+        return self
 
     def logout(self):
         """Logs out from a pod.