mirror of
https://github.com/freedomofpress/dangerzone.git
synced 2025-04-28 18:02:38 +02:00
4d3f2b32c7
This reverts commit 344d6f7bfa.
Stopwatch is no longer needed now that we're removing timeouts.
74 lines
2.5 KiB
Python
74 lines
2.5 KiB
Python
import pathlib
|
|
import platform
|
|
import string
|
|
import subprocess
|
|
import sys
|
|
from typing import Optional
|
|
|
|
import appdirs
|
|
|
|
|
|
def get_config_dir() -> str:
|
|
return appdirs.user_config_dir("dangerzone")
|
|
|
|
|
|
def get_tmp_dir() -> Optional[str]:
|
|
"""Get the parent dir for the Dangerzone temporary dirs.
|
|
|
|
This function returns the parent directory where Dangerzone will store its temporary
|
|
directories. The default behavior is to let Python choose for us (e.g., in `/tmp`
|
|
for Linux), which is why we return None. However, we still need to define this
|
|
function in order to be able to set this dir via mocking in our tests.
|
|
"""
|
|
return None
|
|
|
|
|
|
def get_resource_path(filename: str) -> str:
|
|
if getattr(sys, "dangerzone_dev", False):
|
|
# Look for resources directory relative to python file
|
|
project_root = pathlib.Path(__file__).parent.parent
|
|
prefix = project_root.joinpath("share")
|
|
else:
|
|
if platform.system() == "Darwin":
|
|
bin_path = pathlib.Path(sys.executable)
|
|
app_path = bin_path.parent.parent
|
|
prefix = app_path.joinpath("Resources", "share")
|
|
elif platform.system() == "Linux":
|
|
prefix = pathlib.Path(sys.prefix).joinpath("share", "dangerzone")
|
|
elif platform.system() == "Windows":
|
|
exe_path = pathlib.Path(sys.executable)
|
|
dz_install_path = exe_path.parent
|
|
prefix = dz_install_path.joinpath("share")
|
|
else:
|
|
raise NotImplementedError(f"Unsupported system {platform.system()}")
|
|
resource_path = prefix.joinpath(filename)
|
|
return str(resource_path)
|
|
|
|
|
|
def get_version() -> str:
|
|
try:
|
|
with open(get_resource_path("version.txt")) as f:
|
|
version = f.read().strip()
|
|
except FileNotFoundError:
|
|
# In dev mode, in Windows, get_resource_path doesn't work properly for the container, but luckily
|
|
# it doesn't need to know the version
|
|
version = "unknown"
|
|
return version
|
|
|
|
|
|
def get_subprocess_startupinfo(): # type: ignore [no-untyped-def]
|
|
if platform.system() == "Windows":
|
|
startupinfo = subprocess.STARTUPINFO()
|
|
startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
|
|
return startupinfo
|
|
else:
|
|
return None
|
|
|
|
|
|
def replace_control_chars(untrusted_str: str) -> str:
|
|
"""Remove control characters from string. Protects a terminal emulator
|
|
from obcure control characters"""
|
|
sanitized_str = ""
|
|
for char in untrusted_str:
|
|
sanitized_str += char if char in string.printable else "_"
|
|
return sanitized_str
|