读取requests.POST(Python)得到的excel文件

Question

I use the POST method via the requests library to download an EXCEL file.

payload = json.dumps({
  "data_ids": [
    "124902201",
    "124835548"
  ]
})

headers = {
  'Accept': 'application/vnd.ms-excel', 
  'Content-Type': 'application/json',
  }


response=requests.post(url, headers=headers, data=payload)

file_excel=pandas.read_excel(response.content)

When I tried the query on Postman, it returns me a response in Bytes form. So the idea I had was to read this file and convert it to Excel format and then load it into a dataframe. Is it possible.

Answer 1

Install openpyxl

pip install openpyxl

Pandas can read the response.content directly

import pandas as pd
import io

file_extension = "xlsx"

(...)

if file_extension == 'xlsx':
    excel_data = pd.read_excel(resp.content, engine='openpyxl')
elif file_extension == 'xls':
    excel_data = pd.read_excel(resp.content)

EDIT: Ok, now I've tested that this does the trick.

Answer 2

Thanks, Geoffrey Garrett your suggestion worked for reading both versions of excel files from disk. But I had to modify read_excel() for 'xls', by adding engine='xlrd' and I had to install current version of xlrd. Using current version of Pandas also. Hope this helps others.