【原创】批量区分横屏竖屏照片

 

程序功能：

• 扫描指定目录下的所有图片文件（支持jpg、jpeg、png、gif格式）
• 根据图片宽高比判断是横屏（宽>高）还是竖屏（高≥宽）
• 在源目录下创建两个子文件夹：landscape（横屏）和portrait（竖屏）
• 将图片复制到对应的文件夹中
• 显示处理统计信息

使用方法：

1. 将代码保存为main.go文件
2. 打开命令行终端
3. 运行命令：go run main.go <图片目录路径>

go run main.go /path/to/your/photos

程序会在指定的目录下创建两个子文件夹，并将图片分类复制到对应的文件夹中。原始图片保持不变，所以您不用担心图片丢失。

package main

import (
	"fmt"
	"image"
	_ "image/gif"
	_ "image/jpeg"
	_ "image/png"
	"io/fs"
	"os"
	"path/filepath"
	"strings"
)

func main() {
	// 检查命令行参数
	if len(os.Args) < 2 {
		fmt.Println("用法: go run main.go <图片目录路径>")
		return
	}

	// 获取图片目录路径
	sourcePath := os.Args[1]

	// 创建横屏和竖屏目录
	landscapePath := filepath.Join(sourcePath, "landscape")
	portraitPath := filepath.Join(sourcePath, "portrait")

	// 确保目标目录存在
	os.MkdirAll(landscapePath, os.ModePerm)
	os.MkdirAll(portraitPath, os.ModePerm)

	// 统计
	var totalCount, landscapeCount, portraitCount, errorCount int

	// 遍历目录下的所有文件
	err := filepath.WalkDir(sourcePath, func(path string, d fs.DirEntry, err error) error {
		if err != nil {
			return err
		}

		// 跳过目录
		if d.IsDir() {
			return nil
		}

		// 检查文件是否为图片
		ext := strings.ToLower(filepath.Ext(path))
		if ext != ".jpg" && ext != ".jpeg" && ext != ".png" && ext != ".gif" {
			return nil
		}

		// 跳过已经分类的图片
		if strings.Contains(path, landscapePath) || strings.Contains(path, portraitPath) {
			return nil
		}

		totalCount++

		// 打开图片文件
		file, err := os.Open(path)
		if err != nil {
			fmt.Printf("打开文件 %s 失败: %v\n", path, err)
			errorCount++
			return nil
		}
		defer file.Close()

		// 解码图片以获取尺寸
		img, _, err := image.DecodeConfig(file)
		if err != nil {
			fmt.Printf("解码图片 %s 失败: %v\n", path, err)
			errorCount++
			return nil
		}

		// 判断图片方向
		var targetDir string
		if img.Width > img.Height {
			targetDir = landscapePath
			landscapeCount++
		} else {
			targetDir = portraitPath
			portraitCount++
		}

		// 构建目标路径
		fileName := filepath.Base(path)
		targetPath := filepath.Join(targetDir, fileName)

		// 复制文件到目标目录
		err = copyFile(path, targetPath)
		if err != nil {
			fmt.Printf("复制文件 %s 到 %s 失败: %v\n", path, targetPath, err)
			errorCount++
			return nil
		}

		fmt.Printf("已处理: %s -> %s\n", path, targetPath)
		return nil
	})

	if err != nil {
		fmt.Printf("遍历目录失败: %v\n", err)
		return
	}

	// 打印统计信息
	fmt.Println("\n--- 处理完成 ---")
	fmt.Printf("总共处理图片: %d\n", totalCount)
	fmt.Printf("横屏图片: %d\n", landscapeCount)
	fmt.Printf("竖屏图片: %d\n", portraitCount)
	fmt.Printf("处理失败: %d\n", errorCount)
}

// 复制文件
func copyFile(src, dst string) error {
	// 读取源文件
	data, err := os.ReadFile(src)
	if err != nil {
		return err
	}

	// 写入目标文件
	return os.WriteFile(dst, data, 0644)
}