|
13 | 13 | # limitations under the License.
|
14 | 14 | from bson.codec_options import DEFAULT_CODEC_OPTIONS
|
15 | 15 | from pyarrow import Table, timestamp
|
16 |
| -from pymongoarrow.lib import ( |
17 |
| - BinaryBuilder, |
18 |
| - BoolBuilder, |
19 |
| - CodeBuilder, |
20 |
| - DatetimeBuilder, |
21 |
| - Decimal128Builder, |
22 |
| - DocumentBuilder, |
23 |
| - DoubleBuilder, |
24 |
| - Int32Builder, |
25 |
| - Int64Builder, |
26 |
| - ListBuilder, |
27 |
| - ObjectIdBuilder, |
28 |
| - StringBuilder, |
29 |
| -) |
30 | 16 | from pymongoarrow.types import _BsonArrowTypes, _get_internal_typemap
|
31 | 17 |
|
32 |
| -_TYPE_TO_BUILDER_CLS = { |
33 |
| - _BsonArrowTypes.int32: Int32Builder, |
34 |
| - _BsonArrowTypes.int64: Int64Builder, |
35 |
| - _BsonArrowTypes.double: DoubleBuilder, |
36 |
| - _BsonArrowTypes.datetime: DatetimeBuilder, |
37 |
| - _BsonArrowTypes.objectid: ObjectIdBuilder, |
38 |
| - _BsonArrowTypes.decimal128: Decimal128Builder, |
39 |
| - _BsonArrowTypes.string: StringBuilder, |
40 |
| - _BsonArrowTypes.bool: BoolBuilder, |
41 |
| - _BsonArrowTypes.document: DocumentBuilder, |
42 |
| - _BsonArrowTypes.array: ListBuilder, |
43 |
| - _BsonArrowTypes.binary: BinaryBuilder, |
44 |
| - _BsonArrowTypes.code: CodeBuilder, |
45 |
| -} |
| 18 | +try: |
| 19 | + from pymongoarrow.lib import ( |
| 20 | + BinaryBuilder, |
| 21 | + BoolBuilder, |
| 22 | + CodeBuilder, |
| 23 | + DatetimeBuilder, |
| 24 | + Decimal128Builder, |
| 25 | + DocumentBuilder, |
| 26 | + DoubleBuilder, |
| 27 | + Int32Builder, |
| 28 | + Int64Builder, |
| 29 | + ListBuilder, |
| 30 | + ObjectIdBuilder, |
| 31 | + StringBuilder, |
| 32 | + ) |
| 33 | + |
| 34 | + _TYPE_TO_BUILDER_CLS = { |
| 35 | + _BsonArrowTypes.int32: Int32Builder, |
| 36 | + _BsonArrowTypes.int64: Int64Builder, |
| 37 | + _BsonArrowTypes.double: DoubleBuilder, |
| 38 | + _BsonArrowTypes.datetime: DatetimeBuilder, |
| 39 | + _BsonArrowTypes.objectid: ObjectIdBuilder, |
| 40 | + _BsonArrowTypes.decimal128: Decimal128Builder, |
| 41 | + _BsonArrowTypes.string: StringBuilder, |
| 42 | + _BsonArrowTypes.bool: BoolBuilder, |
| 43 | + _BsonArrowTypes.document: DocumentBuilder, |
| 44 | + _BsonArrowTypes.array: ListBuilder, |
| 45 | + _BsonArrowTypes.binary: BinaryBuilder, |
| 46 | + _BsonArrowTypes.code: CodeBuilder, |
| 47 | + } |
| 48 | +except ImportError: |
| 49 | + import traceback |
| 50 | + import warnings |
| 51 | + |
| 52 | + warnings.warn( |
| 53 | + "Could not find compiled pymongoarrow.lib extension, please install " |
| 54 | + "from source or report the following traceback on the issue tracker:\n" |
| 55 | + f"{traceback.format_exc()}" |
| 56 | + ) |
46 | 57 |
|
47 | 58 |
|
48 | 59 | class PyMongoArrowContext:
|
|
0 commit comments