利用Python獲取文件類型

這里選擇使用使用filetype獲取文件的類型。

使用filetype之前，先用pip安裝filetype。

#!/usr/bin/python3
import filetype
import argparse
import sys

def get_parameter():
    parser=argparse.ArgumentParser(description='該腳本用於獲取文件的類型')
    parser.add_argument('-f',dest='inputFile',type=str,default='',help='輸入待檢測文件')
    args=parser.parse_args()
    inputFile=args.inputFile
    return inputFile

def main():
    inputFile=get_parameter()
    if inputFile=='':
        print('請輸入待檢測文件')
        sys.exit(1)
    ft1=filetype.guess(inputFile)
    if ft1 is None:
        print('無法判斷該文件類型')
    print('文件擴展名為：{}'.format(ft1.extension))
    print('文件類型為：{}'.format(ft1.mime))

if __name__=='__main__':
    main()

本腳本運行環境為python3環境。