您的当前位置:首页正文

Django中如何直接执行SQL语句

2024-11-03 来源:个人技术集锦

Django直接执行SQL语句

今天在django views.py看到同事写的代码里面有段关于数据库查询的语句。

因为涉及多个表的查询,所以django 的models的查询无法满足需求,所以直接执行了SQL语句。

他是按照下面的方法实现的。

    try:
        connection = MySQLdb.connect(host=SQL_IP,
                                     user=SQL_USER,
                                     passwd=SQL_PASSWD,
                                     db=SQL_DB, charset='utf8')
        cursor = connection.cursor()
        cmd_query = query_data(request)
        data_sort = sort_data(request)
        cmd = "set session group_concat_max_len = 8000;"
        cursor.execute(cmd)
        ......

看到这段代码,感觉应该重复造了轮子,数据库的链接应该交由django这个框架处理的。于是查看了下官方文档。

django.db.connection 就是上面的同事写的那个connection了。

后面的方法同他的一样执行就是了。即执行

connection.cursor()
cursor.execute(sql, [params])
cursor.fetchone() or cursor.fetchall()

下面是一个简单的例子。

from django.db import connection
def my_custom_sql(self):
    with connection.cursor() as cursor:
        cursor.execute("UPDATE bar SET foo = 1 WHERE baz = %s", [self.baz])
        cursor.execute("SELECT foo FROM bar WHERE baz = %s", [self.baz])
        row = cursor.fetchone()
    return row

如果你项目中有多个数据库的话,django.db.connection也能很方便的选取数据库。

django.db.connections 是一个类字典的结构,允许你使用数据库别名获取connection。

from django.db import connections
with connections['my_db_alias'].cursor() as cursor:
    # Your code here...

使用with作为上下文管理器:

with connection.cursor() as c:
    c.execute(...)

和下面这段代码等效

c = connection.cursor()
try:
    c.execute(...)
finally:
    c.close()

注意的问题:

如果你做了更新或者插入操作需要在代码中使用 :transaction.commit_unless_managed() 来提交数据。

或者使用事务装饰器(例如 commit_on_success)来修饰视图和提供事务控制数据提交。

这样就不用在代码中调用transaction.commit_unless_managed()。

但是,如果你不手动提交修改,你需要使用 transaction.set_dirty() 将事务标识为已脏。

使用 Django ORM 对数据库进行修改时,Django 会自动调用 set_dirty() 。

但如果你使用了原始 SQL ,Django 就无法获得你的 SQL 是否修改了数据。

只有手动调用 set_dirty() 才能确保 Django 知晓哪些修改必须被提交。

from django.db.transaction import commit_on_success
@commit_on_success
def my_custom_sql_view(request, value):
    from django.db import connection, transaction
    cursor = connection.cursor()
    # Data modifying operation
    cursor.execute("UPDATE bar SET foo = 1 WHERE baz = %s", [value])
    # Since we modified data, mark the transaction as dirty
    transaction.set_dirty()
    # Data retrieval operation. This doesn't dirty the transaction,
    # so no call to set_dirty() is required.
    cursor.execute("SELECT foo FROM bar WHERE baz = %s", [value])
    row = cursor.fetchone()
    #transaction.commit_unless_managed()
    return render_to_response('template.html', {'row': row})

如果不使用

参考文献:1、

Django底层实现sql语句

在django中我们可以自己手写底层sql语句,那么如何写呢?

#导入原生sql模块
from django.db import connection
# 获取当前用户关注课程列表
def get_myflow(request):
    # 获取用户id
    uid = request.GET.get('id')
    # 建立游标对象
    cursor = connection.cursor()
    # 链表查询
    cursor.execute('select a.username,c.title from user a left join flows b on a.id=b.uid left join course c on b.cid=c.id where a.id = %s'% uid)
    # 获取结果集
    result = dictfetchall(cursor)
    print(result)
    # 返回结果
    return HttpResponse(json.dumps(result),content_type='application/json')

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持。

您可能感兴趣的文章:
Top