Go递归修改文件名的实例代码

站长资源 2024/11/22 佚名

2 0 1

在生活中我们往往有这样的需求，就是从网上找的文件资源经常包含了一些无用信息，而且在它的子目录下也同样存在，如果我们手动一个个的修改的话会特别麻烦，也特别耗时，我们可以考虑把这件事交给计算机来做。
如下图，我的桌面有一个名为dir【我爱你】的目录，而且这个目录下的子目录和文件都包含【我爱你】，这个【我爱你】对于我们来说是完全无用的，我们可以用Go的文件操作库来递归对文件进行重命名，把不需要的名字替换为空

示例代码：

package main

import (
	"bufio"
	"fmt"
	"io/ioutil"
	"log"
	"os"
	"strings"
)

var pathSeparator = string(os.PathSeparator)

/**
 * 文件重命名
 */
func rename(path string, old string, new string) (err error) {
	files, err := ioutil.ReadDir(path)
	if err != nil {
		return err
	}
	for _, fileInfo := range files {
		if fileInfo.IsDir() {
			err = rename(path+pathSeparator+fileInfo.Name(), old, new)
			if err != nil {
				return err
			}
			err = os.Rename(path+pathSeparator+fileInfo.Name(), path+pathSeparator+strings.Replace(fileInfo.Name(), old, new, -1))
			if err != nil {
				return err
			}
		} else {
			err = os.Rename(path+pathSeparator+fileInfo.Name(), path+pathSeparator+strings.Replace(fileInfo.Name(), old, new, -1))
			if err != nil {
				return err
			}
		}
	}
	return err
}

func main() {

	//1.获取要被重命名目录(文件)的绝对路径
	fmt.Print("请输入文件的绝对路径：")
	reader := bufio.NewReader(os.Stdin)
	filePath, _ := reader.ReadString('\n')
	filePath = strings.Replace(filePath, "\n", "", -1)

	//2.获取要被替换掉的名称
	fmt.Print("请输入要被替换为空的名称：")
	reader = bufio.NewReader(os.Stdin)
	name, _ := reader.ReadString('\n')
	name = strings.Replace(name, "\n", "", -1)

	//3.递归调用重命名
	err := rename(filePath, name, "")
	if err != nil {
		log.Fatalf("发生错误，错误为：%v\n", err)
	}
	err = os.Rename(filePath, strings.Replace(filePath, name, "", -1))
	if err != nil {
		log.Fatalf("发生错误，错误为：%v\n", err)
	}

	fmt.Println("success")
}

运行代码