45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
import os
|
|
import tempfile
|
|
from io import BytesIO
|
|
|
|
import defusedxml.ElementTree
|
|
from glymur import jp2box, Jp2k
|
|
|
|
# Replace glymur's ElementTree with a safe one
|
|
jp2box.ET = defusedxml.ElementTree
|
|
|
|
|
|
class BufferedJp2k(Jp2k):
|
|
"""
|
|
For manipulating JP2K from within a binary buffer.
|
|
|
|
For many operations glymur expects to be able to re-read from a file
|
|
based on filename, so this is the least brittle approach.
|
|
"""
|
|
|
|
def __init__(self, contents: bytes):
|
|
stream = BytesIO(contents)
|
|
self.temp_file = tempfile.NamedTemporaryFile(delete=False)
|
|
stream.seek(0)
|
|
self.temp_file.write(stream.read())
|
|
# Windows NT can't handle two FHs open on a tempfile at once
|
|
self.temp_file.close()
|
|
|
|
super().__init__(self.temp_file.name)
|
|
|
|
def __del__(self):
|
|
if self.temp_file is not None:
|
|
os.remove(self.temp_file.name)
|
|
self.temp_file = None
|
|
|
|
def _populate_cparams(self, img_array):
|
|
if self._cratios is None:
|
|
self._cratios = (1920.0, 480.0, 120.0, 30.0, 10.0)
|
|
if self._irreversible is None:
|
|
self.irreversible = True
|
|
return super()._populate_cparams(img_array)
|
|
|
|
def __bytes__(self):
|
|
with open(self.temp_file.name, "rb") as f:
|
|
return f.read()
|