Initial Commit

text2image.py

@@ -0,0 +1,72 @@
+from PIL import Image, ImageDraw, ImageFont
+import csv
+import toml
+
+conf = None
+
+with open('.\settings.conf', mode='r') as c:
+    conf = toml.load(c)
+
+inputFile = conf["csv"]["inputFile"]
+separator = conf["csv"]["separator"]
+skipFirstLine = conf["csv"]["skipFirstLine"]
+
+imgWidth = conf["image"]["width"]
+imgHeight = conf["image"]["height"]
+textFont = conf["image"]["font"]
+bgColor = conf["image"]["bgColor"]
+fontColor = conf["image"]["fontcolor"]
+
+if(type(fontColor) == type([])):
+    fontColor = (fontColor[0],fontColor[1],fontColor[2])
+    
+if(type(bgColor) == type([])):
+    bgColor = (bgColor[0],bgColor[1],bgColor[2])
+
+fontSize = conf["image"]["fontsize"]
+alignment = conf["image"]["alignment"]
+verticalAlign = conf["image"]["verticalAlign"]
+outputPath = conf["image"]["outputPath"]
+
+with open(inputFile, mode='r') as file:
+    csvFile = csv.reader(file, delimiter=separator)
+    i = 0
+
+    for lines in csvFile:
+        if(i == 0 and skipFirstLine):
+            i += 1
+            continue
+            
+        text = lines[1]
+        outputFileName = lines[0]
+        
+        image = Image.new("RGB", (imgWidth, imgHeight), bgColor)
+        draw = ImageDraw.Draw(image)
+        # select font
+        font = ImageFont.truetype(textFont, size=fontSize)
+        textlength = draw.textlength(text, font=font)
+        
+        # Horizontal alignment of text
+        if(alignment == "center"):
+            xtext = (imgWidth - textlength) // 2
+        elif(alignment == "left"):
+            xtext = 5
+        elif(alignment == "right"):
+            xtext = (imgWidth - textlength-3)
+        else:
+            xtext = int(alignment)
+            
+        # Vertical alignment of text
+        if(verticalAlign == "center"):
+            htext = (imgHeight+3 - fontSize) // 2
+        elif(verticalAlign == "top"):
+            htext = 2
+        elif(verticalAlign == "bottom"):
+            htext = (imgHeight - fontSize)
+        else:
+            htext = int(verticalAlign)
+            
+        # draw image
+        draw.text((xtext, htext), text, font=font, fill=fontColor)
+        image.save(f"{outputPath}/{outputFileName}.png")
+        i += 1