Use MSGPACK_PUREPYTHON envvar to test fallback module

This commit is contained in:
INADA Naoki 2013-01-29 14:47:16 +09:00
parent 328369e52e
commit cbabeebc95
2 changed files with 9 additions and 5 deletions

View file

@ -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