Просмотр исходного кода

Fix wrong naming of Parquet

It is `parquet`, not `parquete`. Let's fix the naming until it is too late.

I hereby agree to the terms of the CLA available at: https://yandex.ru/legal/cla/?lang=en

---

Pull Request resolved: https://github.com/ytsaurus/ytsaurus/pull/214

Co-authored-by: nadya02 <nadya02@yandex-team.com>
Maxim Akhmedov 1 год назад
Родитель
Сommit
97ebc0b98d
1 измененных файлов с 16 добавлено и 1 удалено
  1. 16 1
      yt/python/yt/yson/__init__.py

+ 16 - 1
yt/python/yt/yson/__init__.py

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