|
|
import base64
|
|
|
import win32api
|
|
|
from pyDes import *
|
|
|
import binascii
|
|
|
from PyQt5.QtCore import QSize, Qt
|
|
|
from PyQt5.QtWidgets import QMainWindow, QApplication, QFileDialog, QWidget, QLineEdit, QPushButton, QHBoxLayout, QVBoxLayout, QMessageBox, QLabel
|
|
|
from PyQt5 import QtCore, QtGui, QtWidgets
|
|
|
import os
|
|
|
import sys
|
|
|
|
|
|
|
|
|
|
|
|
class Window(QWidget):
|
|
|
def __init__(self):
|
|
|
super().__init__()
|
|
|
self.initUI()
|
|
|
# self.setupUi()
|
|
|
|
|
|
def setupUi(self):
|
|
|
|
|
|
layout = QVBoxLayout()
|
|
|
# 第一行
|
|
|
label1 = QLabel('Label 1')
|
|
|
#第二行
|
|
|
input1 = QLineEdit()
|
|
|
|
|
|
# 第三行
|
|
|
label2 = QLabel('Label 2')
|
|
|
# 第四行
|
|
|
input2 = QLineEdit()
|
|
|
# 第五行
|
|
|
button = QPushButton('Button')
|
|
|
|
|
|
layout.addWidget(button, alignment=Qt.AlignCenter)
|
|
|
self.show()
|
|
|
|
|
|
|
|
|
def initUI(self):
|
|
|
# 创建两个 QLineEdit 控件
|
|
|
self.line_edit1 = QLineEdit()
|
|
|
self.line_edit2 = QLineEdit()
|
|
|
self.label1 = QLabel('输入盘符号：')
|
|
|
self.label2 = QLabel('激活码：')
|
|
|
self.reg = register()
|
|
|
|
|
|
self.button2 = QPushButton('生成激活码')
|
|
|
self.button2.clicked.connect(self.check_license)
|
|
|
|
|
|
spacerItem = QtWidgets.QSpacerItem(40, 20, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum)
|
|
|
spacerItem2 = QtWidgets.QSpacerItem(20, 40, QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Expanding)
|
|
|
vbox = QVBoxLayout()
|
|
|
vbox.addItem(spacerItem2)
|
|
|
vbox.addWidget(self.label1)
|
|
|
vbox.addWidget(self.line_edit1)
|
|
|
vbox.addWidget(self.label2)
|
|
|
vbox.addWidget(self.line_edit2)
|
|
|
|
|
|
|
|
|
vbox.addItem(spacerItem2)
|
|
|
vbox1 = QHBoxLayout()
|
|
|
vbox1.addItem(spacerItem)
|
|
|
vbox1.addWidget(self.button2)
|
|
|
vbox1.addItem(spacerItem)
|
|
|
|
|
|
vbox2 = QVBoxLayout()
|
|
|
vbox2.addLayout(vbox)
|
|
|
vbox2.addLayout(vbox1)
|
|
|
vbox2.addItem(spacerItem2)
|
|
|
# 设置窗口布局，并设置窗口的标题和大小
|
|
|
self.setLayout(vbox2)
|
|
|
self.setWindowTitle('产品许可激活')
|
|
|
self.resize(300, 300)
|
|
|
|
|
|
# 显示窗口
|
|
|
self.show()
|
|
|
|
|
|
|
|
|
def check_license(self):
|
|
|
if self.line_edit1.text():
|
|
|
licese = self.line_edit1.text()
|
|
|
code = self.reg.DesEncrypt(licese).decode('utf-8')
|
|
|
self.line_edit2.setText(code)
|
|
|
else:
|
|
|
reply = QMessageBox.question(self.parent(),
|
|
|
'提示',
|
|
|
"请输入盘符号",
|
|
|
QMessageBox.Yes)
|
|
|
return
|
|
|
|
|
|
|
|
|
class register:
|
|
|
def __init__(self):
|
|
|
self.Des_Key = b"BHC#@*AW" # Key（可换）
|
|
|
self.Des_IV = b"\x22\x33\x35\x81\xBC\x38\x5A\xE7" # 自定义IV向量（可换）
|
|
|
|
|
|
# 获取C盘卷序列号
|
|
|
def getCVolumeSerialNumber(self):
|
|
|
CVolumeSerialNumber = win32api.GetVolumeInformation("C:\\")[1]
|
|
|
# print(CVolumeSerialNumber)
|
|
|
if CVolumeSerialNumber:
|
|
|
return str(CVolumeSerialNumber)
|
|
|
else:
|
|
|
return 0
|
|
|
|
|
|
# 使用DES加base64的形式加密
|
|
|
def DesEncrypt(self, str):
|
|
|
# str = str
|
|
|
k = des(self.Des_Key, CBC, self.Des_IV, pad=None, padmode=PAD_PKCS5)
|
|
|
EncryptStr = k.encrypt(str)
|
|
|
print(type(EncryptStr))
|
|
|
return base64.b64encode(EncryptStr) # 转base64编码返回
|
|
|
|
|
|
# 获取注册码，验证
|
|
|
def regist(self, key):
|
|
|
key = key
|
|
|
self.str1 = base64.b64decode(key)
|
|
|
|
|
|
# 由于输入不符合base64规则的字符串会引起异常，所以需要增加输入判断
|
|
|
|
|
|
if key:
|
|
|
content = self.getCVolumeSerialNumber()
|
|
|
|
|
|
key_decrypted = str(self.DesDecrypt())
|
|
|
if content != 0 and key_decrypted != 0:
|
|
|
if content != key_decrypted:
|
|
|
print("wrong register code, please check and input your register code again:")
|
|
|
self.regist()
|
|
|
elif content == key_decrypted:
|
|
|
return True
|
|
|
|
|
|
else:
|
|
|
return False
|
|
|
else:
|
|
|
return False
|
|
|
else:
|
|
|
self.regist()
|
|
|
return False
|
|
|
|
|
|
# des解码
|
|
|
def DesDecrypt(self):
|
|
|
k = des(self.Des_Key, CBC, self.Des_IV, pad=None, padmode=PAD_PKCS5)
|
|
|
DecryptStr = k.decrypt(self.str1)
|
|
|
return DecryptStr.decode('utf-8')
|
|
|
if __name__ == '__main__':
|
|
|
app = QApplication(sys.argv)
|
|
|
window = Window()
|
|
|
sys.exit(app.exec_())
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|