如何通过python运行sql语句

要通过Python运行SQL语句，可以使用多个库来实现，最常见的库有：sqlite3、PyMySQL、psycopg2。其中，sqlite3适用于SQLite数据库，PyMySQL适用于MySQL数据库，psycopg2适用于PostgreSQL数据库。下面我们将详细介绍使用这三个库运行SQL语句的方法，并重点展开sqlite3的使用。

一、使用sqlite3运行SQL语句

1、安装和导入sqlite3库

sqlite3库是Python的标准库，不需要单独安装，直接导入即可：

import sqlite3

2、连接到SQLite数据库

创建或连接到一个SQLite数据库。如果数据库文件不存在，它将被自动创建。

conn = sqlite3.connect('example.db')

3、创建游标对象

使用连接对象创建一个游标对象，以便执行SQL语句。

cursor = conn.cursor()

4、执行SQL语句

使用游标对象执行SQL语句，例如创建表、插入数据、查询数据等。

# 创建表
cursor.execute('''CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY, name TEXT, age INTEGER)''')
插入数据
cursor.execute("INSERT INTO users (name, age) VALUES ('Alice', 30)")
cursor.execute("INSERT INTO users (name, age) VALUES ('Bob', 25)")
查询数据
cursor.execute("SELECT * FROM users")
rows = cursor.fetchall()
for row in rows:
    print(row)

5、提交事务

对于数据插入、更新和删除操作，需要提交事务。

conn.commit()

6、关闭连接

操作完成后，关闭游标和连接。

cursor.close()
conn.close()

二、使用PyMySQL运行SQL语句

1、安装PyMySQL库

使用pip安装PyMySQL库：

pip install PyMySQL

2、导入PyMySQL库

import pymysql

3、连接到MySQL数据库

创建连接对象：

conn = pymysql.connect(
    host='localhost',
    user='yourusername',
    password='yourpassword',
    db='yourdatabase'
)

4、创建游标对象

创建游标对象：

cursor = conn.cursor()

5、执行SQL语句

执行SQL语句，例如创建表、插入数据、查询数据等：

# 创建表
cursor.execute('''CREATE TABLE IF NOT EXISTS users (id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255), age INT)''')
插入数据
cursor.execute("INSERT INTO users (name, age) VALUES ('Alice', 30)")
cursor.execute("INSERT INTO users (name, age) VALUES ('Bob', 25)")
查询数据
cursor.execute("SELECT * FROM users")
rows = cursor.fetchall()
for row in rows:
    print(row)

6、提交事务

对于数据插入、更新和删除操作，需要提交事务。

conn.commit()

7、关闭连接

操作完成后，关闭游标和连接。

cursor.close()
conn.close()

三、使用psycopg2运行SQL语句

1、安装psycopg2库

使用pip安装psycopg2库：

pip install psycopg2

2、导入psycopg2库

import psycopg2

3、连接到PostgreSQL数据库

创建连接对象：

conn = psycopg2.connect(
    dbname='yourdatabase',
    user='yourusername',
    password='yourpassword',
    host='localhost'
)

4、创建游标对象

创建游标对象：

cursor = conn.cursor()

5、执行SQL语句

执行SQL语句，例如创建表、插入数据、查询数据等：

# 创建表
cursor.execute('''CREATE TABLE IF NOT EXISTS users (id SERIAL PRIMARY KEY, name VARCHAR(255), age INT)''')
插入数据
cursor.execute("INSERT INTO users (name, age) VALUES ('Alice', 30)")
cursor.execute("INSERT INTO users (name, age) VALUES ('Bob', 25)")
查询数据
cursor.execute("SELECT * FROM users")
rows = cursor.fetchall()
for row in rows:
    print(row)

6、提交事务

对于数据插入、更新和删除操作，需要提交事务。

conn.commit()

7、关闭连接

操作完成后，关闭游标和连接。

cursor.close()
conn.close()

四、其他常见操作

1、处理异常

在执行数据库操作时，可能会发生各种异常，如连接失败、SQL语法错误等。可以使用try-except块来捕获和处理这些异常：

try:
    conn = sqlite3.connect('example.db')
    cursor = conn.cursor()
    cursor.execute("SELECT * FROM non_existing_table")
except sqlite3.Error as e:
    print(f"An error occurred: {e}")
finally:
    cursor.close()
    conn.close()

2、参数化查询

为了防止SQL注入攻击，建议使用参数化查询：

name = 'Alice'
age = 30
cursor.execute("INSERT INTO users (name, age) VALUES (?, ?)", (name, age))

3、批量插入数据

可以使用executemany方法批量插入数据：

users = [('Charlie', 22), ('David', 28)]
cursor.executemany("INSERT INTO users (name, age) VALUES (?, ?)", users)
conn.commit()

4、使用上下文管理器

使用上下文管理器可以简化连接的打开和关闭操作：

with sqlite3.connect('example.db') as conn:
    cursor = conn.cursor()
    cursor.execute("SELECT * FROM users")
    rows = cursor.fetchall()
    for row in rows:
        print(row)

5、事务处理

在一些复杂的业务逻辑中，可能需要手动控制事务：

try:
    conn = sqlite3.connect('example.db')
    cursor = conn.cursor()
    cursor.execute("BEGIN")
    cursor.execute("UPDATE users SET age = age + 1 WHERE name = 'Alice'")
    cursor.execute("UPDATE users SET age = age - 1 WHERE name = 'Bob'")
    conn.commit()
except sqlite3.Error as e:
    conn.rollback()
    print(f"An error occurred: {e}")
finally:
    cursor.close()
    conn.close()