# -*- coding: UTF-8 -*-

__revision__ = '$Id$'

# Copyright (c) 2007-2011
#
# 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 Library 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
# 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA

# You may use and distribute this software under the terms of the
# GNU General Public License, version 2 or later

import gutils
import movie
import string
import re

plugin_name         = "Filmtipset.se"
plugin_description  = "Filmtipset.se"
plugin_url          = "www.filmtipset.se"
plugin_language     = _("Swedish")
plugin_author       = "Michael Jahn"
plugin_author_email = "<mikej06@hotmail.com>"
plugin_version      = "1.2"

class Plugin(movie.Movie):

    def __init__(self, id):
        self.encode   = 'iso-8859-1'
        self.movie_id = id
        self.url      = 'http://www.filmtipset.se/film/' + str(self.movie_id)

    def get_image(self):
        self.image_url = ''
        tmp = gutils.trim(self.page, 'src="http://images.filmtipset.se/posters/', '"')
        if tmp != '':
            self.image_url = 'http://images.filmtipset.se/posters/' + tmp

    def get_o_title(self):
        self.o_title = gutils.trim(self.page, 'Originaltitel:', '</tr>')
        if self.o_title == '':
            self.o_title = gutils.trim(self.page, '<h1>', '</h1>')

    def get_title(self):
        self.title = gutils.after(gutils.trim(self.page, 'class="movie_header"', '<td'), '>')

    def get_director(self):
        self.director = gutils.trim(self.page, 'Regiss&ouml;r:', '</tr>')

    def get_plot(self):
        self.plot = gutils.regextrim(self.page, '<h2>Om [^>]*[>]', '</tr>')

    def get_year(self):
        self.year = gutils.trim(self.page, 'Utgivnings&aring;r:', '</tr>')

    def get_runtime(self):
        self.runtime = gutils.trim(self.page, 'L&auml;ngd:', ' min').strip()

    def get_genre(self):
        self.genre = gutils.trim(self.page, 'Nyckelord:', '</tr>').strip()
        if self.genre == '':
            self.genre = gutils.trim(self.page, 'Genre:</h2>', '</tr>').strip()

    def get_cast(self):
        self.cast = gutils.regextrim(self.page, 'Sk&aring;despelare[^:]*[:]', '</tr>')
        self.cast = self.cast.replace(', ', '\n')

    def get_classification(self):
        self.classification = ''

    def get_studio(self):
        self.studio = ''

    def get_o_site(self):
        self.o_site = ''
        tmp = gutils.trim(self.page, 'http://www.imdb.com', '"')
        if tmp != '':
            self.o_site = 'http://www.imdb.com' + tmp

    def get_site(self):
        self.site = self.url

    def get_trailer(self):
        self.trailer = ""

    def get_country(self):
        self.country = gutils.trim(self.page, 'Produktionsland:', '</span>').strip()

    def get_rating(self):
        self.rating = "0"
        tmp = gutils.trim(self.page, 'grade_images/grade1', '_')
        try:
            self.rating = str(int(tmp) * 2)
        except:
            pass

    def get_notes(self):
        self.notes = ''
        tmp = gutils.trim(self.page, 'Alt. titel:', '</span>')
        if tmp != '':
            self.notes = self.notes + 'Alt. titel:' + gutils.strip_tags(tmp).strip()

    def get_screenplay(self):
        self.screenplay = gutils.trim(self.page, 'Manus:', '</tr>')

class SearchPlugin(movie.SearchMovie):

    def __init__(self):
        self.original_url_search   = "http://www.filmtipset.se/search.cgi?field=name&field=orgname&show_graded=1&search_value="
        self.translated_url_search = "http://www.filmtipset.se/search.cgi?field=name&field=orgname&show_graded=1&search_value="
        self.encode                = 'iso-8859-1'

    def search(self,parent_window):
        if not self.open_search(parent_window):
            return None
        tmp_page = gutils.trim(self.page, 'Matchning', 'Hittade')
        if tmp_page == '':
            tmp_page = gutils.trim(self.page, 'Matchning', 'Visa fler')
        self.page = tmp_page
        return self.page

    def get_searches(self):
        elements1 = re.split('href="film/', self.page)
        elements1[0] = None
        for element in elements1:
            if element != None:
                searchResult = re.search('["&?]', element)
                if searchResult is None:
                    self.ids.append(gutils.before(element, '"'))
                else:
                    self.ids.append(element[:searchResult.end() - 1])
                self.titles.append(
                    gutils.strip_tags(gutils.trim(element, '>', '</a>')) + ' / ' +
                    gutils.clean(gutils.trim(element, 'Originaltitel:', '</div>').replace('&nbsp;', ' '))
                )

#
# Plugin Test
#
class SearchPluginTest(SearchPlugin):
    #
    # Configuration for automated tests:
    # dict { movie_id -> [ expected result count for original url, expected result count for translated url ] }
    #
    test_configuration = {
        'Rocky' : [ 19, 19 ],
    }

class PluginTest:
    #
    # Configuration for automated tests:
    # dict { movie_id -> dict { arribute -> value } }
    #
    # value: * True/False if attribute only should be tested for any value
    #        * or the expected value
    #
    test_configuration = {
        'rocky-balboa.html' : {
            'title'               : 'Rocky Balboa',
            'o_title'             : 'Rocky Balboa',
            'director'            : 'Sylvester Stallone',
            'plot'                : True,
            'cast'                : 'Sylvester Stallone\n\
Burt Young\n\
Milo Ventimiglia\n\
Geraldine Hughes\n\
Antonio Tarver\n\
James Francis Kelly III\n\
Tony Burton\n\
Henry G Sanders',
            'country'             : 'USA',
            'genre'               : 'Drama, Action',
            'classification'      : False,
            'studio'              : False,
            'o_site'              : 'http://www.imdb.com/title/tt0479143/',
            'site'                : 'http://www.filmtipset.se/film/rocky-balboa.html',
            'trailer'             : False,
            'year'                : 2006,
            'notes'               : False,
            'runtime'             : 102,
            'image'               : True,
            'rating'              : 6,
            'cameraman'           : False,
            'screenplay'          : 'Sylvester Stallone',
            'barcode'             : False
        },
    }
