#!/usr/bin/env python
# coding=utf-8
# 批量重命名圖片名字從-2-01改成-1-01
import os
import time
 
class ImageRename():
 def __init__(self):
  self.path = './'
 def rename(self):
  filelist = os.listdir(self.path)
  total_num = len(filelist)
  i = 0
  for item in filelist:
   if item[-4:].lower() == '.jpg' or item[-4:].lower() == '.png' or item[-5:].lower() == '.jpeg' or item[-4:].lower() == '.gif':
    src = os.path.join(os.path.abspath(self.path), item)
    if "-2-01" in item:
     rename = item.replace("-2-01", "-1-01")
     dst = os.path.join(os.path.abspath(self.path), rename)
     os.rename(src, dst)
     print('更改圖片%s名字中...' % (item))
     i = i + 1
  if i ==0:
   print("沒有符合更改要求的圖片名字，程序?qū)⒃?秒后自動退出！")
  else:
   print ('已完成，該目錄下共 %d個文件修改了%d張圖片，程序?qū)⒃?秒后自動退出！' % (total_num, i))
  time.sleep(5)
 
if __name__ == '__main__':
 newname = ImageRename()
 newname.rename()

# 獲取路徑 D:\后臺數(shù)據(jù)處理\添加pos活動圖標(biāo)\pos_icon
file_path = os.path.dirname(os.getcwd())
# 獲取所有文件
file_list = os.listdir(file_path)
# 如果長度是1那么代表只有一張png圖，才會繼續(xù)執(zhí)行
png_list = []
for file in file_list:
 if file[-4:].lower() == ".png":
  png_list.append(file)
if len(png_list) == 1:
 png_path = file_path + "/" + png_list[0]
 print(png_path)
else:
 print("ERROR:發(fā)現(xiàn)%s張png圖，停止上傳，請核實！" % len(png_list))