Make the most prominent output strings Unicode and fix Unicode title bug

This commit is contained in:
Ricardo Garcia 2008-07-25 13:28:41 +02:00
parent 0c2dc87d9e
commit 76a7f36400

View file

@ -179,12 +179,15 @@ class FileDownloader(object):
def to_stdout(self, message, skip_eol=False): def to_stdout(self, message, skip_eol=False):
"""Print message to stdout if not in quiet mode.""" """Print message to stdout if not in quiet mode."""
if not self._params.get('quiet', False): if not self._params.get('quiet', False):
sys.stdout.write('%s%s' % (message, ['\n', ''][skip_eol])) if skip_eol:
print message,
else:
print message
sys.stdout.flush() sys.stdout.flush()
def to_stderr(self, message): def to_stderr(self, message):
"""Print message to stderr.""" """Print message to stderr."""
sys.stderr.write('%s\n' % message) print >>sys.stderr, message
def fixed_template(self): def fixed_template(self):
"""Checks if the output template is fixed.""" """Checks if the output template is fixed."""
@ -220,16 +223,16 @@ class FileDownloader(object):
def report_destination(self, filename): def report_destination(self, filename):
"""Report destination filename.""" """Report destination filename."""
self.to_stdout('[download] Destination: %s' % filename) self.to_stdout(u'[download] Destination: %s' % filename)
def report_progress(self, percent_str, data_len_str, speed_str, eta_str): def report_progress(self, percent_str, data_len_str, speed_str, eta_str):
"""Report download progress.""" """Report download progress."""
self.to_stdout('\r[download] %s of %s at %s ETA %s' % self.to_stdout(u'\r[download] %s of %s at %s ETA %s' %
(percent_str, data_len_str, speed_str, eta_str), skip_eol=True) (percent_str, data_len_str, speed_str, eta_str), skip_eol=True)
def report_finish(self): def report_finish(self):
"""Report download finished.""" """Report download finished."""
self.to_stdout('') self.to_stdout(u'')
def download(self, url_list): def download(self, url_list):
"""Download a given list of URLs.""" """Download a given list of URLs."""
@ -386,7 +389,7 @@ class InfoExtractor(object):
def to_stderr(self, message): def to_stderr(self, message):
"""Print message to stderr.""" """Print message to stderr."""
sys.stderr.write('%s\n' % message) print >>sys.stderr, message
def _real_initialize(self): def _real_initialize(self):
"""Real initialization process. Redefine in subclasses.""" """Real initialization process. Redefine in subclasses."""
@ -410,23 +413,23 @@ class YoutubeIE(InfoExtractor):
def report_login(self): def report_login(self):
"""Report attempt to log in.""" """Report attempt to log in."""
self.to_stdout('[youtube] Logging in') self.to_stdout(u'[youtube] Logging in')
def report_age_confirmation(self): def report_age_confirmation(self):
"""Report attempt to confirm age.""" """Report attempt to confirm age."""
self.to_stdout('[youtube] Confirming age') self.to_stdout(u'[youtube] Confirming age')
def report_webpage_download(self, video_id): def report_webpage_download(self, video_id):
"""Report attempt to download webpage.""" """Report attempt to download webpage."""
self.to_stdout('[youtube] %s: Downloading video webpage' % video_id) self.to_stdout(u'[youtube] %s: Downloading video webpage' % video_id)
def report_information_extraction(self, video_id): def report_information_extraction(self, video_id):
"""Report attempt to extract video information.""" """Report attempt to extract video information."""
self.to_stdout('[youtube] %s: Extracting video information' % video_id) self.to_stdout(u'[youtube] %s: Extracting video information' % video_id)
def report_video_url(self, video_id, video_real_url): def report_video_url(self, video_id, video_real_url):
"""Report extracted video URL.""" """Report extracted video URL."""
self.to_stdout('[youtube] %s: URL: %s' % (video_id, video_real_url)) self.to_stdout(u'[youtube] %s: URL: %s' % (video_id, video_real_url))
def _real_initialize(self): def _real_initialize(self):
if self._downloader is None: if self._downloader is None:
@ -449,7 +452,7 @@ class YoutubeIE(InfoExtractor):
else: else:
raise netrc.NetrcParseError('No authenticators for %s' % self._NETRC_MACHINE) raise netrc.NetrcParseError('No authenticators for %s' % self._NETRC_MACHINE)
except (IOError, netrc.NetrcParseError), err: except (IOError, netrc.NetrcParseError), err:
self.to_stderr('WARNING: parsing .netrc: %s' % str(err)) self.to_stderr(u'WARNING: parsing .netrc: %s' % str(err))
return return
# No authentication to be performed # No authentication to be performed
@ -469,10 +472,10 @@ class YoutubeIE(InfoExtractor):
self.report_login() self.report_login()
login_results = urllib2.urlopen(request).read() login_results = urllib2.urlopen(request).read()
if re.search(r'(?i)<form[^>]* name="loginForm"', login_results) is not None: if re.search(r'(?i)<form[^>]* name="loginForm"', login_results) is not None:
self.to_stderr('WARNING: unable to log in: bad username or password') self.to_stderr(u'WARNING: unable to log in: bad username or password')
return return
except (urllib2.URLError, httplib.HTTPException, socket.error), err: except (urllib2.URLError, httplib.HTTPException, socket.error), err:
self.to_stderr('WARNING: unable to log in: %s' % str(err)) self.to_stderr(u'WARNING: unable to log in: %s' % str(err))
return return
# Confirm age # Confirm age
@ -485,14 +488,14 @@ class YoutubeIE(InfoExtractor):
self.report_age_confirmation() self.report_age_confirmation()
age_results = urllib2.urlopen(request).read() age_results = urllib2.urlopen(request).read()
except (urllib2.URLError, httplib.HTTPException, socket.error), err: except (urllib2.URLError, httplib.HTTPException, socket.error), err:
self.to_stderr('ERROR: unable to confirm age: %s' % str(err)) self.to_stderr(u'ERROR: unable to confirm age: %s' % str(err))
return return
def _real_extract(self, url): def _real_extract(self, url):
# Extract video id from URL # Extract video id from URL
mobj = re.match(self._VALID_URL, url) mobj = re.match(self._VALID_URL, url)
if mobj is None: if mobj is None:
self.to_stderr('ERROR: invalid URL: %s' % url) self.to_stderr(u'ERROR: invalid URL: %s' % url)
return [None] return [None]
video_id = mobj.group(2) video_id = mobj.group(2)
@ -514,14 +517,14 @@ class YoutubeIE(InfoExtractor):
self.report_webpage_download(video_id) self.report_webpage_download(video_id)
video_webpage = urllib2.urlopen(request).read() video_webpage = urllib2.urlopen(request).read()
except (urllib2.URLError, httplib.HTTPException, socket.error), err: except (urllib2.URLError, httplib.HTTPException, socket.error), err:
self.to_stderr('ERROR: unable to download video webpage: %s' % str(err)) self.to_stderr(u'ERROR: unable to download video webpage: %s' % str(err))
return [None] return [None]
self.report_information_extraction(video_id) self.report_information_extraction(video_id)
# "t" param # "t" param
mobj = re.search(r', "t": "([^"]+)"', video_webpage) mobj = re.search(r', "t": "([^"]+)"', video_webpage)
if mobj is None: if mobj is None:
self.to_stderr('ERROR: unable to extract "t" parameter') self.to_stderr(u'ERROR: unable to extract "t" parameter')
return [None] return [None]
video_real_url = 'http://www.youtube.com/get_video?video_id=%s&t=%s' % (video_id, mobj.group(1)) video_real_url = 'http://www.youtube.com/get_video?video_id=%s&t=%s' % (video_id, mobj.group(1))
if format_param is not None: if format_param is not None:
@ -531,14 +534,14 @@ class YoutubeIE(InfoExtractor):
# uploader # uploader
mobj = re.search(r'More From: ([^<]*)<', video_webpage) mobj = re.search(r'More From: ([^<]*)<', video_webpage)
if mobj is None: if mobj is None:
self.to_stderr('ERROR: unable to extract uploader nickname') self.to_stderr(u'ERROR: unable to extract uploader nickname')
return [None] return [None]
video_uploader = mobj.group(1) video_uploader = mobj.group(1)
# title # title
mobj = re.search(r'(?im)<title>YouTube - ([^<]*)</title>', video_webpage) mobj = re.search(r'(?im)<title>YouTube - ([^<]*)</title>', video_webpage)
if mobj is None: if mobj is None:
self.to_stderr('ERROR: unable to extract video title') self.to_stderr(u'ERROR: unable to extract video title')
return [None] return [None]
video_title = mobj.group(1).decode('utf-8') video_title = mobj.group(1).decode('utf-8')
video_title = re.sub(ur'(?u)&(.+?);', lambda x: unichr(htmlentitydefs.name2codepoint[x.group(1)]), video_title) video_title = re.sub(ur'(?u)&(.+?);', lambda x: unichr(htmlentitydefs.name2codepoint[x.group(1)]), video_title)
@ -550,12 +553,12 @@ class YoutubeIE(InfoExtractor):
# Return information # Return information
return [{ return [{
'id': video_id, 'id': video_id.decode('utf-8'),
'url': video_real_url, 'url': video_real_url.decode('utf-8'),
'uploader': video_uploader, 'uploader': video_uploader.decode('utf-8'),
'title': video_title, 'title': video_title,
'stitle': simple_title, 'stitle': simple_title,
'ext': video_extension, 'ext': video_extension.decode('utf-8'),
}] }]
class MetacafeIE(InfoExtractor): class MetacafeIE(InfoExtractor):
@ -575,19 +578,19 @@ class MetacafeIE(InfoExtractor):
def report_disclaimer(self): def report_disclaimer(self):
"""Report disclaimer retrieval.""" """Report disclaimer retrieval."""
self.to_stdout('[metacafe] Retrieving disclaimer') self.to_stdout(u'[metacafe] Retrieving disclaimer')
def report_age_confirmation(self): def report_age_confirmation(self):
"""Report attempt to confirm age.""" """Report attempt to confirm age."""
self.to_stdout('[metacafe] Confirming age') self.to_stdout(u'[metacafe] Confirming age')
def report_download_webpage(self, video_id): def report_download_webpage(self, video_id):
"""Report webpage download.""" """Report webpage download."""
self.to_stdout('[metacafe] %s: Downloading webpage' % video_id) self.to_stdout(u'[metacafe] %s: Downloading webpage' % video_id)
def report_extraction(self, video_id): def report_extraction(self, video_id):
"""Report information extraction.""" """Report information extraction."""
self.to_stdout('[metacafe] %s: Extracting information' % video_id) self.to_stdout(u'[metacafe] %s: Extracting information' % video_id)
def _real_initialize(self): def _real_initialize(self):
# Retrieve disclaimer # Retrieve disclaimer
@ -596,7 +599,7 @@ class MetacafeIE(InfoExtractor):
self.report_disclaimer() self.report_disclaimer()
disclaimer = urllib2.urlopen(request).read() disclaimer = urllib2.urlopen(request).read()
except (urllib2.URLError, httplib.HTTPException, socket.error), err: except (urllib2.URLError, httplib.HTTPException, socket.error), err:
self.to_stderr('ERROR: unable to retrieve disclaimer: %s' % str(err)) self.to_stderr(u'ERROR: unable to retrieve disclaimer: %s' % str(err))
return return
# Confirm age # Confirm age
@ -609,14 +612,14 @@ class MetacafeIE(InfoExtractor):
self.report_age_confirmation() self.report_age_confirmation()
disclaimer = urllib2.urlopen(request).read() disclaimer = urllib2.urlopen(request).read()
except (urllib2.URLError, httplib.HTTPException, socket.error), err: except (urllib2.URLError, httplib.HTTPException, socket.error), err:
self.to_stderr('ERROR: unable to confirm age: %s' % str(err)) self.to_stderr(u'ERROR: unable to confirm age: %s' % str(err))
return return
def _real_extract(self, url): def _real_extract(self, url):
# Extract id and simplified title from URL # Extract id and simplified title from URL
mobj = re.match(self._VALID_URL, url) mobj = re.match(self._VALID_URL, url)
if mobj is None: if mobj is None:
self.to_stderr('ERROR: invalid URL: %s' % url) self.to_stderr(u'ERROR: invalid URL: %s' % url)
return [None] return [None]
video_id = mobj.group(1) video_id = mobj.group(1)
@ -635,20 +638,20 @@ class MetacafeIE(InfoExtractor):
self.report_download_webpage(video_id) self.report_download_webpage(video_id)
webpage = urllib2.urlopen(request).read() webpage = urllib2.urlopen(request).read()
except (urllib2.URLError, httplib.HTTPException, socket.error), err: except (urllib2.URLError, httplib.HTTPException, socket.error), err:
self.to_stderr('ERROR: unable retrieve video webpage: %s' % str(err)) self.to_stderr(u'ERROR: unable retrieve video webpage: %s' % str(err))
return [None] return [None]
# Extract URL, uploader and title from webpage # Extract URL, uploader and title from webpage
self.report_extraction(video_id) self.report_extraction(video_id)
mobj = re.search(r'(?m)"mediaURL":"(http.*?\.flv)"', webpage) mobj = re.search(r'(?m)"mediaURL":"(http.*?\.flv)"', webpage)
if mobj is None: if mobj is None:
self.to_stderr('ERROR: unable to extract media URL') self.to_stderr(u'ERROR: unable to extract media URL')
return [None] return [None]
mediaURL = mobj.group(1).replace('\\', '') mediaURL = mobj.group(1).replace('\\', '')
mobj = re.search(r'(?m)"gdaKey":"(.*?)"', webpage) mobj = re.search(r'(?m)"gdaKey":"(.*?)"', webpage)
if mobj is None: if mobj is None:
self.to_stderr('ERROR: unable to extract gdaKey') self.to_stderr(u'ERROR: unable to extract gdaKey')
return [None] return [None]
gdaKey = mobj.group(1) gdaKey = mobj.group(1)
@ -656,24 +659,24 @@ class MetacafeIE(InfoExtractor):
mobj = re.search(r'(?im)<meta name="title" content="Metacafe - ([^"]+)"', webpage) mobj = re.search(r'(?im)<meta name="title" content="Metacafe - ([^"]+)"', webpage)
if mobj is None: if mobj is None:
self.to_stderr('ERROR: unable to extract title') self.to_stderr(u'ERROR: unable to extract title')
return [None] return [None]
video_title = mobj.group(1).decode('utf-8') video_title = mobj.group(1).decode('utf-8')
mobj = re.search(r'(?m)<li id="ChnlUsr">.*?Submitter:<br />(.*?)</li>', webpage) mobj = re.search(r'(?m)<li id="ChnlUsr">.*?Submitter:<br />(.*?)</li>', webpage)
if mobj is None: if mobj is None:
self.to_stderr('ERROR: unable to extract uploader nickname') self.to_stderr(u'ERROR: unable to extract uploader nickname')
return [None] return [None]
video_uploader = re.sub(r'<.*?>', '', mobj.group(1)) video_uploader = re.sub(r'<.*?>', '', mobj.group(1))
# Return information # Return information
return [{ return [{
'id': video_id, 'id': video_id.decode('utf-8'),
'url': video_url, 'url': video_url.decode('utf-8'),
'uploader': video_uploader, 'uploader': video_uploader.decode('utf-8'),
'title': video_title, 'title': video_title,
'stitle': simple_title, 'stitle': simple_title,
'ext': video_extension, 'ext': video_extension.decode('utf-8'),
}] }]
class YoutubePlaylistIE(InfoExtractor): class YoutubePlaylistIE(InfoExtractor):
@ -695,7 +698,7 @@ class YoutubePlaylistIE(InfoExtractor):
def report_download_page(self, playlist_id, pagenum): def report_download_page(self, playlist_id, pagenum):
"""Report attempt to download playlist page with given number.""" """Report attempt to download playlist page with given number."""
self.to_stdout('[youtube] PL %s: Downloading page #%s' % (playlist_id, pagenum)) self.to_stdout(u'[youtube] PL %s: Downloading page #%s' % (playlist_id, pagenum))
def _real_initialize(self): def _real_initialize(self):
self._youtube_ie.initialize() self._youtube_ie.initialize()
@ -704,7 +707,7 @@ class YoutubePlaylistIE(InfoExtractor):
# Extract playlist id # Extract playlist id
mobj = re.match(self._VALID_URL, url) mobj = re.match(self._VALID_URL, url)
if mobj is None: if mobj is None:
self.to_stderr('ERROR: invalid url: %s' % url) self.to_stderr(u'ERROR: invalid url: %s' % url)
return [None] return [None]
# Download playlist pages # Download playlist pages
@ -718,7 +721,7 @@ class YoutubePlaylistIE(InfoExtractor):
try: try:
page = urllib2.urlopen(request).read() page = urllib2.urlopen(request).read()
except (urllib2.URLError, httplib.HTTPException, socket.error), err: except (urllib2.URLError, httplib.HTTPException, socket.error), err:
self.to_stderr('ERROR: unable to download webpage: %s' % str(err)) self.to_stderr(u'ERROR: unable to download webpage: %s' % str(err))
return [None] return [None]
# Extract video identifiers # Extract video identifiers
@ -791,21 +794,21 @@ if __name__ == '__main__':
# Conflicting, missing and erroneous options # Conflicting, missing and erroneous options
if len(args) < 1: if len(args) < 1:
sys.exit('ERROR: you must provide at least one URL') sys.exit(u'ERROR: you must provide at least one URL')
if opts.usenetrc and (opts.username is not None or opts.password is not None): if opts.usenetrc and (opts.username is not None or opts.password is not None):
sys.exit('ERROR: using .netrc conflicts with giving username/password') sys.exit(u'ERROR: using .netrc conflicts with giving username/password')
if opts.password is not None and opts.username is None: if opts.password is not None and opts.username is None:
sys.exit('ERROR: account username missing') sys.exit(u'ERROR: account username missing')
if opts.outtmpl is not None and (opts.useliteral or opts.usetitle): if opts.outtmpl is not None and (opts.useliteral or opts.usetitle):
sys.exit('ERROR: using output template conflicts with using title or literal title') sys.exit(u'ERROR: using output template conflicts with using title or literal title')
if opts.usetitle and opts.useliteral: if opts.usetitle and opts.useliteral:
sys.exit('ERROR: using title conflicts with using literal title') sys.exit(u'ERROR: using title conflicts with using literal title')
if opts.username is not None and opts.password is None: if opts.username is not None and opts.password is None:
opts.password = getpass.getpass('Type account password and press return:') opts.password = getpass.getpass(u'Type account password and press return:')
if opts.ratelimit is not None: if opts.ratelimit is not None:
numeric_limit = FileDownloader.parse_bytes(opts.ratelimit) numeric_limit = FileDownloader.parse_bytes(opts.ratelimit)
if numeric_limit is None: if numeric_limit is None:
sys.exit('ERROR: invalid rate limit specified') sys.exit(u'ERROR: invalid rate limit specified')
opts.ratelimit = numeric_limit opts.ratelimit = numeric_limit
# Information extractors # Information extractors
@ -823,10 +826,10 @@ if __name__ == '__main__':
'forcetitle': opts.gettitle, 'forcetitle': opts.gettitle,
'simulate': (opts.simulate or opts.geturl or opts.gettitle), 'simulate': (opts.simulate or opts.geturl or opts.gettitle),
'format': opts.format, 'format': opts.format,
'outtmpl': ((opts.outtmpl is not None and opts.outtmpl) 'outtmpl': ((opts.outtmpl is not None and opts.outtmpl.decode())
or (opts.usetitle and '%(stitle)s-%(id)s.%(ext)s') or (opts.usetitle and u'%(stitle)s-%(id)s.%(ext)s')
or (opts.useliteral and '%(title)s-%(id)s.%(ext)s') or (opts.useliteral and u'%(title)s-%(id)s.%(ext)s')
or '%(id)s.%(ext)s'), or u'%(id)s.%(ext)s'),
'ignoreerrors': opts.ignoreerrors, 'ignoreerrors': opts.ignoreerrors,
'ratelimit': opts.ratelimit, 'ratelimit': opts.ratelimit,
}) })
@ -839,6 +842,6 @@ if __name__ == '__main__':
except DownloadError: except DownloadError:
sys.exit(1) sys.exit(1)
except SameFileError: except SameFileError:
sys.exit('ERROR: fixed output name but more than one file to download') sys.exit(u'ERROR: fixed output name but more than one file to download')
except KeyboardInterrupt: except KeyboardInterrupt:
sys.exit('\nERROR: Interrupted by user') sys.exit(u'\nERROR: Interrupted by user')