"""
parser.http.topBottomParser module (imdb package).

This module provides the classes (and the instances), used to parse the
lists of top 250 and bottom 100 movies.
E.g.:
    http://www.imdb.com/chart/top
    http://www.imdb.com/chart/bottom

Copyright 2009-2017 Davide Alberani <da@erlug.linux.it>

This program 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 2 of the License, or
(at your option) any later version.

This program 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 this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
"""

from imdb.utils import analyze_title

from .utils import Attribute, DOMParserBase, Extractor, analyze_imdbid


class DOMHTMLTop250Parser(DOMParserBase):
    """Parser for the "top 250" page.
    The page should be provided as a string, as taken from
    the www.imdb.com server.  The final result will be a
    dictionary, with a key for every relevant section.

    Example:
        tparser = DOMHTMLTop250Parser()
        result = tparser.parse(top250_html_string)
    """
    label = 'top 250'
    ranktext = 'top 250 rank'

    def _init(self):
        self.extractors = [
            Extractor(
                label=self.label,
                path="//div[@id='main']//div[1]//div//table//tbody//tr",
                attrs=Attribute(
                    key=None,
                    multi=True,
                    path={
                        self.ranktext: "./td[2]/text()",
                        'rating': "./td[3]//strong//text()",
                        'title': "./td[2]//a//text()",
                        'year': "./td[2]//span//text()",
                        'movieID': "./td[2]//a/@href",
                        'votes': "./td[3]//strong/@title"
                    }
                )
            )
        ]

    def postprocess_data(self, data):
        if not data or self.label not in data:
            return []
        mlist = []
        data = data[self.label]
        # Avoid duplicates.  A real fix, using XPath, is auspicabile.
        # XXX: probably this is no more needed.
        seenIDs = []
        for d in data:
            if 'movieID' not in d:
                continue
            if self.ranktext not in d:
                continue
            if 'title' not in d:
                continue
            theID = analyze_imdbid(d['movieID'])
            if theID is None:
                continue
            theID = str(theID)
            if theID in seenIDs:
                continue
            seenIDs.append(theID)
            minfo = analyze_title(d['title'] + ' ' + d['year'])
            try:
                minfo[self.ranktext] = int(d[self.ranktext].replace('.', ''))
            except ValueError:
                pass
            if 'votes' in d:
                try:
                    votes = d['votes'].replace(' user ratings', '')
                    votes = votes.split(' based on ')[1]    # is IndexError possible?
                    minfo['votes'] = int(votes.replace(',', ''))
                except (IndexError, ValueError):
                    pass
            if 'rating' in d:
                try:
                    minfo['rating'] = float(d['rating'])
                except ValueError:
                    pass
            mlist.append((theID, minfo))
        return mlist


class DOMHTMLBottom100Parser(DOMHTMLTop250Parser):
    """Parser for the "bottom 100" page.
    The page should be provided as a string, as taken from
    the www.imdb.com server.  The final result will be a
    dictionary, with a key for every relevant section.

    Example:
        tparser = DOMHTMLBottom100Parser()
        result = tparser.parse(bottom100_html_string)
    """
    label = 'bottom 100'
    ranktext = 'bottom 100 rank'


_OBJECTS = {
    'top250_parser': ((DOMHTMLTop250Parser,), None),
    'bottom100_parser': ((DOMHTMLBottom100Parser,), None)
}
