mirror of
https://gitlab.com/oloturia/fumcaso.git
synced 2024-11-01 01:38:16 +01:00
72e9a43123
Allows cartoons with no text.
88 lines
2.3 KiB
Python
Executable file
88 lines
2.3 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
from PIL import Image
|
|
from PIL import ImageFont
|
|
from PIL import ImageDraw
|
|
import csv
|
|
import random
|
|
|
|
def fetchText(indText):
|
|
with open("rtext.csv") as rtext:
|
|
csvReader = csv.reader(rtext,delimiter=';')
|
|
for row in csvReader:
|
|
if len(row)>1:
|
|
if row[0]==indText:
|
|
return row[1],row[2],row[random.randint(3,len(row)-1)].replace('@','\n')
|
|
else:
|
|
return 0
|
|
|
|
def fetch2Text(indText):
|
|
with open("r2text.csv") as rtext:
|
|
csvReader = csv.reader(rtext,delimiter=';')
|
|
for row in csvReader:
|
|
if len(row)>1:
|
|
if row[0]==indText:
|
|
rand1 = random.randint(5,len(row)-1)
|
|
if rand1 %2 == 0:
|
|
rand1 +=1
|
|
rand2 = rand1+1
|
|
return row[1],row[2],row[3],row[4],row[rand1].replace('@','\n'),row[rand2].replace('@','\n')
|
|
else:
|
|
return 0
|
|
|
|
def fetchVign():
|
|
starts = []
|
|
startdest = []
|
|
nvign = 0
|
|
currVign = "000"
|
|
story = []
|
|
with open("ram.csv") as ram:
|
|
csvReader = csv.reader(ram)
|
|
for row in csvReader:
|
|
starts.append(row[0])
|
|
startdest.append(row)
|
|
while nvign <= 3:
|
|
story.append(startdest[starts.index(currVign)][random.randint(1,len(startdest[starts.index(currVign)])-1)])
|
|
currVign = story[nvign]
|
|
if story[nvign] == "B00":
|
|
story[nvign] += "."
|
|
story[nvign] += str(random.randint(0,2))
|
|
story[nvign]+=".png"
|
|
nvign +=1
|
|
return story
|
|
|
|
def addThing(vign):
|
|
#TODO
|
|
pass
|
|
|
|
def writeStrip(story):
|
|
strip = []
|
|
for indVign in story:
|
|
if indVign!="000":
|
|
vign = Image.open(indVign)
|
|
addtext = ImageDraw.Draw(vign)
|
|
fnt = ImageFont.truetype("ubuntu.ttf",16)
|
|
if indVign[0] == 'A':
|
|
textVign = fetchText(indVign)
|
|
if textVign !=0:
|
|
addtext.multiline_text((int(textVign[0]),int(textVign[1])),textVign[2],fill="#000000",font=fnt,align="center")
|
|
else:
|
|
textVign = fetch2Text(indVign)
|
|
if textVign!=0:
|
|
addtext.multiline_text((int(textVign[0]),int(textVign[1])),textVign[4],fill="#000000",font=fnt,align="center")
|
|
addtext.multiline_text((int(textVign[2]),int(textVign[3])),textVign[5],fill="#000000",font=fnt,align="center")
|
|
obj = addThing(indVign)
|
|
if obj!=(0,0):
|
|
#TODO
|
|
pass
|
|
strip.append(vign)
|
|
image = Image.new('RGB',(2400,500))
|
|
xshift=0
|
|
for vign in strip:
|
|
image.paste(vign,(xshift,0))
|
|
xshift += 600
|
|
return image
|
|
|
|
if __name__ == "__main__":
|
|
story = fetchVign()
|
|
finalStrip = writeStrip(story)
|
|
finalStrip.show()
|