import os

from PyQt5.QtGui import *
from PyQt5.QtCore import *
from PyQt5.QtWidgets import *
import sys

class FileDestory(QWidget):
    def __init__(self):
        super(FileDestory, self).__init__()
        self.init_ui()

    def init_ui(self):
        self.setWindowTitle('電腦垃圾清理器  公眾號(hào)：[Python 集中營(yíng)]')
        self.setWindowIcon(QIcon('垃圾桶.ico'))
        self.setFixedWidth(550)
        self.setFixedHeight(80)

        self.process = QProgressBar()
        self.process.setRange(0, 5)

        self.start_btn = QPushButton()
        self.start_btn.setText('開(kāi)始清理')
        self.start_btn.clicked.connect(self.start_btn_click)

        hbox = QHBoxLayout()
        hbox.addWidget(self.process)
        hbox.addWidget(self.start_btn)

        self.thread_ = WorkThread(self)
        self.thread_.finished.connect(self.finished)
        self.thread_.exec_step.connect(self.set_step)

        self.setLayout(hbox)

    def start_btn_click(self):
        self.start_btn.setEnabled(False)
        self.thread_.start()

    def finished(self, finished):
        if finished is True:
            self.start_btn.setText('清理已完成')
            self.start_btn.setEnabled(False)

    def set_step(self, step):
        self.process.setValue(step)

class WorkThread(QThread):
    # 定義好信號(hào)量用來(lái)向主線程中傳遞變量的變化信息，這樣主線程就可以知道運(yùn)行結(jié)果如何。

    # 子線程是否執(zhí)行完成的信號(hào)變量
    finished = pyqtSignal(bool)

    # 子線程具體的步驟信號(hào)變量
    exec_step = pyqtSignal(int)

    def __init__(self, parent=None):
        '''
        子線程類的初始化函數(shù)
        :param parent: UI界面類對(duì)象
        '''
        super(WorkThread, self).__init__(parent)
        self.working = True
        self.parent = parent

    def __del__(self):
        '''
        線程執(zhí)行是否需要進(jìn)入等待過(guò)程
        :return:
        '''
        self.working = False
        self.wait()

    def run(self):
        '''
        子線程主要執(zhí)行邏輯的業(yè)務(wù)函數(shù)
        :return:
        '''
        file_type = {
            '.tmp': '臨時(shí)文件',
            '._mp': '臨時(shí)文件_mp',
            '.log': '日志文件',
            '.gid': '臨時(shí)幫助文件',
            '.chk': '磁盤檢查文件',
            '.old': '臨時(shí)備份文件',
            '.xlk': 'Excel備份文件',
            '.bak': '臨時(shí)備份文件bak'
        }

        user_pro = os.environ['userprofile']

        def del_file_and_dir(root):
            try:
                if os.path.isfile(root):
                    os.remove(root)
                    print("文件", root, "已經(jīng)被移除！")
                elif os.path.isdir(root):
                    os.rmdir(root)
                    print("文件夾", root, "已經(jīng)被移除！")

            except WindowsError:
                print("該文件", root, "不能被移除！")

        def init_size(b):
            try:
                kb = b // 1024
            except:
                print("傳入字節(jié)格式不對(duì)")
                return "Error"
            if kb > 1024:
                M = kb // 1024
                if M > 1024:
                    G = M // 1024
                    return "%dG" % G
                else:
                    return "%dM" % M
            else:
                return "%dkb" % kb

        class Clean(object):
            def __init__(self):
                self.del_info = {}
                self.del_file_paths = []
                self.total_size = 0
                for i, j in file_type.items():
                    self.del_info[i] = dict(name=j, count=0)

            def count_files(self):
                for roots, dirs, files in os.walk(user_pro):
                    for files_item in files:
                        file_extension = os.path.splitext(files_item)[1]
                        if file_extension in self.del_info:
                            file_full_path = os.path.join(roots, files_item)
                            self.del_file_paths.append(file_full_path)
                            self.del_info[file_extension]['count'] += 1
                            self.total_size += os.path.getsize(file_full_path)

            def show_del_files(self):
                re = init_size(self.total_size)
                for i in self.del_info:
                    print(self.del_info[i]["name"], "共計(jì)", self.del_info[i]["count"], "個(gè)")
                return re

            def delete_files(self):
                for path in self.del_file_paths:
                    print('準(zhǔn)備處理文件路徑：', path)
                    del_file_and_dir(path)

        self.exec_step.emit(1)
        clean = Clean()
        self.exec_step.emit(2)
        clean.count_files()
        self.exec_step.emit(3)
        re = clean.show_del_files()
        self.exec_step.emit(4)
        clean.delete_files()
        self.exec_step.emit(5)
        self.finished.emit(True)

if __name__ == '__main__':
    app = QApplication(sys.argv)
    main = FileDestory()
    main.show()
    sys.exit(app.exec_())