add file extention check

--- Redmine Issue Progress / ID / Time : 100% #44 @1h --- What : 1. add font extention list .ttf, .otf, .ttc 2. add code checking if file is in the font extention list Why : to ignore not font file --- *Signed-off-by: sjang42 <[email protected]>*
twiiks · Oct 12, 2017 · c8c4c23 · c8c4c23
1 parent 1cc9d49
commit c8c4c23
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ttf-to-jpg.py b/ttf-to-jpg.py
@@ -5,6 +5,7 @@
 from PIL import ImageDraw
 from PIL import ImageFont
 
+FONT_EXTENTION = ['.ttf', '.otf', '.ttc']
 
 def parse_args():
     desc = "ttf/otf fonts to jpg images set (JUST KOREAN)"
@@ -44,6 +45,11 @@ def main():
 
     for _, _, ttfs in os.walk(ttf_dir):
         for ttf in ttfs:
+            # check extention type of file and continue if it is not font file
+            ext = os.path.splitext(ttf)[-1]
+            if ext not in FONT_EXTENTION:
+                continue
+
             font = ImageFont.truetype(ttf_dir + '/' + ttf, size=char_size)
             count = 0
             for c in charset: