#!/usr/bin/env python

__license__ = 'GPL v3'

import re
from calibre.web.feeds.news import BasicNewsRecipe


class sportowefakty(BasicNewsRecipe):
    title = u'SportoweFakty'
    __author__ = 'Artur Stachecki <artur.stachecki@gmail.com>, Tomasz Długosz <tomek3d@gmail.com>'
    language = 'pl'
    description = u'Najważniejsze informacje sportowe z kraju i ze świata, relacje, komentarze, wywiady, zdjęcia!'
    oldest_article = 1
    masthead_url = 'http://www.sportowefakty.pl/images/logo.png'
    max_articles_per_feed = 100
    simultaneous_downloads = 5
    use_embedded_content = False
    remove_javascript = True
    no_stylesheets = True
    ignore_duplicate_articles = {'title', 'url'}

    keep_only_tags = [dict(attrs={'class': 'box-article'})]
    remove_tags = []
    remove_tags.append(dict(attrs={'class': re.compile(r'^newsStream')}))
    remove_tags.append(dict(attrs={'target': '_blank'}))

    feeds = [
        (u'Piłka Nożna', u'http://www.sportowefakty.pl/pilka-nozna/index.rss'),
        (u'Koszykówka', u'http://www.sportowefakty.pl/koszykowka/index.rss'),
        (u'Żużel', u'http://www.sportowefakty.pl/zuzel/index.rss'),
        (u'Siatkówka', u'http://www.sportowefakty.pl/siatkowka/index.rss'),
        (u'Zimowe', u'http://www.sportowefakty.pl/zimowe/index.rss'),
        (u'Hokej', u'http://www.sportowefakty.pl/hokej/index.rss'),
        (u'Moto', u'http://www.sportowefakty.pl/moto/index.rss'),
        (u'Tenis', u'http://www.sportowefakty.pl/tenis/index.rss')
    ]

    def get_article_url(self, article):
        link = article.get('link', None)
        if 'utm_source' in link:
            return link.split('?utm')[0]
        else:
            return link

    def print_version(self, url):
        print_url = url + '/drukuj'
        return print_url

    def preprocess_html(self, soup):
        head = soup.find('h1')
        if 'Fotorelacja' in self.tag_to_string(head):
            return None
        else:
            for alink in soup.findAll('a'):
                if alink.string is not None:
                    tstr = alink.string
                    alink.replaceWith(tstr)
            return soup