mirror of
https://github.com/mealie-recipes/mealie.git
synced 2025-11-14 16:02:28 -05:00
84c23765cd
* fix strict optional errors * fix typing in repository * fix backup db files location * update workspace settings
61 lines
1.4 KiB
Python
61 lines
1.4 KiB
Python
import json
|
|
import shutil
|
|
import tempfile
|
|
from pathlib import Path
|
|
|
|
|
|
class BackupContents:
|
|
_tables: dict | None = None
|
|
|
|
def __init__(self, file: Path) -> None:
|
|
self.base = file
|
|
self.data_directory = self.base / "data"
|
|
self.tables = self.base / "database.json"
|
|
|
|
def validate(self) -> bool:
|
|
if not self.base.is_dir():
|
|
return False
|
|
|
|
if not self.data_directory.is_dir():
|
|
return False
|
|
|
|
if not self.tables.is_file():
|
|
return False
|
|
|
|
return True
|
|
|
|
def schema_version(self) -> str:
|
|
tables = self.read_tables()
|
|
|
|
alembic_version = tables.get("alembic_version", [])
|
|
|
|
if not alembic_version:
|
|
return ""
|
|
|
|
return alembic_version[0].get("version_num", "")
|
|
|
|
def read_tables(self) -> dict:
|
|
if self._tables is None:
|
|
with open(self.tables) as f:
|
|
self._tables = json.load(f)
|
|
|
|
return self._tables
|
|
|
|
|
|
class BackupFile:
|
|
temp_dir: Path | None
|
|
|
|
def __init__(self, file: Path) -> None:
|
|
self.zip = file
|
|
|
|
def __enter__(self) -> BackupContents:
|
|
self.temp_dir = Path(tempfile.mkdtemp())
|
|
shutil.unpack_archive(str(self.zip), str(self.temp_dir))
|
|
return BackupContents(self.temp_dir)
|
|
|
|
def __exit__(self, exc_type, exc_val, exc_tb):
|
|
if self.temp_dir and self.temp_dir.is_dir():
|
|
shutil.rmtree(self.temp_dir)
|
|
|
|
self.temp_dir = None
|