如何在python中检查上传的文件是csv还是xls？

如何检查上传文件是CSV还是XLS。 如何在 python 中检查它。我正在将文件导入到 openerp 中的二进制字段，该字段可以作为二进制对象检索。我需要读取文件并将数据导入到表中。用户可以上传 csv 或 xls 文件。只要知道我就可以使用 csv 包或 xlrd 包。

The 的十六进制签名.xls文件如下：

Excel 电子表格子标题 (MS Office)

09 08 10 00 00 06 05 00 [512 byte offset]

您可以阅读有关其他各种签名的信息维基百科.

我相信你可以做这样的事情。这尚未经过测试，但您可以摆弄它直到它起作用。如有任何建议或更改，请留下评论。谢谢！

xls_sig = b'\x09\x08\x10\x00\x00\x06\x05\x00'
offset = 512
size = 8

with open('spreadsheet.xls', 'rb') as f:
    f.seek(offset)       # Seek to the offset.
    bytes = f.read(size) # Capture the specified number of bytes.

    if bytes == xls_sig:
        print 'Uploaded file is an xls.'
    else:
        print 'File is not an xls.'

Update 1

对此进行了测试，我可以验证它是否适用于检测.xls files.

Update 2

我开发了一个程序来确定文件是 xls 还是 xlsx：

import codecs

xlsx_sig = b'\x50\x4B\x05\06'
xls_sig = b'\x09\x08\x10\x00\x00\x06\x05\x00'

filenames = [
    ('spreadsheet.xls', 0, 512, 8),
    ('spreadsheet.xlsx', 2, -22, 4)]

for filename, whence, offset, size in filenames:
    with open(filename, 'rb') as f:
        f.seek(offset, whence) # Seek to the offset.
        bytes = f.read(size)   # Capture the specified number of bytes.

        print codecs.getencoder('hex')(bytes)

        if bytes == xls_sig:
            msg = '"{}" is an xls.'
        elif bytes == xlsx_sig:
            msg = '"{}" is an xlsx.'
        else:
            msg = '"{}" is not an Excel document.'
        print msg.format(filename)

这是输出：

('0908100000060500', 8)
"spreadsheet.xls" is an xls.
('504b0506', 4)
"spreadsheet.xlsx" is an xlsx.