Restore sqlite from latest backup file in Python

import sqlite3
import glob
import os
import shutil

def restore_latest_backup(db_path, backup_dir):
    backups = sorted(glob.glob(os.path.join(backup_dir, "*.db")), key=os.path.getmtime)
    if not backups:
        raise FileNotFoundError("No backup files found")
    latest = backups[-1]
    shutil.copy2(latest, db_path)
    return latest

if __name__ == "__main__":
    backup_dir = "backups"
    os.makedirs(backup_dir, exist_ok=True)

    # Create sample backups
    for name in ["old.db", "mid.db", "newest.db"]:
        path = os.path.join(backup_dir, name)
        conn = sqlite3.connect(path)
        conn.execute("CREATE TABLE data (value TEXT)")
        conn.execute("INSERT INTO data VALUES (?)", (name,))
        conn.commit()
        conn.close()

    # Restore latest backup
    db_path = "restored.db"
    restored_from = restore_latest_backup(db_path, backup_dir)
    print(f"Restored from: {restored_from}")

    conn = sqlite3.connect(db_path)
    value = conn.execute("SELECT value FROM data").fetchone()[0]
    print(f"Restored value: {value}")
    conn.close()

    os.remove(db_path)
    for f in glob.glob(os.path.join(backup_dir, "*.db")):
        os.remove(f)