18 Golang批量修改文件后缀并格式化存储

18 批量修改文件后缀并格式化存储

package main

import (
	"fmt"
	"io"
	"os"
	"path/filepath"
)

func main() {
	dir := "C:/Users/Pip/Desktop/Go初识/"

	// 改变路径
	err := os.Chdir(dir)
	if err != nil {
		fmt.Println("打开文件失败")
		os.Exit(1)
	}

	// 获取文件列表
	fileList, err := filepath.Glob("*")
	if err != nil {
		fmt.Println("获取失败")
		os.Exit(1)
	}
	fmt.Println(fileList)

	// 遍历文件列表修改后缀并新建文件
	for _, name := range fileList {
		// 跳过.md文件
		if string(name[len(name) - 3:]) == ".md"{
			continue
		}
		newName := name[:len(name) - 3] + ".md"

		// 打开.go文件
		oldfile, err := os.OpenFile(name, os.O_RDWR, 0666)
		if err != nil {
			fmt.Printf("打开文件%s失败\n", newName)
			os.Exit(1)
		}
		fmt.Println("Open file:", name, "success")

		// 打开.md文件不存在则新建
		newfile, err := os.OpenFile(newName, os.O_CREATE | os.O_RDWR, 0666)
		if err != nil {
			fmt.Printf("Open file: %s failed\n", newName)
			os.Exit(1)
		}
		fmt.Println("Open file:", newName, "success")

		// 修改格式写入另一个文件
		_, err = newfile.WriteString("# " + name[:len(name)-3] + "\n```\n")
		if err != nil{
			fmt.Println("Write ", newName, "failed")
			os.Exit(1)
		}
		var fileContent = make([] byte, 1024)
		for ;;{
			n, err := oldfile.Read(fileContent)
			if err != nil && err != io.EOF{
				fmt.Println("Write to", newName, "failed")
				os.Exit(1)
			}else if n == 0 {
				break
			}
			newfile.WriteString(string(fileContent)[:n])
		}
		_, err = newfile.WriteString("\n```")
		if err != nil {
			fmt.Println("Write to", newName, "failed")
			os.Exit(1)
		}
		fmt.Println("Write to", string(newName), "success\n")


		//newfile.Sync() // 立即同步,即将内容写入数据
		newfile.Close() // 关闭文件或者程序终止内容才会写入文件
	}
}