Merge pull request #35 from uhwot/automated-login
automated logging in to multiple sessions
This commit is contained in:
Daniel
GitHub
162
interface.py
162
interface.py
@@ -74,86 +74,81 @@ class ModuleInterface:
|
|||||||
if not saved_sessions:
|
if not saved_sessions:
|
||||||
saved_sessions = {}
|
saved_sessions = {}
|
||||||
|
|
||||||
if self.settings['enable_mobile']:
|
if not self.settings['enable_mobile']:
|
||||||
# check all saved session for a session starting with "MOBILE"
|
|
||||||
if not any(session for session in saved_sessions.keys() if session[:6] == 'MOBILE'):
|
|
||||||
confirm = input(' "enable_mobile" is enabled but no MOBILE session was found. Do you want to create a '
|
|
||||||
'MOBILE session (used for AC-4/360RA) [Y/n]? ')
|
|
||||||
if confirm.upper() == 'N':
|
|
||||||
self.available_sessions = [SessionType.TV.name]
|
|
||||||
else:
|
|
||||||
self.available_sessions = [SessionType.TV.name]
|
self.available_sessions = [SessionType.TV.name]
|
||||||
|
|
||||||
username, password = None, None
|
while True:
|
||||||
for session_type in self.available_sessions:
|
login_session = None
|
||||||
# create all sessions with the needed API keys
|
|
||||||
if session_type == SessionType.TV.name:
|
|
||||||
sessions[session_type] = TidalTvSession(self.settings['tv_token'], self.settings['tv_secret'])
|
|
||||||
elif session_type == SessionType.MOBILE_ATMOS.name:
|
|
||||||
sessions[session_type] = TidalMobileSession(self.settings['mobile_atmos_token'])
|
|
||||||
else:
|
|
||||||
sessions[session_type] = TidalMobileSession(self.settings['mobile_default_token'])
|
|
||||||
|
|
||||||
if session_type in saved_sessions:
|
def auth_and_save_session(session, session_type):
|
||||||
logging.debug(f'{module_information.service_name}: {session_type} session found, loading')
|
session = self.auth_session(session, session_type, login_session)
|
||||||
|
|
||||||
# load the dictionary from the temporary_settings_controller inside the TidalSession class
|
|
||||||
sessions[session_type].set_storage(saved_sessions[session_type])
|
|
||||||
else:
|
|
||||||
logging.debug(f'{module_information.service_name}: No {session_type} session found, creating new one')
|
|
||||||
if session_type == SessionType.TV.name:
|
|
||||||
self.print(f'{module_information.service_name}: Creating a TV session')
|
|
||||||
sessions[session_type].auth()
|
|
||||||
else:
|
|
||||||
if not username or not password:
|
|
||||||
self.print(f'{module_information.service_name}: Creating a Mobile session')
|
|
||||||
self.print(f'{module_information.service_name}: Enter your TIDAL username and password:')
|
|
||||||
self.print(f'{module_information.service_name}: (password will not be echoed)')
|
|
||||||
username = input(' Username: ')
|
|
||||||
password = getpass(' Password: ')
|
|
||||||
sessions[session_type].auth(username, password)
|
|
||||||
self.print(f'Successfully logged in, using {session_type} token!')
|
|
||||||
|
|
||||||
# get the dict representation from the TidalSession object and save it into saved_session/loginstorage
|
# get the dict representation from the TidalSession object and save it into saved_session/loginstorage
|
||||||
saved_sessions[session_type] = sessions[session_type].get_storage()
|
saved_sessions[session_type] = session.get_storage()
|
||||||
module_controller.temporary_settings_controller.set('sessions', saved_sessions)
|
module_controller.temporary_settings_controller.set('sessions', saved_sessions)
|
||||||
|
return session
|
||||||
|
|
||||||
# always try to refresh session
|
# ask for login if there are no saved sessions
|
||||||
if not sessions[session_type].valid():
|
if not saved_sessions:
|
||||||
sessions[session_type].refresh()
|
login_session_type = None
|
||||||
# Save the refreshed session in the temporary settings
|
if len(self.available_sessions) == 1:
|
||||||
saved_sessions[session_type] = sessions[session_type].get_storage()
|
login_session_type = self.available_sessions[0]
|
||||||
module_controller.temporary_settings_controller.set('sessions', saved_sessions)
|
else:
|
||||||
|
self.print(f'{module_information.service_name}: Choose a login method:')
|
||||||
|
self.print(f'{module_information.service_name}: 1. TV (browser)')
|
||||||
|
self.print(f"{module_information.service_name}: 2. Mobile (username and password, choose TV if this doesn't work)")
|
||||||
|
|
||||||
|
while not login_session_type:
|
||||||
|
input_str = input(' Login method: ')
|
||||||
|
try:
|
||||||
|
login_session_type = {
|
||||||
|
'1': SessionType.TV.name,
|
||||||
|
'tv': SessionType.TV.name,
|
||||||
|
'2': SessionType.MOBILE_DEFAULT.name,
|
||||||
|
'mobile': SessionType.MOBILE_DEFAULT.name,
|
||||||
|
}[input_str.lower()]
|
||||||
|
except KeyError:
|
||||||
|
self.print(f'{module_information.service_name}: Invalid choice, try again')
|
||||||
|
|
||||||
|
login_session = auth_and_save_session(self.init_session(login_session_type), login_session_type)
|
||||||
|
|
||||||
|
for session_type in self.available_sessions:
|
||||||
|
sessions[session_type] = self.init_session(session_type)
|
||||||
|
|
||||||
|
if session_type in saved_sessions:
|
||||||
|
logging.debug(f'{module_information.service_name}: {session_type} session found, loading')
|
||||||
|
|
||||||
|
# load the dictionary from the temporary_settings_controller inside the TidalSession class
|
||||||
|
sessions[session_type].set_storage(saved_sessions[session_type])
|
||||||
|
else:
|
||||||
|
logging.debug(f'{module_information.service_name}: No {session_type} session found, creating new one')
|
||||||
|
sessions[session_type] = auth_and_save_session(sessions[session_type], session_type)
|
||||||
|
|
||||||
|
# always try to refresh session
|
||||||
|
if not sessions[session_type].valid():
|
||||||
|
sessions[session_type].refresh()
|
||||||
|
# Save the refreshed session in the temporary settings
|
||||||
|
saved_sessions[session_type] = sessions[session_type].get_storage()
|
||||||
|
module_controller.temporary_settings_controller.set('sessions', saved_sessions)
|
||||||
|
|
||||||
while True:
|
|
||||||
# check for a valid subscription
|
# check for a valid subscription
|
||||||
subscription = self.check_subscription(sessions[session_type].get_subscription())
|
subscription = self.check_subscription(sessions[session_type].get_subscription())
|
||||||
if subscription:
|
if not subscription:
|
||||||
|
confirm = input(' Do you want to relogin? [Y/n]: ')
|
||||||
|
|
||||||
|
if confirm.upper() == 'N':
|
||||||
|
self.print('Exiting...')
|
||||||
|
exit()
|
||||||
|
|
||||||
|
# reset saved sessions and loop back to login
|
||||||
|
saved_sessions = {}
|
||||||
break
|
break
|
||||||
|
|
||||||
confirm = input(' Do you want to create a new session? [Y/n]: ')
|
if not login_session:
|
||||||
|
login_session = sessions[session_type]
|
||||||
|
|
||||||
if confirm.upper() == 'N':
|
if saved_sessions:
|
||||||
self.print('Exiting...')
|
break
|
||||||
exit()
|
|
||||||
|
|
||||||
# create a new session finally
|
|
||||||
if session_type == SessionType.TV.name:
|
|
||||||
self.print(f'{module_information.service_name}: Recreating a TV session')
|
|
||||||
sessions[session_type].auth()
|
|
||||||
else:
|
|
||||||
self.print(f'{module_information.service_name}: Recreating a Mobile session')
|
|
||||||
self.print(f'{module_information.service_name}: Enter your TIDAL username and password:')
|
|
||||||
self.print(f'{module_information.service_name}: (password will not be echoed)')
|
|
||||||
username = input('Username: ')
|
|
||||||
password = getpass('Password: ')
|
|
||||||
sessions[session_type].auth(username, password)
|
|
||||||
|
|
||||||
saved_sessions[session_type] = sessions[session_type].get_storage()
|
|
||||||
module_controller.temporary_settings_controller.set('sessions', saved_sessions)
|
|
||||||
|
|
||||||
# reset username and password
|
|
||||||
username, password = None, None
|
|
||||||
|
|
||||||
# only needed for region locked albums where the track is available but force_album_format is used
|
# only needed for region locked albums where the track is available but force_album_format is used
|
||||||
self.album_cache = {}
|
self.album_cache = {}
|
||||||
@@ -161,6 +156,39 @@ class ModuleInterface:
|
|||||||
# load the Tidal session with all saved sessions (TV, Mobile Atmos, Mobile Default)
|
# load the Tidal session with all saved sessions (TV, Mobile Atmos, Mobile Default)
|
||||||
self.session: TidalApi = TidalApi(sessions)
|
self.session: TidalApi = TidalApi(sessions)
|
||||||
|
|
||||||
|
def init_session(self, session_type):
|
||||||
|
session = None
|
||||||
|
# initialize session with the needed API keys
|
||||||
|
if session_type == SessionType.TV.name:
|
||||||
|
session = TidalTvSession(self.settings['tv_token'], self.settings['tv_secret'])
|
||||||
|
elif session_type == SessionType.MOBILE_ATMOS.name:
|
||||||
|
session = TidalMobileSession(self.settings['mobile_atmos_token'])
|
||||||
|
else:
|
||||||
|
session = TidalMobileSession(self.settings['mobile_default_token'])
|
||||||
|
return session
|
||||||
|
|
||||||
|
def auth_session(self, session, session_type, login_session):
|
||||||
|
if login_session:
|
||||||
|
# refresh tokens can be used with any client id
|
||||||
|
# this can be used to switch to any client type from an existing session
|
||||||
|
session.refresh_token = login_session.refresh_token
|
||||||
|
session.user_id = login_session.user_id
|
||||||
|
session.country_code = login_session.country_code
|
||||||
|
session.refresh()
|
||||||
|
elif session_type == SessionType.TV.name:
|
||||||
|
self.print(f'{module_information.service_name}: Creating a TV session')
|
||||||
|
session.auth()
|
||||||
|
else:
|
||||||
|
self.print(f'{module_information.service_name}: Creating a Mobile session')
|
||||||
|
self.print(f'{module_information.service_name}: Enter your TIDAL username and password:')
|
||||||
|
self.print(f'{module_information.service_name}: (password will not be echoed)')
|
||||||
|
username = input(' Username: ')
|
||||||
|
password = getpass(' Password: ')
|
||||||
|
session.auth(username, password)
|
||||||
|
self.print(f'Successfully logged in, using {session_type} token!')
|
||||||
|
|
||||||
|
return session
|
||||||
|
|
||||||
def check_subscription(self, subscription: str) -> bool:
|
def check_subscription(self, subscription: str) -> bool:
|
||||||
# returns true if "disable_subscription_checks" is enabled or subscription is HIFI Plus
|
# returns true if "disable_subscription_checks" is enabled or subscription is HIFI Plus
|
||||||
if not self.disable_subscription_check and subscription not in {'HIFI', 'PREMIUM_PLUS'}:
|
if not self.disable_subscription_check and subscription not in {'HIFI', 'PREMIUM_PLUS'}:
|
||||||
|
|||||||
Reference in New Issue
Block a user