Merge pull request #283 from GioF71/feature/login-session-file-add-op…

…tional-print-function-282

Add optional fn_print to Session::login_session_file

docs/login.rst

@@ -46,7 +46,7 @@ This will print the link, and then wait for the login future to complete, but it
 .. testcode::
 
     # The function is print by default, but you can use anything, here we do it to avoid the print being swallowed
-    session.login_oauth_simple(function=printer)
+    session.login_oauth_simple(fn_print=printer)
     print(session.check_login())
 
 .. testoutput::

tidalapi/session.py

@@ -446,12 +446,15 @@ def login_session_file(
         self,
         session_file: Path,
         do_pkce: Optional[bool] = False,
+        fn_print: Callable[[str], None] = print,
     ) -> bool:
         """Logs in to the TIDAL api using an existing OAuth/PKCE session file. If no
         session json file exists, a new one will be created after successful login.
 
         :param session_file: The session json file
         :param do_pkce: Perform PKCE login. Default: Use OAuth logon
+        :param fn_print: A function which will be called to print the challenge text,
+            defaults to `print()`.
         :return: Returns true if we think the login was successful.
         """
         self.load_session_from_file(session_file)
@@ -460,10 +463,10 @@ def login_session_file(
         if not self.check_login():
             if do_pkce:
                 log.info("Creating new session (PKCE)...")
-                self.login_pkce()
+                self.login_pkce(fn_print=fn_print)
             else:
                 log.info("Creating new session (OAuth)...")
-                self.login_oauth_simple()
+                self.login_oauth_simple(fn_print=fn_print)
 
         if self.check_login():
             log.info("TIDAL Login OK")
@@ -576,17 +579,17 @@ def pkce_get_auth_token(self, url_redirect: str) -> dict[str, Union[str, int]]:
 
         return token
 
-    def login_oauth_simple(self, function: Callable[[str], None] = print) -> None:
+    def login_oauth_simple(self, fn_print: Callable[[str], None] = print) -> None:
         """Login to TIDAL using a remote link. You can select what function you want to
         use to display the link.
 
-        :param function: The function you want to display the link with
+        :param fn_print: The function you want to display the link with
         :raises: TimeoutError: If the login takes too long
         """
 
         login, future = self.login_oauth()
         text = "Visit https://{0} to log in, the code will expire in {1} seconds"
-        function(text.format(login.verification_uri_complete, login.expires_in))
+        fn_print(text.format(login.verification_uri_complete, login.expires_in))
         future.result()
 
     def login_oauth(self) -> Tuple[LinkLogin, concurrent.futures.Future[Any]]: