fix

[audio.git] / src / youtube / thumbnail.py

diff --git a/src/youtube/thumbnail.py b/src/youtube/thumbnail.py
index e40b552..c64e5b5 100644 (file)
--- a/src/youtube/thumbnail.py
+++ b/src/youtube/thumbnail.py
@@ -70,7 +70,7 @@ def draw_box_with_scaling(img, d, context, get_font_path):
 
 def create_thumbnail(background_path, defn, context, get_font_path):
     img = Image.open(background_path)
 
 def create_thumbnail(background_path, defn, context, get_font_path):
     img = Image.open(background_path)

-    d = yaml.load(defn)
+    d = yaml.safe_load(defn)

     for boxdef in d['boxes']:
         if not draw_box_with_scaling(img, boxdef, context, get_font_path):
             raise ValueError()
     for boxdef in d['boxes']:
         if not draw_box_with_scaling(img, boxdef, context, get_font_path):
             raise ValueError()