python 连接数据库

import sqlite3

# 连接到SQLite数据库
# 数据库文件是test.db
# 如果文件不存在，会自动在当前目录创建
conn = sqlite3.connect('test.db')

# 创建一个游标对象
cursor = conn.cursor()

# 执行一条SQL语句，创建user表
cursor.execute('create table user (id varchar(20) primary key, name varchar(20))')

# 继续执行一条SQL语句，插入记录
cursor.execute('insert into user (id, name) values (\'1\', \'Michael\')')

# 获取插入的行数
print(cursor.rowcount)

# 提交事务
conn.commit()

# 关闭游标
cursor.close()

# 关闭连接
conn.close()

解释说明：

1. 导入模块：首先导入 sqlite3 模块，用于连接 SQLite 数据库。
2. 连接数据库：使用 sqlite3.connect() 方法连接到名为 test.db 的 SQLite 数据库。如果该文件不存在，则会自动创建。
3. 创建游标：通过 conn.cursor() 创建一个游标对象，用于执行 SQL 语句。
4. 执行 SQL 语句：
   • 使用 cursor.execute() 方法执行 SQL 语句，这里创建了一个名为 user 的表，并插入了一条记录。
5. 获取插入的行数：通过 cursor.rowcount 获取刚刚插入的行数。
6. 提交事务：使用 conn.commit() 提交事务，确保数据被保存到数据库中。
7. 关闭游标和连接：最后关闭游标和数据库连接，释放资源。

如果你需要连接其他类型的数据库（如 MySQL、PostgreSQL 等），可以使用相应的数据库驱动模块（如 pymysql 或 psycopg2）。