pchtrakt.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Authors: Jonathan Lauwers / Frederic Haumont
# URL: http://github.com/pchtrakt/pchtrakt
#
# This file is part of pchtrakt.
#
# pchtrakt is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# pchtrakt is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with pchtrakt.  If not, see <http://www.gnu.org/licenses/>.

# pchtrakt - Connect your PCH 200 Series to trakt.tv :)
# pchtrakt uses some pyhton lib :
#    - tvdb_api (https://github.com/dbr/tvdb_api)
#    - nbrhttpconnection (another project)
#     - some classes from Sick Beard (http://sickbeard.com/)

import sys
import getopt

if not hasattr(sys, "setdefaultencoding"):
	reload(sys)

import os
import json
from pchtrakt.pch import *
from pchtrakt.scrobble import *
from pchtrakt.config import *
from pchtrakt.movieparser import MovieResultNotFound
from pchtrakt import mediaparser as mp
from time import sleep, time
from lib.tvdb_api import tvdb_api
from lib.tvdb_api import tvdb_exceptions
from lib.utilities import checkSettings

if SyncCheck >= 0:
    from lib.oversight import OversightSyncMain

if YAMJSyncCheck >= 0:
    from lib.yamj2 import YAMJSyncMain

if BetaSeriesScrobbleTvShow:
    from lib import pylast

from xml.etree import ElementTree
from httplib import HTTPException, BadStatusLine

class PchTraktException(Exception):
    pass

tvdb = tvdb_api.Tvdb()
pchtrakt.oPchRequestor = PchRequestor()
pchtrakt.oPchMusicRequestor = PchMusicRequestor()
pchtrakt.mediaparser = mp.MediaParser()


class media():
    def __str__(self):
        if isinstance(self.parsedInfo, mp.MediaParserResultTVShow):
            msg = 'TV Show : {0} - Season:{1} - Episode:{2} ' \
                    '- {3}% - {4} - TvDB: {5}'.format(
                    self.parsedInfo.name,
                    self.parsedInfo.season_number,
                    self.parsedInfo.episode_numbers,
                    self.oStatus.percent,
                    self.oStatus.status,
                    self.parsedInfo.id)
        else:
            msg = 'Movie : {0} - Year : {1} - ' \
                    '{2}% - IMDB: {3}'.format(
                    self.parsedInfo.name,
                    self.parsedInfo.year,
                    self.oStatus.percent,
                    self.parsedInfo.id)
        return msg

myMedia = media()

def printHelp():
    print('Usage {0} <options>'.format('pythpn pchtrak.py'))
    print('Options:')
    print('    -h,--help    :    display this message')
    print('    -d,--daemon  :    launches pchtrakt in the background')

def getParams():
    try:
        opts, args = getopt.getopt(sys.argv[1:],
                                   "dht:",
                                   ['daemon', 'help'])
    except getopt.GetoptError:
        print("Available options: -d, --daemon")
        sys.exit()

    for o, a in opts:
        # Run as a daemon
        if o in ('-d', '--daemon'):
            if sys.platform == 'win32':
                print('Daemonize not supported under Windows, ' \
                      'starting normally')
            else:
                pchtrakt.DAEMON = True

        if o in ('-h', '--help'):
            printHelp()
            sys.exit()

        if o in ('-t'):
            try:
                checkSettings()
            except AuthenticationTraktError:
                pass
            finally:
                sys.exit()

def checkUpdate(when):
    hash = ""
    for row in os.popen('git ls-remote 2>&-').read().split('\n'):
        if row.find('refs/heads/dvp') != -1:
            hash = row.split()[0]
            break
    if hash == PchTraktVersion or hash == "":
        if when == "first":
            pchtrakt.logger.info(' [Pchtrakt] Starting Pchtrakt version = ' + PchTraktVersion[-4:]  + ' Millers Mods (Running latest ' + pchtrakt.chip + ' version)')
            if TraktScrobbleTvShow or TraktScrobbleMovie:
                checkSettings()
    else:
        if AutoUpdate >= 0:
            if when == "first":
                pchtrakt.logger.info(' [Pchtrakt] Starting Pchtrakt version = ' + PchTraktVersion[-4:]  + ' Millers Mods (Running latest ' + pchtrakt.chip + ' version)')
                pchtrakt.logger.info(' [Pchtrakt] A new version is online. Starting update')
            else:
                pchtrakt.logger.info(' [Pchtrakt] Checking for new version.... ' + PchTraktVersion[-4:]  + ' Millers Mods (Running latest ' + pchtrakt.chip + ' version)')
                pchtrakt.logger.info(' [Pchtrakt] A new version is online. Starting update')
            os.system("./daemon.sh update")
        elif AutoUpdate < 0:
            if when == "first":
                pchtrakt.logger.info(' [Pchtrakt] Starting Pchtrakt version = ' + PchTraktVersion[-4:] + ' Millers Mods (' + pchtrakt.chip + ' version)')
                pchtrakt.logger.info(' [Pchtrakt] A new version is online. For manual install, download from https://github.com/cptjhmiller/pchtrakt/archive/dvp.zip')
                if TraktScrobbleTvShow or TraktScrobbleMovie:
                   checkSettings()

def daemonize():
    """
    Fork off as a daemon
    """

    # Make a non-session-leader child process
    try:
        pid = os.fork()  # @UndefinedVariable - only available in UNIX
        if pid != 0:
            sys.exit(0)
    except OSError as e:
        raise RuntimeError("1st fork failed: %s [%d]" %
                   (e.strerror, e.errno))

    os.setsid()  # @UndefinedVariable - only available in UNIX

    # Make sure I can read my own files and shut out others
    prev = os.umask(0)
    os.umask(prev and int('077', 8))

    # Make the child a session-leader by detaching from the terminal
    try:
        pid = os.fork()  # @UndefinedVariable - only available in UNIX
        if pid != 0:
            sys.exit(0)
    except OSError as e:
        raise RuntimeError("2st fork failed: %s [%d]" % (e.strerror, e.errno))

    dev_null = file('/dev/null', 'r')
    os.dup2(dev_null.fileno(), sys.stdin.fileno())

def Reset():
    Debug('[Pchtrakt] RESETTING DATA')
    pchtrakt.StopTrying = 0
    pchtrakt.watched = 0
    pchtrakt.lastPath = ''
    pchtrakt.lastName = ''
    pchtrakt.lastShowName = ''
    pchtrakt.DirtyName = ''
    pchtrakt.isMovie = 0
    pchtrakt.isTvShow = 0
    pchtrakt.Check = 0
    pchtrakt.Ttime = 0
    pchtrakt.CreatedFile = 0
    myMedia.parsedInfo = None
    pchtrakt.logger.info(' [Pchtrakt] Waiting for a file to start.....')

def doWork():
    pchtrakt.problem = ''
    if hasattr(myMedia, 'parsedInfoOld'):
        myMedia.parsedInfoOld.percent = myMedia.oStatus.percent
    myMedia.oStatus = pchtrakt.oPchRequestor.getStatus(ipPch, 10)
    if myMedia.oStatus.status == EnumStatus.NOPLAY and (LastfmNowPlaying is True or LastfmScrobble is True):
        myMedia.oStatus = pchtrakt.oPchMusicRequestor.getStatus(ipPch, 10)
        if myMedia.oStatus.status == EnumStatus.PLAY:
            pchtrakt.logger.info(' [Pchtrakt] Music found playing')
            network = ''
            LastScrobble = 0
            while myMedia.oStatus.status != EnumStatus.NOPLAY:
                if network == '':
                    network = pylast.LastFMNetwork(api_key = "e43fb2fa81a6d8fa85b9e630c90bdf27", api_secret = 
                                                   "c51b8c3fa4b6b5b61cdfe59775b59c07", username = LastfmUsername, password_hash = pylast.md5(LastfmPwd))
                if LastfmNowPlaying is True and (pchtrakt.lastPath != myMedia.oStatus.fullPath and myMedia.oStatus.artist != ''):
                    network.update_now_playing(myMedia.oStatus.artist, myMedia.oStatus.title, duration = int(myMedia.oStatus.totalTime - myMedia.oStatus.currentTime))
                    pchtrakt.lastPath = myMedia.oStatus.fullPath
                    LastScrobble = 0
                    pchtrakt.logger.info('  [Last.fm] Now playing %s - %s '% (myMedia.oStatus.artist, myMedia.oStatus.title))
                if LastfmScrobble is True and LastScrobble == 0:
                    if myMedia.oStatus.percent > 75:
                        network.scrobble(myMedia.oStatus.artist, myMedia.oStatus.title, int(time()))
                        LastScrobble = 1
                        pchtrakt.logger.info('  [Last.fm] Scrobbled %s - %s '% (myMedia.oStatus.artist, myMedia.oStatus.title))
                sleep(sleepTime)
                myMedia.oStatus = pchtrakt.oPchMusicRequestor.getStatus(ipPch, 10)
    if pchtrakt.lastPath != myMedia.oStatus.fullPath and pchtrakt.StopTrying == 0:
        myMedia.parsedInfo = None
    if YamjWatched == True and not pchtrakt.watched and myMedia.oStatus.percent > watched_percent and pchtrakt.CreatedFile == 0:
        watchedFileCreation(myMedia)
    if not pchtrakt.StopTrying:
        if myMedia.oStatus.status not in [EnumStatus.NOPLAY,
                                          EnumStatus.UNKNOWN,
                                          EnumStatus.PAUSE]:
            pchtrakt.allowedPauseTime = TraktMaxPauseTime
            if myMedia.parsedInfo == None:
                Debug('[Pchtrakt] full path: ' + myMedia.oStatus.fullPath)
                pchtrakt.logger.info(' [Pchtrakt] %s File %s' % (myMedia.oStatus.status, myMedia.oStatus.fileName))
                myMedia.parsedInfo = pchtrakt.mediaparser.parse(myMedia.oStatus.fullPath)
                pchtrakt.Ttime = myMedia.oStatus.totalTime
                if hasattr(myMedia, 'parsedInfo'):
                    myMedia.parsedInfoOld = myMedia.parsedInfo
                if hasattr(myMedia.parsedInfo, 'dirty'):
                    pchtrakt.DirtyName = myMedia.parsedInfo.dirty
            if myMedia.oStatus.fullPath != '':
                if isIgnored(myMedia) == True:
                    startWait('it is being ignored')
                else:
                    videoStatusHandle(myMedia)
            if pchtrakt.problem != '':
                startWait(pchtrakt.problem)
        elif (myMedia.oStatus.status == EnumStatus.PAUSE
            and pchtrakt.allowedPauseTime > 0):
            pchtrakt.allowedPauseTime -= sleepTime
        else:
            if pchtrakt.lastPath != '':
                if myMedia.oStatus.status == EnumStatus.NOPLAY:
                    pchtrakt.logger.info(' [Pchtrakt] video/music file has stopped')
                    videoStopped(myMedia)
                if pchtrakt.allowedPauseTime <= 0:
                    pchtrakt.logger.info(' [Pchtrakt] It seems you paused ' \
                                         'the video for more than {0} minutes: ' \
                                         'I say to trakt you stopped watching ' \
                                         'your video'.format(TraktMaxPauseTime/60))
                Reset()

def stopTrying():
    try:
        pchtrakt.StopTrying = 1
        pchtrakt.lastPath = myMedia.oStatus.fullPath
        pchtrakt.lastName = myMedia.oStatus.fileName
        sleep(sleepTime)
    except Exception as e:
        pass

def New_Pin():
    msg = '[traktAPI] You need to autherize a new pin'
    pchtrakt.logger.warning(msg)
    config=ConfigParser.RawConfigParser()
    config.read(r'pchtrakt.ini')
    config.set('Trakt', 'api_token', 'None')
    config.set('Trakt', 'refresh_token', 'None')
    config.set('Trakt', 'api_pin', 'get at http://trakt.tv/pin/361')
    with open(r'pchtrakt.ini', 'wb') as configfile:
        config.write(configfile)
    pchtrakt.stop = 1

def startWait(msg=''):
    if msg != '':
        pchtrakt.logger.info(' [Pchtrakt] waiting for file to stop as %s' % msg)
    else:
        pchtrakt.logger.info(' [Pchtrakt] waiting for file to stop as somthing is wrong with file name')
    waitforstop = pchtrakt.oPchRequestor.getStatus(ipPch, 10)
    NowPlaying = waitforstop.fileName
    while waitforstop.status != 'noplay' and NowPlaying == waitforstop.fileName:
        sleep(sleepTime)
        waitforstop = pchtrakt.oPchRequestor.getStatus(ipPch, 10)
        if YamjWatched == True and not pchtrakt.watched and waitforstop.percent > watched_percent and pchtrakt.CreatedFile == 0:
            try:
                watchedFileCreation(myMedia)
            except BaseException as e:
                pchtrakt.logger.error(e)
    Reset()

def starttvdbWait():
    if pchtrakt.online:
        while urllib.urlopen("http://thetvdb.com").getcode() != 200:
            pchtrakt.StopTrying = 1
            while myMedia.oStatus.status != EnumStatus.NOPLAY:
                sleep(sleepTime)
                myMedia.oStatus = pchtrakt.oPchRequestor.getStatus(ipPch, 10)
                if YamjWatched == True and not pchtrakt.watched and myMedia.oStatus.percent > watched_percent and pchtrakt.CreatedFile == 0:
                    try:
                        watchedFileCreation(myMedia)
                    except BaseException as e:
                        pchtrakt.logger.error(e)
            pchtrakt.StopTrying = 0

def StartUP():
    if pchtrakt.online:
        checkUpdate('first')
        if SyncCheck >= 0:
            OversightSyncMain().OversightSync()
        if YAMJSyncCheck >= 0:
            YAMJSyncMain().YAMJSync()
        if os.path.isfile('missed.scrobbles'):
            pchtrakt.logger.info(' [Pchtrakt] Found missed scrobbles, updating trakt.tv')
            with open('missed.scrobbles','r+') as f:
                pchtrakt.missed = json.load(f)
            new_list = {}
            for xname in pchtrakt.missed:
                pchtrakt.logger.info(u' [Pchtrakt] marking %s watched on trakt.tv' % xname.split('/')[::-1][0])
                myMedia.parsedInfo = pchtrakt.mediaparser.parse(xname.split('/')[::-1][0])
                myMedia.idxEpisode = 0
                myMedia.oStatus = PchStatus()
                myMedia.oStatus.totalTime = pchtrakt.missed[xname]['Totaltime']
                myMedia.oStatus.percent = 100
                if isinstance(myMedia.parsedInfo,mp.MediaParserResultTVShow):
                    pchtrakt.watched = showIsSeen(myMedia, pchtrakt.missed[xname]['Totallength'])
                elif isinstance(myMedia.parsedInfo,mp.MediaParserResultMovie):
                    pchtrakt.watched = movieIsSeen(myMedia, pchtrakt.missed[xname]['Totallength'])
                if not pchtrakt.watched:
                    pchtrakt.logger.info(u' [traktAPI]  %s was NOT marked as watched on trakt.tv' % xname.split('/')[::-1][0])
                    new_list[xname]={"Totaltime": int(pchtrakt.missed[xname]['Totaltime']), "Totallength": int(pchtrakt.missed[xname]['Totallength'])}
            if new_list != {}:
                with open('missed.scrobbles','w') as f:
                    json.dump(new_list, f, separators=(',',':'), indent=4)
            else:
                os.remove('missed.scrobbles')
    else:
        pchtrakt.logger.info(' [Pchtrakt] Pchtrakt START version = ' + PchTraktVersion[-4:] + ' Millers Mods (' + pchtrakt.chip + ' version)')
        pchtrakt.logger.info(' [Pchtrakt] No internet - can not check for updates')
        pchtrakt.logger.info(' [Pchtrakt] .watched files will be created but no xml updating or track.tv scrobbles will happen.')
        pchtrakt.logger.info(' [Pchtrakt] track.tv scrobbles will be saved and processed when next on-line.')
    pchtrakt.logger.info(' [Pchtrakt] Waiting for a file to start.....')
    pchtrakt.Started = time()
    pchtrakt.Started2 = pchtrakt.Started

if __name__ == '__main__':
    getParams()
    if pchtrakt.DAEMON:
        daemonize()

    #Get model
    pchtrakt.chip = os.popen('/opt/syb/sigma/bin/gbus_read_uint32 0x0002fee8 2>&-').read()[-5:-1]
    if pchtrakt.chip == "8911":
        pchtrakt.chip = "A400 series"
    elif pchtrakt.chip == "8647":
        pchtrakt.chip = "300 series"
    elif pchtrakt.chip == "8643":
        pchtrakt.chip = "200 series"
    elif pchtrakt.chip == "8635":
        pchtrakt.chip = "100 series"
    else:
        pchtrakt.chip = ""
    
    #Main Routine
    FirstRun = 0
    while not pchtrakt.stop:
        try:
            if FirstRun == 0:
                StartUP()
                pchtrakt.Started1 = time()
                pchtrakt.Started2 = time()
                pchtrakt.Started3 = time()
                FirstRun = 1
            else:
                doWork()
                sleep(sleepTime)
                if myMedia.oStatus.status == EnumStatus.NOPLAY:
                    if float(time()) > float(pchtrakt.Started1+AutoUpdate) and AutoUpdate > 0:
                        checkUpdate('no')
                        pchtrakt.Started1 = time()
                    if float(time()) > float(pchtrakt.Started2+SyncCheck) and SyncCheck > 0:
                        OversightSync()
                        pchtrakt.Started2 = time()
                    if float(time()) > float(pchtrakt.Started2+YAMJSyncCheck) and YAMJSyncCheck > 0:
                        YAMJSync()
                        pchtrakt.Started3 = time()
        except tvdb_exceptions.tvdb_shownotfound as e:
            stopTrying()
            msg = ('[The TvDB] The show was not found ' \
            '{0} '.format(pchtrakt.lastPath))
            pchtrakt.logger.warning(msg)
            startWait('The show was not found')
            sys.exc_clear()
            pass
        except tvdb_exceptions.tvdb_error, e:
            stopTrying()
            pchtrakt.logger.error('[The TvDB] Site apears to be down')
            starttvdbWait()
        except MovieResultNotFound as e:
            stopTrying()
            msg = '[Pchtrakt] Unable to find match for file - {0}'.format(e.file_name)
            pchtrakt.logger.warning(msg)
            startWait()
        except PchTraktException as e:
            stopTrying()
            msg = '[Pchtrakt] PchTraktException - {0}'.format(e)
            pchtrakt.logger.error(msg)
            sleep(sleepTime)
        except BetaSerieAuthenticationException as e:
            pchtrakt.logger.error(e)
        except IOError, e:
             if hasattr(e, 'reason'):
                 if e.reason == 'Unauthorized':
                    pchtrakt.logger.error('[traktAPI] Login or password incorrect')
                    startWait()
                 else:
                    pchtrakt.logger.error('Reason: %s ' % (e.reason))
             elif hasattr(e, 'code'):
                 pchtrakt.logger.error('Error code: %s' % (e.code))
        except ValueError as e:
            pchtrakt.logger.error('[traktAPI] Problem with trakt.tv site  - {0}'.format(e))
        except AttributeError as e:
            if pchtrakt.online:
                Debug('[Pchtrakt] ID not found will retry in 60 seconds  - {0}'.format(e))
                while not (hasattr(myMedia.parsedInfo, 'id')) and myMedia.oStatus.status == EnumStatus.PLAY:
                    sleep(15)
                    myMedia.parsedInfo = pchtrakt.mediaparser.parse(myMedia.oStatus.fileName)
                    myMedia.oStatus = pchtrakt.oPchRequestor.getStatus(ipPch, 10)
                    Debug('[Pchtrakt] ID not found will retry in 60 seconds')
                videoStatusHandleMovie(myMedia)
        except Exception as e:
            if hasattr(e, 'message') and e.message != '':  # error 401 or 503, possibly others
                if e.message == "global name 'NotFoundError' is not defined":
                    msg = '[traktAPI] Unable to find match for file - {0}'.format(pchtrakt.lastName)
                    pchtrakt.logger.warning(msg)
                    startWait()
                elif e.message == "local variable 'TRAKT_REFRESH_TOKEN' referenced before assignment":
                    New_Pin()
                elif hasattr(e.message, 'message'):
                    if e.message.message == '401 Client Error: Unauthorized':
                        New_Pin()
                    elif e.message.message == '404 Client Error: Not Found':
                        startWait()
                elif hasattr(e.message, 'response'):
                    if e.message.response.json()['error'] == 'invalid_grant':
                        New_Pin()
            elif hasattr(e, 'details'):
                if e.details == 'Invalid authentication token. Please check username/password supplied':
                    pchtrakt.logger.warning('[Last.fm] Please check your username and/or your password')
                else:
                   stopTrying()
                   pchtrakt.logger.exception('This should never happend! Please contact me with the error if you read this')
                   pchtrakt.logger.exception(pchtrakt.lastPath)
                   pchtrakt.logger.exception(e.message)
                   startWait()
            else:
                stopTrying()
                pchtrakt.logger.exception('This should never happend! Please contact me with the error if you read this')
                pchtrakt.logger.exception(pchtrakt.lastPath)
                pchtrakt.logger.exception(e)
                startWait()

    pchtrakt.logger.info('   [Pchtrakt] STOPPED')