mirror of
https://github.com/mealie-recipes/mealie.git
synced 2025-12-18 00:05:12 -05:00
Feature/database backups (#1040)
* add annotations to docs * alchemy data dumper * initial tests * sourcery refactor * db backups/restore * potential postgres fix * potential postgres fix * this is terrible * potential pg fix * cleanup * remove unused import * fix comparison * generate frontend types * update timestamp and add directory filter * rewrite to new admin-api * update backup routers * add file_token response helper * update imports * remove test_backup
This commit is contained in:
95
mealie/routes/admin/admin_backups.py
Normal file
95
mealie/routes/admin/admin_backups.py
Normal file
@@ -0,0 +1,95 @@
|
||||
import operator
|
||||
import shutil
|
||||
from pathlib import Path
|
||||
|
||||
from fastapi import APIRouter, File, HTTPException, UploadFile, status
|
||||
|
||||
from mealie.core.config import get_app_dirs
|
||||
from mealie.core.security import create_file_token
|
||||
from mealie.pkgs.stats.fs_stats import pretty_size
|
||||
from mealie.routes._base import BaseAdminController, controller
|
||||
from mealie.schema.admin.backup import AllBackups, BackupFile
|
||||
from mealie.schema.response.responses import FileTokenResponse, SuccessResponse
|
||||
from mealie.services.backups_v2.backup_v2 import BackupV2
|
||||
|
||||
router = APIRouter(prefix="/backups")
|
||||
|
||||
|
||||
@controller(router)
|
||||
class AdminBackupController(BaseAdminController):
|
||||
def _backup_path(self, name) -> Path:
|
||||
return get_app_dirs().BACKUP_DIR / name
|
||||
|
||||
@router.get("", response_model=AllBackups)
|
||||
def get_all(self):
|
||||
app_dirs = get_app_dirs()
|
||||
imports = []
|
||||
for archive in app_dirs.BACKUP_DIR.glob("*.zip"):
|
||||
backup = BackupFile(
|
||||
name=archive.name, date=archive.stat().st_ctime, size=pretty_size(archive.stat().st_size)
|
||||
)
|
||||
imports.append(backup)
|
||||
|
||||
templates = [template.name for template in app_dirs.TEMPLATE_DIR.glob("*.*")]
|
||||
imports.sort(key=operator.attrgetter("date"), reverse=True)
|
||||
|
||||
return AllBackups(imports=imports, templates=templates)
|
||||
|
||||
@router.post("", status_code=status.HTTP_201_CREATED, response_model=SuccessResponse)
|
||||
def create_one(self):
|
||||
backup = BackupV2()
|
||||
|
||||
try:
|
||||
backup.backup()
|
||||
except Exception as e:
|
||||
raise HTTPException(status.HTTP_500_INTERNAL_SERVER_ERROR) from e
|
||||
|
||||
return SuccessResponse.respond("Backup created successfully")
|
||||
|
||||
@router.get("/{file_name}", response_model=FileTokenResponse)
|
||||
def get_one(self, file_name: str):
|
||||
"""Returns a token to download a file"""
|
||||
file = self._backup_path(file_name)
|
||||
|
||||
if not file.exists():
|
||||
raise HTTPException(status.HTTP_404_NOT_FOUND)
|
||||
|
||||
return FileTokenResponse.respond(create_file_token(file))
|
||||
|
||||
@router.delete("/{file_name}", status_code=status.HTTP_200_OK, response_model=SuccessResponse)
|
||||
def delete_one(self, file_name: str):
|
||||
file = self._backup_path(file_name)
|
||||
|
||||
if not file.is_file():
|
||||
raise HTTPException(status.HTTP_400_BAD_REQUEST)
|
||||
try:
|
||||
file.unlink()
|
||||
except Exception as e:
|
||||
raise HTTPException(status.HTTP_500_INTERNAL_SERVER_ERROR) from e
|
||||
|
||||
return SuccessResponse.respond(f"{file_name} has been deleted.")
|
||||
|
||||
@router.post("/upload", response_model=SuccessResponse)
|
||||
def upload_one(self, archive: UploadFile = File(...)):
|
||||
"""Upload a .zip File to later be imported into Mealie"""
|
||||
app_dirs = get_app_dirs()
|
||||
dest = app_dirs.BACKUP_DIR.joinpath(archive.filename)
|
||||
|
||||
with dest.open("wb") as buffer:
|
||||
shutil.copyfileobj(archive.file, buffer)
|
||||
|
||||
if not dest.is_file:
|
||||
raise HTTPException(status.HTTP_400_BAD_REQUEST)
|
||||
|
||||
@router.post("/{file_name}/restore", response_model=SuccessResponse)
|
||||
def import_one(self, file_name: str):
|
||||
backup = BackupV2()
|
||||
|
||||
file = self._backup_path(file_name)
|
||||
|
||||
try:
|
||||
backup.restore(file)
|
||||
except Exception as e:
|
||||
raise HTTPException(status.HTTP_500_INTERNAL_SERVER_ERROR) from e
|
||||
|
||||
return SuccessResponse.respond("Restore successful")
|
||||
Reference in New Issue
Block a user