From 452ce699ee43284d406c051d9610d02ad5d8ebca Mon Sep 17 00:00:00 2001
From: Fred Boniface <fred@fjla.uk>
Date: Thu, 2 May 2024 12:19:38 +0100
Subject: [PATCH] Add source file to YAML output

---
 src/formatter.py  | 2 ++
 src/local_mode.py | 1 +
 2 files changed, 3 insertions(+)

diff --git a/src/formatter.py b/src/formatter.py
index 6c5972a..9f5e107 100644
--- a/src/formatter.py
+++ b/src/formatter.py
@@ -14,6 +14,7 @@ def humanYaml(pis_list):
                     additional_pis += f'  - code: "{pis["pis"]}"\n'
                     additional_pis += f'    #headcode: {pis["headcode"]}\n'
                     additional_pis += f'    #date: {pis["date"]}\n'
+                    additional_pis += f'    #source_file: {pis["diagram_file"]}\n'
                     additional_pis += f'    stops: [{",".join(crs)}]\n'
             except Exception as err:
                 print(err)
@@ -22,6 +23,7 @@ def humanYaml(pis_list):
             manual_review += f'  - code: "{pis["pis"]}"\n'
             manual_review += f'    #headcode: {pis["headcode"]}\n'
             manual_review += f'    #date: {pis["date"]}\n'
+            manual_review += f'    #source_file: {pis["diagram_file"]}\n'
             for service in pis["services"]:
                 crs = []
                 if service and service['stops']:
diff --git a/src/local_mode.py b/src/local_mode.py
index 117c898..1c2ee6b 100644
--- a/src/local_mode.py
+++ b/src/local_mode.py
@@ -48,6 +48,7 @@ def start():
             'headcode': item['headcode'],
             'date': item['date'],
             'services': [],
+            'diagram_file': item['diagram_file']
         }
         for service in item['services']:
             service_detail = owlboard_connector.get_service_detail(service['trainUid'], item['date'])