利用Python获取文件类型

佚名 4年前 (2020-03-31) Python 2316人围观抢沙发百度已收录

这里选择使用使用filetype获取文件的类型。

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

SRE实战互联网时代守护先锋，助力企业售后服务体系运筹帷幄！一键直达领取阿里云限量特价优惠。

#!/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环境。