Django中直接执行SQL语句

欢迎加入python学习交流群 667279387

今天在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、https://docs.djangoproject.com/en/2.0/topics/db/sql/

欢迎加入python学习交流群 667279387