def preprocessImage(task):
task_tag = "IMAGE PREPROCESSING"
print "\n\n************** %s [START] ******************\n" % task_tag
print "image preprocessing at %s" % task.doc_id
task.setStatus(302)
from lib.Worker.Models.ic_image import InformaCamImage
from conf import DEBUG
from vars import ASSET_TAGS
image = InformaCamImage(_id=task.doc_id)
if image is None:
print "DOC IS NONE"
print "\n\n************** %s [ERROR] ******************\n" % task_tag
task.fail()
return
import os
from conf import getConfig, ANNEX_DIR
try:
J3M_DIR = getConfig('jpeg_tools_dir')
except Exception as e:
if DEBUG: print "NO J3M DIR! %s" % e
print "\n\n************** %s [ERROR] ******************\n" % task_tag
task.fail()
return
import re
from subprocess import Popen, PIPE
from cStringIO import StringIO
from vars import UPLOAD_RESTRICTION
from lib.Worker.Models.uv_task import UnveillanceTask
tiff_txt = StringIO()
obscura_marker_found = False
was_encrypted = False
ic_j3m_txt = None
cmd = [os.path.join(J3M_DIR, "j3mparser.out"),
os.path.join(ANNEX_DIR, image.file_name)]
p = Popen(cmd, stdout=PIPE, close_fds=True)
data = p.stdout.readline()
while data:
data = data.strip()
if re.match(r'^file: .*', data): pass
elif re.match(r'^Generic APPn .*', data): pass
elif re.match(r'^Component.*', data): pass
elif re.match(r'^Didn\'t find .*', data): pass
elif re.match(r'^Got obscura marker.*', data):
if DEBUG:
print "\n\nWE ALSO HAVE J3M DATA\n\n"
obscura_marker_found = True
ic_j3m_txt = StringIO()
else:
if obscura_marker_found: ic_j3m_txt.write(data)
else: tiff_txt.write(data)
data = p.stdout.readline()
p.stdout.close()
image.addAsset(tiff_txt.getvalue(), "file_metadata.txt",
description="tiff metadata as per jpeg redaction library")
tiff_txt.close()
del tiff_txt
if ic_j3m_txt is not None:
from lib.Worker.Utils.funcs import getFileType
from vars import MIME_TYPES, MIME_TYPE_MAP
ic_j3m_txt = ic_j3m_txt.getvalue()
ic_j3m_txt_mime_type = getFileType(ic_j3m_txt, as_buffer=True)
inflate = {}
if ic_j3m_txt_mime_type != MIME_TYPES['json']:
from lib.Core.Utils.funcs import b64decode
un_b64 = b64decode(ic_j3m_txt)
if un_b64 is not None:
un_b64_mime_type = getFileType(un_b64, as_buffer=True)
if un_b64_mime_type in [MIME_TYPES['pgp'], MIME_TYPES['gzip']]:
if DEBUG:
print "MIME TYPE: %s" % un_b64_mime_type
asset_path = "j3m_raw.%s" % MIME_TYPE_MAP[un_b64_mime_type]
image.addAsset(un_b64, asset_path)
if DEBUG:
print "\n\nPGP KEY FILE PATH: %s\n\n" % asset_path
gz = image.addAsset(None, "j3m_raw.gz", tags=[ASSET_TAGS['OB_M']],
description="j3m data extracted from obscura marker")
if un_b64_mime_type == MIME_TYPES['pgp']:
task.put_next([
"PGP.decrypt.decrypt",
"J3M.j3mify.parse_zipped_j3m"
])
inflate.update({
'pgp_file' : os.path.join(image.base_path, asset_path),
'save_as' : gz
})
was_encrypted = True
elif un_b64_mime_type == MIME_TYPES['gzip']:
task.put_next("J3M.j3mify.parse_zipped_j3m")
else:
asset_path = image.addAsset(ic_j3m_txt, "j3m_raw.json", as_literal=False)
task.put_next([
"J3M.j3mify.j3mify",
"J3M.massage_j3m.massageJ3M",
"PGP.verify_signature.verifySignature",
"J3M.verify_visual_content.verifyVisualContent"
])
inflate.update({'j3m_name' : "j3m_raw.json"})
try:
upload_restriction = image.getFileMetadata('uv_restriction')
except Exception as e:
print "could not get metadata for uv_restriction"
print e
else:
print "NO IC J3M TEXT FOUND???"
print "\n\n************** %s [WARN] ******************\n" % task_tag
upload_restriction = UPLOAD_RESTRICTION['for_local_use_only']
if upload_restriction is None or upload_restriction == UPLOAD_RESTRICTION['no_restriction']:
task.put_next("Image.make_derivatives.makeDerivatives")
task.put_next("Image.get_vector.get_vector")
image.addCompletedTask(task.task_path)
task.routeNext(inflate=inflate)
task.finish()
print "\n\n************** %s [END] ******************\n" % task_tag
def makeDerivatives(uv_task):
task_tag = "DERIVATIVES: IMAGE"
print "\n\n************** %s [START] ******************\n" % task_tag
print "making derivatives for %s" % uv_task.doc_id
uv_task.setStatus(302)
from lib.Worker.Models.ic_image import InformaCamImage
from conf import DEBUG
from vars import ASSET_TAGS
image = InformaCamImage(_id=uv_task.doc_id)
if image is None:
print "DOC IS NONE"
print "\n\n************** %s [ERROR] ******************\n" % task_tag
uv_task.fail()
return
import os, Image
from conf import ANNEX_DIR
resolutions = {
'high' : None,
'thumb' : [0.15, 0.15],
'med' : [0.75, 0.75],
'low' : [0.5, 0.5]
}
image.getFile(image.file_name)
for label, res in resolutions.iteritems():
asset_path = image.addAsset(None, "%s_%s.jpg" % (label, image.file_name),
tags=[ASSET_TAGS['M_DERIV'], ASSET_TAGS[label.upper()]],
description="derivative of image in %s resolution" % label)
if asset_path is None:
print "COULD NOT INIT THIS ASSET"
print "\n\n************** %s [ERROR] ******************\n" % task_tag
uv_task.fail()
return
image_obj = Image.open(os.path.join(ANNEX_DIR, image.file_name))
if res is not None:
resolution_tuple = tuple([int(x*y) for x, y in zip(res, list(image_obj.size))])
if DEBUG:
print "NEW DOWNSIZE RESOLUTION:"
print resolution_tuple
image_obj = image_obj.resize(resolution_tuple)
try:
image_obj.save(os.path.join(ANNEX_DIR, asset_path), "JPEG")
image.addFile(asset_path, None, sync=True)
except Exception as e:
print "\n\n************** %s [INFO] ******************\n" % task_tag
print "COULD NOT MAKE %s DERIVATIVE" % label
print e, type(e)
image.addCompletedTask(uv_task.task_path)
uv_task.routeNext()
uv_task.finish()
print "\n\n************** %s [END] ******************\n" % task_tag