feat: Adds support for displaying NAGs in the chess.svg.board AND other optimizations chess.svg.board function

.gitignore

@@ -22,3 +22,4 @@ data/syzygy/giveaway/*.[gs]tb[wz]
 fuzz/corpus
 
 release-v*.txt
+.venv

chess/__init__.py

@@ -1340,7 +1340,7 @@ def transform(self: BaseBoardT, f: Callable[[Bitboard], Bitboard]) -> BaseBoardT
         board.apply_transform(f)
         return board
 
-    def apply_mirror(self: BaseBoardT) -> None:
+    def apply_mirror(self: BaseBoard) -> None:
         self.apply_transform(flip_vertical)
         self.occupied_co[WHITE], self.occupied_co[BLACK] = self.occupied_co[BLACK], self.occupied_co[WHITE]
 
@@ -1561,14 +1561,14 @@ class Board(BaseBoard):
     manipulation.
     """
 
-    def __init__(self: BoardT, fen: Optional[str] = STARTING_FEN, *, chess960: bool = False) -> None:
+    def __init__(self: Board, fen: Optional[str] = STARTING_FEN, *, chess960: bool = False) -> None:
         BaseBoard.__init__(self, None)
 
         self.chess960 = chess960
 
         self.ep_square = None
         self.move_stack = []
-        self._stack: List[_BoardState[BoardT]] = []
+        self._stack: List[_BoardState[Board]] = []
 
         if fen is None:
             self.clear()
@@ -2177,7 +2177,7 @@ def _board_state(self: BoardT) -> _BoardState[BoardT]:
     def _push_capture(self, move: Move, capture_square: Square, piece_type: PieceType, was_promoted: bool) -> None:
         pass
 
-    def push(self: BoardT, move: Move) -> None:
+    def push(self: Board, move: Move) -> None:
         """
         Updates the position with the given *move* and puts it onto the
         move stack.
@@ -2262,6 +2262,7 @@ def push(self: BoardT, move: Move) -> None:
             elif diff == -16 and square_rank(move.from_square) == 6:
                 self.ep_square = move.from_square - 8
             elif move.to_square == ep_square and abs(diff) in [7, 9] and not captured_piece_type:
+                assert ep_square is not None
                 # Remove pawns captured en passant.
                 down = -8 if self.turn == WHITE else 8
                 capture_square = ep_square + down
@@ -2298,7 +2299,7 @@ def push(self: BoardT, move: Move) -> None:
         # Swap turn.
         self.turn = not self.turn
 
-    def pop(self: BoardT) -> Move:
+    def pop(self: Board) -> Move:
         """
         Restores the previous position and returns the last move from the stack.
 
@@ -3696,7 +3697,7 @@ def transform(self: BoardT, f: Callable[[Bitboard], Bitboard]) -> BoardT:
         board.apply_transform(f)
         return board
 
-    def apply_mirror(self: BoardT) -> None:
+    def apply_mirror(self: Board) -> None:
         super().apply_mirror()
         self.turn = not self.turn
 

chess/pgn.py

@@ -92,7 +92,9 @@
     |(\()
     |(\))
     |(\*|1-0|0-1|1/2-1/2)
+    |(\!N)
     |([\?!]{1,2})
+    |(TN)
     """, re.DOTALL | re.VERBOSE)
 
 SKIP_MOVETEXT_REGEX = re.compile(r""";|\{|\}""")
@@ -922,7 +924,7 @@ def without_tag_roster(cls: Type[GameT]) -> GameT:
         return cls(headers={})
 
     @classmethod
-    def builder(cls: Type[GameT]) -> GameBuilder[GameT]:
+    def builder(cls: Type[GameT]) -> GameBuilder[Game]:
         return GameBuilder(Game=cls)
 
     def __repr__(self) -> str:
@@ -1026,7 +1028,7 @@ def __repr__(self) -> str:
             ", ".join("{}={!r}".format(key, value) for key, value in self.items()))
 
     @classmethod
-    def builder(cls: Type[HeadersT]) -> HeadersBuilder[HeadersT]:
+    def builder(cls: Type[HeadersT]) -> HeadersBuilder[Headers]:
         return HeadersBuilder(Headers=cls)
 
 
@@ -1180,7 +1182,7 @@ class GameBuilder(BaseVisitor[GameT]):
     @typing.overload
     def __init__(self: GameBuilder[Game]) -> None: ...
     @typing.overload
-    def __init__(self: GameBuilder[GameT], *, Game: Type[GameT]) -> None: ...
+    def __init__(self: GameBuilder[Game], *, Game: Type[GameT]) -> None: ...
     def __init__(self, *, Game: Any = Game) -> None:
         self.Game = Game
 
@@ -1277,7 +1279,7 @@ class HeadersBuilder(BaseVisitor[HeadersT]):
     @typing.overload
     def __init__(self: HeadersBuilder[Headers]) -> None: ...
     @typing.overload
-    def __init__(self: HeadersBuilder[HeadersT], *, Headers: Type[Headers]) -> None: ...
+    def __init__(self: HeadersBuilder[Headers], *, Headers: Type[Headers]) -> None: ...
     def __init__(self, *, Headers: Any = Headers) -> None:
         self.Headers = Headers
 
@@ -1736,6 +1738,8 @@ def read_game(handle: TextIO, *, Visitor: Any = GameBuilder) -> Any:
                 visitor.visit_nag(NAG_SPECULATIVE_MOVE)
             elif token == "?!":
                 visitor.visit_nag(NAG_DUBIOUS_MOVE)
+            elif token == "TN" or token == "!N":
+                visitor.visit_nag(NAG_NOVELTY)
             elif token in ["1-0", "0-1", "1/2-1/2", "*"] and len(board_stack) == 1:
                 visitor.visit_result(token)
             else: