Skip to content

Commit 3a1a728

Browse files
committed
Update send_protobuf_message method with consistent arguments
1 parent 54c5624 commit 3a1a728

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

src/steam/client/mixins/product_info.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
from typing import Optional, Any, TYPE_CHECKING
44
from steam.enums.emsgs import EMsg
55
from steam.utils.vdf import VDFParser
6+
from google.protobuf.message import Message
67

78
if TYPE_CHECKING:
89
class CMsgClientPICSProductInfoRequest:
@@ -36,7 +37,7 @@ class ProductInfoMixin:
3637

3738
if TYPE_CHECKING:
3839
async def send_protobuf_message(
39-
self, emsg: EMsg, _message: Any) -> None: ...
40+
self, emsg: EMsg, message: Message, steam_id: Optional[int] = None) -> None: ...
4041

4142
async def wait_for(
4243
self, event: Any, timeout: Optional[float] = None, check: Optional[Any] = None) -> Any: ...
@@ -67,8 +68,8 @@ async def get_product_info(self, app_ids: list[int], access_tokens: Optional[dic
6768

6869
try:
6970
packet = await self.wait_for(EMsg.ClientPICSProductInfoResponse, timeout=timeout)
70-
7171
body = packet.body
72+
7273
if isinstance(body, bytes):
7374
response = CMsgClientPICSProductInfoResponse()
7475
response.ParseFromString(body)

0 commit comments

Comments
 (0)