Merge pull request #21 from mentaterasmus/master

[nhentai.git] / nhentai / cmdline.py

diff --git a/nhentai/cmdline.py b/nhentai/cmdline.py
index 6d049ef95212563c1631aaa9e4866a171c8ee3c7..1cb58a065da0920fce73e5575dcec9e3cef981ee 100644 (file)
--- a/nhentai/cmdline.py
+++ b/nhentai/cmdline.py
@@ -34,7 +34,8 @@ def cmd_parser():
                           '\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('--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')
@@ -49,8 +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('--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):
@@ -64,35 +75,45 @@ def cmd_parser():
         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 > 15:
         logger.critical('Maximum number of used threads is 15')
-        exit(0)
+        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