有時(shí)候,我看著我的目錄,或者我想找個(gè)文件時(shí),尤其是目錄很多時(shí),總是有點(diǎn)難受。
比如這個(gè)目錄,是我的一個(gè)教程目錄。
但是我并不能一目了然的知道每個(gè)文件夾下都有啥,并且每次找一個(gè)文件也很費(fèi)事。
所以,基于上述的需求,使用Go語(yǔ)言,終于擼了個(gè)目錄生成器,感覺(jué)還行。
通過(guò)編寫(xiě)的腳本,可以將目錄整合成.txt文件,并且下級(jí)目錄使用4個(gè)空格縮進(jìn)。
先別管怎么實(shí)現(xiàn)的,先看代碼。
package mainimport ( "bytes" "flag" "fmt" "io/ioutil" "os" "path/filepath")func GenderNSymbol(char byte, count int) string { symbolSince := bytes.Repeat([]byte{char}, count) var symbol = string(symbolSince) return symbol}// @title GenderDirTree// @description 生成目錄樹(shù)// @param path string "需要生成的目錄"http:// @param count int "生成相同字符的個(gè)數(shù)"http:// @param char byte "生成相同字符的字符"http:// @param current_tier int "當(dāng)前層數(shù)"http:// @param end_tier int "終止層數(shù)"http:// @param fileObj int "文件對(duì)象"http:// @return 無(wú) 無(wú) "無(wú)"func GenderDirTree(path string, count int, char byte, current_tier int, end_tier int, fileObj *os.File) { if !(current_tier < end_tier) && end_tier != 0 { return} current_tier++ files, err := ioutil.ReadDir(path) if err != nil { fmt.Println("錯(cuò)誤:目錄錯(cuò)誤") return} for _, file := range files { var name = file.Name() //生成指定數(shù)目的相同符號(hào) var plac = GenderNSymbol(char, count) //符號(hào)+目錄 var sname = fmt.Sprintf("%s%s/n", plac, name) //輸出當(dāng)前目錄 fileObj.WriteString(sname) //fmt.Println(sname) //判斷是否為目錄,如果是,繼續(xù)下次遞歸 var isDir = file.IsDir() if isDir { //拼接傳入的目錄和循環(huán)的當(dāng)前目錄 var nerPaht = filepath.Join(path, name) GenderDirTree(nerPaht, count+4, char, current_tier, end_tier, fileObj) }}}func main() { //終止層數(shù),0表示無(wú)限層,>0表示指定層數(shù) var end_tier int //輸入的目錄 //var path = `D:/0_教程/易錦教程` var path string flag.StringVar(&path, "path", "", "目錄") flag.IntVar(&end_tier, "tier", 0, "終止層數(shù)") flag.Parse() //文件對(duì)象 var wDirPath = filepath.Join(path, "目錄.txt") fileObj, _ := os.OpenFile(wDirPath, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0666) defer fileObj.Close() //生成目錄樹(shù) GenderDirTree(path, 0, ' ', 0, end_tier, fileObj)}
去掉注釋,其實(shí)也就30行代碼,就實(shí)現(xiàn)了這個(gè)功能。
go build main.go
采用flag包動(dòng)態(tài)控制參數(shù),參數(shù)如下。
d:>main.exe -hUsage of main.exe: -path string 目錄 -tier int 終止層數(shù)
main.exe -path <要生成的目錄> -tier <終止層級(jí)>注意:-tier,終止層數(shù),0表示無(wú)限層,>0表示指定層數(shù),默認(rèn)為0
此處-tier指定的是0,表示無(wú)限制生成,如果指定是1,就表示生成一層。
會(huì)在要生成的目錄下,多出一個(gè)目錄.txt。
本次更像是日常使用中,為了解決實(shí)際問(wèn)題的腳本,其實(shí)腳本使用Python更好。
但是Go有一個(gè)更大的優(yōu)勢(shì)就是可以編譯為exe。
這次的難點(diǎn)主要是理解遞歸,自己調(diào)自己!
本文鏈接:http://www.tebozhan.com/showinfo-26-57881-0.html手把手教你用Go語(yǔ)言開(kāi)發(fā)一款簡(jiǎn)易目錄生成器
聲明:本網(wǎng)頁(yè)內(nèi)容旨在傳播知識(shí),若有侵權(quán)等問(wèn)題請(qǐng)及時(shí)與本網(wǎng)聯(lián)系,我們將在第一時(shí)間刪除處理。郵件:2376512515@qq.com
上一篇: 掌握Python之道:正確學(xué)習(xí)官方文檔的有效方法
下一篇: 在 Rust 編程中使用多線程