From a0c2cee857365fb03ca2503713b31d5073dacb54 Mon Sep 17 00:00:00 2001
From: stijndcl <declercq.stijn@outlook.com>
Date: Tue, 27 Sep 2022 18:34:07 +0200
Subject: [PATCH] Add extra course for Machine Learning

---
 database/scripts/db02_ufora_courses_22_23.py  |  2 +-
 database/scripts/db03_machine_learning_cse.py | 21 +++++++++++++++++++
 2 files changed, 22 insertions(+), 1 deletion(-)
 create mode 100644 database/scripts/db03_machine_learning_cse.py

diff --git a/database/scripts/db02_ufora_courses_22_23.py b/database/scripts/db02_ufora_courses_22_23.py
index 6d04829..b8880e3 100644
--- a/database/scripts/db02_ufora_courses_22_23.py
+++ b/database/scripts/db02_ufora_courses_22_23.py
@@ -215,7 +215,7 @@ async def main():
         machine_learning = UforaCourse(
             course_id=630807,
             code="C003758",
-            name="Machine Learning",
+            name="Machine Learning (CS)",
             year=4,
             compulsory=True,
             role_id=1023294041825235087,
diff --git a/database/scripts/db03_machine_learning_cse.py b/database/scripts/db03_machine_learning_cse.py
new file mode 100644
index 0000000..35c994a
--- /dev/null
+++ b/database/scripts/db03_machine_learning_cse.py
@@ -0,0 +1,21 @@
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from database.engine import DBSession
+from database.schemas import UforaCourse
+
+__all__ = ["main"]
+
+
+async def main():
+    """Add a missing Ufora course (Machine Learning - CSE)"""
+    session: AsyncSession
+    async with DBSession() as session:
+        mlcse = UforaCourse(
+            code="E061330",
+            name="Machine Learning (CSE)",
+            role_id=1024355572256092170,
+            overarching_role_id=1023300434800164914,
+        )
+
+        session.add(mlcse)
+        await session.commit()