nhentai.py

   1 #coding: utf-8
   2 from hentai.cmdline import cmd_parser, banner
   3 from hentai.parser import dojinshi_parser, search_parser, print_dojinshi
   4 from hentai.dojinshi import Dojinshi
   5 from hentai.downloader import Downloader
   6 from hentai.logger import logger
   7
   8
   9 __version__ = '0.1'
  10
  11
  12 def main():
  13     banner()
  14     options = cmd_parser()
  15
  16     logger.log(15, 'nHentai: あなたも変態。 いいね?')
  17
  18     dojinshi_ids = []
  19     dojinshi_list = []
  20
  21     if options.keyword:
  22         dojinshis = search_parser(options.keyword)
  23         if options.is_download:
  24             dojinshi_ids = map(lambda d: d['id'], dojinshis)
  25         else:
  26             print_dojinshi(dojinshis)
  27     else:
  28         dojinshi_ids = options.ids
  29
  30     if dojinshi_ids:
  31         for id in dojinshi_ids:
  32             dojinshi_info = dojinshi_parser(id)
  33             dojinshi_list.append(Dojinshi(**dojinshi_info))
  34     else:
  35         logger.log(15, 'Nothing has been done.')
  36         raise SystemExit
  37
  38     if options.is_download:
  39         downloader = Downloader(path=options.saved_path, thread=options.threads)
  40         for dojinshi in dojinshi_list:
  41             dojinshi.downloader = downloader
  42             dojinshi.download()
  43     else:
  44         map(lambda dojinshi: dojinshi.show(), dojinshi_list)
  45
  46     logger.log(15, u'🍺 All done.')
  47
  48
  49 if __name__ == '__main__':
  50     main()