当前位置：首页 > 文章列表 > Golang > Go教程 > 在 GO 中整理您的下载

在 GO 中整理您的下载

来源：dev.to 2024-10-30 22:48:59 0浏览收藏

Golang小白一枚，正在不断学习积累知识，现将学习到的知识记录一下，也是将我的所得分享给大家！而今天这篇文章《在 GO 中整理您的下载》带大家来了解一下##content_title##，希望对大家的知识积累有所帮助，从而弥补自己的不足，助力实战开发！

在 GO 中整理您的下载

大家好，离子又来了。

害怕学习人工智能有一天会完成的事情让我感到非常痛苦。但是，如果“解决问题”仍然是对未来人类的要求，为什么不坚持呢？

这次我又带来了一个教程。比第一个没啥用。那么我们来定义一下“问题”的结构，因为我们已经知道一件事：那些没有问题的人，是因为他们看起来不够。对于那些还没有找到它们的人来说，创建它们只是时间问题。

项目结构

程序最简单的结构是：

扫描文件夹（例如下载文件夹或其他目录）
识别相关目录中每个文件的类型
将文件移动到与其类型对应的子文件夹（图像、视频、文档...）

启动项目

创建一个目录并导航到它：

mkdir organizador
cd organizador

创建一个 organizer.go 文件并启动其模块：

touch organizador.go
go mod init organizador.go

你应该有或多或少像这样的东西：

~/organizador
.
├── go.mod
└── organizador.go

第 1 部分：检查目录是否存在

让我们定义源目录dirorigem，我们将在其中执行组织。定义后，我们检查它是否确实存在，否则我们将返回错误：

package main

import (
    "fmt"
    "os"
)

// defina o deretório o qual você quer organizar como variável global
var dirorigem string := "/users/user/downloads" // troque o diretório 

func main() {

    // verificar se o diretório existe, caso contrário, retornar erro
    if _, err := os.stat(dirorigem); os.isnotexist(err) {
        fmt.println("bad dir :( \ndiretório não encontrado: ", dirorigem)
        return
    } else {
        // imprimir mensagem caso o diretório exista
        fmt.println("good dir :) \ndiretório encontrado: ", dirorigem)
    }
}

现在，让我们对上面的代码进行一些考虑：

os.stat函数返回两个类型为os.fileinfo和错误err的值。

fileinfo 是一个返回详细文件信息的接口，在本例中我们不需要。因此，为了忽略这个接口，我们使用_,。但是，我们不想忽略错误： if _, err := ...
我们将错误 err 作为参数传递给 os.isnotexist() 函数，因为如果目录不存在，os.stat() 函数将返回 not null 错误，从而导致os.isnotexist() 返回 true，执行我们的消息：bad dir :(
如果 os.isnotexist() 返回 false，我们将从 else 条件中打印消息：good dir :)

第 2 部分：回调函数概念太疯狂了，伙计！

你有没有注意到，我们在这里一点一点地享受机械键盘声音的比特和字节。 _查卡查卡繁荣！ _

现在我们要创建一个回调函数。这是我从未真正了解过的东西，或者从来没有足够的好奇心来质疑我是否在我之前的 golang 生活中的一些 python 代码中使用过这个概念。

回调函数是一个带有参数传递给另一个函数的函数。

如果您已经熟悉这个概念，那么恭喜您，否则，恭喜您。换句话说，恭喜你！

现在让我们创建一个 filepath.walk 回调函数，它将作为参数传递给另一个函数。

func main() { 
// restante do código
.
.
.

// percorrer e listar os arquivos no diretório dirorigem
    err := filepath.walk(dirorigem, listararquivos)
    if err != nil {
        fmt.println("erro ao percorrer o diretório: ", err)
    }
}

// função que lista os arquivos do diretório
func listararquivos(caminho string, info os.fileinfo, err error) error {
    if err != nil {
        return err
    }

    // ignorar diretórios e exibir apenas arquivos
    if !info.isdir() && !strings.hasprefix(info.name(), ".") {
        fmt.println("arquivo encontrado: ", info.name())
    }
    return nil
}

但是等等，filepath.walk 是如何调用回调函数的？

当您调用 filepath.walk(sourcedir, listfiles) 时，filepath.walk 函数会执行遍历 sourcedir 中的所有文件和子目录的繁重工作。

对于找到的每个文件或目录，它都会使用三个参数调用 listfiles 函数：

path：当前文件或目录的完整路径。
info：一个 os.fileinfo 对象，包含文件/目录的信息（例如名称、是否是目录、大小等）。
err：如果访问此文件或目录时出现问题，则会出现错误。

go 自动理解 listfiles 必须接收这三个参数，因为 filepath.walk 期望一个完全遵循此签名的函数：

// função walk()
func walk(root string, walkfn walkfunc) error

// tipo walkfunc
type walkfunc func(path string, info os.fileinfo, err error) error

请注意，walk 函数返回一个错误！这是相关的！

这就是为什么我们将函数 filepath.walk(dirorigem,listararquivos) 等同于错误：

err := filepath.walk(dirorgiem, listararquivos)

毕竟，因为它返回一个错误，所以它是一个错误xd

实际例子

以下更详细地介绍了每个步骤发生的情况：

//percorrer e listar os arquivos no diretório
err := filepath.walk(dirorigem, listararquivos)

对于 dirorigem 中的每个文件或目录，filepath.walk 将调用 listfiles，就好像它是这样的：

listararquivos("/users/user/downloads/file1.txt", infosobrefile1, nil)
listararquivos("/users/user/downloads/file2.txt", infosobrefile2, nil)
listararquivos("/users/user/downloads/file3.mp3", infosobrefile3, nil)

在此示例中，对于每个调用：

path：接收文件或目录的路径。
信息：包含有关此项目的信息（例如名称和类型）。
err：用于捕获访问文件/目录时的任何特定错误。

回调函数

listfiles 是 filepath.walk 使用这些值自动调用的回调函数。这样，我们就不需要担心设置path、info和err值； filepath.walk 已经为我们做到了这一点。

phe！

现在在你的终端中进行顽皮的测试：

go run organizador.go

你可以得到结果：

good dir :)
diretório encontrado:  /users/user/downloads
arquivo encontrado:  suic_ sos · 4.32pm · 10-25.jpeg
arquivo encontrado:  suic_ sos.jpeg
arquivo encontrado:  quantum-distortion-cyberpunk-music-234058.mp3
arquivo encontrado:  slow-mode-enabled-cyberpunk-music-230623.mp3
arquivo encontrado:  teaching-lp-20152-seminario-go.pdf

或者：

go run organizador.go
bad dir :(
diretório não encontrado:  /users/user/downloadss

在这种情况下，我只是在“下载”中添加了一个额外的“s”，这样原始目录就会不正确。

现在删除 listfiles 函数，因为我们不会使用它。

开个玩笑，我们只是要更改她的名字并添加其他逻辑。

第 3 部分：组织！=组织

组织得很好，组织得很棒。

在我的精彩观察之后，让我们继续我们真正感兴趣的部分：组织整个事情。

人生的一个讽刺是，在整理文件之前，我们必须先整理一下下一步的想法。

我们的下一个功能基本上需要：

根据 dirorigem 目录中每个文件的扩展名创建子文件夹（如果不存在）。
根据扩展名将文件移动到各自的文件夹中。
但是如果文件已经在组织子文件夹中，则不应再次创建它们。

让我们了解一下这段代码的每个部分的作用：

// função que organiza os arquivos do diretório
func organizararquivos(caminho string, info os.fileinfo, err error) error {
    if err != nil {
        return err
    }

    // ignorar diretórios e exibir apenas arquivos
    if !info.isdir() && !strings.hasprefix(info.name(), ".") {
        fmt.println("arquivo encontrado: ", info.name())

        // obter extensões em letras minúsculas
        extensao := strings.tolower(filepath.ext(info.name()))

        // criar nome das subpastas diretamente em dirorigem
        subpasta := filepath.join(dirorigem, extensao[1:])

        // verificar se a subpasta já existe; caso contrário, criar a pasta
        if _, err := os.stat(subpasta); os.isnotexist(err) {
            err := os.mkdir(subpasta, os.modeperm)
            if err != nil {
                fmt.println("erro ao criar subpasta: ", err)
                return err
            }
        }

        // caminho de destino para mover arquivos
        caminhodestino := filepath.join(subpasta, info.name())

        // verificar se o arquivo já está na subpasta
        if caminhodestino == caminho {
            fmt.printf("o arquivo %s já está na subpasta %s. ignorando...\n", info.name(), subpasta)
        } else {
            // mover o arquivo para a subpasta
            err := os.rename(caminho, caminhodestino)
            if err != nil {
                fmt.println("erro ao mover arquivo: ", err)
                return err
            }
            fmt.printf("arquivo %s movido para %s\n", info.name(), subpasta)
        }
    }
    return nil
}

组织文件函数的结构

为目录结构中找到的每个文件或文件夹调用organizefiles函数。它检查根据扩展名组织每个文件的条件，创建子文件夹并在必要时移动文件。

func organizararquivos(caminho string, info os.fileinfo, err error) error {
    if err != nil {
        return err
    }

这里，organizefiles 函数采用三个参数：

path：当前文件或目录的完整路径。
info：文件或目录信息，从类型 os.fileinfo.
err：尝试访问该项目时可能发生的错误。

首先检查访问文件/目录时是否出现错误。如果是，则立即返回。

过滤文件并忽略隐藏目录

if !info.isdir() && !strings.hasprefix(info.name(), ".") {
        fmt.println("arquivo encontrado: ", info.name())

此片段进行两项检查：

!info.isdir()：检查该项目是否不是目录（即，它是文件）。
!strings.hasprefix(info.name(), ".")：检查文件名是否不以“.”开头，忽略基于 unix 的系统上的隐藏文件。

如果两个条件都满足，则文件将显示为 fmt.println。

识别文件扩展名并创建子文件夹名称

extensao := strings.tolower(filepath.ext(info.name()))
subpasta := filepath.join(dirorigem, extensao[1:])

这里：

strings.tolower(filepath.ext(info.name()))：提取文件扩展名（例如.txt）并将其转换为小写以确保一致性。
subfolder := filepath.join(sourcedir, extension[1:])：创建文件将移动到的子文件夹的完整路径。扩展名[1:]去掉扩展名中的起始点(.)，形成子文件夹的名称，如txt。

如果子文件夹尚不存在，则创建它

if _, err := os.stat(subpasta); os.isnotexist(err) {
            err := os.mkdir(subpasta, os.modeperm)
            if err != nil {
                fmt.println("erro ao criar subpasta: ", err)
                return err
            }
        }

这里，函数：

使用 os.stat 检查子文件夹是否已存在。
如果子文件夹不存在 (os.isnotexist(err))，则使用 os.mkdir(subfolder, os.modeperm) 创建它。
os.modeperm 设置新文件夹的默认权限。如果创建文件夹时出错，则会显示并返回。

设置文件目标路径

caminhodestino := filepath.join(subpasta, info.name())

此时，destinationpath 代表文件将被移动的最终路径。它是使用 filepath.join 构造的，用于将子文件夹路径连接到文件名。

检查文件是否已位于目标文件夹中

if caminhoDestino == caminho {
            fmt.Printf("O arquivo %s já está na subpasta %s. Ignorando...\n", info.Name(), subpasta)
        } else {
            // Mover o arquivo para a subpasta
            err := os.Rename(caminho, caminhoDestino)
            if err != nil {
                fmt.Println("Erro ao mover arquivo: ", err)
                return err
            }
            fmt.Printf("Arquivo %s movido para %s\n", info.Name(), subpasta)
        }
    }
    return nil
}