nhentai/doujinshi.py

   1 # coding: utf-8
   2
   3 from tabulate import tabulate
   4
   5 from nhentai.constant import DETAIL_URL, IMAGE_URL
   6 from nhentai.logger import logger
   7 from nhentai.utils import format_filename
   8
   9 EXT_MAP = {
  10     'j': 'jpg',
  11     'p': 'png',
  12     'g': 'gif',
  13 }
  14
  15
  16 class DoujinshiInfo(dict):
  17     def __init__(self, **kwargs):
  18         super(DoujinshiInfo, self).__init__(**kwargs)
  19
  20     def __getattr__(self, item):
  21         try:
  22             return dict.__getitem__(self, item)
  23         except KeyError:
  24             return ''
  25
  26
  27 class Doujinshi(object):
  28     def __init__(self, name=None, id=None, img_id=None, ext='', pages=0, name_format='[%i][%a][%t]', **kwargs):
  29         self.name = name
  30         self.id = id
  31         self.img_id = img_id
  32         self.ext = ext
  33         self.pages = pages
  34         self.downloader = None
  35         self.url = '%s/%d' % (DETAIL_URL, self.id)
  36         self.info = DoujinshiInfo(**kwargs)
  37
  38         name_format = name_format.replace('%i', str(self.id))
  39         name_format = name_format.replace('%a', self.info.artists)
  40         name_format = name_format.replace('%t', self.name)
  41         name_format = name_format.replace('%s', self.info.subtitle)
  42         self.filename = format_filename(name_format)
  43
  44         self.table = [
  45             ["Parodies", self.info.parodies],
  46             ["Doujinshi", self.name],
  47             ["Subtitle", self.info.subtitle],
  48             ["Characters", self.info.characters],
  49             ["Authors", self.info.artists],
  50             ["Languages", self.info.languages],
  51             ["Tags", self.info.tags],
  52             ["URL", self.url],
  53             ["Pages", self.pages],
  54         ]
  55
  56     def __repr__(self):
  57         return '<Doujinshi: {0}>'.format(self.name)
  58
  59     def show(self):
  60
  61         logger.info(u'Print doujinshi information of {0}\n{1}'.format(self.id, tabulate(self.table)))
  62
  63     def download(self):
  64         logger.info('Starting to download doujinshi: %s' % self.name)
  65         if self.downloader:
  66             download_queue = []
  67
  68             if len(self.ext) != self.pages:
  69                 logger.warning('Page count and ext count do not equal')
  70
  71             for i in range(1, min(self.pages, len(self.ext)) + 1):
  72                 download_queue.append('%s/%d/%d.%s' % (IMAGE_URL, int(self.img_id), i, self.ext[i - 1]))
  73
  74             self.downloader.download(download_queue, self.filename)
  75
  76             '''
  77             for i in range(len(self.ext)):
  78                 download_queue.append('%s/%d/%d.%s' % (IMAGE_URL, int(self.img_id), i+1, EXT_MAP[self.ext[i]]))
  79             '''
  80
  81         else:
  82             logger.critical('Downloader has not been loaded')
  83
  84
  85 if __name__ == '__main__':
  86     test = Doujinshi(name='test nhentai doujinshi', id=1)
  87     print(test)
  88     test.show()
  89     try:
  90         test.download()
  91     except Exception as e:
  92         print('Exception: %s' % str(e))