from PyQt5.QtWidgets import QApplication,QPushButton,QLabel,QMainWindow
from PyQt5.QtGui import QFont,QPixmap
import sys,os
import shutil
 
def copyfile(srcfile, dstfile):#用來復(fù)制文件，源文件會保留
 
  if not os.path.isfile(srcfile):
    print("%s not exist!" % srcfile)
  else:
    f_path, f_name = os.path.split(dstfile) # 分離文件名和路徑
    if not os.path.exists(f_path):
      os.makedirs(f_path) # 創(chuàng)建路徑
    shutil.copyfile(srcfile, dstfile) # 復(fù)制文件
    print("copy %s -> %s" % (srcfile, dstfile))
class mainForm(QMainWindow):
  def __init__(self):
    super(mainForm, self).__init__()
 
    self.img_path="faces/" #文件夾和py文件要再同一個目錄下面
    self.img_list=os.listdir(self.img_path) #獲取目錄下的所有文件
    self.idx=0#可以改這里，選擇程序運行的時候第一個顯示的圖片是哪一個
 
    self.initUI()
    self.show()
  def initUI(self):
    font=QFont()
    font.setPixelSize(20)#新建一個字體控件
 
    self.setWindowTitle("label_me")#設(shè)置窗體的標(biāo)題
    self.setGeometry(0,0,900,600)#位置和大小
 
    button_list=["Chandler","Phoebe","Joey","Monica","Rachel","Ross","Others","Thing",]#這里是顯示的按鈕們，也是可能的類別數(shù)
 
    for idx, label_name in enumerate(button_list):
 
      button=QPushButton(label_name,self)
      button.move(idx*110+20,500)
      button.setFont(font)
      button.setFixedHeight(35)
 
      button.clicked.connect(self.classify)#動態(tài)控件綁定同一個事件，根據(jù)事件的sender判斷是哪個按鈕按下
 
    self.lbl_list=[]#存放顯示圖片的label 的list
    for i in range(self.get_remainder()):
 
      self.pix = QPixmap(self.img_path+self.img_list[self.idx+i])
      label_img = QLabel(self)
      label_img.setGeometry(360*i+10, 400-100*(3-i)+40, 100*(3-i)+40,100*(3-i)+40)
      label_img.setStyleSheet("border:2px solid red")
      label_img.setPixmap(self.pix)#設(shè)置label控件要顯示的圖片
      label_img.setScaledContents(True)
      self.lbl_list.append(label_img)
 
  def get_remainder(self):#打算是要顯示3個label圖片，如果是到了最后，顯示不了那么多了。
    r=len(self.img_list)-self.idx
    if r>3:
      r=3
    return r
  def clear_lbls(self):#最后的時候會用到，不顯示某些label
    for i in range(len(self.lbl_list)):
      self.lbl_list[i].hide()
 
  def classify(self):
 
    sender = self.sender()
    dir_path=sender.text()+"/"#獲取按鈕的text屬性
 
    current_img_path=self.img_list[self.idx]#獲取剛剛被分類的圖片的路徑
    copyfile(self.img_path+current_img_path , dir_path+current_img_path)
    self.idx +=1#下一個圖片
    img_full_path = [self.img_path + self.img_list[self.idx + i]  for i in range(self.get_remainder())]
 
    self.clear_lbls()
    for i in range(self.get_remainder()):
 
      pix = QPixmap(img_full_path[i])
      self.lbl_list[i].setPixmap(pix)
      self.lbl_list[i].show()
    self.setWindowTitle("當(dāng)前是第 %d 個圖片"%self.idx)
 
app=QApplication(sys.argv)
f=mainForm()
sys.exit(app.exec())