Python数据库API 2.0 防SQL注入

PEP 249是Python数据库API 2.0的规范,其中定义了一些数据库操作的通用接口。许多数据库驱动都遵循这个规范,包括`pymysql`、`psycopg2`(PostgreSQL)和`sqlite3`等。在使用这些数据库驱动程序时,为了防止SQL注入攻击,建议使用参数化查询。

以下是一个示例,演示如何通过PEP 249接口执行安全的数据库查询,并防止SQL注入。为了说明,我将使用`pymysql`库作为MySQL的示例,但类似的方法也适用于其他遵循PEP 249规范的数据库库。

### 示例代码

```python
import pymysql
import sys

def fetch_user_data(username, age):
    # 数据库连接
    connection = pymysql.connect(
        host='localhost',
        user='your_username',
        password='your_password',
        database='your_database'
    )
    
    try:
        cursor = connection.cursor()
        
        # 使用参数化查询
        query = "SELECT * FROM users WHERE username = %s AND age = %s"
        cursor.execute(query, (username, age))
        
        # 获取并打印结果
        results = cursor.fetchall()
        for row in results:
            print(row)
    
    except Exception as e:
        print(f"An error occurred: {e}", file=sys.stderr)
    
    finally:
        # 关闭游标和连接
        cursor.close()
        connection.close()

# 获取用户输入并查询
if __name__ == "__main__":
    username_input = input("请输入用户名: ")
    age_input = input("请输入年龄: ")
    
    if age_input.isdigit():  # 检查年龄是否为数字
        fetch_user_data(username_input, int(age_input))
    else:
        print("请输入有效的年龄(整数)")
```

### 说明

1. **参数化查询**:
   - 在`cursor.execute(query, (username, age))`中,使用`%s`作为占位符,实现了参数化查询。这里的`(username, age)`元组中的数据将被安全地替换,从而避免SQL注入。

2. **输入验证**:
   - 在许可证中,加入了对年龄输入的基本验证,确保输入为整数,以防止程序崩溃。

3. **异常处理**:
   - 使用`try-except`块来捕获可能的错误,以便在发生错误时能进行处理,而不会导致程序直接崩溃。

4. **最终资源释放**:
   - 使用`finally`块确保无论发生什么情况,游标和数据库连接都会被关闭,防止资源泄露。

这个代码段展示了如何使用符合PEP 249规范的方法安全地与数据库交互,防止SQL注入,同时确保代码的可读性和可维护性。
 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值