0.2.18

requirement.txt missing new line

.travis.yml

@@ -16,3 +16,4 @@ script:
     - NHENTAI=https://nhentai.net nhentai --id=152503,146134 -t 10 --output=/tmp/
     - NHENTAI=https://nhentai.net nhentai -l nhentai_test:nhentai --output=/tmp/
     - NHENTAI=https://nhentai.net nhentai --tag lolicon
+    - NHENTAI=https://nhentai.net nhentai --id 92066 --output=/tmp/ --cbz

nhentai/__init__.py

@@ -1,3 +1,3 @@
-__version__ = '0.2.16'
+__version__ = '0.2.18'
 __author__ = 'RicterZ'
 __email__ = 'ricterzheng@gmail.com'

nhentai/cmdline.py

@@ -68,6 +68,8 @@ def cmd_parser():
 
     parser.add_option('--cbz', dest='is_cbz', action='store_true',
                       help='Generate Comic Book CBZ File')
+    parser.add_option('--rm-origin-dir', dest='rm_origin_dir', action='store_true', default=False,
+                      help='Remove downloaded doujinshi dir when generated CBZ file.')
                       
     try:
         sys.argv = list(map(lambda x: unicode(x.decode(sys.stdin.encoding)), sys.argv))

nhentai/command.py

@@ -21,8 +21,6 @@ def main():
     doujinshi_ids = []
     doujinshi_list = []
 
-    import pdb; pdb.set_trace()
-
     if options.login:
         username, password = options.login.split(':', 1)
         logger.info('Logging in to nhentai using credential pair \'%s:%s\'' % (username, '*' * len(password)))
@@ -61,7 +59,7 @@ def main():
             if not options.is_nohtml and not options.is_cbz:
                 generate_html(options.output_dir, doujinshi)
             elif options.is_cbz:
-                generate_cbz(options.output_dir, doujinshi)
+                generate_cbz(options.output_dir, doujinshi, options.rm_origin_dir)
 
         if not platform.system() == 'Windows':
             logger.log(15, '🍻 All done.')

nhentai/downloader.py

@@ -41,20 +41,27 @@ class Downloader(Singleton):
                                                                                 extension)))
                 return 1, url
 
+            response = None
             with open(os.path.join(folder, base_filename.zfill(3) + extension), "wb") as f:
                 i = 0
                 while i < 10:
                     try:
                         response = request('get', url, stream=True, timeout=self.timeout)
+                        if response.status_code != 200:
+                            raise NhentaiImageNotExistException
+
+                    except NhentaiImageNotExistException as e:
+                        raise e
+
                     except Exception as e:
                         i += 1
                         if not i < 10:
                             logger.critical(str(e))
                             return 0, None
                         continue
+
                     break
-                if response.status_code != 200:
-                    raise NhentaiImageNotExistException
+
                 length = response.headers.get('content-length')
                 if length is None:
                     f.write(response.content)

nhentai/parser.py

@@ -50,7 +50,7 @@ def login_parser(username, password):
     if not count:
         logger.error("Can't get your number of favorited doujins. Did the login failed?")
 
-    count = int(count.text.strip('(').strip(')'))
+    count = int(count.text.strip('(').strip(')').replace(',', ''))
     if count == 0:
         logger.warning('No favorites found')
         return []

nhentai/utils.py

@@ -83,12 +83,11 @@ def generate_html(output_dir='.', doujinshi_obj=None):
         logger.warning('Writen HTML Viewer failed ({})'.format(str(e)))
 
 
-def generate_cbz(output_dir='.', doujinshi_obj=None):
+def generate_cbz(output_dir='.', doujinshi_obj=None, rm_origin_dir=False):
     if doujinshi_obj is not None:
         doujinshi_dir = os.path.join(output_dir, format_filename('%s-%s' % (doujinshi_obj.id,
-                                                                            str(doujinshi_obj.name[:200]))))    
-        cbz_filename = os.path.join(output_dir, format_filename('%s-%s.cbz' % (doujinshi_obj.id,
-                                                                            str(doujinshi_obj.name[:200]))))
+                                                                            doujinshi_obj.name)))
+        cbz_filename = os.path.join(os.path.join(doujinshi_dir, '..'), '%s.cbz' % doujinshi_obj.id)
     else:
         cbz_filename = './doujinshi.cbz'
         doujinshi_dir = '.'
@@ -96,20 +95,18 @@ def generate_cbz(output_dir='.', doujinshi_obj=None):
     file_list = os.listdir(doujinshi_dir)
     file_list.sort()
 
+    logger.info('Writing CBZ file to path: {}'.format(cbz_filename))
     with zipfile.ZipFile(cbz_filename, 'w') as cbz_pf:
         for image in file_list:
             image_path = os.path.join(doujinshi_dir, image)
             cbz_pf.write(image_path, image)
 
+    if rm_origin_dir:
         shutil.rmtree(doujinshi_dir, ignore_errors=True)
+
     logger.log(15, 'Comic Book CBZ file has been write to \'{0}\''.format(doujinshi_dir))
 
 
-
-
-
-
-
 def format_filename(s):
     """Take a string and return a valid filename constructed from the string.
 Uses a whitelist approach: any characters not present in valid_chars are
@@ -124,4 +121,6 @@ an invalid filename.
     valid_chars = "-_.() %s%s" % (string.ascii_letters, string.digits)
     filename = ''.join(c for c in s if c in valid_chars)
     filename = filename.replace(' ', '_')  # I don't like spaces in filenames.
+    if len(filename) > 100:
+        filename = filename[:100]
     return filename

requirements.txt

@@ -2,4 +2,5 @@ requests>=2.5.0
 BeautifulSoup4>=4.0.0
 threadpool>=1.2.7
 tabulate>=0.7.5
-future>=0.15.2threadpool==1.3.2
+future>=0.15.2
+threadpool==1.3.2