Salve a saída da tartaruga como JPEG

Question

Eu tenho um criador de imagens fractais.Ele cria uma árvore fractal aleatória.Quando terminar, solicita ao usuário que salve a árvore.Eu o salvei como .svg agora e funciona, MAS quero salvá-lo em um tipo de arquivo mais conveniente, como jpeg.Alguma ideia?Código:

import turtle
import random
from sys import exit
from time import clock
import canvasvg
turtle.colormode(255)
red = 125
green = 70
blue = 38        
pen = 10
def saveImg():
    print("Done.")
    save = input("Would you like to save this tree? Y/N \n")
    if save.upper() == "Y":
        t.hideturtle()
        name = input("What would you like to name it? \n")
        nameSav = name + ".svg"
        ts = turtle.getscreen().getcanvas()
        canvasvg.saveall(nameSav, ts)
    elif save.upper() == "N":
        def runChk():
            runAgain = input("Would you like to run again? Y/N (N will exit)")
            if runAgain.upper() == "Y":
                print("Running")
                main()
            elif runAgain.upper() == "N":
                print("Exiting...")
                exit()
            else:
                print("Invalid response.")
                runChk()
        runChk()
    else:
        print("Invalid response.")
        saveImg()

def tree(branchLen, t, red, green, blue, pen):
    time = str(round(clock()))
    print("Drawing... " + time)
    if branchLen > 3:
        pen = pen*0.8
        t.pensize(pen)
        if (red > 10 and green < 140):
            red = red - 15
            green = green + 8
    if branchLen > 5:
        angle = random.randrange(18, 55)
        angleTwo = 0.5*angle
        sub = random.randrange(1,16)
        t.color(red, green, blue)
        t.forward(branchLen)
        t.right(angleTwo)
        tree(branchLen-sub,t, red, green, blue, pen)
        t.left(angle)
        tree(branchLen-sub, t, red, green, blue, pen)
        t.right(angleTwo)
        t.backward(branchLen)

def main():
    t = turtle.Turtle()
    myWin = turtle.Screen()
    t.speed(0)
    t.hideturtle()
    t.left(90)
    t.up()
    t.backward(100)
    t.down()
    print("Please wait while I draw...")
    tree(random.randrange(60,95),t,red,green,blue, pen)
    saveImg()
main()

Answer 1

Tem que ser um JPEG?PNG seria suficiente?

Nesse caso, você pode converter de SVG para PNG usando cairosvg.Infelizmente canvasvg.saveall() só permite que você passe um nome de arquivo no qual ele gravará o SVG, então você precisará usar um arquivo temporário para o SVG e depois converter esse arquivo temporário para PNG usando cairosvg.svg2png().Então, algo assim deve funcionar:

import os
import shutil
import tempfile

import canvasvg

name = raw_input("What would you like to name it? \n")
nameSav = name + ".png"
tmpdir = tempfile.mkdtemp()  # create a temporary directory
tmpfile = os.path.join(tmpdir, 'tmp.svg')  # name of file to save SVG to
ts = turtle.getscreen().getcanvas()
canvasvg.saveall(tmpfile, ts)
with open(tmpfile) as svg_input, open(nameSav, 'wb') as png_output:
    cairosvg.svg2png(bytestring=svg_input.read(), write_to=png_output)
shutil.rmtree(tmpdir)  # clean up temp file(s)

Se você gostasse, você poderia escrever o seu próprio saveall() função baseada em canvasvg.saveall() (é bem pequeno) que aceita um objeto semelhante a um arquivo em vez de um nome de arquivo e grava nesse objeto.Então você pode passar em um StringIO objeto e não precisa se preocupar com o arquivo temporário.Ou o seu saveall() poderia simplesmente retornar os dados SVG como uma string de bytes.