Python如何进行数据库操作

在Python中进行数据库操作，通常需要使用数据库适配器或ORM（对象关系映射）库。以下是一些常用的方法和步骤：

使用数据库适配器

1. 安装数据库适配器：

• 对于MySQL，可以使用mysql-connector-python或PyMySQL。
• 对于PostgreSQL，可以使用psycopg2。
• 对于SQLite，Python标准库中已经包含了sqlite3模块。

例如，安装MySQL适配器：

pip install mysql-connector-python

2. 连接数据库：

import mysql.connector

# 连接数据库
mydb = mysql.connector.connect(
  host="localhost",
  user="yourusername",
  password="yourpassword",
  database="yourdatabase"
)

3. 创建游标：

cursor = mydb.cursor()

4. 执行SQL语句：

# 插入数据
sql = "INSERT INTO customers (name, address) VALUES (%s, %s)"
val = ("John", "Highway 21")
cursor.execute(sql, val)

# 提交更改
mydb.commit()

# 查询数据
sql = "SELECT * FROM customers"
cursor.execute(sql)
myresult = cursor.fetchall()
for x in myresult:
  print(x)

5. 关闭连接：

cursor.close()
mydb.close()

使用ORM（对象关系映射）

ORM允许你使用Python类和对象来表示数据库表和记录，而不是直接编写SQL语句。

1. 安装ORM库：

• 对于SQLAlchemy，可以使用pip install sqlalchemy。
• 对于Django ORM，如果你在使用Django框架，则已经内置了ORM。

2. 定义模型：

from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

Base = declarative_base()

class Customer(Base):
    __tablename__ = 'customers'

    id = Column(Integer, primary_key=True)
    name = Column(String)
    address = Column(String)

# 创建数据库引擎
engine = create_engine('sqlite:///example.db')

# 创建所有表
Base.metadata.create_all(engine)

# 创建会话
Session = sessionmaker(bind=engine)
session = Session()

3. 添加数据：

new_customer = Customer(name='John', address='Highway 21')
session.add(new_customer)
session.commit()

4. 查询数据：

customers = session.query(Customer).all()
for customer in customers:
    print(customer.name, customer.address)

5. 关闭会话：

session.close()

注意事项

• 在进行数据库操作时，务必处理好异常情况，如使用try...except块捕获并处理可能的错误。
• 对于生产环境，建议使用连接池来管理数据库连接，以提高性能和可靠性。
• 在执行SQL语句时，注意防止SQL注入攻击，尽量使用参数化查询。

通过以上方法，你可以在Python中方便地进行数据库操作。