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