go入门项目：(2) 基于命令行的文件整理小项目

介绍

假设你之前下载的文件(音频,视频,文档,图片等)都下载到了一个文件夹里, 现在想把它们分类整理到相应的文件夹里去. 本文使用go实现这个小功能.

代码实现

• 从命令行输入想要整理的目录
  main.go
  1. // 获取用户输入 - 目标文件夹需要整理
  2. scanner := bufio.NewScanner(os.Stdin)
  3. fmt.Print("你想整理哪个文件夹 - ")
  4. scanner.Scan()
  6. targetFolder := scanner.Text()

  复制代码
• 判断是否存在该目录
  1. v// 检查文件夹是否存在
  2. _, err := os.Stat(targetFolder)
  3. if os.IsNotExist(err) {
  4.         log.Fatal("文件夹不存在")
  5. } else {
  6.         // 创建默认文件夹，如Images, Docs, Others, Videos
  7.         createDefaultFolders(targetFolder)
  9.         // 整理文件夹
  10.         organizeFolder(targetFolder)
  11. }

  复制代码
• 分类处理
  1. switch fileExt {
  2.         case ".jpg", ".JPG", "png":
  3.         newPath := filepath.Join(targetFolder, "Images", fileInfo.Name())
  4.         err := os.Rename(oldPath, newPath)
  5.         check(err)
  6.         noOfFiles++
  7.         case ".mp4", ".mov", ".avi", ".amv":
  8.         newPath := filepath.Join(targetFolder, "Videos", fileInfo.Name())
  9.         err := os.Rename(oldPath, newPath)
  10.         check(err)
  11.         noOfFiles++
  12.         case ".pdf", ".docx", ".csv", ".xlsx":
  13.         newPath := filepath.Join(targetFolder, "Docs", fileInfo.Name())
  14.         err = os.Rename(oldPath, newPath)
  15.         check(err)
  16.         noOfFiles++
  17.         default:
  18.         newPath := filepath.Join(targetFolder, "Others", fileInfo.Name())
  19.         err = os.Rename(oldPath, newPath)
  20.         check(err)
  21.         noOfFiles++
  22. }

  复制代码

1. 4. 根目录下使用go build生成可执行文件, 通过命令行执行该文件
2. ```bash
3. go build

复制代码

完整代码

1. package main
3. import (
4.         "bufio"
5.         "fmt"
6.         "log"
7.         "os"
8.         "path/filepath"
9. )
11. func main() {
12.         // 获取用户输入 - 目标文件夹需要整理
13.         scanner := bufio.NewScanner(os.Stdin)
14.         fmt.Print("你想整理哪个文件夹 - ")
15.         scanner.Scan()
17.         targetFolder := scanner.Text()
19.         // targetFolder := "/home/cenjw/Downloads/Inspackage"
21.         // 检查文件夹是否存在
22.         _, err := os.Stat(targetFolder)
23.         if os.IsNotExist(err) {
24.                 log.Fatal("文件夹不存在")
25.         } else {
26.                 // 创建默认文件夹，如Images, Docs, Others, Videos
27.                 createDefaultFolders(targetFolder)
29.                 // 整理文件夹
30.                 organizeFolder(targetFolder)
31.         }
32. }
34. // function to Oraganize folders
35. func organizeFolder(targetFolder string) {
36.         // 读取目录
37.         filesAndFolders, err := os.ReadDir(targetFolder)
38.         check(err)
40.         // 统计有多少文件需要被移动
41.         noOfFiles := 0
43.         for _, filesAndFolder := range         filesAndFolders {
44.                 // 如果是文件就移动到相应的文件夹
45.                 if !filesAndFolder.IsDir() {
46.                         fileInfo, err := filesAndFolder.Info()
47.                         check(err)
49.                         // 获取文件的绝对路径
50.                         oldPath := filepath.Join(targetFolder, fileInfo.Name())
51.                         fileExt := filepath.Ext(oldPath) // 返回指定路径使用的文件扩展名 如 ".csv"
53.                         switch fileExt {
54.                         case ".jpg", ".JPG", "png":
55.                                 newPath := filepath.Join(targetFolder, "Images", fileInfo.Name())
56.                                 err := os.Rename(oldPath, newPath)
57.                                 check(err)
58.                                 noOfFiles++
59.                         case ".mp4", ".mov", ".avi", ".amv":
60.                                 newPath := filepath.Join(targetFolder, "Videos", fileInfo.Name())
61.                                 err := os.Rename(oldPath, newPath)
62.                                 check(err)
63.                                 noOfFiles++
64.                         case ".pdf", ".docx", ".csv", ".xlsx":
65.                                 newPath := filepath.Join(targetFolder, "Docs", fileInfo.Name())
66.                                 err = os.Rename(oldPath, newPath)
67.                                 check(err)
68.                                 noOfFiles++
69.                         default:
70.                                 newPath := filepath.Join(targetFolder, "Others", fileInfo.Name())
71.                                 err = os.Rename(oldPath, newPath)
72.                                 check(err)
73.                                 noOfFiles++
74.                         }
75.                 }
76.         }
78.         if noOfFiles > 0 {
79.                 fmt.Printf("%v number of files moved\n", noOfFiles)
80.         } else {
81.                 fmt.Printf("No files moved")
82.         }
83. }
85. func check(err error) {
86.         if err != nil {
87.                 log.Fatal(err)
88.         }
89. }
91. // function to create default folders such as Images, Docs, Others, Videos
92. func createDefaultFolders(targetFolder string) {
93.         defaultFolders := []string{"Videos", "Docs", "Images", "Others"}
95.         for _, folder := range defaultFolders {
96.                 _, err := os.Stat(folder)
97.                 if os.IsNotExist(err) {
98.                         os.Mkdir(filepath.Join(targetFolder, folder), 0755)
99.                 }
100.         }
101. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！