Fix for plistlib dropping Data in python 3.9

Fixes #1184 (plistlib.Data is going to be dropped in python 3.9)
This commit is contained in:
Kovid Goyal 2020-07-02 08:42:18 +05:30
parent 4d3d69ccd1
commit 5b2baa7233
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
2 changed files with 21 additions and 11 deletions

View File

@ -363,29 +363,25 @@ class XMLConfig(dict):
self.update(d)
def __getitem__(self, key):
from polyglot.plistlib import Data
from polyglot.plistlib import unwrap_bytes
try:
ans = dict.__getitem__(self, key)
if isinstance(ans, Data):
ans = ans.data
return ans
return unwrap_bytes(ans)
except KeyError:
return self.defaults.get(key, None)
def get(self, key, default=None):
from polyglot.plistlib import Data
from polyglot.plistlib import unwrap_bytes
try:
ans = dict.__getitem__(self, key)
if isinstance(ans, Data):
ans = ans.data
return ans
return unwrap_bytes(ans)
except KeyError:
return self.defaults.get(key, default)
def __setitem__(self, key, val):
from polyglot.plistlib import Data
from polyglot.plistlib import wrap_bytes
if isinstance(val, bytes):
val = Data(val)
val = wrap_bytes(val)
dict.__setitem__(self, key, val)
self.commit()

View File

@ -7,6 +7,20 @@ from __future__ import absolute_import, division, print_function, unicode_litera
from polyglot.builtins import is_py3
if is_py3:
from plistlib import loads, dumps, Data # noqa
from plistlib import loads, dumps # noqa
def unwrap_bytes(x):
return x
def wrap_bytes(x):
return x
else:
from plistlib import readPlistFromString as loads, writePlistToString as dumps, Data # noqa
def unwrap_bytes(x):
if isinstance(x, Data):
x = x.data
return x
def wrap_bytes(x):
return Data(x)