lua工具类（一）

末蓝、 2022-07-15 11:44 136阅读 0赞

转自：[http://www.cnblogs.com/newlist/p/3595739.html][http_www.cnblogs.com_newlist_p_3595739.html]

--
    -- Author: My Name
    -- Date: 2013-12-16 18:52:11
    -- csv解析 
    -- 
    
    -- 去掉字符串左空白 
    local function trim_left(s)  
        return string.gsub(s, "^%s+", "");  
    end  
    
    -- 去掉字符串右空白 
    local function trim_right(s)  
        return string.gsub(s, "%s+$", "");  
    end  
    
    -- 解析一行 
    local function parseline(line)  
        local ret = {};  
    
        local s = line .. ",";  -- 添加逗号,保证能得到最后一个字段 
    
        while (s ~= "") do  
            --print(0,s); 
            local v = "";  
            local tl = true;  
            local tr = true;  
    
            while(s ~= "" and string.find(s, "^,") == nil) do  
                --print(1,s); 
                if(string.find(s, "^\"")) then  
                    local _,_,vx,vz = string.find(s, "^\"(.-)\"(.*)");  
                    --print(2,vx,vz); 
                    if(vx == nil) then  
                        return nil;  -- 不完整的一行 
                    end  
    
                    -- 引号开头的不去空白 
                    if(v == "") then  
                        tl = false;  
                    end  
    
                    v = v..vx;  
                    s = vz;  
    
                    --print(3,v,s); 
    
                    while(string.find(s, "^\"")) do  
                        local _,_,vx,vz = string.find(s, "^\"(.-)\"(.*)");  
                        --print(4,vx,vz); 
                        if(vx == nil) then  
                            return nil;  
                        end  
    
                        v = v.."\""..vx;  
                        s = vz;  
                        --print(5,v,s); 
                    end  
    
                    tr = true;  
                else  
                    local _,_,vx,vz = string.find(s, "^(.-)([,\"].*)");  
                    --print(6,vx,vz); 
                    if(vx~=nil) then  
                        v = v..vx;  
                        s = vz;  
                    else  
                        v = v..s;  
                        s = "";  
                    end  
                    --print(7,v,s); 
    
                    tr = false;  
                end  
            end  
    
            if(tl) then v = trim_left(v); end  
            if(tr) then v = trim_right(v); end  
    
            ret[table.getn(ret)+1] = v;  
            --print(8,"ret["..table.getn(ret).."]=".."\""..v.."\""); 
    
            if(string.find(s, "^,")) then  
                s = string.gsub(s,"^,", "");  
            end  
    
        end  
    
        return ret;  
    end  
    
    
    
    --解析csv文件的每一行 
    local function getRowContent(file)  
        local content;  
    
        local check = false  
        local count = 0  
        while true do  
            local t = file:read()  
            if not t then  if count==0 then check = true end  break end  
    
            if not content then  
                content = t  
            else  
                content = content..t  
            end  
    
            local i = 1  
            while true do  
                local index = string.find(t, "\"", i)  
                if not index then break end  
                i = index + 1  
                count = count + 1  
            end  
    
            if count % 2 == 0 then check = true break end  
        end  
    
        if not check then  assert(1~=1) end  
        return content  
    end  
    
    
    
    --解析csv文件 
    function LoadCsv(fileName)  
        local ret = {};  
    
        local file = io.open(fileName, "r")  
        assert(file)  
        local content = {}  
        while true do  
            local line = getRowContent(file)  
            if not line then break end  
            table.insert(content, line)  
        end  
    
        for k,v in pairs(content) do  
            ret[table.getn(ret)+1] = parseline(v);  
        end  
    
    
        file:close()  
    
        return ret  
    end  
    
    function Csv2Lua(fileName, filePath, desFilePath )
        local t = LoadCsv(filePath);
        if t then
            t = luautil.serialize(fileName,t);
        end
        if t then
            luautil.writefile(t, desFilePath) 
        end
    end
    
    --test 
    --local t= LoadCsv("csvtesttxt.csv") 
    --for k,v in pairs(t) do 
    -- local tt = v 
    -- local s = "" 
    -- for i,j in pairs(tt) do 
    -- s = string.format("%s,%s",s,j) 
    -- end 
    -- print ("",s) 
    --end

[http_www.cnblogs.com_newlist_p_3595739.html]: http://www.cnblogs.com/newlist/p/3595739.html