mirror of
				https://github.com/RicterZ/nhentai.git
				synced 2025-11-04 02:50:55 +01:00 
			
		
		
		
	fix #31
This commit is contained in:
		@@ -1,3 +1,3 @@
 | 
				
			|||||||
__version__ = '0.2.16'
 | 
					__version__ = '0.2.17'
 | 
				
			||||||
__author__ = 'RicterZ'
 | 
					__author__ = 'RicterZ'
 | 
				
			||||||
__email__ = 'ricterzheng@gmail.com'
 | 
					__email__ = 'ricterzheng@gmail.com'
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -41,20 +41,27 @@ class Downloader(Singleton):
 | 
				
			|||||||
                                                                                extension)))
 | 
					                                                                                extension)))
 | 
				
			||||||
                return 1, url
 | 
					                return 1, url
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            response = None
 | 
				
			||||||
            with open(os.path.join(folder, base_filename.zfill(3) + extension), "wb") as f:
 | 
					            with open(os.path.join(folder, base_filename.zfill(3) + extension), "wb") as f:
 | 
				
			||||||
                i=0
 | 
					                i = 0
 | 
				
			||||||
                while i<10:
 | 
					                while i < 10:
 | 
				
			||||||
                    try:
 | 
					                    try:
 | 
				
			||||||
                        response = request('get', url, stream=True, timeout=self.timeout)
 | 
					                        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:
 | 
					                    except Exception as e:
 | 
				
			||||||
                        i+=1
 | 
					                        i += 1
 | 
				
			||||||
                        if not i<10:
 | 
					                        if not i < 10:
 | 
				
			||||||
                            logger.critical(str(e))
 | 
					                            logger.critical(str(e))
 | 
				
			||||||
                            return 0, None
 | 
					                            return 0, None
 | 
				
			||||||
                        continue
 | 
					                        continue
 | 
				
			||||||
 | 
					
 | 
				
			||||||
                    break
 | 
					                    break
 | 
				
			||||||
                if response.status_code != 200:
 | 
					
 | 
				
			||||||
                    raise NhentaiImageNotExistException
 | 
					 | 
				
			||||||
                length = response.headers.get('content-length')
 | 
					                length = response.headers.get('content-length')
 | 
				
			||||||
                if length is None:
 | 
					                if length is None:
 | 
				
			||||||
                    f.write(response.content)
 | 
					                    f.write(response.content)
 | 
				
			||||||
@@ -96,6 +103,9 @@ class Downloader(Singleton):
 | 
				
			|||||||
        if self.path:
 | 
					        if self.path:
 | 
				
			||||||
            folder = os.path.join(self.path, folder)
 | 
					            folder = os.path.join(self.path, folder)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        if len(folder) > 100:
 | 
				
			||||||
 | 
					            folder = folder[:100]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        if not os.path.exists(folder):
 | 
					        if not os.path.exists(folder):
 | 
				
			||||||
            logger.warn('Path \'{0}\' does not exist, creating.'.format(folder))
 | 
					            logger.warn('Path \'{0}\' does not exist, creating.'.format(folder))
 | 
				
			||||||
            try:
 | 
					            try:
 | 
				
			||||||
 
 | 
				
			|||||||
		Reference in New Issue
	
	Block a user