nhentai/doujinshi.py

   1 # coding: utf-8
   2 from __future__ import print_function, unicode_literals
   3 from tabulate import tabulate
   4 from future.builtins import range
   5
   6 from nhentai.constant import DETAIL_URL, IMAGE_URL
   7 from nhentai.logger import logger
   8 from nhentai.utils import format_filename
   9
  10
  11 EXT_MAP = {
  12     'j': 'jpg',
  13     'p': 'png',
  14 }
  15
  16
  17 class DoujinshiInfo(dict):
  18     def __init__(self, **kwargs):
  19         super(DoujinshiInfo, self).__init__(**kwargs)
  20
  21     def __getattr__(self, item):
  22         try:
  23             return dict.__getitem__(self, item)
  24         except KeyError:
  25             return ''
  26
  27
  28 class Doujinshi(object):
  29     def __init__(self, name=None, id=None, img_id=None, ext='', pages=0, **kwargs):
  30         self.name = name
  31         self.id = id
  32         self.img_id = img_id
  33         self.ext = ext
  34         self.pages = pages
  35         self.downloader = None
  36         self.url = '%s/%d' % (DETAIL_URL, self.id)
  37         self.info = DoujinshiInfo(**kwargs)
  38
  39     def __repr__(self):
  40         return '<Doujinshi: {0}>'.format(self.name)
  41
  42     def show(self):
  43         table = [
  44             ["Doujinshi", self.name],
  45             ["Subtitle", self.info.subtitle],
  46             ["Characters", self.info.characters],
  47             ["Authors", self.info.artists],
  48             ["Language", self.info.language],
  49             ["Tags", self.info.tags],
  50             ["URL", self.url],
  51             ["Pages", self.pages],
  52         ]
  53         logger.info(u'Print doujinshi information of {0}\n{1}'.format(self.id, tabulate(table)))
  54
  55     def download(self):
  56         logger.info('Start download doujinshi: %s' % self.name)
  57         if self.downloader:
  58             download_queue = []
  59             for i in range(len(self.ext)):
  60                 download_queue.append('%s/%d/%d.%s' % (IMAGE_URL, int(self.img_id), i+1, EXT_MAP[self.ext[i]]))
  61
  62             self.downloader.download(download_queue, format_filename('%s-%s' % (self.id, self.name[:200])))
  63         else:
  64             logger.critical('Downloader has not be loaded')
  65
  66
  67 if __name__ == '__main__':
  68     test = Doujinshi(name='test nhentai doujinshi', id=1)
  69     print(test)
  70     test.show()
  71     try:
  72         test.download()
  73     except Exception as e:
  74         print('Exception: %s' % str(e))