CouchPotatoServer/couchpotato/core/notifications/nmj/main.py

from couchpotato.api import addApiView
from couchpotato.core.event import addEvent
from couchpotato.core.helpers.encoding import tryUrlencode
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, host = 'localhost', **kwargs):

        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 {
            '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 {
            'success': False
        }

    def test(self):
        return {
            'success': self.addToLibrary()
        }