From 80b947823b6cfb08d61cd36fce54160af329119e Mon Sep 17 00:00:00 2001
From: Antony Clare <6871661+aclare92@users.noreply.github.com>
Date: Tue, 7 Mar 2023 22:19:00 +0000
Subject: [PATCH] Support for Ted's Notebook

---
 modules/ergast.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/modules/ergast.py b/modules/ergast.py
index a59ec94f..539039da 100644
--- a/modules/ergast.py
+++ b/modules/ergast.py
@@ -29,6 +29,7 @@ terms = {
     "pre": ["pre", "voorbeschouwing"],
     "post": ["post", "nabeschouwing"],
     "quali": ["quali", "kwalificatie"],
+    "notebook": ["notebook", "notitieboekje"],
     "preview": ["preview", "seizoensvoorbeschouwing"],
     "summary": ["summary", "samenvatting"],
     "highlight": ["highlight", "hoogtepunten"],
@@ -43,6 +44,7 @@ names = {
         "Pre-Sprint Build-up": "Sprint Voorbeschouwing",
         "Post-Sprint Analysis": "Sprint Nabeschouwing",
         "Sprint Qualifying": "Sprint Kwalificatie",
+        "Ted's Sprint Notebook": "Ted's Sprint Notitieboekje",
         "Pre-Qualifying Build-up": "Kwalificatie Voorbeschouwing",
         "Post-Qualifying Analysis": "Kwalificatie Nabeschouwing",
         "Qualifying Session": "Kwalificatie",
@@ -52,6 +54,7 @@ names = {
         "Live from the Grid": "Vanaf de grid",
         "Highlights": "Samenvatting",
         "Race Session": "Race",
+        "Ted's Race Notebook": "Ted's Race Notitieboekje",
     }
 }
 
@@ -93,12 +96,14 @@ class Race:
                 output = "Pre-Sprint Build-up"
             elif any([x in title for x in terms["post"]]):
                 output = "Post-Sprint Analysis"
+            elif any([x in title for x in terms["notebook"]]):
+                output = "Ted's Sprint Notebook"
             else:
                 output = "Sprint Qualifying"
         elif any([x in title for x in terms["quali"]]):
             if any([x in title for x in terms["pre"]]):
                 output = "Pre-Qualifying Build-up"
-            elif any([x in title for x in terms["post"]]):
+            elif any([x in title for x in terms["post"] + terms["notebook"]]):
                 output = "Post-Qualifying Analysis"
             else:
                 output = "Qualifying Session"
@@ -112,6 +117,8 @@ class Race:
             output = "Live from the Grid"
         elif any([x in title for x in terms["summary"] + terms["highlight"]]):
             output = "Highlights"
+        elif any([x in title for x in terms["notebook"]]):
+            output = "Ted's Race Notebook"
         else:
             output = "Race Session"
         if "2160" in title or "4K" in title: