Rozdział 15 · Python i akta

Bezpieczne zarządzanie plikami

Zapisywanie do pliku to działanie z konsekwencjami. Rozsądna ostrożność jest warta kilku dodatkowych linijek kodu.

Przed nagraniem – zadaj sobie pytanie

1. Czy chcę zastąpić treść, czy nadpisać?
2. Albo utworzyć tylko, jeśli plik jeszcze nie istnieje?
3. Czy potrzebuję zapasowego przed zmianą?
4. Czy to dokładnie ta ścieżka, którą mam na myśli?"

Praktyka w tym kursie są wyłącznie w piaskownicach

Wyjście z twojego folderu jest path traversal

imya_fajla = input(„Nazwa pliku do zapisu: ”)
# jeśli użytkownik wpisze „.. /.. /important.txt”
path = Path("data") / imya_fajla

Bardziej profesjonalne: Bezpieczne oszczędzanie

Pomysł na zmniejszenie ryzyka pozostawienia głównego pliku w uszkodzonym stanie pośrednim: Najpierw zapisz w tymczasowy plik w tym samym folderze, a następnie zastąp główny plik w całości:

import json
from pathlib import Path

def bezopasno_sohranit(path: Path, data: dict) -> None:
    vremenny = path.with_suffix(path.suffix + ".tmp")
    with vremenny.open("w", encoding="utf-8") as f:
        json.dump(data, f, ensure_ascii=False, indent=2)
    vremenny.replace(path)   # zamień główny plik tylko wtedy, gdy wszystko jest już zapisane

Kopia zapasowa przed wymianą

import shutil
from pathlib import Path

path = Path("save.json")
if path.exists():
    shutil.copy(path, path.with_suffix(".json.bak"))