Python防止SQL注入攻击的方法

在Web开发中，SQL注入是一种常见的安全漏洞，攻击者可以通过在输入框中输入恶意的SQL语句来获取敏感数据或者破坏数据库。Python作为一种流行的编程语言，在处理用户输入时需要特别注意防止SQL注入攻击。本文将介绍Python中防止SQL注入攻击的方法，并给出相应的代码示例。

1. 使用参数化查询
   参数化查询是防止SQL注入的有效方法。在Python中，可以使用数据库模块提供的参数化查询方法来执行SQL语句，例如使用cursor.execute()方法。

import pymysql

# 连接数据库
conn = pymysql.connect(host='localhost', user='root', password='password', database='mydb')
cursor = conn.cursor()

# 参数化查询
sql = "SELECT * FROM users WHERE username=%s AND password=%s"
cursor.execute(sql, (username, password))
result = cursor.fetchall()

# 关闭连接
cursor.close()
conn.close()

在上面的代码中，我们使用了%s作为占位符，并将实际的参数传递给execute()方法。这样可以防止用户输入的内容被解释为SQL语句的一部分。

2. 使用ORM框架
   ORM（Object-Relational Mapping）框架可以帮助开发者避免直接拼接SQL语句，从而有效防止SQL注入攻击。Python中有多种ORM框架可供选择，例如Django的ORM、SQLAlchemy等。

from sqlalchemy import create_engine, Table, MetaData
from sqlalchemy.sql import select

# 创建引擎
engine = create_engine('mysql+pymysql://root:password@localhost/mydb')

# 创建元数据
metadata = MetaData()
users = Table('users', metadata, autoload_with=engine)

# 使用ORM查询
stmt = select([users]).where(users.c.username == username).where(users.c.password == password)
result = engine.execute(stmt).fetchall()

在使用ORM框架时，开发者无需直接编写SQL语句，框架会自动处理参数化查询，从而避免SQL注入攻击。

3. 输入验证和过滤
   除了使用参数化查询和ORM框架外，开发者还应该进行输入验证和过滤，确保用户输入的内容符合预期的格式和范围。例如，可以使用正则表达式对输入进行验证，或者使用内置的字符串处理函数对输入进行过滤。

import re

# 输入验证
if not re.match(r'^[a-zA-Z0-9_]+$', username):
    raise ValueError('Invalid username')

# 输入过滤
username = username.strip()

我们使用正则表达式对用户名进行验证，确保只包含字母、数字和下划线。同时，使用strip()函数对输入进行过滤，去除首尾的空白字符。