Python3如何將源目錄中的圖片用MD5命名並可以設定目標目錄

2022-02-10 13:00:33

嘗試了一下用 python 實現了一個生成密碼的程式。感覺還是比較好上手的。但是那個程式還是非常簡單的一個小程式。這次我需要實現一個更加複雜的程式，就是整理圖片。

列出需求

將圖片檔案用圖片的 md5 值進行重新命名。
可以設定源目錄
可以設定目標目錄
可以設定是移動，還是複製
引數可以預設

列出這些需求，我開始設計我的程式。

最終實現程式碼

#!/usr/bin/env python3
# -*- coding: UTF-8 -*-
import sys
import os
# 獲取檔案型別庫
import imghdr
# MD5庫
import hashlib
# 檔案操作庫
import shutil
# 設定預設引數
sourceDir = './'
targetDir = './'
operation = 'move'
images = []
# 使用者操作說明
def Usage():
    print('USAGE: md5img [source dir] [target dir] [-m | -c]')
    sys.exit(0)
# 根據輸入的引數，進行不同的處理
if len(sys.argv) == 1 or len(sys.argv) > 4:
    Usage()
if len(sys.argv) >= 2:
    sourceDir = sys.argv[1]
    targetDir = sys.argv[1]
if len(sys.argv) >= 3:
    targetDir = sys.argv[2]
if len(sys.argv) == 4:
    if sys.argv[3] == '-m':
        operation = 'move'
    elif sys.argv[3] == '-c':
        operation = 'copy'
    else:
        print('Operation method is not defined')
        Usage()
# 找到源目錄下所有的圖片
def findImage(sourceDir):
    # 如果源目錄不存在，則報錯退出
    if not os.path.exists(sourceDir):
        print('Source directory is not defined')
        Usage()
    # 迴圈目標目錄中的檔案
    for fil in os.listdir(sourceDir):
        # 取得檔案的路徑
        filPath = os.path.join(sourceDir, fil)
        # 判斷檔案是否為目錄
        if not os.path.isdir(filPath):
            # 判斷檔案是否為圖片
            if imghdr.what(filPath) != None:
                # 將圖片插入需要處理的列表
                images.append(filPath)
# 計算 MD5 值函數
def calcMD5(filepath):
    with open(filepath,'rb') as f:
        md5obj = hashlib.md5()
        md5obj.update(f.read())
        hash = md5obj.hexdigest()
        return hash
# 處理所有圖片
def md5img(images):
    # 如果目標目錄不存在，則報錯退出
    if not os.path.exists(targetDir):
        print('Target directory is not defined')
        Usage()
    # 迴圈需要處理的圖片列表
    for img in images:
        # 根據圖片的真實字尾，來確定圖片的字尾，如果是 jpeg 則改成 jpg
        postfix = 'jpg' if imghdr.what(img) == 'jpeg' else imghdr.what(img)
        # os.rename(img, targetDir + '/' + calcMD5(img) + '.' + postfix)
        # 執行復制或者移動操作
        if operation == 'move':
            shutil.move(img, targetDir + '/' + calcMD5(img) + '.' + postfix)
        elif operation == 'copy':
            shutil.copy(img, targetDir + '/' + calcMD5(img) + '.' + postfix)
# 找圖片
findImage(sourceDir)
# 處理圖片
md5img(images)
# 完成
print('Finish')

程式碼說明

首先呢是參照各種需要的庫。sys庫和os庫是非常基本並且常用的。然後就是針對各個需要的功能點，找的不同的庫，進行引入。

然後，設定了各種預設引數。需要源目錄，目標目錄，操作方法，以及圖片列表，一個空的陣列。

定義函數

在 js 中，我們定義一個函數，用的是 function 。在 python 中，用的是 def。我現在還不知道 def 代表的是啥意思，如果你知道，在品論中告訴我哈~

其他的內容，除了格式差異以外，我感覺和 js 是非常類似的。

但是，在 js 中，函數放在哪裡並不會影響執行。但是在 python 中就不一樣了。函數必須在執行的程式碼之前定義。否則就會報錯。

目前我還沒有進行多檔案的編碼。如果是進行多檔案的編碼的話，如何組織檔案，需要我好好的想一下。

使用者使用說明

首先，我定義了一個使用者使用說明的函數。內容非常簡單，就是告訴使用者，這個函數的使用方法。

然後，就是各種終端錄入的引數的判斷處理，根據錄入的引數，重設我的預設值，如果是引數出錯了，則需要告訴使用者，出錯了。

三元運算

python 中沒有三元運算？？？但是有類似的處理方法，只是語法不一樣而已。

比如在 js 中，三元運算如下:

res = a > 5 ? 0 : 1

但是在 python 中沒有三元運算，我們可以這樣寫：

res = 0 if a > 5 else 1

比較怪異，但是比照著，也是能寫的。

判斷一個路徑是否存在

我們可以使用 os.path.exists(path) 來判斷 path 這個路徑是否存在。

找出來一個路徑下的所有檔案或者目錄

os.listdir(path) 就可以列出 path 這個目錄下面所有的檔案或者目錄了。這裡得到的是一個陣列。

我比較習慣 js 的說法。 python 中說這是列表……

判斷一個檔案是否為圖片，如果為圖片，則輸出圖片的型別

這需要使用一個庫 imghdr，我們在程式碼的開頭參照進來即可。

然後我們可以使用 imghdr.what(filPath) 來看檔案是否為圖片，如果輸出的是 None 則表示不是圖片。如果是圖片的話，則會輸出圖片的型別。比如 jpg 圖片會輸出 jpeg 這樣。

其他小結

關鍵是找到對應的庫，然後看庫的方法就可以了。目前我很不熟悉，畢竟是剛剛開始學。但是通過 google 搜尋，是可以很快找到解決各種問題的方法的。

唯一需要注意的是，需要加上關鍵詞 python3 來避免看到 python2 的內容。我就遇到好幾個 2 的程式碼，導致程式跑不起來出錯的問題。

其他的說明我的註釋裡面已經說得非常清楚了。

Python修改圖片md5

import time
writefile = int(time.time * 1000)
with open(path,"a") as f:
    f.write(str(writefile))

以上為個人經驗，希望能給大家一個參考，也希望大家多多支援it145.com。