diff --git a/sender_ui.py b/sender_ui.py index 160c621..cdb0244 100644 --- a/sender_ui.py +++ b/sender_ui.py @@ -53,13 +53,6 @@ class Ui_MainWindow(object): self.mode_box = QtWidgets.QComboBox(self.centralwidget) self.mode_box.setGeometry(QtCore.QRect(560, 200, 121, 41)) self.mode_box.setObjectName("mode_box") - self.key_generate_button = QtWidgets.QPushButton(self.centralwidget) - self.key_generate_button.setGeometry(QtCore.QRect(560, 10, 121, 51)) - self.key_generate_button.setAutoDefault(False) - self.key_generate_button.setObjectName("key_generate_button") - self.send_public_button = QtWidgets.QPushButton(self.centralwidget) - self.send_public_button.setGeometry(QtCore.QRect(560, 60, 121, 51)) - self.send_public_button.setObjectName("send_public_button") self.envelope_button = QtWidgets.QPushButton(self.centralwidget) self.envelope_button.setGeometry(QtCore.QRect(560, 290, 121, 51)) self.envelope_button.setObjectName("envelope_button") @@ -72,8 +65,6 @@ class Ui_MainWindow(object): self.statusbar.setObjectName("statusbar") MainWindow.setStatusBar(self.statusbar) self.retranslateUi(MainWindow) - self.key_generate_button.clicked.connect(self.key_generate) # type: ignore - self.send_public_button.clicked.connect(self.send_publickey) # type: ignore self.choose_button.clicked.connect(self.choose_file) # type: ignore self.enbutton.clicked.connect(self.encrypt) # type: ignore self.envelope_button.clicked.connect(self.send) # type: ignore @@ -95,12 +86,15 @@ class Ui_MainWindow(object): self.mode="" self.algorithm="" self.encrypted_text="" - with open("public_key.pem","rb") as f: - self.public_key=crypto.load_publickey(crypto.FILETYPE_PEM,f.read(),password=None) - with open("private_key.pem","rb") as f: - self.private_key=crypto.load_privatekey(crypto.FILETYPE_PEM,f.read(),password=None) - with open("public_key_other.pem","rb") as f: - self.public_key_other=crypto.load_publickey(crypto.FILETYPE_PEM,f.read(),password=None) + path = QFileDialog.getOpenFileName()[0] + with open(path,"rb") as f: + self.public_key=crypto.load_publickey(crypto.FILETYPE_PEM,f.read()) + path = QFileDialog.getOpenFileName()[0] + with open(path,"rb") as f: + self.private_key=crypto.load_privatekey(crypto.FILETYPE_PEM,f.read()) + path = QFileDialog.getOpenFileName()[0] + with open(path,"rb") as f: + self.public_key_other=crypto.load_publickey(crypto.FILETYPE_PEM,f.read()) self.key="" self.envelope={} #等待接收方连接 @@ -110,8 +104,6 @@ class Ui_MainWindow(object): MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow")) self.enbutton.setText(_translate("MainWindow", "加密并封装")) self.choose_button.setText(_translate("MainWindow", "选择文件")) - self.key_generate_button.setText(_translate("MainWindow", "生成公私钥")) - self.send_public_button.setText(_translate("MainWindow", "发送公钥")) self.envelope_button.setText(_translate("MainWindow", "发送")) self.mode_box.addItems(["ECB", "CBC"]) self.algorithm_box.addItems(["AES","SM4"])