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

Ensure archive directory can be created

Browse source 
Verifies that the archive directory can be created as soon as the
document is set to be archived.
This commit is contained in:
deeplow 2022-11-22 18:47:50 +00:00
parent bbd0d98f50
commit f54446f2fd
No known key found for this signature in database
GPG key ID: 577982871529A52A
2 changed files with 15 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
dangerzone/document.py
View file

@ -77,6 +77,11 @@ class Document:
# in unwriteable directory # in unwriteable directory
raise errors.UnwriteableOutputDirException() raise errors.UnwriteableOutputDirException()
def validate_default_archive_dir(self) -> None:
"""Checks if archive dir can be created"""
if not os.access(self.default_archive_dir.parent, os.W_OK):
raise errors.UnwriteableArchiveDirException()
@property @property
def input_filename(self) -> str: def input_filename(self) -> str:
if self._input_filename is None: if self._input_filename is None:
@ -125,6 +130,7 @@ class Document:
@archive_after_conversion.setter @archive_after_conversion.setter
def archive_after_conversion(self, enabled: bool) -> None: def archive_after_conversion(self, enabled: bool) -> None:
if enabled: if enabled:
self.validate_default_archive_dir()
self._archive = True self._archive = True
else: else:
self._archive = False self._archive = False

9
dangerzone/errors.py
View file

@ -71,6 +71,15 @@ class OutputDirIsNotDirException(DocumentFilenameException):
super().__init__("Specified output directory is actually not a directory") super().__init__("Specified output directory is actually not a directory")
class UnwriteableArchiveDirException(DocumentFilenameException):
"""Exception for when the archive directory cannot be created."""
def __init__(self) -> None:
super().__init__(
"Archive directory for storing unsafe documents cannot be created."
)
class SuffixNotApplicableException(DocumentFilenameException): class SuffixNotApplicableException(DocumentFilenameException):
"""Exception for when the suffix cannot be applied to the output filename.""" """Exception for when the suffix cannot be applied to the output filename."""