almet/dangerzone
1
0
Fork 0
mirror of https://github.com/freedomofpress/dangerzone.git synced 2025-04-28 18:02:38 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 11

isolation_provider: Pass doc when creating doc-to-pixels proc

Browse source 
Pass the Document instance that will be converted to the
`IsolationProvider.start_doc_to_pixels_proc()` method. Concrete classes
can then associate this name with the started process, so that they can
later on kill it.
This commit is contained in:
Alex Pyrgiotis 2024-04-08 19:47:45 +03:00
parent b920de36d1
commit 6850d31edc
No known key found for this signature in database
GPG key ID: B6C15EBA0357C9AA
5 changed files with 13 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
dangerzone/isolation_provider/base.py
View file

@ -69,7 +69,7 @@ class IsolationProvider(ABC):
self.progress_callback = progress_callback
document.mark_as_converting()
try:
conversion_proc = self.start_doc_to_pixels_proc()
conversion_proc = self.start_doc_to_pixels_proc(document)
with tempfile.TemporaryDirectory() as t:
Path(f"{t}/pixels").mkdir()
self.doc_to_pixels(document, t, conversion_proc)
@ -192,7 +192,7 @@ class IsolationProvider(ABC):
return armor_start + conversion_string + armor_end
@abstractmethod
def start_doc_to_pixels_proc(self) -> subprocess.Popen:
def start_doc_to_pixels_proc(self, document: Document) -> subprocess.Popen:
pass

2
dangerzone/isolation_provider/container.py
View file

@ -251,7 +251,7 @@ class Container(IsolationProvider):
)
shutil.move(container_output_filename, document.output_filename)
def start_doc_to_pixels_proc(self) -> subprocess.Popen:
def start_doc_to_pixels_proc(self, document: Document) -> subprocess.Popen:
# Convert document to pixels
command = [
"/usr/bin/python3",

2
dangerzone/isolation_provider/dummy.py
View file

@ -74,7 +74,7 @@ class Dummy(IsolationProvider):
) -> None:
pass
def start_doc_to_pixels_proc(self) -> subprocess.Popen:
def start_doc_to_pixels_proc(self, document: Document) -> subprocess.Popen:
return subprocess.Popen("True")
def get_max_parallel_conversions(self) -> int:

2
dangerzone/isolation_provider/qubes.py
View file

@ -51,7 +51,7 @@ class Qubes(IsolationProvider):
def get_max_parallel_conversions(self) -> int:
return 1
def start_doc_to_pixels_proc(self) -> subprocess.Popen:
def start_doc_to_pixels_proc(self, document: Document) -> subprocess.Popen:
dev_mode = getattr(sys, "dangerzone_dev", False) == True
if dev_mode:
# Use dz.ConvertDev RPC call instead, if we are in development mode.

14
tests/isolation_provider/base.py
View file

@ -37,7 +37,7 @@ class IsolationProviderTest:
provider.progress_callback = mocker.MagicMock()
doc = Document(pdf_11k_pages)
p = provider.start_doc_to_pixels_proc()
p = provider.start_doc_to_pixels_proc(doc)
with pytest.raises(errors.ConverterProcException):
provider.doc_to_pixels(doc, tmpdir, p)
assert provider.get_proc_exception(p) == errors.MaxPagesException
@ -54,7 +54,7 @@ class IsolationProviderTest:
"dangerzone.conversion.errors.MAX_PAGES", 1
) # sample_doc has 4 pages > 1
doc = Document(sample_doc)
p = provider.start_doc_to_pixels_proc()
p = provider.start_doc_to_pixels_proc(doc)
with pytest.raises(errors.MaxPagesException):
provider.doc_to_pixels(doc, tmpdir, p)
@ -67,10 +67,12 @@ class IsolationProviderTest:
tmpdir: str,
) -> None:
provider.progress_callback = mocker.MagicMock()
p = provider.start_doc_to_pixels_proc()
doc = Document(sample_bad_width)
p = provider.start_doc_to_pixels_proc(doc)
with pytest.raises(errors.MaxPageWidthException):
provider.doc_to_pixels(Document(sample_bad_width), tmpdir, p)
provider.doc_to_pixels(doc, tmpdir, p)
p = provider.start_doc_to_pixels_proc()
doc = Document(sample_bad_height)
p = provider.start_doc_to_pixels_proc(doc)
with pytest.raises(errors.MaxPageHeightException):
provider.doc_to_pixels(Document(sample_bad_height), tmpdir, p)
provider.doc_to_pixels(doc, tmpdir, p)