youtube-dl

commit 03f9daab34605f538294fdffb141ef5d9fc670e6
parent a8156c1d2e4b2a7ac5e034c247c6fccaca15a21d
Author: Philipp Hagemeister <phihag@phihag.de>
Date:   Wed, 28 Nov 2012 00:09:17 +0100

Use io.BytesIO instead of StringIO

Diffstat:
M
youtube_dl/InfoExtractors.py
 | 
6
------
M
youtube_dl/utils.py
 | 
10
+++-------

2 files changed, 3 insertions(+), 13 deletions(-)
diff --git a/youtube_dl/InfoExtractors.py b/youtube_dl/InfoExtractors.py
@@ -2,7 +2,6 @@
 # -*- coding: utf-8 -*-
 
 import datetime
-import HTMLParser
 import httplib
 import netrc
 import os
@@ -15,11 +14,6 @@ import random
 import math
 from urlparse import parse_qs
 
-try:
-	import cStringIO as StringIO
-except ImportError:
-	import StringIO
-
 from utils import *
 
 
diff --git a/youtube_dl/utils.py b/youtube_dl/utils.py
@@ -2,6 +2,7 @@
 # -*- coding: utf-8 -*-
 
 import gzip
+import io
 import locale
 import os
 import re
@@ -11,11 +12,6 @@ import email.utils
 import json
 
 try:
-	import cStringIO as StringIO
-except ImportError:
-	import StringIO
-
-try:
 	import urllib.request as compat_urllib_request
 except ImportError: # Python 2
 	import urllib2 as compat_urllib_request
@@ -400,12 +396,12 @@ class YoutubeDLHandler(compat_urllib_request.HTTPHandler):
 		old_resp = resp
 		# gzip
 		if resp.headers.get('Content-encoding', '') == 'gzip':
-			gz = gzip.GzipFile(fileobj=StringIO.StringIO(resp.read()), mode='r')
+			gz = gzip.GzipFile(fileobj=io.BytesIO(resp.read()), mode='r')
 			resp = self.addinfourl_wrapper(gz, old_resp.headers, old_resp.url, old_resp.code)
 			resp.msg = old_resp.msg
 		# deflate
 		if resp.headers.get('Content-encoding', '') == 'deflate':
-			gz = StringIO.StringIO(self.deflate(resp.read()))
+			gz = io.BytesIO(self.deflate(resp.read()))
 			resp = self.addinfourl_wrapper(gz, old_resp.headers, old_resp.url, old_resp.code)
 			resp.msg = old_resp.msg
 		return resp