Глава 15 · Python и файлы

Безопасная работа с файлами

Запись в файл — действие с последствиями. Разумная осторожность стоит нескольких лишних строк кода.

Перед записью — спросите себя

1. Я хочу заменить содержимое или дозаписать?
2. Или создать только если файла ещё нет?
3. Нужна ли резервная копия перед изменением?
4. Это точно тот путь, который я имею в виду?

Практики этого курса — только в песочнице

Путь наружу из своей папки — path traversal

imya_fajla = input("Имя файла для сохранения: ")
# если пользователь введёт "../../important.txt" — куда это укажет?
path = Path("data") / imya_fajla

Профессиональнее: безопасное сохранение

Идея, снижающая риск оставить основной файл в повреждённом промежуточном состоянии: сначала записать во временный файл в той же папке, а затем заменить основной файл целиком:

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)   # заменяем основной файл только когда всё уже записано

Резервная копия перед заменой

import shutil
from pathlib import Path

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