weboob-devel/weboob/backends/downparadise/pages/index.py
2011-04-20 09:26:34 +02:00

59 lines
2 KiB
Python

# -*- coding: utf-8 -*-
# Copyright(C) 2010-2011 Nicolas Duhamel
#
# This file is part of weboob.
#
# weboob is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# weboob 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 Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with weboob. If not, see <http://www.gnu.org/licenses/>.
from weboob.tools.browser import BasePage
from .base import DownparadisePage
from weboob.capabilities.collection import Collection
__all__ = ['IndexPage']
class IndexPage(DownparadisePage):
def on_loaded(self):
self.collections = []
self.parse_forums()
def parse_forums(self):
""" Parse all forums """
def do(id):
self.browser.location(id)
return self.browser.page.iter_threads()
maintable = self.document.xpath("//div[@id='wrapheader']/table")[3]
for line in maintable.xpath("./tr"):
forums = line.xpath(".//a[@class='forumlink']")
for fo in forums:
coll = Collection()
coll.title = fo.text.strip().encode('latin-1')
coll.id = fo.get("href")
for link in line.getiterator('a'):
if "subforum" in link.attrib.get('class', ""):
sub = Collection(title=link.text.strip().encode('latin-1'))
sub.id = link.get("href")
sub.children = do
coll.appendchild(sub)
if not coll.children:
coll.children = do
self.collections.append(coll)
def get_collections(self):
return self.collections