From 9edb6e41ce3ad6ca3c280d71a065f325eae30db9 Mon Sep 17 00:00:00 2001
From: Dominik Boehi <dominik.boehi@gmail.com>
Date: Sat, 12 Jun 2021 18:28:21 +0200
Subject: [PATCH] Make gen.py work without CSafeLoader

---
 embassy-stm32/gen.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/embassy-stm32/gen.py b/embassy-stm32/gen.py
index 2736cd69d..779389cac 100644
--- a/embassy-stm32/gen.py
+++ b/embassy-stm32/gen.py
@@ -4,6 +4,12 @@ import re
 import os
 import re
 
+try:
+    from yaml import CSafeLoader as SafeLoader
+except ImportError:
+    from yaml import SafeLoader
+
+
 abspath = os.path.abspath(__file__)
 dname = os.path.dirname(abspath)
 os.chdir(dname)
@@ -18,11 +24,11 @@ except:
 # ======= load chip
 chip_name = chip_name.upper()
 with open(f'{data_path}/chips/{chip_name}.yaml', 'r') as f:
-    chip = yaml.load(f, Loader=yaml.CSafeLoader)
+    chip = yaml.load(f, Loader=SafeLoader)
 
 # ======= load GPIO AF
 with open(f'{data_path}/gpio_af/{chip["gpio_af"]}.yaml', 'r') as f:
-    af = yaml.load(f, Loader=yaml.CSafeLoader)
+    af = yaml.load(f, Loader=SafeLoader)
 
 # ======= Generate!
 with open(output_file, 'w') as f: