fix: resolve pylint, mypy, cspell errors in PKRange change

- Import PKRange in _routing_map_provider_common.py (fixes all emulator tests)
- Fix namedtuple name mismatch (_PKRangeBase, not PKRange) for mypy
- Use raise-from pattern in PKRange.__getitem__ (pylint W0707)
- Move _locks_lock and _collection_locks init into __init__ (pylint W0201)
- Add 'pkrange' to cspell dictionary

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Author: tvaron3

.vscode/cspell.json

@@ -1719,6 +1719,7 @@
       "filename": "sdk/cosmos/azure-cosmos/**",
       "words": [
         "colls",
+        "pkrange",
         "pkranges",
         "Upserts",
         "sprocs",

sdk/cosmos/azure-cosmos/azure/cosmos/_routing/_routing_map_provider_common.py

@@ -34,6 +34,7 @@
 from .collection_routing_map import CollectionRoutingMap, _build_routing_map_from_ranges
 from . import routing_range
 from .routing_range import (
+    PKRange,
     PartitionKeyRange,
     _is_sorted_and_non_overlapping,
     _subtract_range,

sdk/cosmos/azure-cosmos/azure/cosmos/_routing/aio/routing_map_provider.py

@@ -79,6 +79,8 @@ def __init__(self, client: Any):
             self._collection_routing_map_by_item = _shared_routing_map_cache[self._endpoint]
         # A lock to control access to the locks dictionary itself
         self._locks_lock = asyncio.Lock()
+        # A dictionary to hold a lock for each collection ID
+        self._collection_locks: Dict[str, asyncio.Lock] = {}
 
     def clear_cache(self):
         """Clear the shared routing map cache for this endpoint."""
@@ -87,8 +89,7 @@ def clear_cache(self):
                 _shared_routing_map_cache[self._endpoint] = {}
             self._collection_routing_map_by_item = _shared_routing_map_cache.get(self._endpoint, {})
 
-        # A dictionary to hold a lock for each collection ID
-        self._collection_locks: Dict[str, asyncio.Lock] = {}
+        self._collection_locks = {}
 
     async def _get_lock_for_collection(self, collection_id: str) -> asyncio.Lock:
         """Safely gets or creates a lock for a given collection ID.

sdk/cosmos/azure-cosmos/azure/cosmos/_routing/routing_map_provider.py

@@ -76,17 +76,20 @@ def __init__(self, client: Any):
                 _shared_routing_map_cache[self._endpoint] = {}
             self._collection_routing_map_by_item = _shared_routing_map_cache[self._endpoint]
 
+        # A lock to control access to the locks dictionary itself
+        self._locks_lock = threading.Lock()
+        # A dictionary to hold a lock for each collection ID
+        self._collection_locks: Dict[str, threading.Lock] = {}
+
     def clear_cache(self):
         """Clear the shared routing map cache for this endpoint."""
         with _shared_cache_lock:
             if self._endpoint in _shared_routing_map_cache:
                 _shared_routing_map_cache[self._endpoint] = {}
             self._collection_routing_map_by_item = _shared_routing_map_cache.get(self._endpoint, {})
 
-        # A lock to control access to the locks dictionary itself
         self._locks_lock = threading.Lock()
-        # A dictionary to hold a lock for each collection ID
-        self._collection_locks: Dict[str, threading.Lock] = {}
+        self._collection_locks = {}
 
     def _get_lock_for_collection(self, collection_id: str) -> threading.Lock:
 

sdk/cosmos/azure-cosmos/azure/cosmos/_routing/routing_range.py

@@ -29,7 +29,7 @@
 
 from collections import namedtuple
 
-_PKRangeBase = namedtuple('PKRange', ['id', 'minInclusive', 'maxExclusive', 'parents'])
+_PKRangeBase = namedtuple('_PKRangeBase', ['id', 'minInclusive', 'maxExclusive', 'parents'])
 
 
 class PKRange(_PKRangeBase):
@@ -39,8 +39,8 @@ class PKRange(_PKRangeBase):
     def __getitem__(self, key):
         try:
             return getattr(self, key)
-        except AttributeError:
-            raise KeyError(key)
+        except AttributeError as exc:
+            raise KeyError(key) from exc
 
     def get(self, key, default=None):
         return getattr(self, key, default)