Fix a crash when a game to be analyzed has not move

fsmosca · Mar 20, 2021 · 1e57fb8 · 1e57fb8
1 parent 11636ee
commit 1e57fb8
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/chess_artist.py b/chess_artist.py
@@ -9,7 +9,7 @@
 
 __author__ = 'fsmosca'
 __script_name__ = 'Chess Artist'
-__version__ = 'v2.26.0'
+__version__ = 'v2.27.0'
 __credits__ = ['alxlk', 'ddugovic', 'huytd', 'kennyfrc', 'python-chess']
 
 
@@ -1686,6 +1686,8 @@ def SaveMaterialBalance(game):
         """
         mat = []
         gameNode = game
+        nextNode = None
+
         while gameNode.variations:
             board = gameNode.board()
             fen = board.fen()
@@ -1695,9 +1697,10 @@ def SaveMaterialBalance(game):
             nextNode = gameNode.variation(0)
 
             gameNode = nextNode
-
-        endFen = nextNode.board().fen()
-        mat.append([endFen, Analyze.GetMaterialBalance(endFen)])
+
+        if nextNode is not None:
+            endFen = nextNode.board().fen()
+            mat.append([endFen, Analyze.GetMaterialBalance(endFen)])
 
         return mat