mirror of
https://github.com/RicterZ/nhentai.git
synced 2025-05-05 02:56:16 +02:00
67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
# coding: utf-8
|
|
from __future__ import print_function
|
|
from tabulate import tabulate
|
|
from builtins import range
|
|
|
|
from nhentai.constant import DETAIL_URL, IMAGE_URL
|
|
from nhentai.logger import logger
|
|
|
|
|
|
class DoujinshiInfo(dict):
|
|
def __init__(self, **kwargs):
|
|
super(DoujinshiInfo, self).__init__(**kwargs)
|
|
|
|
def __getattr__(self, item):
|
|
try:
|
|
return dict.__getitem__(self, item)
|
|
except KeyError:
|
|
return ''
|
|
|
|
|
|
class Doujinshi(object):
|
|
def __init__(self, name=None, id=None, img_id=None, ext='jpg', pages=0, **kwargs):
|
|
self.name = name
|
|
self.id = id
|
|
self.img_id = img_id
|
|
self.ext = ext
|
|
self.pages = pages
|
|
self.downloader = None
|
|
self.url = '%s/%d' % (DETAIL_URL, self.id)
|
|
self.info = DoujinshiInfo(**kwargs)
|
|
|
|
def __repr__(self):
|
|
return '<Doujinshi: {0}>'.format(self.name)
|
|
|
|
def show(self):
|
|
table = [
|
|
["Doujinshi", self.name],
|
|
["Subtitle", self.info.subtitle],
|
|
["Characters", self.info.characters],
|
|
["Authors", self.info.artists],
|
|
["Language", self.info.language],
|
|
["Tags", self.info.tags],
|
|
["URL", self.url],
|
|
["Pages", self.pages],
|
|
]
|
|
logger.info(u'Print doujinshi information\n{0}'.format(tabulate(table)))
|
|
|
|
def download(self):
|
|
logger.info('Start download doujinshi: %s' % self.name)
|
|
if self.downloader:
|
|
download_queue = []
|
|
for i in range(1, self.pages + 1):
|
|
download_queue.append('%s/%d/%d.%s' % (IMAGE_URL, int(self.img_id), i, self.ext))
|
|
self.downloader.download(download_queue, self.id)
|
|
else:
|
|
logger.critical('Downloader has not be loaded')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
test = Doujinshi(name='test nhentai doujinshi', id=1)
|
|
print(test)
|
|
test.show()
|
|
try:
|
|
test.download()
|
|
except Exception as e:
|
|
print('Exception: %s' % str(e))
|