python中mysql基础代码

在Python中操作MySQL数据库时，通常使用第三方库如 `pymysql` 或 `mysql-connector-python` 来实现。以下是一个基础代码示例，展示了如何连接数据库、执行查询、插入数据以及处理事务。 ### 安装依赖 在执行代码之前，确保安装了 `pymysql` 模块： ```bash pip install pymysql ``` ### 基础代码示例 #### 1. 连接数据库并执行查询 ```python import pymysql # 创建数据库连接 conn = pymysql.connect( host='localhost', port=3306, user='root', password='your_password', database='test_db', charset='utf8mb4' ) # 获取游标对象 cur = conn.cursor() # 执行查询语句 cur.execute("SELECT VERSION()") # 获取查询结果 data = cur.fetchone() print("Database version : %s " % data) # 关闭连接 conn.close() ``` #### 2. 插入数据并提交事务 ```python import pymysql # 创建数据库连接 conn = pymysql.connect( host='localhost', port=3306, user='root', password='your_password', database='test_db', charset='utf8mb4' ) # 获取游标对象 cur = conn.cursor() # 插入数据 sql = """ INSERT INTO employees (first_name, last_name, age, gender, income) VALUES ('John', 'Doe', 30, 'M', 5000) """ try: # 执行SQL语句 cur.execute(sql) # 提交事务 conn.commit() print("Record inserted successfully.") except Exception as e: # 如果发生错误则回滚事务 conn.rollback() print("Error: %s" % e) # 关闭连接 conn.close() ``` #### 3. 查询数据并遍历结果集 ```python import pymysql # 创建数据库连接 conn = pymysql.connect( host='localhost', port=3306, user='root', password='your_password', database='test_db', charset='utf8mb4' ) # 获取游标对象 cur = conn.cursor() # 执行查询语句 cur.execute("SELECT * FROM employees") # 遍历结果集 for row in cur.fetchall(): print(row) # 关闭连接 conn.close() ``` #### 4. 使用上下文管理器自动管理连接和事务 ```python import pymysql # 使用上下文管理器自动管理连接 with pymysql.connect( host='localhost', port=3306, user='root', password='your_password', database='test_db', charset='utf8mb4' ) as conn: with conn.cursor() as cur: # 执行查询语句 cur.execute("SELECT * FROM employees") # 获取所有记录 rows = cur.fetchall() for row in rows: print(row) ``` 上述代码展示了Python中使用 `pymysql` 模块连接和操作MySQL数据库的基本流程，包括连接数据库、执行SQL语句、处理事务以及遍历结果集等操作。