改进检查

Author: 3-Tokisaki-Kurumi

pydoll/connection/connection_handler.py

@@ -200,6 +200,9 @@ def _is_connection_closed(self) -> bool:
 
     def _check_connection_state(self) -> bool:
         """Check connection state using various methods for compatibility."""
+        # This method is only called when _ws_connection is not None
+        assert self._ws_connection is not None
+
         # For websockets library (newer versions)
         if hasattr(self._ws_connection, 'closed'):
             return self._ws_connection.closed
@@ -217,6 +220,7 @@ def _check_connection_state(self) -> bool:
 
     def _check_state_attribute(self) -> bool:
         """Check connection state attribute."""
+        assert self._ws_connection is not None
         state = self._ws_connection.state
         if hasattr(state, 'name'):
             return state.name in {'CLOSED', 'CLOSING'}

pydoll/fingerprint/browser.py

@@ -110,6 +110,7 @@ async def _inject_fingerprint_script(self, tab: Tab):
         """
         try:
             # Get the JavaScript injection code
+            assert self.fingerprint_manager is not None
             script = self.fingerprint_manager.get_fingerprint_js()
 
             # Inject the script using Page.addScriptToEvaluateOnNewDocument
@@ -241,6 +242,7 @@ async def _inject_fingerprint_script(self, tab: Tab):
         """
         try:
             # Get the JavaScript injection code
+            assert self.fingerprint_manager is not None
             script = self.fingerprint_manager.get_fingerprint_js()
 
             # Inject the script using Page.addScriptToEvaluateOnNewDocument

pydoll/fingerprint/models.py

@@ -83,7 +83,7 @@ def to_dict(self) -> Dict[str, Union[str, int, float, bool, List, None]]:
     @classmethod
     def from_dict(cls, data: Dict[str, Union[str, int, float, bool, List, None]]) -> 'Fingerprint':
         """Create fingerprint from dictionary format."""
-        return cls(**data)
+        return cls(**data)  # type: ignore[arg-type]
 
 
 @dataclass
@@ -135,4 +135,4 @@ def to_dict(self) -> Dict[str, Union[str, int, bool, List, None]]:
     @classmethod
     def from_dict(cls, data: Dict[str, Union[str, int, bool, List, None]]) -> 'FingerprintConfig':
         """Create config from dictionary format."""
-        return cls(**data)
+        return cls(**data)  # type: ignore[arg-type]

pydoll/protocol/base.py

@@ -1,10 +1,12 @@
-from typing import TypedDict
+from typing import Any, Dict, TypedDict, TypeVar
 
 try:
     from typing import NotRequired
 except ImportError:
     from typing_extensions import NotRequired
 
+T = TypeVar('T')
+
 
 class CommandParams(TypedDict, total=False):
     """Base structure for all command parameters."""
@@ -47,4 +49,4 @@ class Event(TypedDict):
     """Base structure for all events."""
 
     method: str
-    params: NotRequired[dict[str, str]]
+    params: NotRequired[Dict[str, Any]]

pyproject.toml

@@ -64,4 +64,7 @@ post_test = 'coverage html'
 [tool.mypy]
 exclude = [
     "tests/",
+]
+disable_error_code = [
+    "type-arg",  # Disable "expects no type arguments" errors
 ]