|
9 | 9 | try: |
10 | 10 | import ibis.backends.base_sqlalchemy.compiler as comp |
11 | 11 | except ImportError: |
12 | | - import ibis.sql.compiler as comp |
| 12 | + try: |
| 13 | + import ibis.sql.compiler as comp |
| 14 | + except ImportError: |
| 15 | + import ibis.backends.base.sql.compiler as comp |
13 | 16 | try: |
14 | 17 | import ibis.common.exceptions as com |
15 | 18 | except ImportError: |
|
24 | 27 | import toolz |
25 | 28 |
|
26 | 29 | try: |
27 | | - # 2.x |
28 | | - from ibis.backends.base.sql import (fixed_arity, literal, |
29 | | - operation_registry, reduction, unary) |
| 30 | + from ibis.backends.base.sql.alchemy.registry import _literal as literal |
| 31 | + from ibis.backends.base.sql.alchemy.registry import fixed_arity, reduction |
| 32 | + from ibis.backends.base.sql.alchemy.registry import \ |
| 33 | + sqlalchemy_operation_registry as operation_registry |
| 34 | + from ibis.backends.base.sql.alchemy.registry import unary |
30 | 35 | except ImportError: |
31 | 36 | try: |
32 | | - # 1.4 |
33 | | - from ibis.backends.base_sql import (fixed_arity, literal, |
| 37 | + # 2.x |
| 38 | + from ibis.backends.base.sql import (fixed_arity, literal, |
34 | 39 | operation_registry, reduction, |
35 | 40 | unary) |
36 | 41 | except ImportError: |
37 | | - # 1.2 |
38 | | - from ibis.impala.compiler import _literal as literal |
39 | | - from ibis.impala.compiler import \ |
40 | | - _operation_registry as operation_registry |
41 | | - from ibis.impala.compiler import _reduction as reduction |
42 | | - from ibis.impala.compiler import fixed_arity, unary |
| 42 | + try: |
| 43 | + # 1.4 |
| 44 | + from ibis.backends.base_sql import (fixed_arity, literal, |
| 45 | + operation_registry, reduction, |
| 46 | + unary) |
| 47 | + except ImportError: |
| 48 | + # 1.2 |
| 49 | + from ibis.impala.compiler import _literal as literal |
| 50 | + from ibis.impala.compiler import \ |
| 51 | + _operation_registry as operation_registry |
| 52 | + from ibis.impala.compiler import _reduction as reduction |
| 53 | + from ibis.impala.compiler import fixed_arity, unary |
43 | 54 |
|
44 | 55 | try: |
45 | 56 | from ibis.backends.base_sql.compiler import (BaseExprTranslator, |
@@ -517,7 +528,18 @@ def _trans_param(self, expr): |
517 | 528 | return '@{}'.format(expr.get_name()) |
518 | 529 |
|
519 | 530 |
|
520 | | -compiles = BigQueryExprTranslator.compiles |
| 531 | +try: |
| 532 | + compiles = BigQueryExprTranslator.compiles |
| 533 | +except AttributeError: |
| 534 | + # https://github.com/ibis-project/ibis/commit/3d5a10 |
| 535 | + def _add_operation(operation): |
| 536 | + def decorator(translation_func): |
| 537 | + BigQueryExprTranslator.add_operation( |
| 538 | + operation, translation_func |
| 539 | + ) |
| 540 | + return decorator |
| 541 | + compiles = _add_operation |
| 542 | + |
521 | 543 | rewrites = BigQueryExprTranslator.rewrites |
522 | 544 |
|
523 | 545 |
|
|
0 commit comments