【python】在Django中,执行原生SQL查询

文章目录

  • 在Django中,执行原生SQL查询
    • [1. 使用 `raw()` 方法](#1. 使用 raw() 方法)
    • [2. 使用数据库连接直接执行SQL](#2. 使用数据库连接直接执行SQL)
    • [3. 使用多个数据库](#3. 使用多个数据库)
    • [4. 执行写操作(INSERT/UPDATE/DELETE)](#4. 执行写操作(INSERT/UPDATE/DELETE))
    • [5. 事务处理](#5. 事务处理)
    • [6. 实用的封装函数](#6. 实用的封装函数)
    • 注意事项

在Django中,执行原生SQL查询

1. 使用 raw() 方法

raw() 方法允许你执行原生SQL查询并返回模型实例:

python 复制代码
from myapp.models import Person

# 基本查询
people = Person.objects.raw('SELECT * FROM myapp_person WHERE age > %s', [18])

# 遍历结果
for person in people:
    print(person.name, person.age)

# 带参数的查询
people = Person.objects.raw(
    'SELECT * FROM myapp_person WHERE age BETWEEN %s AND %s',
    [20, 30]
)

2. 使用数据库连接直接执行SQL

基本用法:

python 复制代码
from django.db import connection

def my_custom_sql():
    with connection.cursor() as cursor:
        # 执行查询
        cursor.execute("SELECT * FROM myapp_person WHERE age > %s", [25])
        
        # 获取所有结果
        rows = cursor.fetchall()
        
        # 或者逐行获取
        # row = cursor.fetchone()
        # rows = cursor.fetchmany(size=10)
    
    return rows

返回字典格式的结果:

python 复制代码
from django.db import connection

def dictfetchall(cursor):
    """将游标结果转换为字典列表"""
    columns = [col[0] for col in cursor.description]
    return [
        dict(zip(columns, row))
        for row in cursor.fetchall()
    ]

def my_custom_sql():
    with connection.cursor() as cursor:
        cursor.execute("SELECT id, name, age FROM myapp_person")
        results = dictfetchall(cursor)
    return results

3. 使用多个数据库

如果你的项目配置了多个数据库:

python 复制代码
from django.db import connections

def query_multiple_dbs():
    # 使用默认数据库
    with connections['default'].cursor() as cursor:
        cursor.execute("SELECT * FROM table1")
        # 处理结果
    
    # 使用其他数据库
    with connections['other_db'].cursor() as cursor:
        cursor.execute("SELECT * FROM table2")
        # 处理结果

4. 执行写操作(INSERT/UPDATE/DELETE)

python 复制代码
from django.db import connection

def update_data():
    with connection.cursor() as cursor:
        # INSERT
        cursor.execute(
            "INSERT INTO myapp_person (name, age) VALUES (%s, %s)",
            ['John', 30]
        )
        
        # UPDATE
        cursor.execute(
            "UPDATE myapp_person SET age = %s WHERE name = %s",
            [31, 'John']
        )
        
        # DELETE
        cursor.execute(
            "DELETE FROM myapp_person WHERE age < %s",
            [18]
        )

5. 事务处理

python 复制代码
from django.db import transaction, connection

@transaction.atomic
def complex_operation():
    with connection.cursor() as cursor:
        try:
            # 一系列SQL操作
            cursor.execute("UPDATE accounts SET balance = balance - %s", [100])
            cursor.execute("UPDATE accounts SET balance = balance + %s", [100])
        except Exception as e:
            # 出错时会自动回滚
            print(f"Error: {e}")

6. 实用的封装函数

python 复制代码
from django.db import connection

def execute_query(query, params=None, return_dict=True):
    """
    执行SQL查询的通用函数
    """
    with connection.cursor() as cursor:
        cursor.execute(query, params or [])
        
        if query.strip().upper().startswith('SELECT'):
            if return_dict:
                columns = [col[0] for col in cursor.description]
                return [dict(zip(columns, row)) for row in cursor.fetchall()]
            else:
                return cursor.fetchall()
        else:
            # 对于非SELECT查询,返回受影响的行数
            return cursor.rowcount

# 使用示例
results = execute_query(
    "SELECT name, age FROM myapp_person WHERE age > %s",
    [25]
)

Django原生SQL查询 raw方法 直接数据库连接 返回模型实例 支持模型方法 自动字段映射 connection.cursor 返回原始数据 手动结果处理 执行查询 SELECT查询 写操作 fetchall fetchone fetchmany INSERT UPDATE DELETE 元组格式 字典格式转换

注意事项

  1. SQL注入防护:始终使用参数化查询,不要直接拼接字符串

    python 复制代码
    # ✅ 安全:使用参数化查询
    cursor.execute("SELECT * FROM users WHERE username = %s", [username])
    
    # ❌ 危险:字符串拼接(SQL注入风险)
    cursor.execute(f"SELECT * FROM users WHERE username = '{username}'")
    
    # ❌ 危险:直接传递用户输入
    cursor.execute("SELECT * FROM users WHERE username = %s" % username)
  2. 表名引用:在SQL中使用实际的数据库表名,而不是模型名

  3. 性能考虑 :原生SQL通常更快,但失去了Django ORM的一些便利功能;使用直接连接:需要最高性能、复杂查询或跨表查询

  4. 数据库兼容性:如果你使用多个数据库后端,要注意SQL语法的差异

这些方法让你在需要优化性能或执行复杂查询时,能够灵活地使用原生SQL,同时保持代码的安全性和可维护性。

相关推荐
cqbzcsq4 小时前
蛋白质功能预测模型DAMPE论文阅读报告
论文阅读·人工智能·python·深度学习·生物信息学
转转技术团队4 小时前
回收团队基于Cursor集成MCP的智能代码修复提示词生成实践
人工智能·python·程序员
石头dhf4 小时前
大模型配置
开发语言·python
南科1号4 小时前
Tushare数据来源分析一例
python
API技术员4 小时前
京东API接口:如何高效获取商品详情与SKU信息
python
sort浅忆5 小时前
deeptest执行接口脚本,添加python脚本断言
开发语言·python
冷月半明5 小时前
trea solo,让我从牛马外包翻身当“甲方”
python·trae
武昌库里写JAVA5 小时前
在iview中使用upload组件上传文件之前先做其他的处理
java·vue.js·spring boot·后端·sql
superman超哥5 小时前
仓颉借用检查器工作原理深度解析
c语言·开发语言·c++·python·仓颉
张彦峰ZYF6 小时前
Python 项目文件组织与工程化实践
python·项目文件组织与工程化实践