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 MAX_FIELD_LENGTH = 100
  10 EXT_MAP = {
  11     'j': 'jpg',
  12     'p': 'png',
  13     'g': 'gif',
  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 def trunk_string(string):
  29     if len(string) >= MAX_FIELD_LENGTH:
  30         string = string[:MAX_FIELD_LENGTH] + u'…'
  31
  32     return string
  33
  34
  35 class Doujinshi(object):
  36     def __init__(self, name=None, pretty_name=None, id=None, img_id=None,
  37                  ext='', pages=0, name_format='[%i][%a][%t]', **kwargs):
  38         self.name = name
  39         self.pretty_name = pretty_name
  40         self.id = id
  41         self.img_id = img_id
  42         self.ext = ext
  43         self.pages = pages
  44         self.downloader = None
  45         self.url = '%s/%d' % (DETAIL_URL, self.id)
  46         self.info = DoujinshiInfo(**kwargs)
  47
  48         name_format = name_format.replace('%i', str(self.id))
  49         name_format = name_format.replace('%a', self.info.artists)
  50
  51         name_format = name_format.replace('%t', trunk_string(self.name))
  52         name_format = name_format.replace('%p', trunk_string(self.pretty_name))
  53         name_format = name_format.replace('%s', trunk_string(self.info.subtitle))
  54         self.filename = format_filename(name_format)
  55
  56         self.table = [
  57             ["Parodies", self.info.parodies],
  58             ["Doujinshi", self.name],
  59             ["Subtitle", self.info.subtitle],
  60             ["Characters", self.info.characters],
  61             ["Authors", self.info.artists],
  62             ["Languages", self.info.languages],
  63             ["Tags", self.info.tags],
  64             ["URL", self.url],
  65             ["Pages", self.pages],
  66         ]
  67
  68     def __repr__(self):
  69         return '<Doujinshi: {0}>'.format(self.name)
  70
  71     def show(self):
  72
  73         logger.info(u'Print doujinshi information of {0}\n{1}'.format(self.id, tabulate(self.table)))
  74
  75     def download(self):
  76         logger.info('Starting to download doujinshi: %s' % self.name)
  77         if self.downloader:
  78             download_queue = []
  79             if len(self.ext) != self.pages:
  80                 logger.warning('Page count and ext count do not equal')
  81
  82             for i in range(1, min(self.pages, len(self.ext)) + 1):
  83                 download_queue.append('%s/%d/%d.%s' % (IMAGE_URL, int(self.img_id), i, self.ext[i - 1]))
  84
  85             self.downloader.download(download_queue, self.filename)
  86         else:
  87             logger.critical('Downloader has not been loaded')
  88
  89
  90 if __name__ == '__main__':
  91     test = Doujinshi(name='test nhentai doujinshi', id=1)
  92     print(test)
  93     test.show()
  94     try:
  95         test.download()
  96     except Exception as e:
  97         print('Exception: %s' % str(e))