mirror of
https://github.com/freedomofpress/dangerzone.git
synced 2025-05-01 19:22:23 +02:00
bf38c24d99
stdout_callback is used to flow progress information from the conversion to some front-end. It was always used in tandem with printing to the terminal (which is kind of a front-end). So it made sense to put them always together.
72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
import logging
|
|
import os
|
|
import shutil
|
|
import sys
|
|
import time
|
|
from typing import Callable, Optional
|
|
|
|
from ..document import Document
|
|
from ..util import get_resource_path
|
|
from .base import IsolationProvider
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class Dummy(IsolationProvider):
|
|
"""Dummy Isolation Provider (FOR TESTING ONLY)
|
|
|
|
"Do-nothing" converter - the sanitized files are the same as the input files.
|
|
Useful for testing without the need to use docker.
|
|
"""
|
|
|
|
def __init__(self) -> None:
|
|
# Sanity check
|
|
if not getattr(sys, "dangerzone_dev", False):
|
|
raise Exception(
|
|
"Dummy isolation provider is UNSAFE and should never be "
|
|
+ "called in a non-testing system."
|
|
)
|
|
|
|
def install(self) -> bool:
|
|
return True
|
|
|
|
def _convert(
|
|
self,
|
|
document: Document,
|
|
ocr_lang: Optional[str],
|
|
) -> bool:
|
|
log.debug("Dummy converter started:")
|
|
log.debug(
|
|
f" - document: {os.path.basename(document.input_filename)} ({document.id})"
|
|
)
|
|
log.debug(f" - ocr : {ocr_lang}")
|
|
log.debug("\n(simulating conversion)")
|
|
|
|
success = True
|
|
|
|
progress = [
|
|
[False, "Converting to PDF using GraphicsMagick", 0.0],
|
|
[False, "Separating document into pages", 3.0],
|
|
[False, "Converting page 1/1 to pixels", 5.0],
|
|
[False, "Converted document to pixels", 50.0],
|
|
[False, "Converting page 1/1 from pixels to PDF", 50.0],
|
|
[False, "Merging 1 pages into a single PDF", 95.0],
|
|
[False, "Compressing PDF", 97.0],
|
|
[False, "Safe PDF created", 100.0],
|
|
]
|
|
|
|
for error, text, percentage in progress:
|
|
self.print_progress(document, error, text, percentage) # type: ignore [arg-type]
|
|
if error:
|
|
success = False
|
|
time.sleep(0.2)
|
|
|
|
if success:
|
|
shutil.copy(
|
|
get_resource_path("dummy_document.pdf"), document.output_filename
|
|
)
|
|
|
|
return success
|
|
|
|
def get_max_parallel_conversions(self) -> int:
|
|
return 1
|