removed the conditional

pytube/request.py

@@ -22,7 +22,8 @@ def get(
     if streaming:
         return stream_response(response, chunk_size)
     elif headers:
-        return dict(response.info().items())
+        # https://github.com/nficano/pytube/issues/160
+        return {k.lower(): v for k, v in response.info().items()}
     return (
         response
         .read()

pytube/streams.py

@@ -135,10 +135,7 @@ class Stream(object):
     def filesize(self):
         """File size of the media stream in bytes."""
         headers = request.get(self.url, headers=True)
-        # https://github.com/nficano/pytube/issues/160
-        if 'content-length' in headers:
-            return int(headers['content-length'])
-        return int(headers['Content-Length'])
+        return int(headers['content-length'])
 
     @property
     def default_filename(self):

tests/test_request.py

@@ -27,10 +27,10 @@ def test_get_streaming(mock_urlopen):
 @mock.patch('pytube.request.urlopen')
 def test_get_headers(mock_urlopen):
     response = mock.Mock()
-    response.info.return_value = {'Content-Length': '16384'}
+    response.info.return_value = {'content-length': '16384'}
     mock_urlopen.return_value = response
     response = request.get('fakeassurl.gov', headers=True)
-    assert response == {'Content-Length': '16384'}
+    assert response == {'content-length': '16384'}
 
 
 @mock.patch('pytube.request.urlopen')

tests/test_streams.py

@@ -9,7 +9,7 @@ from pytube import Stream
 
 def test_filesize(cipher_signature, mocker):
     mocker.patch.object(request, 'get')
-    request.get.return_value = {'Content-Length': '6796391'}
+    request.get.return_value = {'content-length': '6796391'}
     assert cipher_signature.streams.first().filesize == 6796391
 
 
@@ -22,8 +22,8 @@ def test_default_filename(cipher_signature):
 def test_download(cipher_signature, mocker):
     mocker.patch.object(request, 'get')
     request.get.side_effect = [
-        {'Content-Length': '16384'},
-        {'Content-Length': '16384'},
+        {'content-length': '16384'},
+        {'content-length': '16384'},
         iter([str(random.getrandbits(8 * 1024))]),
     ]
     with mock.patch('pytube.streams.open', mock.mock_open(), create=True):
@@ -47,8 +47,8 @@ def test_on_progress_hook(cipher_signature, mocker):
 
     mocker.patch.object(request, 'get')
     request.get.side_effect = [
-        {'Content-Length': '16384'},
-        {'Content-Length': '16384'},
+        {'content-length': '16384'},
+        {'content-length': '16384'},
         iter([str(random.getrandbits(8 * 1024))]),
     ]
     with mock.patch('pytube.streams.open', mock.mock_open(), create=True):
@@ -67,8 +67,8 @@ def test_on_complete_hook(cipher_signature, mocker):
 
     mocker.patch.object(request, 'get')
     request.get.side_effect = [
-        {'Content-Length': '16384'},
-        {'Content-Length': '16384'},
+        {'content-length': '16384'},
+        {'content-length': '16384'},
         iter([str(random.getrandbits(8 * 1024))]),
     ]
     with mock.patch('pytube.streams.open', mock.mock_open(), create=True):