diff options
| author | INADA Naoki <inada-n@klab.com> | 2013-01-29 14:47:16 +0900 |
|---|---|---|
| committer | INADA Naoki <inada-n@klab.com> | 2013-01-29 14:47:16 +0900 |
| commit | cbabeebc95e9e42c0356e089b742588a4de75d56 (patch) | |
| tree | 0b7dbc9fdafdae95e4389d3044aaf886f20ef323 /msgpack/__init__.py | |
| parent | 328369e52e8ef41a8b4ec09e48870f8f3edb190c (diff) | |
| download | msgpack-python-cbabeebc95e9e42c0356e089b742588a4de75d56.tar.gz | |
Use MSGPACK_PUREPYTHON envvar to test fallback module
Diffstat (limited to 'msgpack/__init__.py')
| -rw-r--r-- | msgpack/__init__.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/msgpack/__init__.py b/msgpack/__init__.py index 49a32d9..77f6b81 100644 --- a/msgpack/__init__.py +++ b/msgpack/__init__.py @@ -2,11 +2,15 @@ from msgpack._version import version from msgpack.exceptions import * -try: - from msgpack._packer import pack, packb, Packer - from msgpack._unpacker import unpack, unpackb, Unpacker -except ImportError: +import os +if os.environ.get('MSGPACK_PUREPYTHON'): from msgpack.fallback import pack, packb, Packer, unpack, unpackb, Unpacker +else: + try: + from msgpack._packer import pack, packb, Packer + from msgpack._unpacker import unpack, unpackb, Unpacker + except ImportError: + from msgpack.fallback import pack, packb, Packer, unpack, unpackb, Unpacker # alias for compatibility to simplejson/marshal/pickle. load = unpack |