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
   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('Login to nhentai use credential \'%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.keyword:
  31         doujinshis = search_parser(options.keyword, options.page)
  32         print_doujinshi(doujinshis)
  33         if options.is_download:
  34             doujinshi_ids = map(lambda d: d['id'], doujinshis)
  35     else:
  36         doujinshi_ids = options.id
  37
  38     if doujinshi_ids:
  39         for id_ in doujinshi_ids:
  40             doujinshi_info = doujinshi_parser(id_)
  41             doujinshi_list.append(Doujinshi(**doujinshi_info))
  42
  43     if not options.is_show:
  44         downloader = Downloader(path=options.output_dir,
  45                                 thread=options.threads, timeout=options.timeout)
  46
  47         for doujinshi in doujinshi_list:
  48             doujinshi.downloader = downloader
  49             doujinshi.download()
  50             if not options.is_nohtml and not options.is_cbz:
  51                 generate_html(options.output_dir, doujinshi)
  52             elif options.is_cbz:
  53                 generate_cbz(options.output_dir, doujinshi)
  54
  55         if not platform.system() == 'Windows':
  56             logger.log(15, '🍻 All done.')
  57         else:
  58             logger.log(15, 'All done.')
  59
  60     else:
  61         [doujinshi.show() for doujinshi in doujinshi_list]
  62
  63
  64 def signal_handler(signal, frame):
  65     logger.error('Ctrl-C signal received. Quit.')
  66     exit(1)
  67
  68
  69 signal.signal(signal.SIGINT, signal_handler)
  70
  71 if __name__ == '__main__':
  72     main()