欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

在django項目中導(dǎo)出數(shù)據(jù)到excel文件并實現(xiàn)下載的功能

 更新時間:2020年03月13日 10:41:11   作者:matrix_theOne  
這篇文章主要介紹了在django項目中導(dǎo)出數(shù)據(jù)到excel文件并實現(xiàn)下載的功能,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧

依賴模塊

xlwt下載:pip install xlwt

后臺模塊

view.py

# 導(dǎo)出Excel文件
def export_excel(request):
  city = request.POST.get('city')
  print(city)
  list_obj=place.objects.filter(city=city)
  # 設(shè)置HTTPResponse的類型
  response = HttpResponse(content_type='application/vnd.ms-excel')
  response['Content-Disposition'] = 'attachment;filename='+city+'.xls'
  """導(dǎo)出excel表"""
  if list_obj:
    # 創(chuàng)建工作簿
    ws = xlwt.Workbook(encoding='utf-8')
    # 添加第一頁數(shù)據(jù)表
    w = ws.add_sheet('sheet1') # 新建sheet(sheet的名稱為"sheet1")
    # 寫入表頭
    w.write(0, 0, u'地名')
    w.write(0, 1, u'次數(shù)')
    w.write(0, 2, u'經(jīng)度')
    w.write(0, 3, u'緯度')
    # 寫入數(shù)據(jù)
    excel_row = 1
    for obj in list_obj:
      name = obj.place
      sum = obj.sum
      lng = obj.lng
      lat = obj.lat
      # 寫入每一行對應(yīng)的數(shù)據(jù)
      w.write(excel_row, 0, name)
      w.write(excel_row, 1, sum)
      w.write(excel_row, 2, lng)
      w.write(excel_row, 3, lat)
      excel_row += 1
    # 寫出到IO
    output = BytesIO()
    ws.save(output)
    # 重新定位到開始
    output.seek(0)
    response.write(output.getvalue())
  return response

前端模塊

<button id="export_excel" type="button" class="btn btn-primary col-sm-5" style="margin-left: 10px" >導(dǎo)出excel</button>

$("#export_excel").click(function () {
     var csrf=$('input[name="csrfmiddlewaretoken"]').val();
     const req = new XMLHttpRequest();
     req.open('POST', '/export_excel/', true);
     req.responseType = 'blob';
     req.setRequestHeader('Content-Type', 'application/x-www-form-urlencoded'); //設(shè)置請求頭
     req.send('city='+$('#city').val()+"&&csrfmiddlewaretoken="+csrf); //輸入?yún)?shù)
     req.onload = function() {
       const data = req.response;
       const a = document.createElement('a');
       const blob = new Blob([data]);
       const blobUrl = window.URL.createObjectURL(blob);
       download(blobUrl) ;
     };

   });
function download(blobUrl) {
 var city = $("input[name='city']").val();
 const a = document.createElement('a');
 a.style.display = 'none';
 a.download = '<文件命名>';
 a.href = blobUrl;
 a.click();
 document.body.removeChild(a);
}

補充知識:Python Django實現(xiàn)MySQL百萬、千萬級的數(shù)據(jù)量下載:解決memoryerror、nginx time out

前文

在用Django寫項目的時候時常需要提供文件下載的功能,而Django也是貼心提供了幾種方法:FileResponse、StreamingHttpResponse、HttpResponse,其中FileResponse和StreamingHttpResponse都是使用迭代器迭代生成數(shù)據(jù)的方法,所以適合傳輸文件比較大的情況;而HttpResponse則是直接取得數(shù)據(jù)返回給用戶,所以容易造成memoryerror和nginx time out(一次性取得數(shù)據(jù)和返回的數(shù)據(jù)過多,導(dǎo)致nginx超時或者內(nèi)存不足),關(guān)于這三者,DJango的官網(wǎng)也是寫的非常清楚,連接如下:https://docs.djangoproject.com/en/1.11/ref/request-response/

那正常我們使用的是FileResponse和StreamingHttpResponse,因為它們流式傳輸(迭代器)的特點,可以使得數(shù)據(jù)一條條的返回給客戶端,文件隨時中斷和復(fù)傳,并且保持文件的一致性。

FileResponse和StreamingHttpResponse

FileResponse顧名思義,就是打開文件然后進行傳輸,并且可以指定一次能夠傳輸?shù)臄?shù)據(jù)chunk。所以適用場景:從服務(wù)端返回大文件。缺點是無法實時獲取數(shù)據(jù)庫的內(nèi)容并傳輸給客戶端。舉例如下:

def download(request):
 file=open('path/demo.py','rb')
  response =FileResponse(file)
  response['Content-Type']='application/octet-stream'
  response['Content-Disposition']='attachment;filename="demo.py"'
  return response

從上可以發(fā)現(xiàn),文件打開后作為參數(shù)傳入FileResponse,隨后指定傳輸頭即可,但是很明顯用這個來傳輸數(shù)據(jù)庫就不太方便了,所以這邊推介用StreamingHttpResponse的方式來傳輸。

這里就用PyMysql來取得數(shù)據(jù),然后指定為csv的格式返回,具體代碼如下:

# 通過pymysql取得數(shù)據(jù)
import pymysql
field_types = {
    1: 'tinyint',
    2: 'smallint',
    3: 'int'} #用于后面的字段名匹配,這里省略了大多數(shù)
conn = pymysql.connect(host='127.0.0.1',port=3306,database='demo',user='root',password='root')
cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
cursor.execute(sql)
#獲取所有數(shù)據(jù)
data = cursor.fetchall()
cols = {}
#獲取所有字段
for i,row in enumerate(self.cursor.description):
 if row[0] in cols:
   cols[str(i)+row[0]] = field_types.get(row[1], str(row[1])) #這里的field_type是類型和數(shù)字的匹配
 cols[row[0]] = field_types.get(row[1], str(row[1]))
cursor.close()
conn.close()

#通過StreamingHttpResponse指定返回格式為csv
response = StreamingHttpResponse(get_result_fromat(data, cols))
response['Content-Type'] = 'application/octet-stream'
response['Content-Disposition'] = 'attachment;filename="{0}"'.format(out_file_name)
return response

#循環(huán)所有數(shù)據(jù),然后加到字段上返回,注意的是要用迭代器來控制
def get_result_fromat(data, cols):
 tmp_str = ""
 # 返回文件的每一列列名
  for col in cols:
    tmp_str += '"%s",' % (col)
  yield tmp_str.strip(",") + "\n"
  for row in data:
    tmp_str = ""
    for col in cols:
      tmp_str += '"%s",' % (str(row[col]))
    yield tmp_str.strip(',') + "\n"

整個代碼如上,大致分為三部分:從mysql取數(shù)據(jù),格式化成我們想要的格式:excel、csv、txt等等,這邊指定的是csv,如果對其他格式也有興趣的可以留言,最后就是用StreamingHttpResponse指定返回的格式返回。

實現(xiàn)百萬級數(shù)據(jù)量下載

上面的代碼下載可以支持幾萬行甚至十幾萬行的數(shù)據(jù),但是如果超過20萬行以上的數(shù)據(jù),那就比較困難了,我這邊的剩余內(nèi)存大概是1G的樣子,當(dāng)超過15萬行數(shù)據(jù)(大概)的時候,就報memoryerror了,問題就是因為fetchall,雖然我們StreamingHttpResponse是一條條的返回,但是我們的數(shù)據(jù)時一次性批量的取得!

如何解決?以下是我的解決方法和思路:

用fetchone來代替fetchall,迭代生成fetchone

發(fā)現(xiàn)還是memoryerror,因為execute是一次性執(zhí)行,后來發(fā)現(xiàn)可以用流式游標(biāo)來代替原來的普通游標(biāo),即SSDictCursor代替DictCursor

于是整個代碼需要修改的地方如下:

cursor = conn.cursor(cursor=pymysql.cursors.DictCursor) ===>
cursor = conn.cursor(cursor=pymysql.cursors.SSDictCursor)

data = cursor.fetchall() ===>
row = cursor.fetchone()

def get_result_fromat(data, cols):
 tmp_str = ""
 # 返回文件的每一列列名
  for col in cols:
    tmp_str += '"%s",' % (col)
  yield tmp_str.strip(",") + "\n"
  for row in data:
    tmp_str = ""
    for col in cols:
      tmp_str += '"%s",' % (str(row[col]))
    yield tmp_str.strip(',') + "\n" 
    
    =====>
    
def get_result_fromat(data, cols):
 tmp_str = ""
  for col in cols:
    tmp_str += '"%s",' % (col)
  yield tmp_str.strip(",") + "\n"
  while True:
    tmp_str = ""
    for col in cols:
      tmp_str += '"%s",' % (str(row[col]))
    yield tmp_str.strip(',') + "\n"
    row = db.cursor.fetchone()
    if row is None:
      break

可以看到就是通過while True來實現(xiàn)不斷地取數(shù)據(jù)下載,有效避免一次性從MySQL取出內(nèi)存不足報錯,又或者取得過久導(dǎo)致nginx超時!

總結(jié)

關(guān)于下載就分享到這了,還是比較簡單的,謝謝觀看~希望能給大家一個參考,也希望大家多多支持腳本之家。

相關(guān)文章

  • Python3爬蟲里關(guān)于Splash負載均衡配置詳解

    Python3爬蟲里關(guān)于Splash負載均衡配置詳解

    在本篇文章里小編給大家分享了關(guān)于Python3爬蟲里關(guān)于Splash負載均衡配置的相關(guān)內(nèi)容,需要的朋友們可以學(xué)習(xí)參考下。
    2020-07-07
  • 解讀FastAPI異步化為transformers模型打造高性能接口

    解讀FastAPI異步化為transformers模型打造高性能接口

    這篇文章主要介紹了解讀FastAPI異步化為transformers模型打造高性能接口問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教
    2024-06-06
  • Python辦公自動化之發(fā)送電子郵件和Outlook集成

    Python辦公自動化之發(fā)送電子郵件和Outlook集成

    Python辦公?動化是利?Python編程語?來創(chuàng)建腳本和程序,以簡化、加速和?動化?常辦公任務(wù)和?作流程的過程,本文主要介紹一下如何利用Python實現(xiàn)發(fā)送電子郵件和Outlook集成,需要的可以參考下
    2023-12-12
  • Pandas-DataFrame知識點匯總

    Pandas-DataFrame知識點匯總

    這篇文章主要介紹了Pandas-DataFrame知識點匯總,DataFrame是一種表格型數(shù)據(jù)結(jié)構(gòu),它含有一組有序的列,每列可以是不同的值,下面我們一起進入文章了解更多詳細內(nèi)容吧,需要的小伙伴也可以參考一下
    2022-03-03
  • Python如何爬取51cto數(shù)據(jù)并存入MySQL

    Python如何爬取51cto數(shù)據(jù)并存入MySQL

    這篇文章主要介紹了Python如何爬取51cto數(shù)據(jù)并存入MySQL,文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下
    2020-08-08
  • Python模塊相關(guān)知識點小結(jié)

    Python模塊相關(guān)知識點小結(jié)

    這篇文章主要介紹了Python模塊相關(guān)知識點,總結(jié)分析了Python模塊的功能、原理、使用方法與操作注意事項,需要的朋友可以參考下
    2020-03-03
  • 利用Python Django實現(xiàn)簡單博客系統(tǒng)

    利用Python Django實現(xiàn)簡單博客系統(tǒng)

    這篇文章主要介紹了利用Python Django實現(xiàn)簡單博客系統(tǒng),文中有非常詳細的代碼示例,對正在學(xué)習(xí)python的小伙伴們有很好地幫助,需要的朋友可以參考下
    2021-05-05
  • pandas.DataFrame.iloc的具體使用詳解

    pandas.DataFrame.iloc的具體使用詳解

    本文主要介紹了pandas.DataFrame.iloc的具體使用詳解,文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
    2022-06-06
  • python之文件的讀寫和文件目錄以及文件夾的操作實現(xiàn)代碼

    python之文件的讀寫和文件目錄以及文件夾的操作實現(xiàn)代碼

    這篇文章主要介紹了python之文件的讀寫和文件目錄以及文件夾的操作實現(xiàn)代碼,需要的朋友可以參考下
    2016-08-08
  • 淺析使用Python操作文件

    淺析使用Python操作文件

    文件操作對編程語言的重要性不用多說,如果數(shù)據(jù)不能持久保存,信息技術(shù)也就失去了意義。按照本人經(jīng)驗,IO也是蠻頭疼的一件事,因為不會用得太多,所以總是記不住API,每次都要重新google就會打斷思路,還不一定每次都快速得到正確的文章。
    2017-07-07

最新評論