from couchpotato.core.event import addEvent
from couchpotato.core.helpers.encoding import simplifyString, toUnicode
from couchpotato.core.logger import CPLog
from couchpotato.core.providers.show.base import ShowProvider
from thetvdb.tvdb_api import Tvdb 
import traceback

log = CPLog(__name__)


class TheTVDb(ShowProvider):
    
    def __init__(self):
        #addEvent('show.by_hash', self.byHash)
        addEvent('show.search', self.search, priority = 1)
        addEvent('show.info', self.getInfo, priority = 1)
        addEvent('show.episodes', self.getEpisodes, priority = 1)
        #addEvent('show.info_by_thetvdb', self.getInfoByTheTVDBId)

        # Use base wrapper
        #thetvdbtmdb.configure(self.conf('api_key'))
        self.tvdb = Tvdb(apikey="7966C02F860586D2", banners=True)

    #def byHash(self, file):
        #''' Find show by hash '''

        #if self.isDisabled():
            #return False

        #cache_key = 'tmdb.cache.%s' % simplifyString(file)
        #results = self.getCache(cache_key)

        #if not results:
            #log.debug('Searching for show by hash: %s', file)
            #try:
                #raw = tmdb.searchByHashingFile(file)

                #results = []
                #if raw:
                    #try:
                        #results = self.parseShow(raw)
                        #log.info('Found: %s', results['titles'][0] + ' (' + str(results.get('year', 0)) + ')')

                        #self.setCache(cache_key, results)
                        #return results
                    #except SyntaxError, e:
                        #log.error('Failed to parse XML response: %s', e)
                        #return False
            #except:
                #log.debug('No shows known by hash for: %s', file)
                #pass

        #return results

    def search(self, q, limit = 12):
        ''' Find show by name 
        show = {    'id': 74713,
                    'language': 'en',
                    'lid': 7,
                    'seriesid': '74713',
                    'seriesname': u'Breaking Bad',}
        '''

        if self.isDisabled():
            return False

        search_string = simplifyString(q)
        cache_key = 'thetvdb.cache.%s.%s' % (search_string, limit)
        results = self.getCache(cache_key)

        if not results:
            log.debug('Searching for show: %s', q)

            raw = None
            try:
                raw = self.tvdb.search(search_string)
                
            except:  # XXX: Make more specific
                log.error('Failed searching TheTVDB for "%s": %s', (search_string, traceback.format_exc()))

            results = []
            if raw:
                try:
                    nr = 0

                    for show in raw:
                        show = self.tvdb[int(show['id'])]
                        results.append(self.parseShow(show))

                        nr += 1
                        if nr == limit:
                            break

                    log.info('Found: %s', [result['titles'][0] + ' (' + str(result.get('year', 0)) + ')' for result in results])

                    self.setCache(cache_key, results)
                    return results
                except SyntaxError, e:
                    log.error('Failed to parse XML response: %s', e)
                    return False

        return results

    def getEpisodes(self, identifier=None):
        if not identifier:
            return []
        
        try:
            show = self.tvdb[int(identifier)]
        except:
            return []
        
        result = []
        for season in show.values():
            for episode in season.values():
                # Consider cache
                result.append(self.parseEpisode(episode))
                
        return result
                
    def getInfo(self, identifier = None):
        cache_key = 'thetvdb.cache.%s' % identifier
        result = self.getCache(cache_key)

        if not result:
            result = {}
            show = None

            try:
                log.debug('Getting info: %s', cache_key)
                #show = thetvdb.imdbLookup(id = identifier)
                show = self.tvdb[int(identifier)]
            except:
                pass

            if show:
                #result = self.parseShow(show[0])
                result = self.parseShow(show)
                self.setCache(cache_key, result)

        return result

    #def getInfoByTheTVDBId(self, id = None):

        #cache_key = 'thetvdb.cache.%s' % id
        #result = self.getCache(cache_key)

        #if not result:
            #result = {}
            #show = None

            #try:
                #log.debug('Getting info: %s', cache_key)
                #show = tmdb.getShowInfo(id = id)
            #except:
                #pass

            #if show:
                #result = self.parseShow(show)
                #self.setCache(cache_key, result)

        #return result

    def parseShow(self, show):
        """    
        show[74713] = {
                    'actors': u'|Bryan Cranston|Aaron Paul|Dean Norris|RJ Mitte|Betsy Brandt|Anna Gunn|Laura Fraser|Jesse Plemons|Christopher Cousins|Steven Michael Quezada|Jonathan Banks|Giancarlo Esposito|Bob Odenkirk|',
                    'added': None,
                    'addedby': None,
                    'airs_dayofweek': u'Sunday',
                    'airs_time': u'9:00 PM',
                    'banner': u'http://thetvdb.com/banners/graphical/81189-g13.jpg',
                    'contentrating': u'TV-MA',
                    'fanart': u'http://thetvdb.com/banners/fanart/original/81189-28.jpg',
                    'firstaired': u'2008-01-20',
                    'genre': u'|Crime|Drama|Suspense|',
                    'id': u'81189',
                    'imdb_id': u'tt0903747',
                    'language': u'en',
                    'lastupdated': u'1376620212',
                    'network': u'AMC',
                    'networkid': None,
                    'overview': u"Walter White, a struggling high school chemistry teacher is diagnosed with advanced lung cancer. He turns to a life of crime, producing and selling methamphetamine accompanied by a former student, Jesse Pinkman with the aim of securing his family's financial future before he dies.",
                    'poster': u'http://thetvdb.com/banners/posters/81189-22.jpg',
                    'rating': u'9.3',
                    'ratingcount': u'473',
                    'runtime': u'60',
                    'seriesid': u'74713',
                    'seriesname': u'Breaking Bad',
                    'status': u'Continuing',
                    'zap2it_id': u'SH01009396'}  
        """
        
        ## Images
        poster = self.getImage(show, type = 'poster', size = 'cover')
        backdrop = self.getImage(show, type = 'fanart', size = 'w1280')
        #poster_original = self.getImage(show, type = 'poster', size = 'original')
        #backdrop_original = self.getImage(show, type = 'backdrop', size = 'original')
        
        ## Genres
        genres = [] if show['genre'] is None else show['genre'].strip('|').split('|')

        ##  Year (not really needed for show)
        year = None

        show_data = {
            'via_thetvdb': True,
            'thetvdb_id': int(show['id']),
            'titles': [show['seriesname'], ],
            'original_title': show['seriesname'],
            'images': {
                'poster': [poster] if poster else [],
                'backdrop': [backdrop] if backdrop else [],
                'poster_original': [],
                'backdrop_original': [],
            },
            'imdb': show['imdb_id'],
            'runtime': show['runtime'],
            'released': show['firstaired'],
            'year': year,
            'plot': show['overview'],
            'genres': genres,
        }

        show_data = dict((k, v) for k, v in show_data.iteritems() if v)

        ## Add alternative names
        #for alt in ['original_name', 'alternative_name']:
            #alt_name = toUnicode(show.get(alt))
            #if alt_name and not alt_name in show_data['titles'] and alt_name.lower() != 'none' and alt_name != None:
                #show_data['titles'].append(alt_name)

        return show_data
    
    def parseEpisode(self, episode):
        """    
        ('episodenumber', u'1'),
        ('thumb_added', None),
        ('rating', u'7.7'),
        ('overview',
         u'Experienced waitress Max Black meets her new co-worker, former rich-girl Caroline Channing, and puts her skills to the test at an old but re-emerging Brooklyn diner. Despite her initial distaste for Caroline, Max eventually softens and the two team up for a new business venture.'),
        ('dvd_episodenumber', None),
        ('dvd_discid', None),
        ('combined_episodenumber', u'1'),
        ('epimgflag', u'7'),
        ('id', u'4099506'),
        ('seasonid', u'465948'),
        ('thumb_height', u'225'),
        ('tms_export', u'1374789754'),
        ('seasonnumber', u'1'),
        ('writer', u'|Michael Patrick King|Whitney Cummings|'),
        ('lastupdated', u'1371420338'),
        ('filename', u'http://thetvdb.com/banners/episodes/248741/4099506.jpg'),
        ('absolute_number', u'1'),
        ('ratingcount', u'102'),
        ('combined_season', u'1'),
        ('thumb_width', u'400'),
        ('imdb_id', u'tt1980319'),
        ('director', u'James Burrows'),
        ('dvd_chapter', None),
        ('dvd_season', None),
        ('gueststars',
         u'|Brooke Lyons|Noah Mills|Shoshana Bush|Cale Hartmann|Adam Korson|Alex Enriquez|Matt Cook|Bill Parks|Eugene Shaw|Sergey Brusilovsky|Greg Lewis|Cocoa Brown|Nick Jameson|'),
        ('seriesid', u'248741'),
        ('language', u'en'),
        ('productioncode', u'296793'),
        ('firstaired', u'2011-09-19'),
        ('episodename', u'Pilot')]
        """
        
        ## Images
        #poster = self.getImage(episode, type = 'poster', size = 'cover')
        #backdrop = self.getImage(episode, type = 'fanart', size = 'w1280')
        ##poster_original = self.getImage(episode, type = 'poster', size = 'original')
        ##backdrop_original = self.getImage(episode, type = 'backdrop', size = 'original')
        poster = []
        backdrop = []
        
        ## Genres
        genres = []

        ##  Year (not really needed for episode)
        year = None

        episode_data = {
            'via_thetvdb': True,
            'thetvdb_id': int(episode['id']),
            'titles': [episode['episodename'], ],
            'original_title': episode['episodename'],
            'images': {
                'poster': [poster] if poster else [],
                'backdrop': [backdrop] if backdrop else [],
                'poster_original': [],
                'backdrop_original': [],
            },
            'imdb': episode['imdb_id'],
            'runtime': None,
            'released': episode['firstaired'],
            'year': year,
            'plot': episode['overview'],
            'genres': genres,
        }

        episode_data = dict((k, v) for k, v in episode_data.iteritems() if v)

        ## Add alternative names
        #for alt in ['original_name', 'alternative_name']:
            #alt_name = toUnicode(episode.get(alt))
            #if alt_name and not alt_name in episode_data['titles'] and alt_name.lower() != 'none' and alt_name != None:
                #episode_data['titles'].append(alt_name)

        return episode_data

    def getImage(self, show, type = 'poster', size = 'cover'):
        """"""
        # XXX: Need to implement size
        image_url = ''
                
        for res, res_data in show['_banners'].get(type, {}).items():
            for bid, banner_info in res_data.items():
                image_url = banner_info.get('_bannerpath', '')
                break  

        return image_url

    def isDisabled(self):
        if self.conf('api_key') == '':
            log.error('No API key provided.')
            True
        else:
            False