python 如何添加数据库数据

Python 添加数据库数据的方法有多种，包括使用SQLite、MySQL、PostgreSQL等数据库。主要步骤包括：连接数据库、创建表、插入数据、提交事务。 在这里，我将详细介绍如何使用Python添加数据库数据的步骤和注意事项。

一、连接数据库

在开始添加数据之前，首先需要连接到数据库。不同的数据库有不同的连接方式。这里以SQLite和MySQL为例，介绍如何连接数据库。

1.1、连接SQLite数据库

SQLite是一个轻量级的嵌入式数据库，不需要单独的服务器进程，适用于小型应用程序。Python内置了SQLite库，可以直接使用。

import sqlite3
连接到SQLite数据库
conn = sqlite3.connect('example.db')
cursor = conn.cursor()

1.2、连接MySQL数据库

MySQL是一个流行的开源关系数据库管理系统。要连接MySQL数据库，需要安装mysql-connector-python库。

import mysql.connector
连接到MySQL数据库
conn = mysql.connector.connect(
    host="localhost",
    user="yourusername",
    password="yourpassword",
    database="yourdatabase"
)
cursor = conn.cursor()

二、创建表

在向数据库中插入数据之前，需要先创建一个表。如果表已经存在，可以跳过这一步。以下是创建表的示例代码。

2.1、创建SQLite表

# 创建表
cursor.execute('''CREATE TABLE IF NOT EXISTS users
                  (id INTEGER PRIMARY KEY AUTOINCREMENT,
                   name TEXT NOT NULL,
                   age INTEGER NOT NULL)''')

2.2、创建MySQL表

# 创建表
cursor.execute('''CREATE TABLE IF NOT EXISTS users (
                  id INT AUTO_INCREMENT PRIMARY KEY,
                  name VARCHAR(255) NOT NULL,
                  age INT NOT NULL)''')

三、插入数据

插入数据是向数据库中添加记录的过程。可以使用INSERT语句插入数据。

3.1、插入单条数据

插入单条数据是最常见的操作。以下是插入单条数据的示例代码。

# 插入单条数据
cursor.execute("INSERT INTO users (name, age) VALUES (?, ?)", ('John Doe', 30))
conn.commit()

3.2、插入多条数据

有时需要一次性插入多条数据，可以使用executemany()方法。

# 插入多条数据
users = [('Alice', 25), ('Bob', 22), ('Charlie', 28)]
cursor.executemany("INSERT INTO users (name, age) VALUES (?, ?)", users)
conn.commit()

四、提交事务

在插入数据后，需要提交事务才能使更改生效。对于SQLite和MySQL，提交事务的方式相同。

# 提交事务
conn.commit()

五、处理异常

在数据库操作过程中，可能会遇到各种异常情况，如连接失败、插入数据失败等。需要使用try-except块来捕获和处理异常。

try:
    conn = sqlite3.connect('example.db')
    cursor = conn.cursor()
    cursor.execute("INSERT INTO users (name, age) VALUES (?, ?)", ('John Doe', 30))
    conn.commit()
except sqlite3.Error as e:
    print(f"An error occurred: {e}")
finally:
    if conn:
        conn.close()

六、示例代码

综合以上步骤，以下是一个完整的示例代码，展示了如何使用Python添加数据库数据。

import sqlite3
def create_connection(db_file):
    conn = None
    try:
        conn = sqlite3.connect(db_file)
        return conn
    except sqlite3.Error as e:
        print(e)
    return conn
def create_table(conn):
    try:
        cursor = conn.cursor()
        cursor.execute('''CREATE TABLE IF NOT EXISTS users
                          (id INTEGER PRIMARY KEY AUTOINCREMENT,
                           name TEXT NOT NULL,
                           age INTEGER NOT NULL)''')
    except sqlite3.Error as e:
        print(e)
def insert_user(conn, user):
    try:
        cursor = conn.cursor()
        cursor.execute("INSERT INTO users (name, age) VALUES (?, ?)", user)
        conn.commit()
    except sqlite3.Error as e:
        print(e)
def main():
    database = "example.db"
    # 创建数据库连接
    conn = create_connection(database)
    # 创建表
    if conn is not None:
        create_table(conn)
        # 插入数据
        insert_user(conn, ('John Doe', 30))
        insert_user(conn, ('Alice', 25))
        insert_user(conn, ('Bob', 22))
        # 关闭连接
        conn.close()
    else:
        print("Error! cannot create the database connection.")
if __name__ == '__main__':
    main()