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')
总结
以上为个人经验,希望能给大家一个参考,也希望大家多多支持。