Django+pdfkit+wkhtmltopdf生成PDF

需求：将txt或其他格式的文件中的数据转换成表格，并生成PDF文件，以供下载。（这里以txt文件为列）

依赖

• 安装wkhtmltopdf, Mac安装方式如下：

brew install wkhtmltopdf

记录下wkhtmltopdf的安装路径,以供代码中使用，这里是：/usr/local/bin/wkhtmltopdf

• Django项目中，需要安装pdfkit依赖，直接使用pip安装

pip install pdfkit

代码编写

第一步：数据转换

txt文件中的数据如下，需要将其取出，处理成list

sdsd.png

代码如下：

读取文件内容

def get_file_content(file_dir, file_name):
    """
    获取文件内容 
    file_dir: txt文件目录
    file_name：txt文件文件名称
    """
    file_path = file_dir + file_name
    if not os.path.exists(file_path):
        raise Exception('文件目录：%s不存在' % file_dir)
    with open(file_path, "r", encoding='gbk') as f:
        data = f.read()
        return data

数据处理

def handle_file_data(data):
      """
      对文件内容进行处理：str->list
      """
      content = data.strip('\n')
        if not content:
            raise Exception('当前文件数据为空')
        lines = content.split('\n')
        key_str = lines.pop(0)
        keys = key_str.split(',')
        items = []
        for line in lines:
            valus = line.split(',')
            # valus的长度不够，补齐
            if len(keys) > len(valus):
                n = len(keys) - len(valus)
                for i in range(n):
                    valus.append('')
            # 数据组装
            item = {}
            for index in range(len(keys)):
                item[keys[index]] = valus[index]
            items.append(item)
        return items, keys

返回数据如下：

[
  {
    "APP_NAME": "测试1",
    "TABLE_NAME": "hei",
    "DATE_NAME": "20220902",
    "COUNT_NUM_SC": "1799454",
    "COUNT_NUM_ZB": "1799454",
    "YES_FLAG": "YES"
  },
  {
    "APP_NAME": "测试1",
    "TABLE_NAME": "hhh",
    "DATE_NAME": "20220903",
    "COUNT_NUM_SC": "1598208",
    "COUNT_NUM_ZB": "1598208",
    "YES_FLAG": "YES"
  },
  {
    "APP_NAME": "测试2",
    "TABLE_NAME": "wewdf",
    "DATE_NAME": "20220904",
    "COUNT_NUM_SC": "1615662",
    "COUNT_NUM_ZB": "1615662",
    "YES_FLAG": "YES"
  }
]

第二步：模版生成

这里使用的是Mako模版, 在项目的/templates目录下创建pdf.html文件，在里面写前端代码

    
    
    



    
                % for name in tableHeaders:
                
                % endfor
            
            % for item in data:
            
                %if mergeData["APP_NAME"][loop.index] > 0:
                  
                %endif
                %if mergeData["TABLE_NAME"][loop.index] > 0:
                  
                %endif
                
            % endfor
        

        

            
${name}
        
        
${ item["APP_NAME"] }
${ item["TABLE_NAME"] }
${ item["DATE_NAME"] }
                
${ item["COUNT_NUM_SC"] }
                
${ item["COUNT_NUM_ZB"] }
                
${ item["YES_FLAG"] }
            
    

使用方法：

t = render_mako_context(request, 'pdf.html', {'data': data, 'mergeData': merge_data,
                                                          'tableHeaders': ch_headers})

content = t.content.decode(encoding="utf-8")

content为生成PDF提供数据

第三步：生成PDF

def pdf_create(wk_path, pdf_name, content, pdf_path):
    """
    生成PDF文件
    wk_path: wkhtmltopdf安装路径:`/usr/local/bin/wkhtmltopdf`
    pdf_path： PDF存放路径
    pdf_name: 生成的PDF文件名称
    content：模版内容
    """
    file_path = pdf_path + pdf_name + '.pdf'
    try:
        config = pdfkit.configuration(wkhtmltopdf=wk_path)
        result = pdfkit.from_string(content, file_path, options={'encoding': "utf-8"},
                           configuration=config)
    except Exception as e:
        logger.error('生成PDF文件异常：%s' % e)
        result = False

扩展：将url转换成PDF

config = pdfkit.configuration(wkhtmltopdf=wk_path)

url = 'http://www.baidu.com/'

pdfkit.from_url(url, file_path, options={'encoding': "utf-8"},
                           configuration=config)

第四步：下载PDF

def export_pdf(self, request, *args, **kwargs):
      """
      file_path: 下载的PDf文件路径
      """
      file = open(file_path, 'rb')
      response = FileResponse(file)
      response['Content-Type'] = 'application/octet-stream'
      response['Content-Disposition'] = 'attachment;filename="{}.pdf"'.format(file_name)
      return response

结果如果：