如何检查传入文件的类型

时间:2017-12-27 10:01:35

标签: excel python-3.x csv text types

我写函数与一些数据一起运行。我从读取文件中的数据开始,将其放入结构中,然后进行一些计算。 我的问题 - 传入的文件可以是txt,csv或xls(xlsx)。 对于不同类型的文件,我需要使用不同的方法。

所以,我的问题是 - 如何检查传入文件的类型。

现在我使用下一种方法:

def main(filename):
    if '.txt' in filename:
        # use read_txt
    elif '.xls' in filename:
        print('there is no txt method')
    else:
        print('etc.')

但它并不完美:例如,如果我创建文件My.txt.xls则失败。 我想到try并逐个使用所有的阅读功能,但看起来有一些更明显的方式......

0 个答案:

没有答案