Skip to content

Commit a920026

Browse files
committed
fix: player_id coerce from player_reference
1 parent 943cff2 commit a920026

1 file changed

Lines changed: 8 additions & 3 deletions

File tree

‎hslog/export.py‎

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -340,9 +340,9 @@ class FriendlyPlayerExporter(BaseExporter):
340340
def __init__(self, packet_tree):
341341
super().__init__(packet_tree)
342342
self._controller_map: Dict[int, int] = {}
343-
self.friendly_player = None
343+
self.friendly_player: Optional[int] = None
344344
self._ai_player = None
345-
self._non_ai_players = []
345+
self._non_ai_players: List[int] = []
346346

347347
def export(self):
348348
for packet in self.packet_tree:
@@ -362,7 +362,12 @@ def handle_player(self, packet):
362362
if packet.lo == 0:
363363
self._ai_player = packet.player_id
364364
else:
365-
self._non_ai_players.append(packet.player_id)
365+
player_id_or_player = packet.player_id
366+
if isinstance(player_id_or_player, PlayerReference):
367+
self._non_ai_players.append(player_id_or_player.player_id)
368+
else:
369+
self._non_ai_players.append(player_id_or_player)
370+
366371

367372
def handle_tag_change(self, packet):
368373
if packet.tag == GameTag.CONTROLLER:

0 commit comments

Comments
 (0)