Issue #15207: Fix mimetypes to read from correct area in Windows registry (Original patch by Dave Chambers)

This commit is contained in:
Tim Golden 2013-10-22 19:27:34 +01:00
parent 49e61806f5
commit 27a856495e
5 changed files with 21 additions and 11 deletions

View File

@ -85,6 +85,9 @@ behavior of the module.
:const:`knownfiles` takes precedence over those named before it. Calling
:func:`init` repeatedly is allowed.
Specifying an empty list for *files* will prevent the system defaults from
being applied: only the well-known values will be present from a built-in list.
.. versionchanged:: 3.2
Previously, Windows registry settings were ignored.

View File

@ -249,19 +249,21 @@ class MimeTypes:
yield ctype
i += 1
with _winreg.OpenKey(_winreg.HKEY_CLASSES_ROOT,
r'MIME\Database\Content Type') as mimedb:
for ctype in enum_types(mimedb):
with _winreg.OpenKey(_winreg.HKEY_CLASSES_ROOT, '') as hkcr:
for subkeyname in enum_types(hkcr):
try:
with _winreg.OpenKey(mimedb, ctype) as key:
suffix, datatype = _winreg.QueryValueEx(key,
'Extension')
with _winreg.OpenKey(hkcr, subkeyname) as subkey:
# Only check file extensions
if not subkeyname.startswith("."):
continue
# raises EnvironmentError if no 'Content Type' value
mimetype, datatype = _winreg.QueryValueEx(
subkey, 'Content Type')
if datatype != _winreg.REG_SZ:
continue
self.add_type(mimetype, subkeyname, strict)
except EnvironmentError:
continue
if datatype != _winreg.REG_SZ:
continue
self.add_type(ctype, suffix, strict)
def guess_type(url, strict=True):
"""Guess the type of a file based on its URL.

View File

@ -98,7 +98,8 @@ class Win32MimeTypesTestCase(unittest.TestCase):
# Use file types that should *always* exist:
eq = self.assertEqual
eq(self.db.guess_type("foo.txt"), ("text/plain", None))
eq(self.db.guess_type("image.jpg"), ("image/jpeg", None))
eq(self.db.guess_type("image.png"), ("image/png", None))
def test_main():
support.run_unittest(MimeTypesTestCase,

View File

@ -201,6 +201,7 @@ Per Cederqvist
Matej Cepl
Carl Cerecke
Octavian Cerna
Dave Chambers
Pascal Chambon
John Chandler
Hye-Shik Chang

View File

@ -81,6 +81,9 @@ Core and Builtins
Library
-------
- Issue #15207: Fix mimetypes to read from correct part of Windows registry
Original patch by Dave Chambers
- Issue #8964: fix platform._sys_version to handle IronPython 2.6+.
Patch by Martin Matusiak.