|
@@ -38,8 +38,10 @@ from . import parser # noqa
|
|
|
from . import yson_types # noqa
|
|
|
|
|
|
TYPE = None
|
|
|
+HAS_PARQUET = False
|
|
|
+
|
|
|
try:
|
|
|
- from yt_yson_bindings import load, loads, dump, dumps, dump_parquete # noqa
|
|
|
+ from yt_yson_bindings import load, loads, dump, dumps # noqa
|
|
|
TYPE = "BINARY"
|
|
|
except ImportError as error:
|
|
|
# XXX(asaitgalin): Sometimes module can't be imported because
|
|
@@ -50,6 +52,19 @@ except ImportError as error:
|
|
|
import sys as _sys
|
|
|
print("Warning! Failed to import YSON bindings: " + message, file=_sys.stderr)
|
|
|
|
|
|
+try:
|
|
|
+ from yt_yson_bindings import dump_parquet # noqa
|
|
|
+ HAS_PARQUET = True
|
|
|
+except ImportError:
|
|
|
+ try:
|
|
|
+ from yt_yson_bindings import dump_parquete as dump_parquet # noqa
|
|
|
+ HAS_PARQUET = True
|
|
|
+ except ImportError as error:
|
|
|
+ message = str(error)
|
|
|
+ if "No module named" not in message:
|
|
|
+ import sys as _sys
|
|
|
+ print("Warning! Failed to import dump_parquet binding: " + message, file=_sys.stderr)
|
|
|
+
|
|
|
if TYPE is None:
|
|
|
from .parser import load, loads # noqa
|
|
|
from .writer import dump, dumps # noqa
|