caff-previewer-wrapper/caff_previewer_wrapper/converter.py
2020-11-27 05:44:32 +01:00

42 lines
1.7 KiB
Python

import subprocess
from flask import current_app
import werkzeug.exceptions
def run_abstract_converter(converter: str, source: str, destination: str) -> int:
"""
Just runs a binary and gives it two arguments
:param converter: the converter binary to run
:param source: source file
:param destination: destination file
:returns: exitcode of the converter
"""
completed_process = subprocess.run([converter, source, destination],
timeout=current_app.config['CONVERSION_TIMEOUT'], env={})
return completed_process.returncode
def convert_caff_to_tga(source: str, destination: str):
"""
This function uses caff_previewer to convert a CAFF file into a TGA file
:param source: path of the source TGA file (must exists)
:param destination: path of the destination TGA file (will be created)
"""
INTERNAL_ERROR_CODES = [0x01, 0x03, 0x04, 0x05, 0x32, 0x51]
ret = run_abstract_converter(current_app.config['CAFF_PREVIEWER_BINARY'], source, destination)
if ret in INTERNAL_ERROR_CODES:
raise RuntimeError(f"Caff Previewer returned an unexpected error code: {ret}")
elif ret != 0:
raise werkzeug.exceptions.BadRequest("CAFF format violation")
def convert_tga_to_png(source: str, destination: str):
"""
This function uses ImageMagick to convert a TGA file into a PNG file
:param source: path of the source TGA file (must exists)
:param destination: path of the destination TGA file (will be created)
"""
ret = run_abstract_converter(current_app.config['IMAGEMAGICK_CONVERT_BINARY'], source, destination)
if ret != 0:
raise RuntimeError(f"Image magick convert returned an unexpected error code: {ret}")