Lua获取文件的后缀及根据后缀判断文件类型

Code如下:

--根据文件名判断文件类型
--其实有文件对象的化直接 Type(fileNmae) 可以更方便快速获取文件类型
local function judgeFileType(fileName)
    local strLen = #fileName
    local po = fileName:find("%.", -4)
    local substr = fileName:sub(po + 1, strLen)
    if substr == "bmp" or substr == "png" or substr == "jpg" then
        print("this file is image!")
    elseif substr == "txt" then
        print("this file is text!")
    elseif substr == "exe" then
        print("this file is program!")
    end
end

local str, str1 = "I think,故我在.bmp", "notText.txt"  --示例文件名
judgeFileType(str)  judgeFileType(str1)                --获取文件类型
local po, po1 = string.find(str, "%.", -4), str1:find("%.", -4)  --反向查找文件 “.” 的位置
local substr, substr1 = str:sub(po + 1, #str), str1:sub(po1 + 1, #str1) --根据 “.” 的位置得到文件后缀
print(string.format("file suffix of name str: %s\nfile suffix of name str1: %s", substr, substr1)) --输出文件后缀

输出如下：