hcornet
506716e703
Some checks failed
Deployment Verification / deploy-and-test (push) Failing after 29s
65 lines
2.6 KiB
Python
65 lines
2.6 KiB
Python
#!/usr/bin/env python3
|
|
#
|
|
# IRIS Source Code
|
|
# Copyright (C) 2021 - Airbus CyberSecurity (SAS)
|
|
# contact@dfir-iris.org
|
|
# Created by Lukas Zurschmiede @LukyLuke
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU Lesser General Public
|
|
# License as published by the Free Software Foundation; either
|
|
# version 3 of the License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
# Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public License
|
|
# along with this program; if not, write to the Free Software Foundation,
|
|
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
import logging
|
|
import os
|
|
import shutil
|
|
import uuid
|
|
import re
|
|
|
|
from pathlib import Path
|
|
from docxtpl import DocxTemplate
|
|
|
|
from docx_generator.globals.picture_globals import PictureGlobals
|
|
|
|
from app.datamgmt.datastore.datastore_db import datastore_get_local_file_path
|
|
|
|
|
|
class ImageHandler(PictureGlobals):
|
|
def __init__(self, template: DocxTemplate, base_path: str):
|
|
self._logger = logging.getLogger(__name__)
|
|
PictureGlobals.__init__(self, template, base_path)
|
|
|
|
def _process_remote(self, image_path: str) -> str:
|
|
"""
|
|
Checks if the given Link is a datastore-link and if so, save the image locally for further processing.
|
|
:
|
|
A Datastore Links looks like this: https://localhost:4433/datastore/file/view/2?cid=1
|
|
"""
|
|
res = re.search(r'datastore\/file\/view\/(\d+)\?cid=(\d+)', image_path)
|
|
if not res:
|
|
return super()._process_remote(image_path)
|
|
|
|
if image_path[:4] == 'http' and len(res.groups()) == 2:
|
|
file_id = res.groups(0)[0]
|
|
case_id = res.groups(0)[1]
|
|
has_error, dsf = datastore_get_local_file_path(file_id, case_id)
|
|
|
|
if has_error:
|
|
raise RenderingError(self._logger, f'File-ID {file_id} does not exist in Case {case_id}')
|
|
if not Path(dsf.file_local_name).is_file():
|
|
raise RenderingError(self._logger, f'File {dsf.file_local_name} does not exists on the server. Update or delete virtual entry')
|
|
|
|
file_ext = os.path.splitext(dsf.file_original_name)[1]
|
|
file_name = os.path.join(self._output_path, str(uuid.uuid4())) + file_ext
|
|
return_value = shutil.copy(dsf.file_local_name, file_name)
|
|
return return_value
|
|
return super()._process_remote(image_path)
|