You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
121 lines
3.7 KiB
121 lines
3.7 KiB
from couchpotato.api import addApiView
|
|
from couchpotato.core.event import addEvent
|
|
from couchpotato.core.helpers.encoding import tryUrlencode
|
|
from couchpotato.core.helpers.request import getParams, jsonified
|
|
from couchpotato.core.logger import CPLog
|
|
from couchpotato.core.notifications.base import Notification
|
|
import re
|
|
import telnetlib
|
|
|
|
try:
|
|
import xml.etree.cElementTree as etree
|
|
except ImportError:
|
|
import xml.etree.ElementTree as etree
|
|
|
|
log = CPLog(__name__)
|
|
|
|
|
|
class NMJ(Notification):
|
|
|
|
def __init__(self):
|
|
addEvent('renamer.after', self.addToLibrary)
|
|
addApiView(self.testNotifyName(), self.test)
|
|
addApiView('notify.nmj.auto_config', self.autoConfig)
|
|
|
|
def autoConfig(self):
|
|
|
|
params = getParams()
|
|
host = params.get('host', 'localhost')
|
|
|
|
database = ''
|
|
mount = ''
|
|
|
|
try:
|
|
terminal = telnetlib.Telnet(host)
|
|
except Exception:
|
|
log.error('Warning: unable to get a telnet session to %s', (host))
|
|
return self.failed()
|
|
|
|
log.debug('Connected to %s via telnet', (host))
|
|
terminal.read_until('sh-3.00# ')
|
|
terminal.write('cat /tmp/source\n')
|
|
terminal.write('cat /tmp/netshare\n')
|
|
terminal.write('exit\n')
|
|
tnoutput = terminal.read_all()
|
|
|
|
match = re.search(r'(.+\.db)\r\n?(.+)(?=sh-3.00# cat /tmp/netshare)', tnoutput)
|
|
|
|
if match:
|
|
database = match.group(1)
|
|
device = match.group(2)
|
|
log.info('Found NMJ database %s on device %s', (database, device))
|
|
else:
|
|
log.error('Could not get current NMJ database on %s, NMJ is probably not running!', (host))
|
|
return self.failed()
|
|
|
|
if device.startswith('NETWORK_SHARE/'):
|
|
match = re.search('.*(?=\r\n?%s)' % (re.escape(device[14:])), tnoutput)
|
|
|
|
if match:
|
|
mount = match.group().replace('127.0.0.1', host)
|
|
log.info('Found mounting url on the Popcorn Hour in configuration: %s', (mount))
|
|
else:
|
|
log.error('Detected a network share on the Popcorn Hour, but could not get the mounting url')
|
|
return self.failed()
|
|
|
|
return jsonified({
|
|
'success': True,
|
|
'database': database,
|
|
'mount': mount,
|
|
})
|
|
|
|
def addToLibrary(self, message = None, group = {}):
|
|
if self.isDisabled(): return
|
|
|
|
host = self.conf('host')
|
|
mount = self.conf('mount')
|
|
database = self.conf('database')
|
|
|
|
if mount:
|
|
log.debug('Try to mount network drive via url: %s', (mount))
|
|
try:
|
|
data = self.urlopen(mount)
|
|
except:
|
|
return False
|
|
|
|
params = {
|
|
'arg0': 'scanner_start',
|
|
'arg1': database,
|
|
'arg2': 'background',
|
|
'arg3': '',
|
|
}
|
|
params = tryUrlencode(params)
|
|
UPDATE_URL = 'http://%(host)s:8008/metadata_database?%(params)s'
|
|
updateUrl = UPDATE_URL % {'host': host, 'params': params}
|
|
|
|
try:
|
|
response = self.urlopen(updateUrl)
|
|
except:
|
|
return False
|
|
|
|
try:
|
|
et = etree.fromstring(response)
|
|
result = et.findtext('returnValue')
|
|
except SyntaxError, e:
|
|
log.error('Unable to parse XML returned from the Popcorn Hour: %s', (e))
|
|
return False
|
|
|
|
if int(result) > 0:
|
|
log.error('Popcorn Hour returned an errorcode: %s', (result))
|
|
return False
|
|
else:
|
|
log.info('NMJ started background scan')
|
|
return True
|
|
|
|
def failed(self):
|
|
return jsonified({'success': False})
|
|
|
|
def test(self):
|
|
return jsonified({'success': self.addToLibrary()})
|
|
|
|
|
|
|