데스크탑 정리: Go에서 파일 정리 도구를 구축하세요.

데스크톱이 혼란스럽습니까? 데스크탑이나 다운로드 디렉토리를 어지럽히는 온갖 종류의 파일이 있습니까? 간단한 스크립트로 이를 수정해 보겠습니다.

모든 Go 프로젝트 시작 시와 마찬가지로 "go mod init" 지시문을 사용하여 go.mod 파일을 생성합니다. 작업을 단순하게 유지하기 위해 main.go 파일에 모든 로직을 작성하겠습니다.

코드를 작성하기 전에 스크립트가 어떻게 작동할지 잠시 이야기해 보겠습니다. 우리는 파일 유형이나 생성 날짜를 나타내는 디렉터리로 파일을 구성할 수 있기를 원합니다. 간단히 말해서, 우리는 스크립트가 비디오 파일을 비디오 디렉토리로, 음악 파일을 음악 디렉토리로 정렬하기를 원합니다. 또는 특정 날짜에 생성된 모든 파일을 동일한 디렉토리에 정렬할 수 있습니다.

이제 코딩해 보겠습니다.

main.go 파일을 만들고 다음을 작성하세요.

package main

type fileAnalyzer interface {
    analyzeAndSort() error
}

func analyze(fa fileAnalyzer) error {
    return fa.analyzeAndSort()
}

프로그램이 다양한 기준에 따라 파일을 정렬하기를 원하기 때문에 analyzeAndSort라는 메소드를 정의하는 fileAnalyzer 인터페이스를 만듭니다.
그런 다음 분석 함수를 작성합니다. 이 함수는 fileAnalyzer 인터페이스를 인수로 구현하는 모든 구조체를 취하고 해당 analyzeAndSort 메서드를 실행합니다.

이 프로그램에서 볼 수 있는 것과 같은 경우에는 이동하고 싶지 않은 특정 파일이 있을 수 있습니다. 예를 들어, 스크립트를 테스트하는 동안 프로그램이 Go 파일이나 실행 파일/바이너리를 다른 디렉터리로 이동하는 것을 원하지 않습니다. 이러한 일이 발생하지 않도록 하려면 그대로 유지하고 싶은 모든 파일이 포함된 블랙리스트를 만들어야 합니다.

main.go 파일에 다음을 작성하세요.

var blacklist = []string{
    "go",
    "mod",
    "exe",
    "ps1",
}

여기서 정렬되지 않은 상태로 유지하려는 파일의 파일 확장자를 추가했습니다. ".go" 및 ".mod"는 go 파일의 확장자입니다. 나는 Windows 컴퓨터를 사용하기 때문에 내 바이너리의 확장자는 ".exe"입니다. 보시다시피 개발 과정에서 Powershell 스크립트를 사용하여 작업하는 것을 좋아하기 때문에 ".ps1"도 포함했습니다.

다음으로 몇 가지 도우미 함수를 작성합니다.

func getFileExtension(name string) string {
    return strings.TrimPrefix(filepath.Ext(name), ".")
}

func listFiles(dirname string) ([]string, error) {
    var files []string

    list, err := os.ReadDir(dirname)
    if err != nil {
        return nil, err
    }

    for _, file := range list {
        if !file.IsDir() {
            files = append(files, file.Name())
        }
    }

    return files, nil
}

func listDirs(dirname string) ([]string, error) {
    var dirs []string

    list, err := os.ReadDir(dirname)
    if err != nil {
        return nil, err
    }

    for _, file := range list {
        if file.IsDir() {
            dirs = append(dirs, file.Name())
        }
    }

    return dirs, nil
}

func mkdir(dirname string) error {
    err := os.Mkdir(dirname, 0644)

    if err != nil && os.IsExist(err) {
        return nil
    }

    var e *os.PathError

    if err != nil && errors.As(err, &e) {
        return nil
    }

    return err
}

func moveFile(name string, dst string) error {
    return os.Rename(name, filepath.Join(dst, name))
}

func getCurrentDate(t time.Time) string {
    return t.Format("2006-01-02")
}

func filter[T any](ts []T, fn func(T) bool) []T {
    filtered := make([]T, 0)

    for i := range ts {
        if fn(ts[i]) {
            filtered = append(filtered, ts[i])
        }
    }

    return filtered
}

대부분은 설명이 필요 없지만 "mkdir" 기능에 대해 이야기하고 싶습니다. "mkdir" 함수는 이름이 인수로 전달된 디렉터리를 생성합니다. 그러나 디렉터리가 이미 존재하거나 "os.PathError"가 있는 경우 함수는 오류를 반환하지 않습니다.

이제 fileAnalyzer 인터페이스를 구현하는 구조체를 만들어 보겠습니다.

package main

type fileAnalyzer interface {
    analyzeAndSort() error
}

func analyze(fa fileAnalyzer) error {
    return fa.analyzeAndSort()
}

fileTypeAnalyzer 구조체에는 현재 작업 디렉터리의 이름을 보유하는 wd와 매퍼라는 두 가지 속성이 있습니다. 매퍼의 키는 감지된 파일 유형이고 해당 값은 키와 연관된 파일 확장자 목록입니다. 그런 다음 생성자 함수를 생성하고 파일 유형과 연관 파일 확장자를 매퍼에 제공합니다. 목록에 더 많은 파일 형식과 확장자를 자유롭게 추가하세요. AnaylyzeAndSort 메소드는 파일 확장자를 파일 유형에 매핑하고 파일 유형 디렉토리를 생성하며 해당 디렉토리로 파일을 이동하는 몇 가지 도우미 함수 및 메소드를 호출합니다. 또한 매퍼가 캡처하지 않은 파일을 보관하기 위해 "misc" 폴더를 추가했습니다. 물론 블랙리스트에 있는 파일은 제외됩니다.

또한 생성 날짜별로 파일을 정리할 수 있기를 원합니다. fileAnalyzer 인터페이스를 구현하지만 파일을 날짜별로 정리하는 또 다른 구조체를 만들어 보겠습니다.

var blacklist = []string{
    "go",
    "mod",
    "exe",
    "ps1",
}

많은 로직이 fileTypeAnalyzer의 로직과 동일합니다. 주요 차이점은 매퍼를 제공하지 않는다는 것입니다. 대신 파일 정보에서 생성 날짜를 가져와 그에 따라 디렉토리를 생성합니다.

이제 모든 것을 main 함수에 넣어보겠습니다.

func getFileExtension(name string) string {
    return strings.TrimPrefix(filepath.Ext(name), ".")
}

func listFiles(dirname string) ([]string, error) {
    var files []string

    list, err := os.ReadDir(dirname)
    if err != nil {
        return nil, err
    }

    for _, file := range list {
        if !file.IsDir() {
            files = append(files, file.Name())
        }
    }

    return files, nil
}

func listDirs(dirname string) ([]string, error) {
    var dirs []string

    list, err := os.ReadDir(dirname)
    if err != nil {
        return nil, err
    }

    for _, file := range list {
        if file.IsDir() {
            dirs = append(dirs, file.Name())
        }
    }

    return dirs, nil
}

func mkdir(dirname string) error {
    err := os.Mkdir(dirname, 0644)

    if err != nil && os.IsExist(err) {
        return nil
    }

    var e *os.PathError

    if err != nil && errors.As(err, &e) {
        return nil
    }

    return err
}

func moveFile(name string, dst string) error {
    return os.Rename(name, filepath.Join(dst, name))
}

func getCurrentDate(t time.Time) string {
    return t.Format("2006-01-02")
}

func filter[T any](ts []T, fn func(T) bool) []T {
    filtered := make([]T, 0)

    for i := range ts {
        if fn(ts[i]) {
            filtered = append(filtered, ts[i])
        }
    }

    return filtered
}

로거를 구성합니다. fileAnalyzer 구현에 인수로 전달할 현재 작업 디렉터리를 가져옵니다. 애플리케이션에 플래그로 전달된 값을 보유하는 모드 변수와 정렬 방법을 제어하는 ​​스위치 문을 만듭니다. 마지막으로 분석 함수를 호출하고 fileAnalyzer 구현을 인수로 전달합니다.

그게 다입니다. 바이너리를 빌드하고 테스트해 보겠습니다. 나는 내 분류기에 전화했습니다. "go build -o [name]"을 사용하여 원하는 대로 이름을 지정할 수 있습니다.

다음은 다양한 유형의 파일로 가득 찬 폴더입니다.

파일 형식별로 정리해보자:

파일 생성 날짜별로 정리해보자:

보너스로 Windows 시스템과 Powershell을 사용하는 경우 프로그램 테스트를 쉽게 만드는 데 도움이 되는 스크립트가 있습니다.

task.ps1 파일을 만들고 다음을 입력하세요.

type fileTypeAnalyzer struct {
    wd     string
    mapper map[string][]string
}

func newFileTypeAnalyzer(wd string) *fileTypeAnalyzer {
    return &fileTypeAnalyzer{
        wd: wd,
        mapper: map[string][]string{
            "video":  {"mp4", "mkv", "3gp", "wmv", "flv", "avi", "mpeg", "webm"},
            "music":  {"mp3", "aac", "wav", "flac"},
            "images": {"jpg", "jpeg", "png", "gif", "svg", "tiff"},
            "docs":   {"docx", "csv", "txt", "xlsx"},
            "books":  {"pdf", "epub"},
        },
    }
}

func (f fileTypeAnalyzer) analyzeAndSort() error {
    files, err := listFiles(f.wd)
    if err != nil {
        return fmt.Errorf("could not list files: %w", err)
    }

    if err := f.createFileTypeDirs(files...); err != nil {
        return err
    }

    return f.moveFileToTypeDir(files...)
}

func (f fileTypeAnalyzer) moveFileToTypeDir(files ...string) error {
    dirs, err := listDirs(f.wd)
    if err != nil {
        return fmt.Errorf("could not list directories: %w", err)
    }

    for _, dir := range dirs {
        for _, file := range files {
            if slices.Contains(f.mapper[dir], strings.ToLower(getFileExtension(file))) {
                if err := moveFile(file, dir); err != nil {
                    return err
                }
            }
        }
    }

    files, err = listFiles(f.wd)
    if err != nil {
        return err
    }

    if len(files) == 0 {
        return nil
    }

    files = filter(files, func(f string) bool {
        return !slices.Contains(blacklist, getFileExtension(f))
    })

    for i := range files {
        if err := f.moveToMisc(files[i]); err != nil {
            return err
        }
    }

    return nil
}

func (f fileTypeAnalyzer) moveToMisc(file string) error {
    if err := mkdir("misc"); err != nil {
        return err
    }

    return moveFile(file, "misc")
}

func (f fileTypeAnalyzer) createFileTypeDirs(files ...string) error {
    for ftype, fvalues := range f.mapper {
        for _, file := range files {
            if slices.Contains(fvalues, getFileExtension(file)) {
                if err := mkdir(ftype); err != nil {
                    return fmt.Errorf("could not create folder: %w", err)
                }
            }
        }
    }

    return nil
}

스크립트를 사용하여 바이너리를 빌드하려면:

스크립트로 파일을 정리하려면:

스크립트로 디렉토리를 삭제하려면:

위 내용은 데스크탑 정리: Go에서 파일 정리 도구를 구축하세요.의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!