E-Hentai backend

weboob/backends/ehentai/backend.py

@@ -0,0 +1,79 @@
+# -*- 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
+
+import datetime
+import re
+import urllib
+
+from weboob.capabilities.gallery import ICapGallery
+from weboob.tools.backend import BaseBackend
+from weboob.tools.misc import to_unicode
+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.7'
+    DESCRIPTION = 'E-hentai galleries'
+    LICENSE = 'GPLv3'
+    BROWSER = EHentaiBrowser
+    CONFIG = ValuesDict(
+        Value('domain', label='Domain', default='g.e-hentai.org'),
+        Value('username', label='Username', default=None),
+        Value('password', label='Password', default=None, 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 }