You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
69 lines
2.7 KiB
Python
69 lines
2.7 KiB
Python
1 week ago
|
from django.core.management.base import BaseCommand, CommandError
|
||
|
|
from call.models import Creature
|
||
|
|
import weasyprint as wsp
|
||
|
|
import subprocess
|
||
|
|
from django.urls import reverse
|
||
|
|
from urllib.parse import urljoin, urlencode
|
||
|
|
from PIL import Image
|
||
|
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
class Command(BaseCommand):
|
||
|
|
help = "Generates an animation"
|
||
|
|
|
||
|
|
def add_arguments(self, parser):
|
||
|
|
parser.add_argument("--baseurl", default="http://localhost:8000/")
|
||
|
|
parser.add_argument("--output", default="animation_{width}x{height}.gif")
|
||
|
|
parser.add_argument("--width", type=int, default=500)
|
||
|
|
parser.add_argument("--height", type=int, default=500)
|
||
|
|
parser.add_argument("--scale", type=int, default=6)
|
||
|
|
|
||
|
|
def handle(self, *args, **options):
|
||
|
|
baseurl = options['baseurl']
|
||
|
|
width = options['width']
|
||
|
|
height = options['height']
|
||
|
|
scale = options['scale']
|
||
|
|
output_filename = options['output'].format(**{'width': width, 'height': height})
|
||
|
|
# url = "http://localhost:8000/creatures/39/"
|
||
|
|
# w = wsp.HTML(url=url)
|
||
|
|
# w.write_pdf("test.pdf")
|
||
|
|
#
|
||
|
|
# frames = []
|
||
|
|
frameno = 0
|
||
|
|
convert_args = ["convert", "-delay", "25"]
|
||
|
|
qvars = {'width': width, 'height': height, 'scale': scale}
|
||
|
|
for creature in Creature.objects.filter(published=True).order_by("-created"):
|
||
|
|
url = urljoin(baseurl, reverse("creature_detail", args=[creature.id,]))
|
||
|
|
print (creature, url)
|
||
|
|
qvars['cell'] = 0
|
||
|
|
w = wsp.HTML(url=f"{url}?{urlencode(qvars)}")
|
||
|
|
w.write_pdf("temp.pdf")
|
||
|
|
subprocess.run(["convert", "-density", "96", "temp.pdf", "temp.png"])
|
||
|
|
# frames.append(Image.open("temp.png"))
|
||
|
|
fn1 = f"frame{frameno:04}.png"
|
||
|
|
os.rename("temp.png", fn1)
|
||
|
|
frameno += 1
|
||
|
|
convert_args.append(fn1)
|
||
|
|
|
||
|
|
qvars['cell'] = 3
|
||
|
|
w = wsp.HTML(url=f"{url}?{urlencode(qvars)}")
|
||
|
|
w.write_pdf("temp.pdf")
|
||
|
|
subprocess.run(["convert", "-density", "96", "temp.pdf", "temp.png"])
|
||
|
|
# frames.append(Image.open("temp.png"))
|
||
|
|
fn2 = f"frame{frameno:04}.png"
|
||
|
|
os.rename("temp.png", fn2)
|
||
|
|
frameno += 1
|
||
|
|
convert_args.append(fn2)
|
||
|
|
for i in range(7):
|
||
|
|
convert_args.append(fn1)
|
||
|
|
convert_args.append(fn2)
|
||
|
|
|
||
|
|
# convert -delay 100 "frame????.png" "animation.imc.gif"
|
||
|
|
convert_args.append(output_filename)
|
||
|
|
subprocess.run(convert_args)
|
||
|
|
subprocess.run("rm temp.pdf frame*.png", shell=True)
|
||
|
|
|
||
|
|
# # self.stdout.write(
|
||
|
|
# # self.style.SUCCESS('Successfully closed poll "%s"' % poll_id)
|
||
|
|
# # )
|