diff options
| -rw-r--r-- | mpm/python/usrp_mpm/mpmutils.py | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/mpm/python/usrp_mpm/mpmutils.py b/mpm/python/usrp_mpm/mpmutils.py index ad4457419..64474d3ff 100644 --- a/mpm/python/usrp_mpm/mpmutils.py +++ b/mpm/python/usrp_mpm/mpmutils.py @@ -8,6 +8,7 @@ Miscellaneous utilities for MPM """ import time +import sys def poll_with_timeout(state_check, timeout_ms, interval_ms): """ @@ -36,3 +37,44 @@ def poll_with_timeout(state_check, timeout_ms, interval_ms): time.sleep(interval_s) return False +def to_native_str(str_or_bstr): + """ + Returns a native string, regardless of the input string type (binary or + UTF-8), and the Python version (2 or 3). + Note that the native string type is actually not the same in Python 2 and + 3: In the former, it's a binary string, in the latter, it's Unicode. + >>> to_native_str(b'foo') + 'foo' + >>> to_native_str(u'foo') + 'foo' + """ + if isinstance(str_or_bstr, str): + return str_or_bstr + if sys.version_info.major >= 3: + return str(str_or_bstr, encoding='ascii') + else: + return str(str_or_bstr) + +def to_binary_str(str_or_bstr): + """ + Returns a binary string, regardless of the input string type (binary or + UTF-8), and the Python version (2 or 3). + Note that in Python 2, a binary string is the native string type. + """ + try: + return bytes(str_or_bstr.encode('utf-8')) + except AttributeError: + return bytes(str_or_bstr) + + +def to_utf8_str(str_or_bstr): + """ + Returns a unicode string, regardless of the input string type (binary or + UTF-8), and the Python version (2 or 3). + Note that in Python 2, a unicode string is not the native string type. + """ + try: + return str_or_bstr.decode('utf-8') + except AttributeError: + return str_or_bstr + |