mirror of
https://github.com/freedomofpress/dangerzone.git
synced 2025-04-30 10:42:37 +02:00
bbd0d98f50
Fixes #251 by implementing the logic for archiving a document after conversion into a default sub-directory.
68 lines
2 KiB
Python
68 lines
2 KiB
Python
import json
|
|
import logging
|
|
import os
|
|
from typing import TYPE_CHECKING, Any, Dict, Optional
|
|
|
|
from .document import SAFE_EXTENSION
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
if TYPE_CHECKING:
|
|
from .logic import DangerzoneCore
|
|
|
|
|
|
class Settings:
|
|
settings: Dict[str, Any]
|
|
|
|
def __init__(self, dangerzone: "DangerzoneCore") -> None:
|
|
self.dangerzone = dangerzone
|
|
self.settings_filename = os.path.join(
|
|
self.dangerzone.appdata_path, "settings.json"
|
|
)
|
|
self.default_settings: Dict[str, Any] = {
|
|
"save": True,
|
|
"archive": True,
|
|
"ocr": True,
|
|
"ocr_language": "English",
|
|
"open": True,
|
|
"open_app": None,
|
|
"safe_extension": SAFE_EXTENSION,
|
|
}
|
|
|
|
self.load()
|
|
|
|
def get(self, key: str) -> Any:
|
|
return self.settings[key]
|
|
|
|
def set(self, key: str, val: Any) -> None:
|
|
self.settings[key] = val
|
|
|
|
def load(self) -> None:
|
|
if os.path.isfile(self.settings_filename):
|
|
self.settings = self.default_settings
|
|
|
|
# If the settings file exists, load it
|
|
try:
|
|
with open(self.settings_filename, "r") as settings_file:
|
|
self.settings = json.load(settings_file)
|
|
|
|
# If it's missing any fields, add them from the default settings
|
|
for key in self.default_settings:
|
|
if key not in self.settings:
|
|
self.settings[key] = self.default_settings[key]
|
|
|
|
except:
|
|
log.error("Error loading settings, falling back to default")
|
|
self.settings = self.default_settings
|
|
|
|
else:
|
|
# Save with default settings
|
|
log.info("Settings file doesn't exist, starting with default")
|
|
self.settings = self.default_settings
|
|
|
|
self.save()
|
|
|
|
def save(self) -> None:
|
|
os.makedirs(self.dangerzone.appdata_path, exist_ok=True)
|
|
with open(self.settings_filename, "w") as settings_file:
|
|
json.dump(self.settings, settings_file, indent=4)
|