from PyQt5 import QtWidgets, QtCore
from rememberUI import Ui_MainWindow
from PyQt5.QtWidgets import QMessageBox
import json
import os
import shutil


# import json


class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
    def __init__(self, table, parent=None):
        super(MainWindow, self).__init__(parent)
        self.table = table
        self.index = 0
        self.setupUi(self)
        self.resetupUi()
        self.wrong_count = 0
        self.pushButton.clicked.connect(self.resetupUi)

    def resetupUi(self):
        if self.index != 0:
            if self.lineEdit.text() != self.table[self.index - 1][1]:
                QMessageBox.warning(None, "Warning", "错了，正确答案：" + self.table[self.index - 1][1])
                self.wrong_count += 1
        if self.index != len(self.table):
            self.label_3.setText(self.table[self.index][0])
            self.lineEdit.setText("")
            self.index += 1
        else:
            QMessageBox.information(None, "Information", "You have finished all the"
                                                         " words\nYou have got " + str(self.wrong_count) + " wrong")
            self.close()
            return True