Python中如何在main()函数中实现数据库操作

在Python中，我们可以使用第三方库来实现数据库操作，最常用的是sqlite3库。sqlite3是Python内置的轻量级数据库，可以用于创建和管理本地数据库文件。

下面是一个实现了数据库操作的例子：

import sqlite3

def create_table():
    # 连接到数据库
    conn = sqlite3.connect('test.db')
    # 创建一个游标对象
    cursor = conn.cursor()
    # 创建表格
    cursor.execute('''CREATE TABLE IF NOT EXISTS students
                      (id INTEGER PRIMARY KEY AUTOINCREMENT,
                      name TEXT,
                      age INTEGER)''')
    # 提交改变
    conn.commit()
    # 关闭游标和连接
    cursor.close()
    conn.close()

def insert_data(name, age):
    # 连接到数据库
    conn = sqlite3.connect('test.db')
    # 创建一个游标对象
    cursor = conn.cursor()
    # 插入数据
    cursor.execute("INSERT INTO students (name, age) VALUES (?, ?)", (name, age))
    # 提交改变
    conn.commit()
    # 关闭游标和连接
    cursor.close()
    conn.close()

def query_data():
    # 连接到数据库
    conn = sqlite3.connect('test.db')
    # 创建一个游标对象
    cursor = conn.cursor()
    # 查询数据
    cursor.execute("SELECT * FROM students")
    # 获取所有数据
    rows = cursor.fetchall()
    # 打印数据
    for row in rows:
        print("ID:", row[0])
        print("Name:", row[1])
        print("Age:", row[2])
        print("
")
    # 关闭游标和连接
    cursor.close()
    conn.close()

if __name__ == "__main__":
    create_table()
    insert_data("Alice", 20)
    insert_data("Bob", 22)
    query_data()

在上述例子中，我们定义了三个函数create_table()、insert_data()和query_data()来实现数据库操作。

create_table()函数用于创建一个名为students的表格，表格包含id、name和age三列。如果表格已经存在，则不会重复创建。

insert_data(name, age)函数用于向表格中插入数据。它接收两个参数，分别是姓名name和年龄age。

query_data()函数用于查询表格中的数据，并打印出来。它不需要任何参数。

在main()函数中，我们首先调用create_table()函数创建表格，然后调用insert_data()函数向表格中插入两条数据，最后调用query_data()函数查询并打印出数据。

通过运行以上代码，你将在控制台上看到输出结果，显示出插入的两条数据。

请注意，每次运行该程序，都会覆盖掉之前的数据库文件test.db。如果你想保留之前的数据，请备份数据库文件或修改数据库文件名。

希望以上示例代码对你有所帮助，如果你要使用其他数据库，可以使用对应的数据库连接库来实现数据库操作。