class App(QWidget):
def __init__(self):
super().__init__()
self.title = 'VAPT Report Generator'
self.left = 10
self.top = 10
self.width = 1280
self.height = 720
self.__Img = None
self.doc = Print_document()
self.doc.start_doc()
self.doc.initialize_doc()
self.initUI()
def initUI(self):
self.setWindowTitle(self.title)
self.setWindowIcon(QIcon(""))
self.setGeometry(self.left, self.top, self.width, self.height)
self.Vname = QLabel('Vulnerabilty Name:',self)
self.Vname.move(20, 5)
self.Vname.resize(250,50)
self.Vnamebox = QLineEdit(self)
self.Vnamebox.move(260, 10)
self.Vnamebox.resize(600,30)
self.VDesc = QLabel('Vulnerabilty Description:',self)
self.VDesc.move(20, 80)
self.VDesc.resize(250,50)
self.VDescbox = QPlainTextEdit(self)
self.VDescbox.move(260, 50)
self.VDescbox.resize(600,100)
self.Vurl = QLabel('Vulnearable URL:',self)
self.Vurl.move(20, 155)
self.Vurl.resize(250,50)
self.Vurlbox = QLineEdit(self)
self.Vurlbox.move(260, 160)
self.Vurlbox.resize(600,30)
self.VSeverity = QLabel('Severity:',self)
self.VSeverity.move(20, 205)
self.VSeverity.resize(250,50)
self.VSeveritybox = QComboBox(self)
self.VSeveritybox.addItem('Critical',1)
self.VSeveritybox.addItem('High',2)
self.VSeveritybox.addItem('Medium',3)
self.VSeveritybox.addItem('Low',4)
self.VSeveritybox.addItem('Informational',5)
self.VSeveritybox.move(260, 210)
self.VSeveritybox.resize(600,30)
#self.VSeveritybox.activated.connect(self.handleActivated)
self.VImpact = QLabel('Impact:',self)
self.VImpact.move(20, 280)
self.VImpact.resize(250,80)
self.VImpactBox = QPlainTextEdit(self)
self.VImpactBox.move(260, 250)
self.VImpactBox.resize(600,130)
self.VRemediation = QLabel('Remediation:',self)
self.VRemediation.move(20, 425)
self.VRemediation.resize(250,50)
self.VRemediationBox = QPlainTextEdit(self)
self.VRemediationBox.move(260, 400)
self.VRemediationBox.resize(600,100)
self.SaveButton = QPushButton('Save', self)
self.SaveButton.move(480,580)
self.SaveButton.clicked.connect(self.on_click)
self.saveReport = QPushButton('Add Vulnerabilty',self)
self.saveReport.move(560,580)
self.saveReport.clicked.connect(self.on_report)
self.ImgButton = QPushButton('Browse', self)
self.ImgButton.move(400,580)
self.ImgButton.clicked.connect(self.on_browse)
self.show()
@pyqtSlot()
def on_browse(self):
self.__Img = QFileDialog.getOpenFileNames(self,'Open Files','/',("Images(*.png)"))
@pyqtSlot()
def on_report(self):
vname = self.Vnamebox.text()
self.doc.setVname(vname)
self.Vnamebox.setText("")
severity = self.VSeveritybox.currentText()
self.doc.setVSeverity(severity)
VDesc = self.VDescbox.toPlainText()
self.doc.SetVdesc(VDesc)
self.VDescbox.setPlainText('')
Vurl = self.Vurlbox.text()
self.doc.setVurl(Vurl)
self.Vurlbox.setText('')
self.doc.setImg(self.__Img)
VImpact = self.VImpactBox.toPlainText()
self.doc.setImpact(VImpact)
self.VImpactBox.setPlainText('')
Vrem = self.VRemediationBox.toPlainText()
self.doc.setVremed(Vrem)
self.VRemediationBox.setPlainText('')
self.doc.pageBreak()
@pyqtSlot()
def on_click(self):
name = QFileDialog.getSaveFileName(self,'Save File','/')
self.doc.Savedoc(name)
class App(QWidget):
def __init__(self):
super().__init__()
self.title = 'VAPT Report Generator'
self.left = 10
self.top = 10
self.width = 1280
self.height = 720
self.__Img = None
self.__doc = None
self.doc = Print_document()
if self.doc.start_doc():
self.doc.reinitialize_doc()
else:
self.doc.start_doc()
self.doc.initialize_doc()
def TechUI(self, MainWindow):
self.setWindowTitle(self.title)
self.setWindowIcon(QIcon("Pristine.png"))
self.setGeometry(self.left, self.top, self.width, self.height)
self.font = QFont()
self.font.setFamily('Helvetica')
self.font.setPointSize(16)
self.Vname = QLabel('Vulnerabilty Name:', self)
self.Vname.move(20, 5)
self.Vname.resize(250, 50)
self.Vname.setFont(self.font)
self.Vnamebox = QLineEdit(self)
self.Vnamebox.move(260, 10)
self.Vnamebox.resize(600, 30)
self.Vnamebox.setFont(self.font)
self.VDesc = QLabel('Vulnerabilty Description:', self)
self.VDesc.move(20, 80)
self.VDesc.resize(250, 50)
self.VDesc.setFont(self.font)
self.VDescbox = QPlainTextEdit(self)
self.VDescbox.move(260, 50)
self.VDescbox.resize(600, 100)
self.Vurl = QLabel('Vulnerable URL:', self)
self.Vurl.move(20, 155)
self.Vurl.resize(250, 50)
self.Vurl.setFont(self.font)
self.Vurlbox = QLineEdit(self)
self.Vurlbox.move(260, 160)
self.Vurlbox.resize(600, 30)
self.Vurlbox.setText('')
self.VPort = QLabel('Vulnerable Port:', self)
self.VPort.move(20, 195)
self.VPort.resize(250, 50)
self.VPort.setFont(self.font)
self.VPortBox = QLineEdit(self)
self.VPortBox.move(260, 200)
self.VPortBox.resize(600, 30)
self.VSeverity = QLabel('Severity:', self)
self.VSeverity.move(20, 245)
self.VSeverity.resize(250, 50)
self.VSeverity.setFont(self.font)
self.VSeveritybox = QComboBox(self)
self.VSeveritybox.addItem('Critical', 1)
self.VSeveritybox.addItem('High', 2)
self.VSeveritybox.addItem('Medium', 3)
self.VSeveritybox.addItem('Low', 4)
self.VSeveritybox.addItem('Informational', 5)
self.VSeveritybox.move(260, 250)
self.VSeveritybox.resize(600, 30)
self.VImpact = QLabel('Impact:', self)
self.VImpact.move(20, 320)
self.VImpact.resize(250, 80)
self.VImpact.setFont(self.font)
self.VImpactBox = QPlainTextEdit(self)
self.VImpactBox.move(260, 290)
self.VImpactBox.resize(600, 130)
self.VRemediation = QLabel('Remediation:', self)
self.VRemediation.move(20, 465)
self.VRemediation.resize(250, 50)
self.VRemediation.setFont(self.font)
self.VRemediationBox = QPlainTextEdit(self)
self.VRemediationBox.move(260, 440)
self.VRemediationBox.resize(600, 100)
self.SaveButton = QPushButton('Save', self)
self.SaveButton.move(480, 580)
self.SaveButton.clicked.connect(self.back)
self.SaveButton.setFont(self.font)
self.saveReport = QPushButton('Add Vulnerabilty', self)
self.saveReport.move(560, 580)
self.saveReport.clicked.connect(self.on_report)
self.saveReport.setFont(self.font)
self.ImgButton = QPushButton('Browse', self)
self.ImgButton.move(400, 580)
self.ImgButton.clicked.connect(self.on_browse)
self.ImgButton.setFont(self.font)
self.show()
@pyqtSlot()
def on_report(self):
vname = self.Vnamebox.text()
self.doc.setVname(vname)
self.Vnamebox.setText(" ")
severity = self.VSeveritybox.currentText()
self.doc.setVSeverity(severity)
VDesc = self.VDescbox.toPlainText()
self.doc.SetVdesc(VDesc)
self.VDescbox.setPlainText(" ")
Vurl = self.Vurlbox.text()
self.doc.setVurl(Vurl)
self.Vurlbox.setText(" ")
Vport = self.VPortBox.text()
self.doc.setVport(Vport)
self.VPortBox.setText(" ")
self.doc.setImg(self.__Img)
VImpact = self.VImpactBox.toPlainText()
self.doc.setImpact(VImpact)
self.VImpactBox.setPlainText(" ")
Vrem = self.VRemediationBox.toPlainText()
self.doc.setVremed(Vrem)
self.VRemediationBox.setPlainText(" ")
self.doc.pageBreak()
@pyqtSlot()
def on_browse(self):
self.__Img = QFileDialog.getOpenFileNames(self, 'Open Files', '/',
("Images(*.png)"))
@pyqtSlot()
def on_click(self):
self.doc.Savereport()
@pyqtSlot()
def back(self):
self.doc.Savereport()
self.ui = MainWindow.App1()
self.ui.initUI()
App.hide(self)