42 lines
1.7 KiB
Python
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}")
|