From cc7c0afbd8d98cf5d1966d4d60b3d9081b4b3471 Mon Sep 17 00:00:00 2001
From: DanTheComputerMan <47550063+DanTheComputerMan@users.noreply.github.com>
Date: Tue, 24 Dec 2024 21:39:49 -0800
Subject: [PATCH] Update merge-pgn.py

Preserves header information (from first pgn), keeps all comment information (including arrows, eval, etc) and NAGs,
---
 merge-pgn.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/merge-pgn.py b/merge-pgn.py
index 4bda416..ebc65bc 100644
--- a/merge-pgn.py
+++ b/merge-pgn.py
@@ -22,6 +22,7 @@ def main():
             game = chess.pgn.read_game(pgn)
 
     master_node = chess.pgn.Game()
+    master_node.headers = games[0].headers
 
     mlist = []
     for game in games:
@@ -40,6 +41,10 @@ def main():
                     continue
                 elif node.move not in list(newmoves):
                     nvnode = vnode.add_variation(node.move)
+                    if len(node.comment) > 0:
+                        nvnode.comment = node.comment
+                    if len(node.nags) > 0:
+                        nvnode.nags = node.nags
                     if len(node.variations) > 0:
                         done = False
                     newvars.append((nvnode, node.variations))