Susun desktop anda: Bina pengatur fail dalam Go.

Adakah desktop anda huru-hara? Adakah anda mempunyai pelbagai jenis fail yang mengacaukan desktop atau direktori muat turun anda? Mari kita betulkan dengan skrip mudah.

Seperti yang kami lakukan pada permulaan sebarang projek go, kami menjana fail go.mod kami dengan arahan "go mod init". Untuk memastikan perkara mudah, kami akan menulis semua logik kami dalam fail main.go kami.

Mari kita bercakap sedikit tentang bagaimana kita akan menyukai skrip untuk berkelakuan sebelum kita menulis sebarang kod. Kami mahu dapat menyusun fail kami ke dalam direktori yang menunjukkan jenis fail atau tarikh penciptaan. Ringkasnya, kami mahu skrip kami mengisih fail video ke dalam direktori video, fail muzik ke dalam direktori muzik dan sebagainya; atau mengisih semua fail yang dibuat pada tarikh tertentu ke dalam direktori yang sama.

Sekarang mari kita kod:

Buat fail main.go dan tulis yang berikut:

package main

type fileAnalyzer interface {
    analyzeAndSort() error
}

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

Oleh kerana kami mahu program kami mengisih fail mengikut kriteria yang berbeza, kami mencipta antara muka Penganalisis fail yang mentakrifkan kaedah: analyzeAndSort.
Kemudian kami menulis fungsi: analisis - yang mengambil mana-mana struct yang melaksanakan antara muka fileAnalyzer sebagai hujah dan melaksanakan kaedah analisisAndSortnya.

Dalam beberapa kes seperti yang akan kita lihat dalam program ini, mungkin terdapat fail tertentu yang anda tidak mahu dialihkan. Contohnya, semasa menguji skrip, kami tidak mahu atur cara mengalihkan fail go kami atau boleh laku/perduaan ke dalam direktori lain. Untuk mengelakkan perkara ini berlaku, kami perlu membuat senarai hitam yang merangkumi semua fail yang kami ingin kekal tidak disentuh.

Dalam fail main.go kami, tulis yang berikut:

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

Di sini, saya menambahkan sambungan fail untuk fail yang saya ingin kekalkan tidak diisih. ".go" dan ".mod" ialah sambungan untuk fail go. Kerana saya menggunakan mesin windows, binari saya akan mempunyai ".exe" sebagai sambungannya. Saya turut sertakan ".ps1" kerana saya suka bekerja dengan skrip powershell dalam pembangunan - seperti yang anda akan lihat.

Seterusnya, kami menulis beberapa fungsi pembantu.

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
}

Kebanyakan ini adalah penjelasan sendiri tetapi saya ingin bercakap tentang fungsi "mkdir". Fungsi "mkdir" mencipta direktori dengan nama yang dihantar kepadanya sebagai argumen - tetapi fungsi itu tidak mengembalikan ralat jika direktori itu sudah wujud atau jika terdapat "os.PathError".

Sekarang mari buat struct yang melaksanakan antara muka Penganalisis fail:

package main

type fileAnalyzer interface {
    analyzeAndSort() error
}

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

Struktur fileTypeAnalyzer mempunyai dua sifat: wd yang memegang nama direktori kerja semasa dan pemeta. Kekunci pemeta akan menjadi jenis fail yang dikesan manakala nilainya ialah senarai sambungan fail yang dikaitkan dengan kunci. Kami kemudian mencipta fungsi pembina dan menyediakan jenis fail serta sambungan fail bersekutunya kepada pemeta. Jangan ragu untuk menambah lebih banyak jenis fail dan sambungan pada senarai. Kaedah anaylyzeAndSort memanggil beberapa fungsi pembantu dan kaedah yang memetakan sambungan fail kepada jenis fail, mencipta direktori jenis fail dan mengalihkan fail ke dalam direktori tersebut. Saya juga menambahkan folder "misc" untuk menyimpan fail yang tidak ditangkap oleh pemeta - sudah tentu tidak termasuk fail yang disenaraihitamkan.

Kami juga mahu dapat menyusun fail mengikut tarikh penciptaan. Mari buat satu lagi struct yang melaksanakan antara muka fileAnalyzer tetapi menyusun fail mengikut tarikh.

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

Banyak logiknya adalah sama seperti dari fileTypeAnalyzer. Perbezaan utama ialah kami tidak menyediakan pemeta - sebaliknya kami mendapat tarikh penciptaan daripada maklumat fail dan mencipta direktori dengan sewajarnya.

Sekarang mari kita susun semuanya dalam fungsi utama kita:

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
}

Kami mengkonfigurasi pembalak; dapatkan direktori kerja semasa untuk diluluskan sebagai hujah kepada pelaksanaan fileAnalyzer kami; buat pembolehubah mod untuk menahan nilai yang dihantar sebagai bendera kepada aplikasi dan penyataan suis untuk mengawal cara kita ingin mengisih. Akhirnya kami memanggil fungsi analisis dan lulus pelaksanaan fileAnalyzer kami sebagai hujah.

Itu sahaja. Mari bina perduaan kami dan uji. Saya memanggil penyusun saya. Anda boleh memanggil nama anda apa sahaja yang anda mahu panggil dengan "go build -o [name]"

Berikut ialah folder yang bersepah dengan fail pelbagai jenis:

Mari kita susun mengikut jenis fail:

Mari kita susun mengikut tarikh penciptaan fail:

Sebagai bonus, jika anda menggunakan mesin windows dan anda menggunakan powershell - berikut ialah skrip untuk membantu menjadikan ujian program anda tidak kelihatan.

Buat fail task.ps1 dan taip yang berikut:

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
}

Untuk membina binari anda dengan skrip:

Untuk membatalkan penyusunan fail dengan skrip:

Untuk memadam direktori dengan skrip:

Atas ialah kandungan terperinci Susun desktop anda: Bina pengatur fail dalam Go.. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!