diff options
author | Raphaël Barrois <raphael.barrois@polytechnique.org> | 2014-01-22 00:03:10 +0100 |
---|---|---|
committer | Raphaël Barrois <raphael.barrois@polytechnique.org> | 2014-01-22 00:24:29 +0100 |
commit | 6f5169ab871e7bedb098c9ebcafcb46decad59ee (patch) | |
tree | 13ecdc0b890bb4a88f0fbd212942bc78d799f45e /factory | |
parent | 3feb4f51a58e39a7aaf82222bff27ba181920b2e (diff) | |
download | factory-boy-6f5169ab871e7bedb098c9ebcafcb46decad59ee.tar factory-boy-6f5169ab871e7bedb098c9ebcafcb46decad59ee.tar.gz |
Fix log_pprint (Closes #123).
Diffstat (limited to 'factory')
-rw-r--r-- | factory/utils.py | 21 |
1 files changed, 19 insertions, 2 deletions
diff --git a/factory/utils.py b/factory/utils.py index 48c6eed..b27fd77 100644 --- a/factory/utils.py +++ b/factory/utils.py @@ -20,6 +20,8 @@ # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN # THE SOFTWARE. +from __future__ import unicode_literals + import collections #: String for splitting an attribute name into a @@ -96,11 +98,26 @@ def import_object(module_name, attribute_name): return getattr(module, attribute_name) +def _safe_repr(obj): + try: + obj_repr = repr(obj) + except UnicodeError: + return '<bad_repr object at %s>' % id(obj) + + try: # Convert to "text type" (= unicode) + return '%s' % obj_repr + except UnicodeError: # non-ascii bytes repr on Py2 + return obj_repr.decode('utf-8') + + def log_pprint(args=(), kwargs=None): kwargs = kwargs or {} return ', '.join( - [str(arg) for arg in args] + - ['%s=%r' % item for item in kwargs.items()] + [repr(arg) for arg in args] + + [ + '%s=%s' % (key, _safe_repr(value)) + for key, value in kwargs.items() + ] ) |