Make user_id and country_code optional when using load_session()

Author: morguldir

tests/test_api.py

@@ -128,3 +128,15 @@ def test_get_track_url(session):
 def test_get_video_url(session):
     video = session.get_video(108046194)
     newurl =  session.get_video_url(video.id)
+
+def test_load_session(session):
+    """
+    Test loading a session from a session id without supplying country code and user_id
+    """
+    user_id = session.user.id
+    country_code = session.country_code
+    session_id = session.session_id
+    session = tidalapi.Session()
+    session.load_session(session_id)
+    assert user_id == session.user.id
+    assert country_code == session.country_code

tidalapi/__init__.py

@@ -60,8 +60,13 @@ def __init__(self, config=Config()):
         self._config = config
         """:type _config: :class:`Config`"""
 
-    def load_session(self, session_id, country_code, user_id):
+    def load_session(self, session_id, country_code=None, user_id=None):
         self.session_id = session_id
+        if not user_id or not country_code:
+            request = self.request('GET', 'sessions').json()
+            country_code = request['countryCode']
+            user_id = request['userId']
+
         self.country_code = country_code
         self.user = User(self, id=user_id)