from __future__ import absolute_import from kodi_six import xbmc from kodi_six import xbmcgui from plexnet import plexapp from plexnet.exceptions import UserSwitchForbiddenException from lib import util from lib.util import T from . import busy from . import dropdown from . import kodigui class UserSelectWindow(kodigui.BaseWindow): xmlFile = 'script-plex-user_select.xml' path = util.ADDON.getAddonInfo('path') theme = 'Main' res = '1080i' width = 1920 height = 1080 USER_LIST_ID = 101 PIN_ENTRY_GROUP_ID = 400 SHUTDOWN_BUTTON_ID = 500 def __init__(self, *args, **kwargs): self.task = None self.selected = False kodigui.BaseWindow.__init__(self, *args, **kwargs) def onFirstInit(self): self.userList = kodigui.ManagedControlList(self, self.USER_LIST_ID, 6) self.start() def onAction(self, action): try: ID = action.getId() if 57 < ID < 68: if not xbmc.getCondVisibility('ControlGroup({0}).HasFocus(0)'.format(self.PIN_ENTRY_GROUP_ID)): item = self.userList.getSelectedItem() if not item.dataSource.isProtected: return self.setFocusId(self.PIN_ENTRY_GROUP_ID) self.pinEntryClicked(ID + 142) return elif 142 <= ID <= 149: # JumpSMS action if not xbmc.getCondVisibility('ControlGroup({0}).HasFocus(0)'.format(self.PIN_ENTRY_GROUP_ID)): item = self.userList.getSelectedItem() if not item.dataSource.isProtected: return self.setFocusId(self.PIN_ENTRY_GROUP_ID) self.pinEntryClicked(ID + 60) return elif ID in (xbmcgui.ACTION_NAV_BACK, xbmcgui.ACTION_PREVIOUS_MENU, xbmcgui.ACTION_BACKSPACE): item = self.userList.getSelectedItem() if xbmc.getCondVisibility('ControlGroup({0}).HasFocus(0)'.format(self.PIN_ENTRY_GROUP_ID)): if item.getProperty('editing.pin'): self.pinEntryClicked(211) else: self.setFocusId(self.USER_LIST_ID) return # return to selected user self.selected = 'cancel' self.doClose() return except: util.ERROR() kodigui.BaseWindow.onAction(self, action) def onClick(self, controlID): if controlID == self.USER_LIST_ID: item = self.userList.getSelectedItem() if item.dataSource.isProtected: self.setFocusId(self.PIN_ENTRY_GROUP_ID) # refresh clicked elif not item.dataSource: from lib import localmode if localmode.isAccountLess(): # account-less: no plex.tv user list to refresh - re-pick the local profiles localmode.seedUsersFromServer(reselect=True) with self.propertyContext('busy'): self.userList.reset() self.setProperty('initialized', '') self.start(with_busy=False) return # refresh user list with self.propertyContext('busy'): self.userList.reset() self.setProperty('initialized', '') plexapp.ACCOUNT.updateHomeUsers(refreshSubscription=True) self.start(with_busy=False) else: from lib import localmode # dialogs must not run inside userSelected's busy dialog if localmode.needsProfileToken(item.dataSource): localmode.promptProfileToken(item.dataSource) self.userSelected(item) elif 200 < controlID < 212: self.pinEntryClicked(controlID) elif controlID == self.SHUTDOWN_BUTTON_ID: self.shutdownClicked() def onFocus(self, controlID): if controlID == self.USER_LIST_ID: item = self.userList.getSelectedItem() item.setProperty('editing.pin', '') def start(self, with_busy=True): if with_busy: self.setProperty('busy', '1') try: users = plexapp.ACCOUNT.homeUsers items = [] selectIndex = None for idx, user in enumerate(users): thumb = plexapp.ACCOUNT.safeUserThumb(user.id, thumb=user.thumb) mli = kodigui.ManagedListItem(user.title, user.title[0].upper(), thumbnailImage=thumb, data_source=user) mli.setProperty('back.image', user.id) mli.setProperty('pin', user.title) mli.setProperty('protected', user.isProtected and '1' or '') mli.setProperty('admin', user.isAdmin and '1' or '') if plexapp.ACCOUNT.ID == user.id: selectIndex = idx items.append(mli) # append refresh button mli = kodigui.ManagedListItem("", "", data_source=kodigui.EmptyDataSource(), properties={"empty": '1'}) items.append(mli) self.userList.addItems(items) self.setFocusId(self.USER_LIST_ID) if selectIndex is not None: self.userList.setSelectedItemByPos(selectIndex) self.setProperty('initialized', '1') finally: self.setProperty('busy', '') def shutdownClicked(self): options = [] options.append({'key': 'sign_out', 'display': T(32421, 'Sign Out')}) options.append({'key': 'exit', 'display': T(32422, 'Exit')}) if util.getSetting('kiosk.mode'): if xbmc.getCondVisibility('System.CanPowerDown'): options.append({'key': 'shutdown', 'display': T(32423, 'Shutdown')}) if xbmc.getCondVisibility('System.CanSuspend'): options.append({'key': 'suspend', 'display': T(32424, 'Suspend')}) if xbmc.getCondVisibility('System.CanHibernate'): options.append({'key': 'hibernate', 'display': T(32425, 'Hibernate')}) if xbmc.getCondVisibility('System.CanReboot'): options.append({'key': 'reboot', 'display': T(32426, 'Reboot')}) with self.propertyContext('dropdown'): choice = dropdown.showDropdown(options, (60, 101)) if not choice: return if choice['key'] == 'sign_out': self.selected = 'signout' self.doClose() elif choice['key'] == 'exit': self.doClose() elif choice['key'] == 'shutdown': xbmc.executebuiltin('Powerdown()') elif choice['key'] == 'suspend': xbmc.executebuiltin('Suspend()') elif choice['key'] == 'hibernate': xbmc.executebuiltin('Hibernate()') elif choice['key'] == 'reboot': xbmc.executebuiltin('Reset()') def pinEntryClicked(self, controlID): item = self.userList.getSelectedItem() if item.getProperty('editing.pin'): pin = item.getProperty('editing.pin') else: pin = '' if len(pin) > 3: return if controlID < 210: pin += str(controlID - 200) elif controlID == 210: pin += '0' elif controlID == 211: pin = pin[:-1] if pin: item.setProperty('pin', ' '.join(list(u"\u2022" * len(pin)))) item.setProperty('editing.pin', pin) if len(pin) > 3: self.userSelected(item, pin) else: item.setProperty('pin', item.dataSource.title) item.setProperty('editing.pin', '') @busy.dialog() def userSelected(self, item, pin=None): user = item.dataSource # xbmc.sleep(500) util.DEBUG_LOG('Home user selected: {0}', user) from lib import plex with plex.CallbackEvent(plexapp.util.APP, 'account:response') as e: try: if plexapp.ACCOUNT.switchHomeUser(user.id, pin) and plexapp.ACCOUNT.switchUser: util.DEBUG_LOG('Waiting for user change...') else: e.close() item.setProperty('pin', item.dataSource.title) item.setProperty('editing.pin', '') util.messageDialog(T(32427, 'Failed'), T(32926, 'Wrong pin entered!')) return except UserSwitchForbiddenException: # re-fetch users plexapp.ACCOUNT.updateHomeUsers() self.selected = 'retry' self.doClose() return self.selected = True self.doClose() def finished(self): if self.task: self.task.cancel() def start(base_win_id): w = UserSelectWindow.create() if w.waitForOpen(base_win_id=base_win_id): w.modal() selected = w.selected del w return selected