INTPYTHON-497 Extend rather than replace TypeRegistry in write function

Author: blink1073

.github/workflows/test-python.yml

@@ -143,3 +143,25 @@ jobs:
         run: just docs
       - name: Run linkcheck
         run: just docs-linkcheck
+  test_minimum:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v5
+        with:
+          persist-credentials: false
+      - name: Setup Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: 3.9
+          cache: 'pip'
+          cache-dependency-path: '**/pyproject.toml'
+      - name: Install Deps
+        run: |
+          python -m pip install uv
+      - name: Run tests
+        shell: bash
+        run: |
+          uv venv
+          source .venv/bin/activate
+          uv pip install -e ".[test]" --resolution=lowest-direct
+          pytest

bindings/python/pymongoarrow/api.py

@@ -49,6 +49,7 @@
     is_uint32,
     is_uint64,
 )
+from pymongo.collection import Collection
 from pymongo.common import MAX_WRITE_BATCH_SIZE
 
 from pymongoarrow.context import PyMongoArrowContext
@@ -488,7 +489,9 @@ def transform_python(self, value):
         return Decimal128(value)
 
 
-def write(collection, tabular, *, exclude_none: bool = False, auto_convert: bool = True):
+def write(
+    collection: Collection, tabular, *, exclude_none: bool = False, auto_convert: bool = True
+):
     """Write data from `tabular` into the given MongoDB `collection`.
 
     :Parameters:
@@ -559,10 +562,13 @@ def write(collection, tabular, *, exclude_none: bool = False, auto_convert: bool
 
     # Add handling for special case types.
     codec_options = collection.codec_options
+    base_codecs = []
+    if hasattr(codec_options.type_registry, "codecs"):
+        base_codecs = codec_options.type_registry.codecs
     if pd is not None:
-        type_registry = TypeRegistry([_PandasNACodec(), _DecimalCodec()])
+        type_registry = TypeRegistry([*base_codecs, _PandasNACodec(), _DecimalCodec()])
     else:
-        type_registry = TypeRegistry([_DecimalCodec()])
+        type_registry = TypeRegistry([*base_codecs, _DecimalCodec()])
     codec_options = codec_options.with_options(type_registry=type_registry)
 
     while cur_offset < tab_size: