parent
							
								
									ff21a710ae
								
							
						
					
					
						commit
						55e7c75e12
					
				
					 1 changed files with 39 additions and 34 deletions
				
			
		
							
								
								
									
										73
									
								
								youtube-dl
									
										
									
									
									
								
							
							
						
						
									
										73
									
								
								youtube-dl
									
										
									
									
									
								
							| 
						 | 
					@ -305,7 +305,6 @@ class FileDownloader(object):
 | 
				
			||||||
			template_dict = dict(info_dict)
 | 
								template_dict = dict(info_dict)
 | 
				
			||||||
			template_dict['epoch'] = unicode(long(time.time()))
 | 
								template_dict['epoch'] = unicode(long(time.time()))
 | 
				
			||||||
			filename = self.params['outtmpl'] % template_dict
 | 
								filename = self.params['outtmpl'] % template_dict
 | 
				
			||||||
			self.report_destination(filename)
 | 
					 | 
				
			||||||
		except (ValueError, KeyError), err:
 | 
							except (ValueError, KeyError), err:
 | 
				
			||||||
			self.trouble('ERROR: invalid output template or system charset: %s' % str(err))
 | 
								self.trouble('ERROR: invalid output template or system charset: %s' % str(err))
 | 
				
			||||||
		if self.params['nooverwrites'] and os.path.exists(filename):
 | 
							if self.params['nooverwrites'] and os.path.exists(filename):
 | 
				
			||||||
| 
						 | 
					@ -319,17 +318,8 @@ class FileDownloader(object):
 | 
				
			||||||
			return
 | 
								return
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		try:
 | 
							try:
 | 
				
			||||||
			outstream = open(filename, 'ab')
 | 
								success = self._do_download(filename, info_dict['url'])
 | 
				
			||||||
		except (OSError, IOError), err:
 | 
							except (OSError, IOError), err:
 | 
				
			||||||
			self.trouble('ERROR: unable to open for writing: %s' % str(err))
 | 
					 | 
				
			||||||
			return
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		try:
 | 
					 | 
				
			||||||
			self._do_download(outstream, info_dict['url'])
 | 
					 | 
				
			||||||
			outstream.close()
 | 
					 | 
				
			||||||
		except (OSError, IOError), err:
 | 
					 | 
				
			||||||
			outstream.close()
 | 
					 | 
				
			||||||
			os.remove(filename)
 | 
					 | 
				
			||||||
			raise UnavailableFormatError
 | 
								raise UnavailableFormatError
 | 
				
			||||||
		except (urllib2.URLError, httplib.HTTPException, socket.error), err:
 | 
							except (urllib2.URLError, httplib.HTTPException, socket.error), err:
 | 
				
			||||||
			self.trouble('ERROR: unable to download video data: %s' % str(err))
 | 
								self.trouble('ERROR: unable to download video data: %s' % str(err))
 | 
				
			||||||
| 
						 | 
					@ -338,11 +328,12 @@ class FileDownloader(object):
 | 
				
			||||||
			self.trouble('ERROR: content too short (expected %s bytes and served %s)' % (err.expected, err.downloaded))
 | 
								self.trouble('ERROR: content too short (expected %s bytes and served %s)' % (err.expected, err.downloaded))
 | 
				
			||||||
			return
 | 
								return
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		try:
 | 
							if success:
 | 
				
			||||||
			self.post_process(filename, info_dict)
 | 
								try:
 | 
				
			||||||
		except (PostProcessingError), err:
 | 
									self.post_process(filename, info_dict)
 | 
				
			||||||
			self.trouble('ERROR: postprocessing: %s' % str(err))
 | 
								except (PostProcessingError), err:
 | 
				
			||||||
			return
 | 
									self.trouble('ERROR: postprocessing: %s' % str(err))
 | 
				
			||||||
 | 
									return
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	def download(self, url_list):
 | 
						def download(self, url_list):
 | 
				
			||||||
		"""Download a given list of URLs."""
 | 
							"""Download a given list of URLs."""
 | 
				
			||||||
| 
						 | 
					@ -379,32 +370,36 @@ class FileDownloader(object):
 | 
				
			||||||
			if info is None:
 | 
								if info is None:
 | 
				
			||||||
				break
 | 
									break
 | 
				
			||||||
	
 | 
						
 | 
				
			||||||
	def _do_download(self, stream, url):
 | 
						def _do_download(self, filename, url):
 | 
				
			||||||
 | 
							stream = None
 | 
				
			||||||
 | 
							open_mode = 'ab'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		basic_request = urllib2.Request(url, None, std_headers)
 | 
							basic_request = urllib2.Request(url, None, std_headers)
 | 
				
			||||||
		request = urllib2.Request(url, None, std_headers)
 | 
							request = urllib2.Request(url, None, std_headers)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		# Resume transfer if filesize is non-zero
 | 
							# Attempt to resume download with "continuedl" option
 | 
				
			||||||
		resume_len = stream.tell()
 | 
							if os.path.isfile(filename):
 | 
				
			||||||
 | 
								resume_len = os.path.getsize(filename)
 | 
				
			||||||
 | 
							else:
 | 
				
			||||||
 | 
								resume_len = 0
 | 
				
			||||||
		if self.params['continuedl'] and resume_len != 0:
 | 
							if self.params['continuedl'] and resume_len != 0:
 | 
				
			||||||
			self.report_resuming_byte(resume_len)
 | 
								self.report_resuming_byte(resume_len)
 | 
				
			||||||
			request.add_header('Range','bytes=%d-' % resume_len)
 | 
								request.add_header('Range','bytes=%d-' % resume_len)
 | 
				
			||||||
		else:
 | 
					
 | 
				
			||||||
			stream.close()
 | 
							# Establish connection
 | 
				
			||||||
			stream = open(stream.name,'wb')
 | 
					 | 
				
			||||||
		try:
 | 
							try:
 | 
				
			||||||
			data = urllib2.urlopen(request)
 | 
								data = urllib2.urlopen(request)
 | 
				
			||||||
		except urllib2.HTTPError, e:
 | 
							except (urllib2.HTTPError, ), err:
 | 
				
			||||||
			if not e.code == 416: #  416 is 'Requested range not satisfiable'
 | 
								if err.code != 416: #  416 is 'Requested range not satisfiable'
 | 
				
			||||||
				raise
 | 
									raise
 | 
				
			||||||
			data = urllib2.urlopen(basic_request)
 | 
								data = urllib2.urlopen(basic_request)
 | 
				
			||||||
			content_length = data.info()['Content-Length']
 | 
								content_length = data.info()['Content-Length']
 | 
				
			||||||
			if content_length is not None and long(content_length) == resume_len:
 | 
								if content_length is not None and long(content_length) == resume_len:
 | 
				
			||||||
				self.report_file_already_downloaded(stream.name)
 | 
									self.report_file_already_downloaded(filename)
 | 
				
			||||||
				return
 | 
									return True
 | 
				
			||||||
			else:
 | 
								else:
 | 
				
			||||||
				self.report_unable_to_resume()
 | 
									self.report_unable_to_resume()
 | 
				
			||||||
				stream.close()
 | 
									open_mode = 'wb'
 | 
				
			||||||
				stream = open(stream.name,'wb')
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
		data_len = data.info().get('Content-length', None)
 | 
							data_len = data.info().get('Content-length', None)
 | 
				
			||||||
		data_len_str = self.format_bytes(data_len)
 | 
							data_len_str = self.format_bytes(data_len)
 | 
				
			||||||
| 
						 | 
					@ -412,12 +407,6 @@ class FileDownloader(object):
 | 
				
			||||||
		block_size = 1024
 | 
							block_size = 1024
 | 
				
			||||||
		start = time.time()
 | 
							start = time.time()
 | 
				
			||||||
		while True:
 | 
							while True:
 | 
				
			||||||
			# Progress message
 | 
					 | 
				
			||||||
			percent_str = self.calc_percent(byte_counter, data_len)
 | 
					 | 
				
			||||||
			eta_str = self.calc_eta(start, time.time(), data_len, byte_counter)
 | 
					 | 
				
			||||||
			speed_str = self.calc_speed(start, time.time(), byte_counter)
 | 
					 | 
				
			||||||
			self.report_progress(percent_str, data_len_str, speed_str, eta_str)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
			# Download and write
 | 
								# Download and write
 | 
				
			||||||
			before = time.time()
 | 
								before = time.time()
 | 
				
			||||||
			data_block = data.read(block_size)
 | 
								data_block = data.read(block_size)
 | 
				
			||||||
| 
						 | 
					@ -426,15 +415,31 @@ class FileDownloader(object):
 | 
				
			||||||
			if data_block_len == 0:
 | 
								if data_block_len == 0:
 | 
				
			||||||
				break
 | 
									break
 | 
				
			||||||
			byte_counter += data_block_len
 | 
								byte_counter += data_block_len
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								# Open file just in time
 | 
				
			||||||
 | 
								if stream is None:
 | 
				
			||||||
 | 
									try:
 | 
				
			||||||
 | 
										stream = open(filename, open_mode)
 | 
				
			||||||
 | 
										self.report_destination(filename)
 | 
				
			||||||
 | 
									except (OSError, IOError), err:
 | 
				
			||||||
 | 
										self.trouble('ERROR: unable to open for writing: %s' % str(err))
 | 
				
			||||||
 | 
										return False
 | 
				
			||||||
			stream.write(data_block)
 | 
								stream.write(data_block)
 | 
				
			||||||
			block_size = self.best_block_size(after - before, data_block_len)
 | 
								block_size = self.best_block_size(after - before, data_block_len)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								# Progress message
 | 
				
			||||||
 | 
								percent_str = self.calc_percent(byte_counter, data_len)
 | 
				
			||||||
 | 
								eta_str = self.calc_eta(start, time.time(), data_len, byte_counter)
 | 
				
			||||||
 | 
								speed_str = self.calc_speed(start, time.time(), byte_counter)
 | 
				
			||||||
 | 
								self.report_progress(percent_str, data_len_str, speed_str, eta_str)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
			# Apply rate limit
 | 
								# Apply rate limit
 | 
				
			||||||
			self.slow_down(start, byte_counter)
 | 
								self.slow_down(start, byte_counter)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		self.report_finish()
 | 
							self.report_finish()
 | 
				
			||||||
		if data_len is not None and str(byte_counter) != data_len:
 | 
							if data_len is not None and str(byte_counter) != data_len:
 | 
				
			||||||
			raise ContentTooShortError(byte_counter, long(data_len))
 | 
								raise ContentTooShortError(byte_counter, long(data_len))
 | 
				
			||||||
 | 
							return True
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class InfoExtractor(object):
 | 
					class InfoExtractor(object):
 | 
				
			||||||
	"""Information Extractor class.
 | 
						"""Information Extractor class.
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue