Merge pull request #21 from mentaterasmus/master

[nhentai.git] / nhentai / cmdline.py

diff --git a/nhentai/cmdline.py b/nhentai/cmdline.py
index c10435d1d6403bec418f2743451d33e9d9f7f619..1cb58a065da0920fce73e5575dcec9e3cef981ee 100644 (file)
--- a/nhentai/cmdline.py
+++ b/nhentai/cmdline.py
@@ -8,15 +8,19 @@ except ImportError:
     pass
 
 import nhentai.constant as constant
-from nhentai.utils import urlparse
+from nhentai.utils import urlparse, generate_html
 from nhentai.logger import logger
 
-reload(sys)
-sys.setdefaultencoding(sys.stdin.encoding)
+try:
+    reload(sys)
+    sys.setdefaultencoding(sys.stdin.encoding)
+except NameError:
+    # python3
+    pass
 
 
 def banner():
-    logger.info('''nHentai: あなたも変態。 いいね?
+    logger.info(u'''nHentai: あなたも変態。 いいね?
        _   _            _        _
  _ __ | | | | ___ _ __ | |_ __ _(_)
 | '_ \| |_| |/ _ \ '_ \| __/ _` | |
@@ -26,9 +30,13 @@ def banner():
 
 
 def cmd_parser():
-    parser = OptionParser()
-    parser.add_option('--download', dest='is_download', action='store_true', help='download doujinshi or not')
-    parser.add_option('--show-info', dest='is_show', action='store_true', help='just show the doujinshi information.')
+    parser = OptionParser('\n  nhentai --search [keyword] --download'
+                          '\n  NHENTAI=http://h.loli.club nhentai --id [ID ...]'
+                          '\n\nEnvironment Variable:\n'
+                          '  NHENTAI                 nhentai mirror url')
+    parser.add_option('--download', dest='is_download', action='store_true',
+                      help='download doujinshi (for search result)')
+    parser.add_option('--show-info', dest='is_show', action='store_true', help='just show the doujinshi information')
     parser.add_option('--id', type='string', dest='id', action='store', help='doujinshi ids set, e.g. 1,2,3')
     parser.add_option('--search', type='string', dest='keyword', action='store', help='search doujinshi by keyword')
     parser.add_option('--page', type='int', dest='page', action='store', default=1,
@@ -42,7 +50,18 @@ def cmd_parser():
                       help='timeout of download doujinshi')
     parser.add_option('--proxy', type='string', dest='proxy', action='store', default='',
                       help='use proxy, example: http://127.0.0.1:1080')
+    parser.add_option('--html', dest='html_viewer', action='store_true',
+                      help='generate a html viewer at current directory')
+
+    parser.add_option('--login', '-l', type='str', dest='login', action='store',
+                      help='username:password pair of nhentai account')
 
+    parser.add_option('--nohtml', dest='is_nohtml', action='store_true',
+                      help='Don\'t generate HTML')
+
+    parser.add_option('--cbz', dest='is_cbz', action='store_true',
+                      help='Generate Comic Book CBZ File')                      
+                      
     try:
         sys.argv = list(map(lambda x: unicode(x.decode(sys.stdin.encoding)), sys.argv))
     except (NameError, TypeError):
@@ -52,35 +71,49 @@ def cmd_parser():
 
     args, _ = parser.parse_args(sys.argv[1:])
 
+    if args.html_viewer:
+        generate_html()
+        exit(0)
+
+    if args.login:
+        try:
+            _, _ = args.login.split(':', 1)
+        except ValueError:
+            logger.error('Invalid `username:password` pair.')
+            exit(1)
+
+        if not args.is_download:
+            logger.warning('YOU DO NOT SPECIFY `--download` OPTION !!!')
+
     if args.tags:
         logger.warning('`--tags` is under construction')
-        exit(0)
+        exit(1)
 
     if args.id:
         _ = map(lambda id: id.strip(), args.id.split(','))
-        args.id = set(map(int, filter(lambda id: id.isdigit(), _)))
+        args.id = set(map(int, filter(lambda id_: id_.isdigit(), _)))
 
-    if (args.is_download or args.is_show) and not args.id and not args.keyword:
+    if (args.is_download or args.is_show) and not args.id and not args.keyword and not args.login:
         logger.critical('Doujinshi id(s) are required for downloading')
         parser.print_help()
-        exit(0)
+        exit(1)
 
-    if not args.keyword and not args.id:
+    if not args.keyword and not args.id and not args.login:
         parser.print_help()
-        exit(0)
+        exit(1)
 
     if args.threads <= 0:
         args.threads = 1
 
-    elif args.threads > 10:
-        logger.critical('Maximum number of used threads is 10')
-        exit(0)
+    elif args.threads > 15:
+        logger.critical('Maximum number of used threads is 15')
+        exit(1)
 
     if args.proxy:
         proxy_url = urlparse(args.proxy)
         if proxy_url.scheme not in ('http', 'https'):
             logger.error('Invalid protocol \'{0}\' of proxy, ignored'.format(proxy_url.scheme))
         else:
-            constant.PROXY = {proxy_url.scheme: args.proxy}
+            constant.PROXY = {'http': args.proxy, 'https': args.proxy}
 
     return args