Python基於mediainfo批量重命名圖片文件

	+---Input_Image  #批量放入待命名的圖片文件
	|    1.jpg
	|    2.png
	|    
	+---Output_Image  #批量輸出已命名的圖片文件
	|    JPEG_1920x1080_32bit_jpg.jpg
	|	PNG_1280x720_32bit_png.png
	|
    \image_info.py  # 獲取圖片文件info信息的模塊，
	\rename_image.py #調用image_info.py並實現重名，可雙擊運行

# coding=utf-8
 
import os
import re
import subprocess
 
 
class ImageInfoGetter():
  '''獲取圖片文件的Formate, 分辨率，位深度'''
 
  def __init__(self, image_file):
    '''判斷文件是否存在，如果存在獲取其mediainfo信息'''
    if os.path.exists(image_file):
      self.image_file = image_file
      p_obj = subprocess.Popen('mediainfo "%s"' % self.image_file, shell=True, stdout=subprocess.PIPE,
                   stderr=subprocess.PIPE)
      self.info = p_obj.stdout.read().decode("utf-8") # 解決非英文字符的編碼問題
    else:
      raise FileNotFoundError("Not this File!") # 如果多媒體文件路徑不存在，必須中斷
 
  def get_image_format(self):
    '''獲取圖片的格式，比如JPEG, PNG, BMP等'''
    try:
      image_codec = re.findall(r"Format\s+:\s(.*)", self.info)[-1] # 取第最後一個Format字段
      image_codec = image_codec.strip() # 去除前後的空格
      if image_codec == "RGB":
        image_codec = "BMP"
    except:
      image_codec = "undef" # 防止程序因為異常而中斷
    return image_codec
 
  def get_image_resolution(self):
    '''獲取圖片的分辨率'''
    try:
      image_widget = re.findall(r'Width\s+:\s(.*)pixels', self.info)[-1]
      image_widget = image_widget.replace(" ", "")
      image_height = re.findall(r'Height\s+:\s(.*)pixels', self.info)[-1]
      image_height = image_height.replace(" ", "")
      image_resolution = image_widget + "x" + image_height
    except:
      image_resolution = "undef" # 防止程序因為異常而中斷
    return image_resolution
 
  def get_image_bit_depth(self):
    '''獲取圖片的位深度'''
    try:
      image_bit_depth = re.findall(r"Bit depth\s+:\s(.*bit)s", self.info)[-1].strip()
      image_bit_depth = image_bit_depth.replace(" ", "") # 去空格
    except:
      image_bit_depth = "undef" # 防止程序因為異常而中斷
    return image_bit_depth
 
  def get_image_container(self):
    '''獲取圖片容器，即文件後綴名'''
    _, image_container = os.path.splitext(self.image_file)
    if not image_container:
      raise NameError("This file no extension")
    image_container = image_container.replace(".", "")
    image_container = image_container.lower() # 全部轉成小寫
    return image_container
 
 
if __name__ == '__main__':
  # 以下代碼塊，隻是用來測試本模塊的，一般不建議直接在這裡大面積調用本模塊'''
  i_obj = ImageInfoGetter("C:\\img.jpg")
  image_format = i_obj.get_image_format()
  print(image_format)
  image_resolution = i_obj.get_image_resolution()
  print(image_resolution)
  image_bit_depth = i_obj.get_image_bit_depth()
  print(image_bit_depth)
  image_container = i_obj.get_image_container()
  print(image_container)

# coding=utf-8
 
import os
import image_info
from shutil import copyfile
 
curdir = os.getcwd()
 
# 輸入文件夾，放入待重命名的圖片
input_image_path = os.path.join(curdir, "Input_Image")
filelist = os.listdir(input_image_path) # 獲取文件列表
 
# 輸出文件夾，已命名的圖片存放在這裡
output_image_path = os.path.join(curdir, "Output_Image")
 
# 如果沒有Output_Image這個文件夾，則創建這個文件夾
if not os.path.exists(output_image_path):
  os.mkdir(output_image_path)
 
if filelist: # 如果文件列表不為空
  for i in filelist: # 遍歷文件列表
    # 以下代碼塊，隻是用來測試本模塊的，一般不建議直接在這裡大面積調用本模塊'''
    image_file = os.path.join(input_image_path, i)
    i_obj = image_info.ImageInfoGetter(image_file)
    image_format = i_obj.get_image_format()
    image_resolution = i_obj.get_image_resolution()
    image_bit_depth = i_obj.get_image_bit_depth()
    image_container = i_obj.get_image_container()
    new_image_name = image_format + "_" + image_resolution + "_" + image_bit_depth + "_" \
             + image_container + "." + image_container
    print(new_image_name)
    new_image_file = os.path.join(output_image_path, new_image_name)
    copyfile(image_file, new_image_file) # 復制文件
else:
  print("It's a Empty folder, please input the image files which need to be renamed firstly!!!")
os.system("pause")