如果发现错误，继续执行 sql 语句 python sqlite3

Question

我有一个包含 sql 命令的文件，我想执行文件中给出的命令，如果任何命令抛出任何错误，则忽略该错误并执行下一个命令。 这是一个示例文件：

drop table department;
drop table classroom;

create table classroom
    (building       varchar(15),
     room_number        varchar(7),
     capacity       numeric(4,0),
     primary key (building, room_number)
    );

create table department
    (dept_name      varchar(20), 
     building       varchar(15), 
     budget             numeric(12,2) check (budget > 0),
     primary key (dept_name)
    );

例如。如果教室表不存在，那么 drop table 命令将产生错误并且程序将终止。我希望程序继续运行并执行文件中的所有命令。

我面临的问题是create table 命令在多行中，所以我不知道如何执行。

Answer 1

看看sqlite3 manual:

import sqlite3
con = sqlite3.connect(":memory:")
cur = con.cursor()
with open('PATH_TO_SQL_FILE', 'r') as fp:
    text = fp.read().split(';')
    for command in text:
       try:
           cur.execute(command)
       except sqlite3.Error:
           pass

Answer 2

import sqlite3

#define connection and cursor
connection = sqlite3.connect ('test_db')
cursor =connection.cursor()


command1 ="""create table classroom
    (building       varchar(15),
    room_number        varchar(7),
    capacity       numeric(4,0),
    primary key (building, room_number)
    )"""

cursor.execute(command1)


command2 ="""create table department
    (dept_name      varchar(20), 
    building       varchar(15), 
    budget             numeric(12,2) check (budget > 0),
    primary key (dept_name)
    )"""
cursor.execute(command2)