Enable reporting unknown types

changes/333.internal.md

@@ -0,0 +1 @@
+Enable reporting unknown types for basedpyright.

mcproto/auth/account.py

@@ -45,6 +45,10 @@ class Account:
 
     __slots__ = ("access_token", "username", "uuid")
 
+    username: str
+    uuid: McUUID
+    access_token: str
+
     def __init__(self, username: str, uuid: McUUID, access_token: str) -> None:
         self.username = username
         self.uuid = uuid

mcproto/auth/microsoft/xbox.py

@@ -4,7 +4,7 @@
 from typing import NamedTuple
 
 import httpx
-from typing_extensions import override
+from typing_extensions import Any, override
 
 __all__ = [
     "XSTSErrorType",
@@ -67,7 +67,7 @@ def __init__(self, exc: httpx.HTTPStatusError):
     @property
     def msg(self) -> str:
         """Produce a message for this error."""
-        msg_parts = []
+        msg_parts: list[str] = []
         if self.err_type is not XSTSErrorType.UNKNOWN:
             msg_parts.append(f"{self.err_type.name}: {self.err_type.value!r}")
         else:
@@ -96,7 +96,7 @@ async def xbox_auth(client: httpx.AsyncClient, microsoft_access_token: str, bedr
     See [`full_microsoft_oauth`][mcproto.auth.microsoft.oauth.] for info on `microsoft_access_token`.
     """
     # Obtain XBL token
-    payload = {
+    payload: dict[str, Any] = {
         "Properties": {
             "AuthMethod": "RPS",
             "SiteName": "user.auth.xboxlive.com",

mcproto/auth/msa.py

@@ -48,7 +48,7 @@ def __init__(self, exc: httpx.HTTPStatusError):
     @property
     def msg(self) -> str:
         """Produce a message for this error."""
-        msg_parts = []
+        msg_parts: list[str] = []
         msg_parts.append(f"HTTP {self.code} from {self.url}:")
         msg_parts.append(f"type={self.err_type.name!r}")
 

mcproto/auth/yggdrasil.py

@@ -5,7 +5,7 @@
 from uuid import uuid4
 
 import httpx
-from typing_extensions import Self, override
+from typing_extensions import Any, Self, override
 
 from mcproto.auth.account import Account
 from mcproto.types.uuid import UUID as McUUID  # noqa: N811
@@ -89,7 +89,7 @@ def __init__(self, exc: httpx.HTTPStatusError):
     @property
     def msg(self) -> str:
         """Produce a message for this error."""
-        msg_parts = []
+        msg_parts: list[str] = []
         msg_parts.append(f"HTTP {self.code} from {self.url}:")
         msg_parts.append(f"type={self.err_type.name!r}")
 
@@ -126,7 +126,7 @@ async def refresh(self, client: httpx.AsyncClient) -> None:
         having to go through a complete re-login. This can happen after some time period, or
         for example when someone else logs in to this minecraft account elsewhere.
         """
-        payload = {
+        payload: dict[str, Any] = {
             "accessToken": self.access_token,
             "clientToken": self.client_token,
             "selectedProfile": {"id": str(self.uuid), "name": self.username},
@@ -197,7 +197,7 @@ async def authenticate(cls, client: httpx.AsyncClient, login: str, password: str
         # Any random string, we use a random v4 uuid, needs to remain same in further communications
         client_token = str(uuid4())
 
-        payload = {
+        payload: dict[str, Any] = {
             "agent": {
                 "name": "Minecraft",
                 "version": 1,

mcproto/buffer.py

@@ -12,7 +12,7 @@ class Buffer(BaseSyncWriter, BaseSyncReader, bytearray):
 
     __slots__ = ("pos",)
 
-    def __init__(self, *args, **kwargs):
+    def __init__(self, *args: object, **kwargs: object):
         super().__init__(*args, **kwargs)
         self.pos = 0
 

mcproto/connection.py

@@ -149,7 +149,7 @@ def __enter__(self) -> Self:
             raise IOError("Connection already closed.")
         return self
 
-    def __exit__(self, *a, **kw) -> None:
+    def __exit__(self, *a: object, **kw: object) -> None:
         self.close()
 
 
@@ -272,7 +272,7 @@ async def __aenter__(self) -> Self:
             raise IOError("Connection already closed.")
         return self
 
-    async def __aexit__(self, *a, **kw) -> None:
+    async def __aexit__(self, *a: object, **kw: object) -> None:
         await self.close()
 
 
@@ -379,7 +379,7 @@ def socket(self) -> socket.socket:
         """Obtain the underlying socket behind the [`Transport`][?asyncio.]."""
         # TODO: This should also have pyright: ignore[reportPrivateUsage]
         # See: https://github.com/DetachHead/basedpyright/issues/494
-        return self.writer.transport._sock  # pyright: ignore[reportAttributeAccessIssue]
+        return self.writer.transport._sock  # pyright: ignore[reportAttributeAccessIssue,reportUnknownMemberType,reportUnknownVariableType]
 
 
 class UDPSyncConnection(SyncConnection, Generic[T_SOCK]):

mcproto/multiplayer.py

@@ -65,7 +65,7 @@ def __init__(self, exc: httpx.HTTPStatusError):
     @property
     def msg(self) -> str:
         """Produce a message for this error."""
-        msg_parts = []
+        msg_parts: list[str] = []
         msg_parts.append(f"HTTP {self.code} from {self.url}:")
         msg_parts.append(f"type={self.err_type.name!r}")
 

mcproto/packets/packet.py

@@ -129,7 +129,7 @@ def from_packet_class(cls, packet_class: type[Packet], buffer: Buffer, message:
     @property
     def msg(self) -> str:
         """Produce a message for this error."""
-        msg_parts = []
+        msg_parts: list[str] = []
 
         if self.direction is PacketDirection.CLIENTBOUND:
             msg_parts.append("Clientbound")

mcproto/types/nbt.py

@@ -344,6 +344,13 @@ class that has a `to_nbt` method.
         Returns:
             The NBT tag created from the python object.
         """
+        # TODO: There are a lot of isinstance checks for dict/list, however, the FromObjectType/FromObjectSchema
+        # type alias declares a Sequence/Mapping type for these collections. The isinstance checks here should
+        # probably be replaced with these types (they should support runtime comparison).
+
+        # TODO: Consider splitting this function up into smaller functions, that each parse a specific type of schema
+        # i.e. _from_object_dict, _from_object_list, _from_object_tag, ...
+
         # Case 0 : schema is an object with a `to_nbt` method (could be a subclass of NBTag for all we know, as long
         # as the data is an instance of the schema it will work)
         if isinstance(schema, type) and hasattr(schema, "to_nbt") and isinstance(data, schema):
@@ -353,23 +360,27 @@ class that has a `to_nbt` method.
         if isinstance(schema, type) and issubclass(schema, NBTag):
             if schema in (CompoundNBT, ListNBT):
                 raise ValueError("Use a list or a dictionary in the schema to create a CompoundNBT or a ListNBT.")
+
             # Check if the data contains the name (if it is a dictionary)
             if isinstance(data, dict):
                 if len(data) != 1:
                     raise ValueError("Expected a dictionary with a single key-value pair.")
                 # We also check if the name isn't already set
                 if name:
                     raise ValueError("The name is already set.")
+
                 key, value = next(iter(data.items()))
                 # Recursive call to go to the next part
                 return NBTag.from_object(value, schema, name=key)
+
             # Else we check if the data can be a payload for the tag
             if not isinstance(data, (bytes, str, int, float, list)):
                 raise TypeError(f"Expected one of (bytes, str, int, float, list), but found {type(data).__name__}.")
+
             # Check if the data is a list of integers
             if isinstance(data, list) and not all(isinstance(item, int) for item in data):
                 raise TypeError("Expected a list of integers, but a non-integer element was found.")
-            data = cast("Union[bytes, str, int, float, list[int]]", data)
+            data = cast("bytes | str | int | float | list[int]", data)
             # Create the tag with the data and the name
             return schema(data, name=name)  # pyright: ignore[reportCallIssue] # The schema is a subclass of NBTag
 
@@ -399,6 +410,7 @@ class that has a `to_nbt` method.
         # as there are only dicts, or only lists in the schema
         if not isinstance(data, list):
             raise TypeError(f"Expected a list, but found {type(data).__name__}.")
+
         if len(schema) == 1:
             # We have two cases here, either the schema supports an unknown number of elements of a single type ...
             children_schema = schema[0]

pyproject.toml

@@ -95,11 +95,6 @@ reportImplicitStringConcatenation = false
 reportUnreachable = "hint"
 reportUnusedParameter = "hint"
 reportUnannotatedClassAttribute = false
-reportUnknownArgumentType = false             # consider enabling
-reportUnknownVariableType = false             # consider enabling
-reportUnknownMemberType = false               # consider enabling
-reportUnknownParameterType = false            # consider enabling
-reportUnknownLambdaType = false               # consider enabling
 reportMissingTypeStubs = "information"        # consider bumping to warning/error
 reportUninitializedInstanceVariable = false   # until https://github.com/DetachHead/basedpyright/issues/491
 reportIncompatibleUnannotatedOverride = false # until https://github.com/DetachHead/basedpyright/issues/1349
@@ -131,8 +126,6 @@ ignore = [
   "D301", # Use r""" if any backslashes in a docstring
   "D417", # Missing argument descrition in the docstring
 
-  "ANN002", # Missing type annotation for *args
-  "ANN003", # Missing type annotation for **kwargs
   "ANN204", # Missing return type annotation for special method
   "ANN401", # Dynamically typed expressions (typing.Any) disallowed
 

tests/helpers.py

@@ -141,7 +141,7 @@ class be of/return the same class.
     _mock_sealed: bool
     _extract_mock_name: Callable[[], str]
 
-    def _get_child_mock(self, **kwargs) -> T_Mock:
+    def _get_child_mock(self, **kwargs: object) -> T_Mock:
         """Make [`child_mock_type`][..] instances instead of instances of the same class.
 
         By default, this method creates a new mock instance of the same original class, and passes

tests/mcproto/protocol/helpers.py

@@ -8,7 +8,7 @@
 class WriteFunctionMock(Mock):
     """Mock write function, storing the written data."""
 
-    def __init__(self, *a, **kw):
+    def __init__(self, *a: object, **kw: object):
         super().__init__(*a, **kw)
         self.combined_data = bytearray()
 
@@ -41,7 +41,7 @@ class WriteFunctionAsyncMock(WriteFunctionMock, AsyncMock):  # pyright: ignore[r
 class ReadFunctionMock(Mock):
     """Mock read function, giving pre-defined data."""
 
-    def __init__(self, *a, combined_data: bytearray | None = None, **kw):
+    def __init__(self, *a: object, combined_data: bytearray | None = None, **kw: object):
         super().__init__(*a, **kw)
         if combined_data is None:
             combined_data = bytearray()

tests/mcproto/test_connection.py

@@ -22,7 +22,7 @@ class MockSocket(CustomMockMixin[MagicMock], MagicMock):  # pyright: ignore[repo
 
     spec_set = socket.socket
 
-    def __init__(self, *args, read_data: bytearray | None = None, **kwargs) -> None:
+    def __init__(self, *args: object, read_data: bytearray | None = None, **kwargs: object) -> None:
         super().__init__(*args, **kwargs)
         self.mock_add_spec(["_recv", "_send", "_closed"])
         self._recv = ReadFunctionMock(combined_data=read_data)
@@ -60,7 +60,7 @@ class MockStreamWriter(CustomMockMixin[MagicMock], MagicMock):  # pyright: ignor
 
     spec_set = asyncio.StreamWriter
 
-    def __init__(self, *args, **kwargs):
+    def __init__(self, *args: object, **kwargs: object):
         super().__init__(*args, **kwargs)
         self.mock_add_spec(["_white", "_closed"])
         self._write = WriteFunctionMock()
@@ -86,7 +86,7 @@ class MockStreamReader(CustomMockMixin[MagicMock], MagicMock):  # pyright: ignor
 
     spec_set = asyncio.StreamReader
 
-    def __init__(self, *args, read_data: bytearray | None = None, **kwargs) -> None:
+    def __init__(self, *args: object, read_data: bytearray | None = None, **kwargs: object) -> None:
         super().__init__(*args, **kwargs)
         self.mock_add_spec(["_read"])
         self._read = ReadFunctionAsyncMock(combined_data=read_data)

tests/mcproto/test_multiplayer.py

@@ -82,7 +82,7 @@ async def test_join_request_invalid(
         ("172.17.0.1"),
     ],
 )
-async def test_join_check_valid(client_ip, httpx_mock: HTTPXMock):
+async def test_join_check_valid(client_ip: str | None, httpx_mock: HTTPXMock):
     """Test making a join check, getting back a valid response."""
     client_username = "ItsDrike"
     server_hash = "-745fc7fdb2d6ae7c4b20e2987770def8f3dd1105"

tests/mcproto/types/test_chat.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+from typing import cast
+
 import pytest
 
 from mcproto.types.chat import ChatMessage, RawChatMessage, RawChatMessageDict
@@ -8,20 +10,23 @@
 
 @pytest.mark.parametrize(
     ("raw", "expected_dict"),
-    [
-        (
-            {"text": "A Minecraft Server"},
-            {"text": "A Minecraft Server"},
-        ),
-        (
-            "A Minecraft Server",
-            {"text": "A Minecraft Server"},
-        ),
-        (
-            [{"text": "hello", "bold": True}, {"text": "there"}],
-            {"extra": [{"text": "hello", "bold": True}, {"text": "there"}]},
-        ),
-    ],
+    cast(
+        "list[tuple[RawChatMessage, RawChatMessageDict]]",
+        [
+            (
+                {"text": "A Minecraft Server"},
+                {"text": "A Minecraft Server"},
+            ),
+            (
+                "A Minecraft Server",
+                {"text": "A Minecraft Server"},
+            ),
+            (
+                [{"text": "hello", "bold": True}, {"text": "there"}],
+                {"extra": [{"text": "hello", "bold": True}, {"text": "there"}]},
+            ),
+        ],
+    ),
 )
 def test_as_dict(raw: RawChatMessage, expected_dict: RawChatMessageDict):
     """Test converting raw ChatMessage input into dict produces expected dict."""

tests/mcproto/types/test_nbt.py

@@ -13,6 +13,8 @@
     DoubleNBT,
     EndNBT,
     FloatNBT,
+    FromObjectSchema,
+    FromObjectType,
     IntArrayNBT,
     IntNBT,
     ListNBT,
@@ -477,7 +479,7 @@ def test_nbt_bigfile():
     data = bytes.fromhex(data)
     buffer = Buffer(data)
 
-    expected_object = {  # Name ! Level
+    expected_object: FromObjectType = {  # Name ! Level
         "longTest": 9223372036854775807,
         "shortTest": 32767,
         "stringTest": "HELLO WORLD THIS IS A TEST STRING ÅÄÖ!",
@@ -499,7 +501,7 @@ def test_nbt_bigfile():
         ),
         "doubleTest": 0.4921875,
     }
-    expected_schema = {
+    expected_schema: FromObjectSchema = {
         "longTest": LongNBT,
         "shortTest": ShortNBT,
         "stringTest": StringNBT,