nhentai/command.py

   1 #!/usr/bin/env python2.7
   2 # coding: utf-8
   3 from __future__ import unicode_literals, print_function
   4 import signal
   5 import platform
   6
   7 from nhentai.cmdline import cmd_parser, banner
   8 from nhentai.parser import doujinshi_parser, search_parser, print_doujinshi, login_parser, tag_guessing, tag_parser
   9 from nhentai.doujinshi import Doujinshi
  10 from nhentai.downloader import Downloader
  11 from nhentai.logger import logger
  12 from nhentai.constant import BASE_URL
  13 from nhentai.utils import generate_html, generate_cbz
  14
  15
  16 def main():
  17     banner()
  18     logger.info('Using mirror: {0}'.format(BASE_URL))
  19     options = cmd_parser()
  20
  21     doujinshi_ids = []
  22     doujinshi_list = []
  23
  24     if options.login:
  25         username, password = options.login.split(':', 1)
  26         logger.info('Logging in to nhentai using credential pair \'%s:%s\'' % (username, '*' * len(password)))
  27         for doujinshi_info in login_parser(username=username, password=password):
  28             doujinshi_list.append(Doujinshi(**doujinshi_info))
  29
  30     if options.tag:
  31         tag_id = tag_guessing(options.tag)
  32         if tag_id:
  33             doujinshis = tag_parser(tag_id, max_page=options.max_page)
  34             print_doujinshi(doujinshis)
  35             if options.is_download:
  36                 doujinshi_ids = map(lambda d: d['id'], doujinshis)
  37
  38     if options.keyword:
  39         doujinshis = search_parser(options.keyword, options.page)
  40         print_doujinshi(doujinshis)
  41         if options.is_download:
  42             doujinshi_ids = map(lambda d: d['id'], doujinshis)
  43
  44     if doujinshi_ids:
  45         for id_ in doujinshi_ids:
  46             doujinshi_info = doujinshi_parser(id_)
  47             doujinshi_list.append(Doujinshi(**doujinshi_info))
  48
  49     if not options.is_show:
  50         downloader = Downloader(path=options.output_dir,
  51                                 thread=options.threads, timeout=options.timeout)
  52
  53         for doujinshi in doujinshi_list:
  54             doujinshi.downloader = downloader
  55             doujinshi.download()
  56             if not options.is_nohtml and not options.is_cbz:
  57                 generate_html(options.output_dir, doujinshi)
  58             elif options.is_cbz:
  59                 generate_cbz(options.output_dir, doujinshi)
  60
  61         if not platform.system() == 'Windows':
  62             logger.log(15, '🍻 All done.')
  63         else:
  64             logger.log(15, 'All done.')
  65
  66     else:
  67         [doujinshi.show() for doujinshi in doujinshi_list]
  68
  69
  70 def signal_handler(signal, frame):
  71     logger.error('Ctrl-C signal received. Stopping...')
  72     exit(1)
  73
  74
  75 signal.signal(signal.SIGINT, signal_handler)
  76
  77 if __name__ == '__main__':
  78     main()