补全：转移一条也要原子

👁️ 2 人浏览 💬 0 人评论 ❤️ 添加收藏

（每道题开头都有同一段：open_db() 建一个内存 SQLite 库、含 users / notes / files 三张表、交回连接；执行 SQL 用 conn.execute(SQL, 参数元组)，取一行用 .fetchone()、取全部用 .fetchall()、新行主键用 .lastrowid。判题机自带 sqlite3。）

（这一节还有：batch_add(conn, uid, texts, fail_at) 一批留言要么全进要么全不进——中途出错就 rollback、成功才 commit；fail_at 指定第几条故意失败。）

补全 move_note：把某条留言从一个人名下改到另一个人名下，用事务保证改完就落定、出错就回滚。

提交你的答案

请登录后提交答案。

去登录

← 补全：中途失败就回滚补全：包一层通用事务 →

更多题目

代码编辑器

语言： Ctrl + Enter 运行

import sqlite3

def open_db():
    """建一个内存 SQLite 库，含 users / notes / files 三张表，交回连接。"""
    conn = sqlite3.connect(":memory:")
    conn.execute("CREATE TABLE users(id INTEGER PRIMARY KEY, name TEXT UNIQUE, pw TEXT)")
    conn.execute("CREATE TABLE notes(id INTEGER PRIMARY KEY, uid INTEGER, text TEXT)")
    conn.execute("CREATE TABLE files(id INTEGER PRIMARY KEY, uid INTEGER, name TEXT, size INTEGER)")
    return conn

def add_note(conn, uid, text):
    """插一条留言，交回它的新 id（自增主键）。"""
    cur = conn.execute("INSERT INTO notes(uid, text) VALUES(?, ?)", (uid, text))
    return cur.lastrowid

def get_text(conn, nid):
    """按 id 取一条留言的正文；没有交回 None。"""
    row = conn.execute("SELECT text FROM notes WHERE id=?", (nid,)).fetchone()
    return row[0] if row else None

def update_note(conn, nid, text):
    """把某条留言的正文改掉。"""
    conn.execute("UPDATE notes SET text=? WHERE id=?", (text, nid))

def delete_note(conn, nid):
    """删掉某条留言。"""
    conn.execute("DELETE FROM notes WHERE id=?", (nid,))

def list_texts(conn, uid):
    """按 id 顺序交回某人所有留言正文的列表。"""
    return [r[0] for r in conn.execute("SELECT text FROM notes WHERE uid=? ORDER BY id", (uid,))]

def count_notes(conn, uid):
    """数某人有几条留言。"""
    return conn.execute("SELECT COUNT(*) FROM notes WHERE uid=?", (uid,)).fetchone()[0]

def batch_add(conn, uid, texts, fail_at=None):
    """一批留言要么全进、要么全不进：中途出错就 rollback，成功才 commit。交回是否成功。
    fail_at 不为 None 时，插到第 fail_at 条故意抛错（模拟中途失败）。"""
    try:
        for i, t in enumerate(texts):
            if fail_at is not None and i == fail_at:
                raise ValueError("boom")
            conn.execute("INSERT INTO notes(uid, text) VALUES(?, ?)", (uid, t))
        conn.commit()
        return True
    except Exception:
        conn.rollback()
        return False

def move_note(conn, nid, to_uid):
    try:
        # TODO: 把 nid 那条的 uid 改成 to_uid，然后落定
        return True
    except Exception:
        conn.rollback()
        return False

c = open_db()
n = add_note(c, 1, "x")
move_note(c, n, 2)
print(str(count_notes(c, 1)) + "/" + str(count_notes(c, 2)))

本次输入：

输出：

👩‍🏫

AI

💬 题目评论

补全：转移一条也要原子

更多题目

全部评论