diff options
Diffstat (limited to 'catapult/common/py_utils/py_utils/camel_case.py')
-rw-r--r-- | catapult/common/py_utils/py_utils/camel_case.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/catapult/common/py_utils/py_utils/camel_case.py b/catapult/common/py_utils/py_utils/camel_case.py index 9a768902..dbebb227 100644 --- a/catapult/common/py_utils/py_utils/camel_case.py +++ b/catapult/common/py_utils/py_utils/camel_case.py @@ -2,7 +2,11 @@ # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. +from __future__ import absolute_import +from __future__ import division +from __future__ import print_function import re +import six def ToUnderscore(obj): @@ -11,7 +15,7 @@ def ToUnderscore(obj): Descends recursively into lists and dicts, converting all dict keys. Returns a newly allocated object of the same structure as the input. """ - if isinstance(obj, basestring): + if isinstance(obj, six.string_types): return re.sub('(?!^)([A-Z]+)', r'_\1', obj).lower() elif isinstance(obj, list): @@ -19,7 +23,7 @@ def ToUnderscore(obj): elif isinstance(obj, dict): output = {} - for k, v in obj.iteritems(): + for k, v in six.iteritems(obj): if isinstance(v, list) or isinstance(v, dict): output[ToUnderscore(k)] = ToUnderscore(v) else: |