Better naming

This commit is contained in:
Kovid Goyal 2019-04-29 21:29:47 +05:30
parent 1744a2b2fd
commit a2fa719545
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
2 changed files with 6 additions and 5 deletions

View File

@ -12,7 +12,7 @@ from itertools import chain
is_ci = os.environ.get('CI', '').lower() == 'true' is_ci = os.environ.get('CI', '').lower() == 'true'
from setup import Command, basenames, __appname__, download_securely, dump_json from setup import Command, basenames, __appname__, download_securely, dump_json
from polyglot.builtins import codepoint_to_chr, itervalues, iteritems, as_unicode_recursive from polyglot.builtins import codepoint_to_chr, itervalues, iteritems, only_unicode_recursive
def get_opts_from_parser(parser): def get_opts_from_parser(parser):
@ -358,7 +358,7 @@ class Resources(Command): # {{{
get_opts_from_parser(p)] get_opts_from_parser(p)]
with open(dest, 'wb') as f: with open(dest, 'wb') as f:
f.write(msgpack_dumps(as_unicode_recursive(complete))) f.write(msgpack_dumps(only_unicode_recursive(complete)))
self.info('\tCreating template-functions.json') self.info('\tCreating template-functions.json')
dest = self.j(self.RESOURCES, 'template-functions.json') dest = self.j(self.RESOURCES, 'template-functions.json')

View File

@ -32,15 +32,16 @@ def as_unicode(x, encoding='utf-8', errors='strict'):
return unicode_type(x) return unicode_type(x)
def as_unicode_recursive(x, encoding='utf-8', errors='strict'): def only_unicode_recursive(x, encoding='utf-8', errors='strict'):
# Convert any bytestrings in lists/tuples/dicts to unicode
if isinstance(x, bytes): if isinstance(x, bytes):
return x.decode(encoding, errors) return x.decode(encoding, errors)
if isinstance(x, unicode_type): if isinstance(x, unicode_type):
return x return x
if isinstance(x, (list, tuple)): if isinstance(x, (list, tuple)):
return type(x)(as_unicode_recursive(i, encoding, errors) for i in x) return type(x)(only_unicode_recursive(i, encoding, errors) for i in x)
if isinstance(x, dict): if isinstance(x, dict):
return {as_unicode_recursive(k, encoding, errors): as_unicode_recursive(v, encoding, errors) for k, v in iteritems(x)} return {only_unicode_recursive(k, encoding, errors): only_unicode_recursive(v, encoding, errors) for k, v in iteritems(x)}
return x return x