Better, safer dialogs to enter network credentials
This commit is contained in:
parent
097cc79c5a
commit
cffebf2f65
5 changed files with 214 additions and 123 deletions
|
@ -95,7 +95,8 @@ class Main():
|
||||||
entrypoint.toggle_plex_tv_sign_in()
|
entrypoint.toggle_plex_tv_sign_in()
|
||||||
|
|
||||||
elif mode == 'passwords':
|
elif mode == 'passwords':
|
||||||
utils.passwords_xml()
|
from resources.lib.windows import direct_path_sources
|
||||||
|
direct_path_sources.start()
|
||||||
|
|
||||||
elif mode == 'switchuser':
|
elif mode == 'switchuser':
|
||||||
entrypoint.switch_plex_user()
|
entrypoint.switch_plex_user()
|
||||||
|
|
|
@ -306,6 +306,36 @@ msgctxt "#30198"
|
||||||
msgid "Search"
|
msgid "Search"
|
||||||
msgstr ""
|
msgstr ""
|
||||||
|
|
||||||
|
# For setting up direct paths and adding network credentials
|
||||||
|
msgctxt "#30200"
|
||||||
|
msgid "In the following window, enter the server's hostname (or IP) where your Plex media resides. Mind the case!"
|
||||||
|
msgstr ""
|
||||||
|
|
||||||
|
# For setting up direct paths and adding network credentials - input window for hostname
|
||||||
|
msgctxt "#30201"
|
||||||
|
msgid "Enter server hostname (or IP)"
|
||||||
|
msgstr ""
|
||||||
|
|
||||||
|
# For setting up direct paths and adding network credentials
|
||||||
|
msgctxt "#30202"
|
||||||
|
msgid "In the following window, enter the network protocol you would like to use. This is likely 'smb'."
|
||||||
|
msgstr ""
|
||||||
|
|
||||||
|
# For setting up direct paths and adding network credentials - input window protocol
|
||||||
|
msgctxt "#30203"
|
||||||
|
msgid "Enter network protocol"
|
||||||
|
msgstr ""
|
||||||
|
|
||||||
|
# For setting up direct paths and adding network credentials
|
||||||
|
msgctxt "#30204"
|
||||||
|
msgid "The hostname or IP '{0}' that you entered is not valid."
|
||||||
|
msgstr ""
|
||||||
|
|
||||||
|
# For setting up direct paths and adding network credentials
|
||||||
|
msgctxt "#30205"
|
||||||
|
msgid "The protocol '{0}' that you entered is not supported."
|
||||||
|
msgstr ""
|
||||||
|
|
||||||
# Video node naming for random e.g. movies
|
# Video node naming for random e.g. movies
|
||||||
msgctxt "#30227"
|
msgctxt "#30227"
|
||||||
msgid "Random"
|
msgid "Random"
|
||||||
|
|
|
@ -602,7 +602,8 @@ class InitialSetup(object):
|
||||||
# Go to network credentials?
|
# Go to network credentials?
|
||||||
if utils.yesno_dialog(utils.lang(39029), utils.lang(39030)):
|
if utils.yesno_dialog(utils.lang(39029), utils.lang(39030)):
|
||||||
LOG.debug("Presenting network credentials dialog.")
|
LOG.debug("Presenting network credentials dialog.")
|
||||||
utils.passwords_xml()
|
from .windows import direct_path_sources
|
||||||
|
direct_path_sources.start()
|
||||||
# Disable Plex music?
|
# Disable Plex music?
|
||||||
if utils.yesno_dialog(utils.lang(29999), utils.lang(39016)):
|
if utils.yesno_dialog(utils.lang(29999), utils.lang(39016)):
|
||||||
LOG.debug("User opted to disable Plex music library.")
|
LOG.debug("User opted to disable Plex music library.")
|
||||||
|
|
|
@ -14,7 +14,6 @@ import xml.etree.ElementTree as etree
|
||||||
import defusedxml.ElementTree as defused_etree # etree parse unsafe
|
import defusedxml.ElementTree as defused_etree # etree parse unsafe
|
||||||
from xml.etree.ElementTree import ParseError
|
from xml.etree.ElementTree import ParseError
|
||||||
from functools import wraps
|
from functools import wraps
|
||||||
from urllib import quote_plus
|
|
||||||
import hashlib
|
import hashlib
|
||||||
import re
|
import re
|
||||||
import gc
|
import gc
|
||||||
|
@ -824,126 +823,6 @@ class XmlKodiSetting(object):
|
||||||
return element
|
return element
|
||||||
|
|
||||||
|
|
||||||
def passwords_xml():
|
|
||||||
"""
|
|
||||||
To add network credentials to Kodi's password xml
|
|
||||||
"""
|
|
||||||
path = path_ops.translate_path('special://userdata/')
|
|
||||||
xmlpath = "%spasswords.xml" % path
|
|
||||||
try:
|
|
||||||
xmlparse = defused_etree.parse(xmlpath)
|
|
||||||
except IOError:
|
|
||||||
# Document is blank or missing
|
|
||||||
root = etree.Element('passwords')
|
|
||||||
skip_find = True
|
|
||||||
except ParseError:
|
|
||||||
LOG.error('Error parsing %s', xmlpath)
|
|
||||||
# "Kodi cannot parse {0}. PKC will not function correctly. Please visit
|
|
||||||
# {1} and correct your file!"
|
|
||||||
messageDialog(lang(29999), lang(39716).format(
|
|
||||||
'passwords.xml', 'http://forum.kodi.tv/'))
|
|
||||||
return
|
|
||||||
else:
|
|
||||||
root = xmlparse.getroot()
|
|
||||||
skip_find = False
|
|
||||||
|
|
||||||
credentials = settings('networkCreds')
|
|
||||||
if credentials:
|
|
||||||
# Present user with options
|
|
||||||
option = dialog('select',
|
|
||||||
"Modify/Remove network credentials",
|
|
||||||
["Modify", "Remove"])
|
|
||||||
|
|
||||||
if option < 0:
|
|
||||||
# User cancelled dialog
|
|
||||||
return
|
|
||||||
|
|
||||||
elif option == 1:
|
|
||||||
# User selected remove
|
|
||||||
success = False
|
|
||||||
for paths in root.getiterator('passwords'):
|
|
||||||
for path in paths:
|
|
||||||
if path.find('.//from').text == "smb://%s/" % credentials:
|
|
||||||
paths.remove(path)
|
|
||||||
LOG.info("Successfully removed credentials for: %s",
|
|
||||||
credentials)
|
|
||||||
etree.ElementTree(root).write(xmlpath,
|
|
||||||
encoding="UTF-8")
|
|
||||||
success = True
|
|
||||||
if not success:
|
|
||||||
LOG.error("Failed to find saved server: %s in passwords.xml",
|
|
||||||
credentials)
|
|
||||||
dialog('notification',
|
|
||||||
heading='{plex}',
|
|
||||||
message="%s not found" % credentials,
|
|
||||||
icon='{warning}',
|
|
||||||
sound=False)
|
|
||||||
return
|
|
||||||
settings('networkCreds', value="")
|
|
||||||
dialog('notification',
|
|
||||||
heading='{plex}',
|
|
||||||
message="%s removed from passwords.xml" % credentials,
|
|
||||||
icon='{plex}',
|
|
||||||
sound=False)
|
|
||||||
return
|
|
||||||
|
|
||||||
elif option == 0:
|
|
||||||
# User selected to modify
|
|
||||||
server = dialog('input',
|
|
||||||
"Modify the computer name or ip address",
|
|
||||||
credentials)
|
|
||||||
if not server:
|
|
||||||
return
|
|
||||||
else:
|
|
||||||
# No credentials added
|
|
||||||
messageDialog("Network credentials",
|
|
||||||
'Input the server name or IP address as indicated in your plex '
|
|
||||||
'library paths. For example, the server name: '
|
|
||||||
'\\\\SERVER-PC\\path\\ or smb://SERVER-PC/path is SERVER-PC')
|
|
||||||
server = dialog('input', "Enter the server name or IP address")
|
|
||||||
if not server:
|
|
||||||
return
|
|
||||||
server = quote_plus(server)
|
|
||||||
|
|
||||||
# Network username
|
|
||||||
user = dialog('input', "Enter the network username")
|
|
||||||
if not user:
|
|
||||||
return
|
|
||||||
user = quote_plus(user)
|
|
||||||
# Network password
|
|
||||||
password = dialog('input',
|
|
||||||
"Enter the network password",
|
|
||||||
'', # Default input
|
|
||||||
type='{alphanum}',
|
|
||||||
option='{hide}')
|
|
||||||
# Need to url-encode the password
|
|
||||||
password = quote_plus(password)
|
|
||||||
# Add elements. Annoying etree bug where findall hangs forever
|
|
||||||
if skip_find is False:
|
|
||||||
skip_find = True
|
|
||||||
for path in root.findall('.//path'):
|
|
||||||
if path.find('.//from').text.lower() == "smb://%s/" % server.lower():
|
|
||||||
# Found the server, rewrite credentials
|
|
||||||
path.find('.//to').text = ("smb://%s:%s@%s/"
|
|
||||||
% (user, password, server))
|
|
||||||
skip_find = False
|
|
||||||
break
|
|
||||||
if skip_find:
|
|
||||||
# Server not found, add it.
|
|
||||||
path = etree.SubElement(root, 'path')
|
|
||||||
etree.SubElement(path, 'from', {'pathversion': "1"}).text = \
|
|
||||||
"smb://%s/" % server
|
|
||||||
topath = "smb://%s:%s@%s/" % (user, password, server)
|
|
||||||
etree.SubElement(path, 'to', {'pathversion': "1"}).text = topath
|
|
||||||
|
|
||||||
# Add credentials
|
|
||||||
settings('networkCreds', value="%s" % server)
|
|
||||||
LOG.info("Added server: %s to passwords.xml", server)
|
|
||||||
# Prettify and write to file
|
|
||||||
indent(root)
|
|
||||||
etree.ElementTree(root).write(xmlpath, encoding="UTF-8")
|
|
||||||
|
|
||||||
|
|
||||||
def playlist_xsp(mediatype, tagname, viewid, viewtype="", delete=False):
|
def playlist_xsp(mediatype, tagname, viewid, viewtype="", delete=False):
|
||||||
"""
|
"""
|
||||||
Feed with tagname as unicode
|
Feed with tagname as unicode
|
||||||
|
|
180
resources/lib/windows/direct_path_sources.py
Normal file
180
resources/lib/windows/direct_path_sources.py
Normal file
|
@ -0,0 +1,180 @@
|
||||||
|
#!/usr/bin/env python
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
"""
|
||||||
|
:module: plexkodiconnect.userselect
|
||||||
|
:synopsis: Prompts the user to add network paths and username passwords for
|
||||||
|
e.g. smb paths
|
||||||
|
"""
|
||||||
|
from __future__ import absolute_import, division, unicode_literals
|
||||||
|
from logging import getLogger
|
||||||
|
import re
|
||||||
|
import socket
|
||||||
|
import urllib
|
||||||
|
|
||||||
|
import xbmc
|
||||||
|
|
||||||
|
from .. import path_ops, utils
|
||||||
|
|
||||||
|
LOG = getLogger('PLEX.direct_path_sources')
|
||||||
|
|
||||||
|
SUPPORTED_PROTOCOLS = ('smb', 'nfs', 'http', 'https', 'ftp', 'sftp')
|
||||||
|
PATH = path_ops.translate_path('special://userdata/')
|
||||||
|
|
||||||
|
|
||||||
|
def get_etree(topelement):
|
||||||
|
try:
|
||||||
|
xml = utils.defused_etree.parse(
|
||||||
|
path_ops.path.join(PATH, '%s.xml' % topelement))
|
||||||
|
except IOError:
|
||||||
|
# Document is blank or missing
|
||||||
|
LOG.info('%s.xml is missing or blank, creating it', topelement)
|
||||||
|
root = utils.etree.Element(topelement)
|
||||||
|
except utils.ParseError:
|
||||||
|
LOG.error('Error parsing %s', topelement)
|
||||||
|
# "Kodi cannot parse {0}. PKC will not function correctly. Please visit
|
||||||
|
# {1} and correct your file!"
|
||||||
|
utils.messageDialog(utils.lang(29999), utils.lang(39716).format(
|
||||||
|
'%s.xml' % topelement, 'http://forum.kodi.tv/'))
|
||||||
|
return
|
||||||
|
else:
|
||||||
|
root = xml.getroot()
|
||||||
|
return root
|
||||||
|
|
||||||
|
|
||||||
|
def is_valid_hostname(hostname):
|
||||||
|
if len(hostname) > 255:
|
||||||
|
return False
|
||||||
|
if hostname[-1] == ".":
|
||||||
|
# strip exactly one dot from the right, if present
|
||||||
|
hostname = hostname[:-1]
|
||||||
|
allowed = re.compile("(?!-)[A-Z\d-]{1,63}(?<!-)$", re.IGNORECASE)
|
||||||
|
return all(allowed.match(x) for x in hostname.split("."))
|
||||||
|
|
||||||
|
|
||||||
|
def is_valid_ip(ip):
|
||||||
|
try:
|
||||||
|
socket.inet_aton(ip)
|
||||||
|
# legal
|
||||||
|
except socket.error:
|
||||||
|
return False
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
def start():
|
||||||
|
"""
|
||||||
|
Hit this function to start entering network credentials
|
||||||
|
"""
|
||||||
|
LOG.info('Editing sources.xml and passwords.xml')
|
||||||
|
# Fix for:
|
||||||
|
# DEBUG: Activating window ID: 13000
|
||||||
|
# INFO: Activate of window '13000' refused because there are active modal dialogs
|
||||||
|
# DEBUG: Activating window ID: 13000
|
||||||
|
xbmc.executebuiltin("Dialog.Close(all, true)")
|
||||||
|
# "In the following window, enter the server's hostname (or IP) where your
|
||||||
|
# Plex media resides. Mind the case!"
|
||||||
|
utils.messageDialog(utils.lang(29999), utils.lang(30200))
|
||||||
|
# "Enter server hostname (or IP)"
|
||||||
|
hostname = utils.dialog('input', utils.lang(30201))
|
||||||
|
if not hostname:
|
||||||
|
return
|
||||||
|
hostname = hostname.decode('utf-8').strip()
|
||||||
|
if not is_valid_hostname(hostname) and not is_valid_ip(hostname):
|
||||||
|
LOG.error('Entered invalid hostname or IP: %s', hostname)
|
||||||
|
# "The hostname or IP '{0}' that you entered is not valid"
|
||||||
|
utils.messageDialog(utils.lang(29999),
|
||||||
|
utils.lang(30204).format(hostname))
|
||||||
|
return
|
||||||
|
# "In the following window, enter the network protocol you would like to
|
||||||
|
# use. This is likely 'smb'."
|
||||||
|
utils.messageDialog(utils.lang(29999), utils.lang(30202))
|
||||||
|
# "Enter network protocol"
|
||||||
|
protocol = utils.dialog('input', utils.lang(30203))
|
||||||
|
if not protocol:
|
||||||
|
return
|
||||||
|
protocol = protocol.decode('utf-8').lower().strip()
|
||||||
|
if protocol not in SUPPORTED_PROTOCOLS:
|
||||||
|
LOG.error('Entered invalid protocol %s', protocol)
|
||||||
|
# "The protocol '{0}' that you entered is not supported."
|
||||||
|
utils.messageDialog(utils.lang(29999),
|
||||||
|
utils.lang(30205).format(protocol))
|
||||||
|
return
|
||||||
|
path = '%s://%s' % (protocol, hostname)
|
||||||
|
# Trailing slash at the end
|
||||||
|
paths = (path, '%s/' % path)
|
||||||
|
# Add hostname to sources.xml, if not already there
|
||||||
|
LOG.info('Hostname we are adding to sources.xml and passwords.xml: %s',
|
||||||
|
path)
|
||||||
|
try:
|
||||||
|
with utils.XmlKodiSetting('sources.xml',
|
||||||
|
force_create=True,
|
||||||
|
top_element='sources') as xml:
|
||||||
|
files = xml.root.find('files')
|
||||||
|
if files is None:
|
||||||
|
files = utils.etree.SubElement(xml.root, 'files')
|
||||||
|
utils.etree.SubElement(files,
|
||||||
|
'default',
|
||||||
|
attrib={'pathversion': '1'})
|
||||||
|
for source in files:
|
||||||
|
entry = source.find('path')
|
||||||
|
if entry is None:
|
||||||
|
LOG.debug('Entry is None')
|
||||||
|
continue
|
||||||
|
LOG.debug('entry found: %s', entry.text)
|
||||||
|
if entry.text in paths:
|
||||||
|
LOG.debug('Already have %s in sources.xml', path)
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
# Need to add an element for our hostname
|
||||||
|
LOG.debug('Adding subelement to sources.xml for %s', hostname)
|
||||||
|
source = utils.etree.SubElement(files, 'source')
|
||||||
|
utils.etree.SubElement(source, 'name').text = 'PKC %s' % hostname
|
||||||
|
utils.etree.SubElement(source,
|
||||||
|
'path',
|
||||||
|
attrib={'pathversion': '1'}).text = '%s/' % path
|
||||||
|
utils.etree.SubElement(source, 'allowsharing').text = 'false'
|
||||||
|
xml.write_xml = True
|
||||||
|
except utils.ParseError:
|
||||||
|
return
|
||||||
|
# Add or change username and password in passwords.xml
|
||||||
|
try:
|
||||||
|
with utils.XmlKodiSetting('passwords.xml',
|
||||||
|
force_create=True,
|
||||||
|
top_element='passwords') as xml:
|
||||||
|
for entry in xml.root:
|
||||||
|
source = entry.find('from')
|
||||||
|
if source is None:
|
||||||
|
continue
|
||||||
|
if source.text in paths:
|
||||||
|
LOG.debug('Found an existing passwords.xml entry for %s, '
|
||||||
|
'replacing it',
|
||||||
|
path)
|
||||||
|
xml.root.remove(entry)
|
||||||
|
entry = utils.etree.SubElement(xml.root, 'path')
|
||||||
|
# "Username"
|
||||||
|
user = utils.dialog('input', utils.lang(1014))
|
||||||
|
if user is None:
|
||||||
|
xml.write_xml = False
|
||||||
|
return
|
||||||
|
user = user.strip()
|
||||||
|
user = urllib.quote(user)
|
||||||
|
user = user.decode('utf-8')
|
||||||
|
# "Password"
|
||||||
|
# May also be blank!! (=user aborts dialog)
|
||||||
|
password = utils.dialog('input',
|
||||||
|
utils.lang(733),
|
||||||
|
'',
|
||||||
|
type='{alphanum}',
|
||||||
|
option='{hide}')
|
||||||
|
password = urllib.quote(password)
|
||||||
|
password = password.decode('utf-8')
|
||||||
|
utils.etree.SubElement(entry,
|
||||||
|
'from',
|
||||||
|
attrib={'pathversion': '1'}).text = '%s/' % path
|
||||||
|
login = '%s://%s:%s@%s/' % (protocol, user, password, hostname)
|
||||||
|
utils.etree.SubElement(entry,
|
||||||
|
'to',
|
||||||
|
attrib={'pathversion': '1'}).text = login
|
||||||
|
xml.write_xml = True
|
||||||
|
except utils.ParseError:
|
||||||
|
return
|
||||||
|
LOG.info('Successfully completed editing sources.xml and padsswords.xml')
|
Loading…
Reference in a new issue