背景描述
编程或者文档处理过程, 经常遇到需要将一个单词修改为另外一个单词的情况, 例如 命名为 shall 修改 为 should。
使用工具实现, 则比较方便,不容易出错, 解放双手。
需求规格
对于某个文件夹中的所有文本文件(txt), 将某个单词替换为目标单词。
实现思路
对于替换的单词映射, 在配置文件config.lua进行设置, 存储一个表,表中每一行 对应 src vocanbulary 和 dest vocanbulary
对应工具的主题逻辑代码在 replace.lua中实现,
待替换的文本文件存储在 replaceFiles文件夹下。
总体目录结果如下:
│ config.lua
│ replace.lua
│
└─replaceFiles
test.txt
代码说明
代码实现路径:
https://github.com/fanqingsong/code-snippet/tree/master/lua/replace
config.lua
-- ttanslating table, in every line first word is source, second word is destination
trans_table_string = [[
you lucy
]]
待替换文件 test.txt
I love you
replace.lua工具逻辑代码实现
--[[
/*******************************************************************************
* Author:
* Date:
* Description: set config for replace, replace by config in files of target path
* Changes:
*******************************************************************************/
]]local require = require
local io = io
local ipairs = ipairs
local assert = assert
local print = print
local string = string
local lfs = require"lfs"local transFilePath = "./replaceFiles"
string.split = function(str, pat, max, regex)
pat = pat or "\n"
max = max or #strlocal t = {}
local c = 1if #str == 0 then
return {""}
endif #pat == 0 then
return nil
endif max == 0 then
return str
endrepeat
local s, e = str:find(pat, c, not regex)
max = max - 1
if s and max < 0 then
t[#t+1] = str:sub(c)
else
t[#t+1] = str:sub(c, s and s - 1)
end
c = e and e + 1 or #str + 1
until not s or max < 0return t
end------------------------------------------ parse start ------------------------------------------
local trans_vocabulary_table = {
--["sourcevocabulary"] = "destvocabulary"
}local function construct_vocabulary_table()
require("config")
print("parse trans_table starting .....")
-- 禄帽募镁?拢卢 禄禄?路没 \n 潞?\r\n
local lineSep = "\r\n"
if string.find(trans_table_string, "\r\n") then
lineSep = "\r\n"
elseif string.find(trans_table_string, "\n") then
lineSep = "\n"
elseif string.find(trans_table_string, "\r") then
lineSep = "\r"
endlocal lines = trans_table_string:split(lineSep)
for _,line in ipairs(lines) do
print("line="..line)
local src, dest = string.match(line, "([%w_]+)%s+([%w_]+)")if src then
print("well formed line="..line)
trans_vocabulary_table[src] = dest
end
end
print("parse trans_table ending .....")
end-- parse table
construct_vocabulary_table()------------------------------------------ parse end ------------------------------------------
------------------------------------------ read file list start ------------------------------------------
local targetFiles = {}local function infilter(file, filters)
if filters == nil or filters == "*" then
return true
endfor _, v in pairs(filters) do
if string.find(file, "%."..v.."$") then
return true
end
end
return false
endlocal function splitonlast (path, sep)
local dir, file = string.match(path,"^(.-)([^:/\\]*)$")
return dir, file
endfunction readdir(dir, filelist, filters)
for file in lfs.dir(dir) do
if file ~= ".." and file ~= "." then
local f = dir.."/"..file
if lfs.attributes(f).mode == "directory" then
readdir(f, filelist, filters)
else
if infilter(file, filters) then
table.insert(filelist, f)
end
end
end
end
endreaddir(transFilePath, targetFiles, {"*"})
for _,file in ipairs(targetFiles) do
--print("c file =".. file)
end------------------------------------------ read file list end ------------------------------------------
------------------------------------------ handle file start ------------------------------------------local function handle_file(file)
local lineBuff = {}-- ?赂?搂?
local fh = assert(io.open (file, "rb"))
local contents = fh:read("*a")
fh:close()
--print(contents)
for src,dest in pairs(trans_vocabulary_table) do
print(src.."==>"..dest)
contents = string.gsub(contents, src, dest)
end--[[
-- 禄帽募镁?拢卢 禄禄?路没 \n 潞?\r\n
local lineSep = "\r\n"
if string.find(contents, "\r\n") then
lineSep = "\r\n"
elseif string.find(contents, "\n") then
lineSep = "\n"
elseif string.find(contents, "\r") then
lineSep = "\r"
endlocal fileLines = string.split(contents, lineSep)
for _,line in ipairs(fileLines) do
--print(" handle_file line= "..line)
local gotPattern = false
for src,dest in pairs(trans_vocabulary_table) do
--print("src="..src.."----")
local s, e = string.find(line, "%s-%(%s-"..src.."%s-,%s-%\"")
if s then
print("!!!! ------- gotPattern ------- src ="..src)gotPattern = true
-- the part before OssUsersrc
local head = string.sub(line, 1, s-1)
-- tail part = now");
--print(head)
local tail = string.sub(line, e+1)
--print("tail="..tail)
--print("tail[1]="..string.sub(tail, 1,1))-- OssUserLogType(LOG_LEVEL_NOTICE, LOG_TYPE_SYSTEM, "the system will reboot now");
local level = dest["level"]
local types = dest["types"]
local msg = dest["msg"]local sep = " "
if msg == "" then
sep = ""
elseif string.sub(tail, 1,1) == "\"" then
sep = ""
end
--print("msg="..msg.."sep="..sep.."--")
local transLine = head .. "OssUserLogType(" .. level ..", " .. types .. ", \"" .. msg .. sep .. tailtable.insert(lineBuff, transLine)
if gotPattern then
break
endend
endif not gotPattern then
table.insert(lineBuff, line)
end
end
]]
--write buff to orig file
local fh = assert(io.open(file, "wb"))
fh:write(contents)
fh:close()
endfor _,file in ipairs(targetFiles) do
print("handling file =".. file)
handle_file(file)
end------------------------------------------ handle file end ------------------------------------------
运行结果
root@fqs:/home/share/luascript/replace# cat ./replaceFiles/test.txt
I love you
root@fqs:/home/share/luascript/replace# lua replace.lua
parse trans_table starting .....
line= you lucy
well formed line= you lucy
line=
parse trans_table ending .....
handling file =./replaceFiles/test.txt
you==>lucy
root@fqs:/home/share/luascript/replace#
root@fqs:/home/share/luascript/replace#
root@fqs:/home/share/luascript/replace# cat ./replaceFiles/test.txtI love lucy
root@fqs:/home/share/luascript/replace#