当前位置:主页 > 脚本语言 > Golang >

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

时间:2021-06-02 08:22:59 | 栏目:Golang | 点击:

在生活中我们往往有这样的需求,就是从网上找的文件资源经常包含了一些无用信息,而且在它的子目录下也同样存在,如果我们手动一个个的修改的话会特别麻烦,也特别耗时,我们可以考虑把这件事交给计算机来做。
如下图,我的桌面有一个名为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")
}

运行代码

查看结果

您可能感兴趣的文章:

相关文章