def yesMethod(self):
from home_window import HomeWindow
with open('file.pkl', 'rb') as fid:
file = cPickle.load(fid)
dfile = file[0][0]
self.g = HomeWindow(self)
self.hide()
import os
for i in range(0, len(dfile)):
if os.path.isfile(dfile[i]):
os.remove(dfile[i])
else: ## Show an error ##
print("Error: %s file not found" % dfile[i])
self.g.show()
def main(ip):
varibles.logout = False
print(ip)
d = Database(ip)
varibles.Host = str(ip)
HomeWindow(d.get_my_database(), d.get_my_cursor())
if varibles.logout:
main(ip)
def submitMethod(self):
from home_window import HomeWindow
self.d = HomeWindow(self)
self.hide()
import os
with open('delete.pkl', 'rb') as fid:
delete = cPickle.load(fid)
with open('folder.pkl', 'rb') as fid:
folder = cPickle.load(fid)
path = folder[0]
for i in range(0, len(delete)):
delete[i] = path + delete[i]
## if file exists, delete it ##
for i in range(0, len(delete)):
if os.path.isfile(delete[i]):
os.remove(delete[i])
else: ## Show an error ##
print("Error: %s file not found" % delete[i])
self.d.show()
class FileWindow(QWidget):
def __init__(self, parent=None):
QMainWindow.__init__(self, parent=None)
# self.setWindowFlags( QtCore.Qt.Window | QtCore.Qt.CustomizeWindowHint | QtCore.Qt.WindowCloseButtonHint | QtCore.Qt.WindowMinimizeButtonHint)
#
self.setFixedSize(QSize(800, 500))
with open('file.pkl', 'rb') as fid:
file = cPickle.load(fid)
if (file[1][0] == 0):
predict = False
else:
predict = True
if (predict):
self.threat()
else:
self.safe()
def threat(self):
label = QLabel(self)
pixmap = QPixmap('image/image.png')
label.setPixmap(pixmap)
label.move(280, 125)
ltext = QLabel(self)
text = "This file is predicted as infected. Do you want to delete it?"
ltext.setText(text)
ltext.move(80, 380)
font = ltext.font()
font.setPixelSize(25)
ltext.setFont(font)
ltext = QLabel(self)
text = "Infected!!!"
ltext.setText(text)
ltext.move(320, 30)
font = ltext.font()
font.setPixelSize(40)
ltext.setFont(font)
ltext.setStyleSheet("color:red;")
backbutton3 = QPushButton('Home', self)
backbutton3.setStyleSheet(
"background-image: url(image/back1.png);color: transparent;")
backbutton3.clicked.connect(self.backMethod3)
backbutton3.resize(60, 36)
backbutton3.move(20, 20)
yesbutton = QPushButton('Yes', self)
yesbutton.clicked.connect(self.yesMethod)
yesbutton.resize(50, 30)
yesbutton.move(550, 440)
nobutton = QPushButton('No', self)
nobutton.clicked.connect(self.noMethod)
nobutton.resize(50, 30)
nobutton.move(650, 440)
self.show()
def safe(self):
label = QLabel(self)
pixmap = QPixmap('image/safe.png')
label.setPixmap(pixmap)
label.move(300, 160)
ltext = QLabel(self)
text = "This file is predicted as benign."
ltext.setText(text)
ltext.move(210, 380)
font = ltext.font()
font.setPixelSize(25)
ltext.setFont(font)
ltext = QLabel(self)
text = "Safe!"
ltext.setText(text)
ltext.move(320, 40)
font = ltext.font()
font.setPixelSize(40)
ltext.setFont(font)
ltext.setStyleSheet("color:green;")
backbutton3 = QPushButton('Home', self)
backbutton3.setStyleSheet(
"background-image: url(image/back1.png);color: transparent;")
backbutton3.clicked.connect(self.backMethod3)
backbutton3.resize(60, 36)
backbutton3.move(20, 20)
self.show()
def backMethod3(self):
from home_window import HomeWindow
self.f = HomeWindow(self)
self.hide()
self.f.show()
def yesMethod(self):
from home_window import HomeWindow
with open('file.pkl', 'rb') as fid:
file = cPickle.load(fid)
dfile = file[0][0]
self.g = HomeWindow(self)
self.hide()
import os
for i in range(0, len(dfile)):
if os.path.isfile(dfile[i]):
os.remove(dfile[i])
else: ## Show an error ##
print("Error: %s file not found" % dfile[i])
self.g.show()
def noMethod(self):
from home_window import HomeWindow
self.h = HomeWindow(self)
self.hide()
self.h.show()
def noMethod(self):
from home_window import HomeWindow
self.h = HomeWindow(self)
self.hide()
self.h.show()
def backMethod3(self):
from home_window import HomeWindow
self.f = HomeWindow(self)
self.hide()
self.f.show()
def backMethod1(self):
from home_window import HomeWindow
self.b = HomeWindow(self)
self.hide()
self.b.show()
class Resolve(QWidget):
def __init__(self, parent=None):
QMainWindow.__init__(self, parent=None)
self.setWindowFlags(QtCore.Qt.Window | QtCore.Qt.CustomizeWindowHint
| QtCore.Qt.WindowCloseButtonHint
| QtCore.Qt.WindowMinimizeButtonHint)
self.setFixedSize(QSize(800, 500))
self.setWindowTitle('VPS')
with open('folder.pkl', 'rb') as fid:
folder = cPickle.load(fid)
sum = 0
for i in folder[2]:
sum = sum + i
percent = (sum / len(folder[2])) * 100
j = sum
backbutton1 = QPushButton('Click me', self)
backbutton1.setStyleSheet(
"background-image: url(image/back1.png);color: transparent;")
backbutton1.clicked.connect(self.backMethod1)
backbutton1.resize(60, 36)
backbutton1.move(30, 30)
resolvebutton = QPushButton('Resolve', self)
# resolvebutton.setStyleSheet("background-image: url(image/back1.png);color: transparent;")
resolvebutton.clicked.connect(self.resolveMethod)
resolvebutton.resize(60, 36)
resolvebutton.move(680, 430)
if (percent == 0):
resolvebutton.setEnabled(False)
ltext = QLabel(self)
s = " files are infected do you want to resolve this?"
s = str(j) + s
text = s
ltext.setText(text)
ltext.move(80, 380)
font = ltext.font()
font.setPixelSize(25)
ltext.setFont(font)
#Progressbar
i = percent
progress = QRoundProgressBar(self)
progress.setBarStyle(QRoundProgressBar.BarStyle.DONUT)
progress.resize(200, 200)
progress.move(320, 120)
palette = QPalette()
brush = QBrush(QColor(255, 105 - i, 100 - i))
brush.setStyle(Qt.SolidPattern)
palette.setBrush(QPalette.Active, QPalette.Highlight, brush)
progress.setPalette(palette)
progress.setValue(i)
qApp.processEvents()
self.show()
def backMethod1(self):
from home_window import HomeWindow
self.b = HomeWindow(self)
self.hide()
self.b.show()
def resolveMethod(self):
self.c = FolderWindow(self)
self.hide()
self.c.show()
class FolderWindow(QWidget):
def __init__(self, parent=None):
# super(HelloWindow, self).__init__(parent)
QMainWindow.__init__(self, parent=None)
self.setWindowFlags(QtCore.Qt.Window | QtCore.Qt.CustomizeWindowHint
| QtCore.Qt.WindowCloseButtonHint
| QtCore.Qt.WindowMinimizeButtonHint)
self.setFixedSize(QSize(800, 500))
# self.setMinimumSize(QSize(800,500))
self.setWindowTitle("VP")
self.check_box()
backbutton2 = QPushButton('back', self)
backbutton2.setStyleSheet(
"background-image: url(image/back1.png);color: transparent;")
backbutton2.clicked.connect(self.backMethod2)
backbutton2.resize(60, 36)
backbutton2.move(30, 30)
submitbutton = QPushButton('Submit', self)
# submitbutton.setStyleSheet("background-image: url(image/back1.png);color: transparent;")
submitbutton.clicked.connect(self.submitMethod)
submitbutton.resize(60, 36)
submitbutton.move(680, 430)
homebutton = QPushButton('Home', self)
# submitbutton.setStyleSheet("background-image: url(image/back1.png);color: transparent;")
homebutton.clicked.connect(self.homeMethod)
homebutton.resize(60, 36)
homebutton.move(680, 30)
ltext = QLabel(self)
text = "Delete"
ltext.setText(text)
ltext.move(540, 90)
# font = ltext.font()
# font.setPixelSize(25)
# ltext.setFont(font)
# ltext = QLabel(self)
# text= "Mark safe"
# ltext.setText(text)
# ltext.move(580,90)
# font = ltext.font()
# font.setPixelSize(25)
# ltext.setFont(font)
def backMethod2(self):
from percentage_window import Resolve
self.c = Resolve(self)
self.hide()
self.c.show()
def submitMethod(self):
from home_window import HomeWindow
self.d = HomeWindow(self)
self.hide()
import os
with open('delete.pkl', 'rb') as fid:
delete = cPickle.load(fid)
with open('folder.pkl', 'rb') as fid:
folder = cPickle.load(fid)
path = folder[0]
for i in range(0, len(delete)):
delete[i] = path + delete[i]
## if file exists, delete it ##
for i in range(0, len(delete)):
if os.path.isfile(delete[i]):
os.remove(delete[i])
else: ## Show an error ##
print("Error: %s file not found" % delete[i])
self.d.show()
def homeMethod(self):
from home_window import HomeWindow
self.e = HomeWindow(self)
self.hide()
self.e.show()
def check_box(self):
with open('folder.pkl', 'rb') as fid:
folder = cPickle.load(fid)
path = folder[0]
list = folder[1][:]
pred = folder[2]
final_list = []
for i in range(0, len(pred)):
if (pred[i] == 1):
final_list.append(list[i])
global l, g, d, flag_d, flag_s, s, l1
l = final_list
g = l[:]
l1 = l[:]
d = []
s = []
flag_d = [0] * len(l)
flag_s = [0] * len(l)
h = 50
for i in range(0, len(l)):
l[i] = QPushButton(g[i] + 'd', self)
l[i].setCheckable(True)
l[i].resize(20, 20)
l[i].move(550, 70 + h)
l[i].clicked[bool].connect(self.selected)
# l[i].setStyleSheet("background-color: black; color: black; ")
l[i].setStyleSheet(
"background-image: url(image/uncheck.png);color: transparent;")
# l[i].hide()
# l1[i] = QPushButton(g[i]+'s', self)
# l1[i].setCheckable(True)
# l1[i].resize(20,20)
# l1[i].move(580,70+h)
#
# l1[i].clicked[bool].connect(self.selected)
## l[i].setStyleSheet("background-color: black; color: black; ")
# l1[i].setStyleSheet("background-image: url(image/uncheck.png);color: transparent;")
ltext = QLabel(self)
text = g[i]
ltext.setText(text)
ltext.move(200, 70 + h)
font = ltext.font()
font.setPixelSize(15)
ltext.setFont(font)
h = h + 30
def selected(self, pressed):
source = self.sender()
# m=[]
# print(source.text)
for i in range(0, len(g)):
if source.text() == g[i] + 'd':
if (flag_d[i] == 0):
d.append(g[i])
flag_d[i] = flag_d[i] + 1
l[i].setStyleSheet(
"background-image: url(image/check.png); color: transparent;"
)
# l[i].setStyleSheet("background-color: black; ")
else:
d.remove(g[i])
l[i].setStyleSheet(
"background-image: url(image/uncheck.png);color: transparent;"
)
# l[i].setStyleSheet("background-color: black color:none; ")
flag_d[i] = flag_d[i] - 1
# for i in range(0,len(g)):
# if source.text() == g[i]+'s':
# if(flag_s[i]==0):
# s.append(g[i])
# flag_s[i]=flag_s[i]+1
# l1[i].setStyleSheet("background-image: url(image/check.png); color: transparent;")
#
## l[i].setStyleSheet("background-color: black; ")
# else:
# s.remove(g[i])
# l1[i].setStyleSheet("background-image: url(image/uncheck.png);color: transparent;")
#
## l[i].setStyleSheet("background-color: black color:none; ")
# flag_s[i]=flag_s[i]-1
print(d)
# print("\n")
# print(s)
print("\n")
# d=['a','b','c','d']
with open('delete.pkl', 'wb') as fid:
cPickle.dump(d, fid)
def homeMethod(self):
from home_window import HomeWindow
self.e = HomeWindow(self)
self.hide()
self.e.show()
def main():
varibles.logout = False
d = Database()
HomeWindow(d.get_my_database(), d.get_my_cursor())
if varibles.logout:
main()
from tkinter import *
import sqlite3
from random_calculator import RandomCalculator
from database import Database
from home_window import HomeWindow
root = Tk()
root.title("Kreator Badaczy Tajemnic")
root.geometry("400x500")
calculator = RandomCalculator()
database = Database()
home_window = HomeWindow(root)
root.mainloop()