weboob-devel/weboob/backends/ehentai/backend.py
2011-04-14 16:19:51 +02:00

74 lines
2.4 KiB
Python

# -*- coding: utf-8 -*-
# Copyright(C) 2010-2011 Roger Philibert
#
# 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 __future__ import with_statement
from weboob.capabilities.gallery import ICapGallery
from weboob.tools.backend import BaseBackend
from weboob.tools.value import Value, ValuesDict
from .browser import EHentaiBrowser
from .gallery import EHentaiGallery, EHentaiImage
__all__ = ['EHentaiBackend']
class EHentaiBackend(BaseBackend, ICapGallery):
NAME = 'ehentai'
MAINTAINER = 'Roger Philibert'
EMAIL = 'roger.philibert@gmail.com'
VERSION = '0.8'
DESCRIPTION = 'E-hentai galleries'
LICENSE = 'AGPLv3+'
BROWSER = EHentaiBrowser
CONFIG = ValuesDict(
Value('domain', label='Domain', default='g.e-hentai.org'),
Value('username', label='Username', default=''),
Value('password', label='Password', default='', masked=True))
def create_default_browser(self):
return self.create_browser(
self.config['domain'],
self.config['username'],
self.config['password'])
def iter_search_results(self, pattern=None, sortby=None, max_results=None):
with self.browser:
return self.browser.iter_search_results(pattern)
def iter_gallery_images(self, gallery):
self.fillobj(gallery, ('url',))
with self.browser:
return self.browser.iter_gallery_images(gallery)
def get_gallery(self, _id):
return EHentaiGallery(_id)
def fill_gallery(self, gallery, fields):
with self.browser:
self.browser.fill_gallery(gallery, fields)
def fill_image(self, image, fields):
with self.browser:
image.url = self.browser.get_image_url(image)
OBJECTS = {
EHentaiGallery: fill_gallery,
EHentaiImage: fill_image }