import sqlite3
import pymysql
# 使用SQLite
def sqlite_example():
conn = sqlite3.connect('example.db')
cursor = conn.cursor()
# 创建表
cursor.execute('''CREATE TABLE IF NOT EXISTS user (
id INTEGER PRIMARY KEY,
name TEXT NOT NULL)''')
# 插入数据
cursor.execute("INSERT INTO user (name) VALUES (?)", ("Alice",))
# 查询数据
cursor.execute("SELECT * FROM user")
rows = cursor.fetchall()
for row in rows:
print(row)
conn.commit()
conn.close()
# 使用MySQL
def mysql_example():
conn = pymysql.connect(host='localhost', user='user', password='passwd', db='example')
cursor = conn.cursor()
# 创建表
cursor.execute('''CREATE TABLE IF NOT EXISTS user (
id INT AUTO_INCREMENT PRIMARY KEY,
name VARCHAR(255) NOT NULL)''')
# 插入数据
cursor.execute("INSERT INTO user (name) VALUES (%s)", ("Alice",))
# 查询数据
cursor.execute("SELECT * FROM user")
rows = cursor.fetchall()
for row in rows:
print(row)
conn.commit()
conn.close()
# 调用函数
sqlite_example()
mysql_example()
这段代码展示了如何使用Python内置的sqlite3库和pymysql库来分别进行SQLite和MySQL的数据库操作。代码中创建了一个名为"user"的表,并展示了如何插入和查询数据。在实际应用中,你需要根据自己的数据库配置(如主机地址、用户名、密码、数据库名等)和需求来修改连接参数。