fix: correctly count parameters for callback function used to persist session info

Fixes #10

Author: gboudreau

setup.py

@@ -3,7 +3,7 @@
 setuptools.setup(
     name = 'WS-API',
     packages = ['ws_api'],
-    version = '0.18.0',
+    version = '0.18.1',
     license = 'GPL-3.0',
     description = 'Access your Wealthsimple account using their (GraphQL) API.',
     author = 'Guillaume Boudreau',

ws_api/wealthsimple_api.py

@@ -156,7 +156,7 @@ def check_oauth_token(self, persist_session_fct: Optional[Callable[[WSAPISession
             self.session.access_token = response['access_token']
             self.session.refresh_token = response['refresh_token']
             if persist_session_fct:
-                if len(signature(persist_session_fct).parameters) == 3:
+                if len(signature(persist_session_fct).parameters) == 2:
                     persist_session_fct(self.session.to_json(), username)
                 else:
                     persist_session_fct(self.session.to_json())
@@ -206,7 +206,7 @@ def login_internal(self, username: str, password: str, otp_answer: str = None,
 
         # Persist the session if a persist function is provided
         if persist_session_fct:
-            if len(signature(persist_session_fct).parameters) == 3:
+            if len(signature(persist_session_fct).parameters) == 2:
                 persist_session_fct(self.session.to_json(), username)
             else:
                 persist_session_fct(self.session.to_json())