Randall Lawrence
39 lines
1.5 KiB
Python
39 lines
1.5 KiB
Python
# Generated by Netbeans
|
|
# Author: Eramde
|
|
# Date: 09.2019
|
|
import tempfile
|
|
import logging
|
|
import gzip
|
|
import subprocess
|
|
import os
|
|
from io import BytesIO
|
|
from typing import Optional, Tuple
|
|
|
|
|
|
LOG: logging.Logger = logging.getLogger("mau.util.tgs")
|
|
|
|
TYPE_TO_MIME = {"image": "image/png", "gif": "image/gif", "video": "video/mp4"}
|
|
TYPE_TO_FORMAT = {"image": ".png", "gif": ".gif", "video": ".mp4"}
|
|
|
|
|
|
def convert_tgs_to(file: bytes, convert_to: str, width: int = 200, height: int = 200) \
|
|
-> Tuple[str, bytes, Optional[int], Optional[int]]:
|
|
if convert_to in TYPE_TO_FORMAT:
|
|
file_ext = TYPE_TO_FORMAT[convert_to]
|
|
mime = TYPE_TO_MIME[convert_to]
|
|
lottie = gzip.open(BytesIO(file))
|
|
with tempfile.NamedTemporaryFile(mode="w+b", suffix=".json") as json_out:
|
|
with tempfile.NamedTemporaryFile(mode="r+b", suffix=file_ext) as tmp:
|
|
tmp_output_file = tmp.name
|
|
json_out.write(lottie.read())
|
|
json_out.flush()
|
|
subprocess.run(["puppeteer-lottie", "-q", "-i", json_out.name, "-o", tmp_output_file,
|
|
"-w", str(width), "-h", str(height)], capture_output=True)
|
|
with open(tmp_output_file, mode="r+b") as out_file:
|
|
out = out_file.read()
|
|
os.remove(tmp_output_file)
|
|
return mime, out, width, height
|
|
else:
|
|
LOG.warning(f"Unable to convert animated sticker, type {convert_to} not supported")
|
|
return "application/gzip", file, None, None
|