Force FullEntity.entity to be an int and not sometimes a string

HearthSim · Aug 7, 2021 · 665a7c3 · 665a7c3
1 parent 6dc1e04
commit 665a7c3
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/hslog/parser.py b/hslog/parser.py
@@ -466,7 +466,7 @@ def block_start(
 		ps.current_block = block
 		return block
 
-	def full_entity(self, ps: ParsingState, ts, entity_id: int, card_id: str):
+	def _full_entity(self, ps: ParsingState, ts, entity_id: int, card_id: str):
 		ps.entity_packet = packets.FullEntity(ts, entity_id, card_id)
 		ps.register_packet(ps.entity_packet)
 
@@ -482,9 +482,11 @@ def full_entity(self, ps: ParsingState, ts, entity_id: int, card_id: str):
 
 		return ps.entity_packet
 
-	def full_entity_update(self, ps: ParsingState, ts, entity, card_id):
-		entity_id = ps.parse_entity_id(entity)
-		return self.full_entity(ps, ts, entity_id, card_id)
+	def full_entity(self, ps: ParsingState, ts, entity_id: str, card_id: str):
+		return self._full_entity(ps, ts, int(entity_id), card_id)
+
+	def full_entity_update(self, ps: ParsingState, ts, entity_id: str, card_id: str):
+		return self._full_entity(ps, ts, ps.parse_entity_id(entity_id), card_id)
 
 	@staticmethod
 	def show_entity(ps: ParsingState, ts, entity, card_id):