diff --git a/rustplus/api/remote/rustplus_proto/rustplus.py b/rustplus/api/remote/rustplus_proto/rustplus.py index 175cdaa..83d882b 100644 --- a/rustplus/api/remote/rustplus_proto/rustplus.py +++ b/rustplus/api/remote/rustplus_proto/rustplus.py @@ -352,6 +352,9 @@ class AppTeamInfoNote(betterproto.Message): type: int = betterproto.int32_field(2) x: float = betterproto.float_field(3) y: float = betterproto.float_field(4) + icon: int = betterproto.int32_field(5) + colour_index: int = betterproto.int32_field(6) + label: str = betterproto.string_field(7) @dataclass diff --git a/rustplus/api/remote/rustws.py b/rustplus/api/remote/rustws.py index 81be523..10e4bbf 100644 --- a/rustplus/api/remote/rustws.py +++ b/rustplus/api/remote/rustws.py @@ -80,7 +80,10 @@ async def connect( ) address += f"?v={str(self.magic_value)}" self.connection = await connect( - address, close_timeout=0, ping_interval=None, max_size=1_000_000_000 + address, + close_timeout=0, + ping_interval=None, + max_size=1_000_000_000, ) self.connected_time = time.time() diff --git a/rustplus/api/structures/rust_team_info.py b/rustplus/api/structures/rust_team_info.py index 49c9ccc..5a6b241 100644 --- a/rustplus/api/structures/rust_team_info.py +++ b/rustplus/api/structures/rust_team_info.py @@ -64,6 +64,9 @@ def __init__(self, data: AppTeamInfoNote) -> None: self._type: int = data.type self._x: float = data.x self._y: float = data.y + self._icon: int = data.icon + self._colour_index: int = data.colour_index + self._label: str = data.label @property def type(self) -> int: @@ -77,8 +80,22 @@ def x(self) -> float: def y(self) -> float: return self._y + @property + def icon(self) -> int: + return self._icon + + @property + def colour_index(self) -> int: + return self._colour_index + + @property + def label(self) -> str: + return self._label + def __str__(self) -> str: - return "RustTeamNote[type={}, x={}, y={}]".format(self._type, self._x, self._y) + return "RustTeamNote[type={}, x={}, y={}, icon={}, colour_index={}, label={}]".format( + self._type, self._x, self._y, self._icon, self._colour_index, self._label + ) class RustTeamInfo: