app.py

#       -
#
#       https://github.com/hunterjreid/HomeScan
#
#       Yoobee Project
#       HomeScan
#       https://github.com/hunterjreid/HomeScan
#
#       Copyright @ 2022 Hunter Reid Catherine Daniels.
#       Any illegal reproduction of this content will result in immediate legal action.
#

#Prerequisites
import psutil
from PyQt5 import QtCore, QtWidgets, uic
from PyQt5.QtCore import Qt
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from random import randint
from datetime import datetime
import pyqtgraph as pg
import sys, socket, os, ipaddress, webbrowser, time, ctypes, speedtest, platform
global privateMode,VpnMode,isSound,warnMessageOnExit,alerts
#Settings !! 
#
#   If you wish to change the default settings you can so here.
#
privateMode = False
VpnMode = False
isSound = False
warnMessageOnExit = True
showMainMessage = True
alerts = ["Welcome To HomeScan!"]
#ingore this var !! DONT CHNAGE 
is_Started = False
#mainwindow.ui start
class HomeScanMain(QMainWindow):
    def __init__(self):
        super(HomeScanMain,self).__init__()
        #load the mainwindow.ui as main Window
        uic.loadUi('mainwindow.ui',self)
        #GLOBAL
        global hostname, IPAddr, conn
        conn=psutil.net_connections()
        hostname=socket.gethostname()   
        IPAddr=socket.gethostbyname(hostname) 
        #GLOBAL END
        # #Connecting routes and also buttons All buttons listed below.
        self.minn.clicked.connect(self.mintab)
        self.full.clicked.connect(self.toggleFull)
        self.exit.clicked.connect(self.quit)
        self.livepanelBtn.clicked.connect(self.goToLivePanel)
        self.arpBtn.clicked.connect(self.goToArp_scan)
        self.connectedIpBtn.clicked.connect(self.goToDevices)
        self.angryScanBtn.clicked.connect(self.advanced_module)
        self.help.clicked.connect(self.goToHelp)
        self.settings.clicked.connect(self.goToSettings)
        self.SCAN.clicked.connect(self.goToScan)
        self.netOverviewBtn.clicked.connect(self.goToOverview)
        self.alertBtn.clicked.connect(self.goToAlerts)
        self.basicScanBtn.clicked.connect(self.goToAScan)
        self.qScanBtn.clicked.connect(self.goToQscan)
        self.hashtagbtn.clicked.connect(self.goHashhtag)
        self.refreshGraph.clicked.connect(self.showrefresh)
        self.popup.clicked.connect(self.popupNetworkConns)
        self.warningBtn.clicked.connect(self.goToScan)
        # LOAD THE TABLES
        self.loadtable_CONNECTION_TABLE()
        self.loadtable_NETWORK_INTERFACETABLE()
        #graphs PRESET
        pen = pg.mkPen(color=(255, 165, 0), width=4, style=QtCore.Qt.DashLine)
        pen2 = pg.mkPen(color=(191, 32, 32), width=5, style=QtCore.Qt.DashLine)
        self.graphicsView2.setBackground(QColor(12, 12, 12))
        self.graphicsView.setBackground(QColor(12, 12, 12))
        #CHART TITLES
        self.label_TimerText.setText(str(IPAddr))
        self.label_3.setHidden(not showMainMessage)
        #set graph
        row = 0
        self.x = []
        self.y = []
        stats=psutil.net_if_stats()
        for key in stats:
            self.x.append(stats[key][3])  
            self.y.append(stats[key][2])
            row=row+1
        #define ranges
        self.x3 = list(range(20)) # 100 time points
        self.y3 = [randint(0,0) for _ in range(20)]  # 100 data points
        self.y5 = [randint(0,0) for _ in range(20)]  # 100 data points
        #set draw lines
        self.data_line = self.graphicsView.plot(self.x3, self.y3, pen=pen)
        self.data_line_2 = self.graphicsView2.plot(self.x3, self.y5, pen=pen2)
        # Start Timer 30ms interval
        self.timer4 = QtCore.QTimer()
        self.timer4.stop()
        self.timer4.setInterval(30)
        self.timer4.timeout.connect(self.update_plot_data) # Update plot function
        self.timer4.start()
    # Functions Start here - Update plot function
    def update_plot_data(self):
        conn_nol=psutil.net_connections()
        self.x3 = self.x3[1:]  # Remove the first y element.
        self.x3.append(self.x3[-1] + 1)  # Add a new value 1 higher than the last.
        self.y3 = self.y3[1:]  # Remove the first
        self.y3.append((len(conn_nol)))  # Add a new random value.
        self.graphicsView.setRange(yRange=[int(len(conn_nol)-5),int(len(conn_nol)+5)]) # Set comfortable spacing around the graph to make it proportional
        theval = round(round(psutil.net_io_counters().bytes_sent / 1024.0 / 1024, 2) + round(psutil.net_io_counters().bytes_recv / 1024.0 / 1024, 2), 2)
        self.y5 = self.y5[1:]  # Remove the first
        self.graphicsView2.setRange(yRange=[(theval-0.7),(theval+0.7)]) # Set comfortable spacing around the graph to make it proportional
        self.y5.append(round(theval, 2))  # Add a new random value.
        self.data_line.setData(self.x3, self.y3)  # Update the data.
        self.data_line_2.setData(self.x3, self.y5) # Update the data.
        #UPDATE TEXT ASWELL
        self.label_7.setText("Net Conns Live: " + str(len(conn_nol)))
        self.label_9.setText("Total transfer: " + str(theval) + " Mbs")
        self.label_15.setText(str(len(conn_nol)))
        self.label_16.setText("Sent: " + str(round(psutil.net_io_counters().bytes_sent / 1024.0 / 1024, 2)) + "Mbs - Recv: " + str(round(psutil.net_io_counters().bytes_recv / 1024.0 / 1024, 2)) + "Mbs")
    # x btn
    def closeEvent(self, event):
        if not self.isStarted:
            self.animation.start()
            self.isStarted = True
            event.ignore()
        else:   
            QWidget.closeEvent(self, event)
    # - btn
    def toggleFull(self):
            if widget.windowState() & QtCore.Qt.WindowFullScreen:
                #Fade In and out
                for i in range(10):
                    i = i / 10
                    widget.setWindowOpacity(1 - i)
                    time.sleep(0.024)
                widget.showNormal()
                for i in range(10):
                    i = i * 10
                    widget.setWindowOpacity(1 + i)
                    time.sleep(0.014)
                widget.setWindowOpacity(1)
            else:
                #Fade In and out
                for i in range(10):
                    i = i / 10
                    widget.setWindowOpacity(1 - i)
                    time.sleep(0.024)
                widget.showFullScreen()
                for i in range(10):
                    i = i * 10
                    widget.setWindowOpacity(1 + i)
                    time.sleep(0.014)
                widget.setWindowOpacity(1)
    # [] btn full screen not full screen btn
    def mintab(self):
        widget.showMinimized()
    #yellow Hashtag Button
    def goHashhtag(self):
        #calc
        last_received = psutil.net_io_counters().bytes_recv
        last_sent = psutil.net_io_counters().bytes_sent
        #bytes to Mbs
        mb_new_received = last_received / 1024 / 1024
        mb_new_sent = last_sent / 1024 / 1024
        #show popup
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText(f"{mb_new_received:.2f} MB received, {mb_new_sent:.2f} MB sent")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Ok)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
    #help refresh
    def showrefresh(self):
        #show popup
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText(f"Please hover over the graph and click the little 'A' in the corner to recenter the graph")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Ok)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
    #RED BUTTON Show conns
    def popupNetworkConns(self):
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText(f"{conn}")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Yes | QMessageBox.No)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
    #exit 
    def quit(self):
        global warnMessageOnExit
        if (warnMessageOnExit == False):
            QtWidgets.qApp.quit()
        else:
            msgBox = QMessageBox()
            msgBox.setIcon(QMessageBox.Information)
            msgBox.setText("Are you sure you want to exit HomeScan?")
            msgBox.setWindowTitle("HomeScan")
            msgBox.setStandardButtons(QMessageBox.Yes | QMessageBox.No)
            msgBox.setWindowIcon(QIcon("assets/icon.ico"))
            returnValue = msgBox.exec()
            if returnValue == QMessageBox.Yes:
                for i in range(20):
                    i = i / 20
                    widget.setWindowOpacity(1 - i)
                    time.sleep(0.1)

                QtWidgets.qApp.quit()
    #load table CONNECTION TABLE
    def loadtable_CONNECTION_TABLE(self): 
        #load CONNECTION TABLE
        people=psutil.net_connections()
        self.tableWidget.setRowCount(len(people))
        #style header
        header_style = "::section {""background-color: #0C0C0C; color: lightgrey; }"
        self.tableWidget.verticalHeader().setStyleSheet(header_style)
        self.tableWidget.horizontalHeader().setStyleSheet(header_style)
        #loop through array
        row = 0
        for person in people: 
            self.tableWidget.setItem(row, 0, QtWidgets.QTableWidgetItem(str(people[row][3][0])))
            self.tableWidget.setItem(row, 1, QtWidgets.QTableWidgetItem(str(person[6])))
            self.tableWidget.setItem(row, 2, QtWidgets.QTableWidgetItem(str(person[5])))
            self.tableWidget.setItem(row, 3, QtWidgets.QTableWidgetItem(str(people[row][3][1])))
            # CHANGE BACKGROUND COLOR BASED ON TYPE
            if str(person[5]) == "NONE":
                self.tableWidget.item(row, 2).setForeground(QColor(255,255,255))
            elif str(person[5]) == "CLOSE_WAIT":
                self.tableWidget.item(row, 2).setBackground(QColor(255,255,102))
                self.tableWidget.item(row, 2).setForeground(QColor(0,0,0))
            elif str(person[5]) == "ESTABLISHED":
                self.tableWidget.item(row, 2).setBackground(QColor(255,215,0))
                self.tableWidget.item(row, 2).setForeground(QColor(0,0,0))
            elif str(person[5]) == "LISTEN":
                self.tableWidget.item(row, 2).setBackground(QColor(255,99,71))
            elif str(person[5]) == "SYN_SENT":
                self.tableWidget.item(row, 2).setBackground(QColor(255,127,80))
            elif str(person[5]) == "FIN_WAIT2":
                self.tableWidget.item(row, 2).setBackground(QColor(150,145,251))
            row=row+1
        self.tableWidget.resizeRowsToContents()
        self.tableWidget.resizeColumnsToContents()  
    #load table 2 NETWORK INTERFACE TABLE
    def loadtable_NETWORK_INTERFACETABLE(self):
        people=psutil.net_if_stats()
        row = 0
        self.tableWidget_2.setRowCount(len(people))
        header_style = "::section {""background-color: #0C0C0C; color: lightgrey; }"
        self.tableWidget_2.verticalHeader().setStyleSheet(header_style)
        self.tableWidget_2.horizontalHeader().setStyleSheet(header_style)
        app.setStyleSheet(' QTableWidget QTableCornerButton::section {background-color: #0C0C0C; }')
        for key in people:
            self.tableWidget_2.setItem(row, 0, QtWidgets.QTableWidgetItem(str(key)))
            self.tableWidget_2.setItem(row, 1, QtWidgets.QTableWidgetItem(str(people[key][0])))
            self.tableWidget_2.setItem(row, 2, QtWidgets.QTableWidgetItem(str(people[key][2])))
            self.tableWidget_2.setItem(row, 3, QtWidgets.QTableWidgetItem(str(people[key][3])))
            if people[key][0] == True:
                self.tableWidget_2.item(row, 1).setBackground(QColor(191, 32, 32))
            row=row+1
        row = 0
        self.tableWidget_2.resizeRowsToContents()
        self.tableWidget_2.resizeColumnsToContents()  
    # router - Changing screen functions
    #
    #   Each function is the same, but only displaying a new route for the mainwindow widget.
    # 
    def goToLivePanel(self):
        livePanel = LivePanel()
        widget.addWidget(livePanel)
        widget.setCurrentIndex(widget.currentIndex()+1)
    def advanced_module(self):
        Advancedscreen = AdvancedScreen()
        widget.addWidget(Advancedscreen)
        widget.setCurrentIndex(widget.currentIndex()+1)
    def goToSettings(self):
        settings = Settings()
        widget.addWidget(settings)
        widget.setCurrentIndex(widget.currentIndex()+1)
    def goToQscan(self):
        qscan = Qscan()
        widget.addWidget(qscan)
        widget.setCurrentIndex(widget.currentIndex()+1)
    def goToArp_scan(self):
        arp_scan = ARP()
        widget.addWidget(arp_scan)
        widget.setCurrentIndex(widget.currentIndex()+1)
    def goToAlerts(self):
        alerts = Alerts()
        widget.addWidget(alerts)
        widget.setCurrentIndex(widget.currentIndex()+1)
    def goToDevices(self):
        devices = Devices()
        widget.addWidget(devices)
        widget.setCurrentIndex(widget.currentIndex()+1)
    def goToScan(self):
        scan = Scan()
        widget.addWidget(scan)
        widget.setCurrentIndex(widget.currentIndex()+1)
    def goToAScan(self):
        ascan = AScan()
        widget.addWidget(ascan)
        widget.setCurrentIndex(widget.currentIndex()+1)
    def goToHelp(self):
        help = Help()
        widget.addWidget(help)
        widget.setCurrentIndex(widget.currentIndex()+1)
    def goToOverview(self):
        overview = Overview()
        widget.addWidget(overview)
        widget.setCurrentIndex(widget.currentIndex()+1)
# Live Panel - Top bar 1st
class LivePanel(QMainWindow):
    def __init__(self):
        super(LivePanel,self).__init__()
        uic.loadUi('router/top_bar/live.ui',self)
        # connect btns
        self.pushButton.clicked.connect(self.goBack)
        self.help.clicked.connect(self.goHelp)
        self.warning.clicked.connect(self.inconsitancy_msgBox)
        # grab global var
        global _last_net_io_meta
        _last_net_io_meta = 0
        #define 4 lines
        self.x = list(range(150))  
        self.y1 = [randint(0,0) for _ in range(150)]  
        self.y3z =  [randint(0,0) for _ in range(150)]
        self.y3zerrin =  [randint(0,0) for _ in range(150)]
        self.y3zerrout =  [randint(0,0) for _ in range(150)]
        #set pen colors for those lines
        pen = pg.mkPen(color=(255,4,111), width=4, style=QtCore.Qt.DashLine)
        pen_2 = pg.mkPen(color=(255,191,0), width=4, style=QtCore.Qt.DashLine)
        pen_3 = pg.mkPen(color=(251,206,177), width=4, style=QtCore.Qt.DashLine)
        pen_4 = pg.mkPen(color=(111,4,111), width=4, style=QtCore.Qt.DashLine)
        #set data lines
        self.data_line3 = self.graphicsView_2.plot(self.x,  self.y3z, name="MB sent", pen=pen)
        self.data_line1 = self.graphicsView_2.plot(self.x, self.y1, name="MB recv", pen=pen_2)
        self.data_line4 = self.graphicsView_2.plot(self.x,  self.y3zerrin, name="Error in", pen=pen_3)
        self.data_line5 = self.graphicsView_2.plot(self.x, self.y3zerrout, name="Error out", pen=pen_4)
        #other graph settings
        self.graphicsView_2.addLegend()
        self.graphicsView_2.showGrid(x=True, y=True)
        self.graphicsView_2.setBackground(QColor(12, 12, 12))
        #init timer
        self.timer = QtCore.QTimer()
        self.timer.setInterval(10)
        self.timer.timeout.connect(self.update_plot_data)
        #start timer
        self.timer.start()
        #set first values for live panel
        bytes_recv_ = round(psutil.net_io_counters().bytes_recv / 1024.0 / 1024, 2)
        bytes_sent_ = round(psutil.net_io_counters().bytes_sent / 1024.0 / 1024, 2)
        errin_ = round(psutil.net_io_counters().errin / 1024.0 / 1024, 2)
        errout_ = round(psutil.net_io_counters().errout / 1024.0 / 1024, 2)
        self.diff_1 = bytes_recv_
        self.diff_2 = bytes_sent_
        self.diff_3 = errin_
        self.diff_4 = errout_
    #UPDATE LIVE PANEL
    def update_plot_data(self):
        self.x = self.x[1:]  # Remove the first y element.
        self.x.append(self.x[-1] + 1)  # Add a new value 1 higher than the last.
        # grab network details
        theval2 = round(psutil.net_io_counters().bytes_recv / 1024.0 / 1024, 2)
        theval3 = round(psutil.net_io_counters().bytes_sent / 1024.0 / 1024, 2)
        theval4 = round(psutil.net_io_counters().errin / 1024.0 / 1024, 2)
        theval5 = round(psutil.net_io_counters().errout / 1024.0 / 1024, 2)
        # remove first val
        self.y1 = self.y1[1:]
        self.y3z = self.y3z[1:] 
        self.y3zerrin = self.y3zerrin[1:] 
        self.y3zerrout = self.y3zerrout[1:]
        #calc diff
        self.y1.append(theval2 - self.diff_1)
        self.y3z.append(theval3 - self.diff_2)
        self.y3zerrin.append(theval4 - self.diff_3)
        self.y3zerrout.append(theval5 - self.diff_4)
        #DRAW NEW LINES
        self.data_line3.setData(self.x, self.y3z)
        self.data_line1.setData(self.x, self.y1)
        self.data_line4.setData(self.x, self.y3zerrin)
        self.data_line5.setData(self.x, self.y3zerrout)
        self.diff_1 = theval2
        self.diff_2 = theval3
        self.diff_3 = theval4
        self.diff_4 = theval5
    # Calculate net usage.
    def net_io_usage():
        global _last_net_io_meta
        net_counters = psutil.net_io_counters()
        tst = time.time()
        send_bytes =  psutil.net_io_counters().bytes_sent
        recv_bytes = net_counters.bytes_recv
        if _last_net_io_meta is None:
            _last_net_io_meta = (send_bytes, recv_bytes, tst)
            return None
        last_send_bytes, last_recv_bytes, last_time = _last_net_io_meta
        delta_time = tst - last_time
        recv_speed = (recv_bytes - last_recv_bytes) / delta_time
        send_speed = (send_bytes - last_send_bytes) / delta_time
        _last_net_io_meta = (send_bytes, recv_bytes, tst)
        print(recv_speed, send_speed)
        return recv_speed, send_speed 
    # help box
    def goHelp(self):
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText("Here is the live panel, you can see in real time your download and upload aswell as your error rates ! \n\n If you have any questions about the live panel then please contact the github.\nhttps://github.com/hunterjreid/HomeScan")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Ok)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
    # warning box
    def inconsitancy_msgBox(self):
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText("This graph may contain inconsitancys when first loading due to the way the graph is calculated, it is possiable that you may only see your network results after a few seconds")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Ok)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
    # go back
    def goBack(self):
        homeScanMain = HomeScanMain()
        widget.addWidget(homeScanMain)
        widget.setCurrentIndex(widget.currentIndex()+1)
# Network Overview - Left bar 1st
class Overview(QMainWindow):
    def __init__(self):
        super(Overview,self).__init__()
        #LOAD UI FROM ROUTER
        uic.loadUi('router/overview.ui',self)
        #LINK BUTTONS TO UI
        self.help.clicked.connect(self.goHelp)
        self.pushButton.clicked.connect(self.goBack)
        self.export_2.clicked.connect(self.export)
        #MORE UI SETTINGS
        self.graphicsView_2.setBackground(QColor(12,12,12,0))
        #CONN ADD
        global conn
        conn = psutil.net_connections()
        pen = pg.mkPen(color=(191, 32, 32), width=0.5, style=QtCore.Qt.DashLine)
        #For Loop append IP and Port to graph
        row = 0
        port_x = []
        PID_y = []
        people=psutil.net_connections()
        for person in people:
            port_x.append(person[6])  
            PID_y.append(people[row][3][1])
            row=row+1
        #append
        self.graphicsView_2.plot(port_x, PID_y, pen=pen, symbol='+', symbolSize=15, symbolBrush=('red'))
    #Return to mainwindow   
    def goBack(self):
        homeScanMain = HomeScanMain()
        widget.addWidget(homeScanMain)
        widget.setCurrentIndex(widget.currentIndex()+1)
    #Help info box
    def goHelp(self):
        Connections=psutil.net_connections()
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText("This screen is the overview screen and \n\nY axis = Port\n\nX axis = UID\n\nThere are " + str(len(Connections)) + " showed nodes in the graph.\n\nA tpyical network ill have an even spread\n\nWhere ever all the strings go to is the hotspot")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Ok)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
    #exports
    def export(self):
        now = datetime.now()
        #grab time
        current_time = now.strftime("%m-%d-%Y--%H-%M-%S")
        #digest exports
        row = 0
        port_x = []
        PID_y = []
        people=psutil.net_connections()
        for person in people:
            port_x.append(person[6])  
            PID_y.append(people[row][3][1])
            row=row+1
        #append to file
        name_of_file = "NetworkOverview-" + str(current_time)
        completeName = 'exports/'+ name_of_file + ".log"
        file1 = open(completeName , "w")
        toFile = str(current_time) +" \nx: " +  str(port_x) + "\ny: " +  str(PID_y)
        file1.write(toFile)
        file1.close()
        #export box popup sucess
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText("Thank you for using HomeScan.\n\nYour exports has been saved under NetworkOverview-" + str(current_time) + ".log \n\nDo you think there is something wrong with your exports? or have you found a bug? Email support@homescan.com to open a support ticket! Show the github some aswell\nhttps://github.com/hunterjreid/HomeScan")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Ok)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
# Settings screen - left bar 2nd down
class Settings(QMainWindow):
    def __init__(self):
        super(Settings,self).__init__()
        uic.loadUi('router/settings.ui',self)
        #connect buttons
        self.back.clicked.connect(self.goBack)
        self.help.clicked.connect(self.goHelp)
        self.export_2.clicked.connect(self.export)
        #
        # These buttons need to be renamed
        # Settings buttons
        #
        self.pushButton_11.clicked.connect(self.togglePrivateMode)
        self.pushButton_9.clicked.connect(self.togglePrivateMode)
        self.pushButton_5.clicked.connect(self.toggleVpnMode)
        self.pushButton_6.clicked.connect(self.toggleVpnMode)
        self.pushButton.clicked.connect(self.togglesound)
        self.pushButton_2.clicked.connect(self.togglesound)
        self.pushButton_3.clicked.connect(self.togglemainwarnMessageOnExit)
        self.pushButton_4.clicked.connect(self.togglemainwarnMessageOnExit)
        self.pushButton_8.clicked.connect(self.togshowMainMessage)
        self.pushButton_7.clicked.connect(self.togshowMainMessage)
        global privateMode, VpnMode, isSound, showMainMessage
        if (not privateMode):
            self.pushButton_9.setStyleSheet("background:rgb(171, 0, 0);font-size:13px;padding:10px;color:white;")
            self.pushButton_11.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
        else:
            self.pushButton_9.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
            self.pushButton_11.setStyleSheet("background-color: rgb(0, 139, 0);color:white;font-size:13px;padding:10px;")
        if (not VpnMode):
            self.pushButton_6.setStyleSheet("background:rgb(171, 0, 0);font-size:13px;padding:10px;color:white;")
            self.pushButton_5.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
        else:
            self.pushButton_6.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
            self.pushButton_5.setStyleSheet("background-color: rgb(0, 139, 0);color:white;font-size:13px;padding:10px;")
        if (not isSound):
            self.pushButton_2.setStyleSheet("background:rgb(171, 0, 0);font-size:13px;padding:10px;color:white;")
            self.pushButton.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
        else:
            self.pushButton_2.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
            self.pushButton.setStyleSheet("background-color: rgb(0, 139, 0);color:white;font-size:13px;padding:10px;")
        if (not warnMessageOnExit):
            self.pushButton_4.setStyleSheet("background:rgb(171, 0, 0);font-size:13px;padding:10px;color:white;")
            self.pushButton_3.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
        else:
            self.pushButton_4.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
            self.pushButton_3.setStyleSheet("background-color: rgb(0, 139, 0);color:white;font-size:13px;padding:10px;")
        if (not showMainMessage):
            self.pushButton_8.setStyleSheet("background:rgb(171, 0, 0);font-size:13px;padding:10px;color:white;")
            self.pushButton_7.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
        else:
            self.pushButton_8.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
            self.pushButton_7.setStyleSheet("background-color: rgb(0, 139, 0);color:white;font-size:13px;padding:10px;")
    #help screen
    def goHelp(self):
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText("You may change any setting you want ! \n\n If you have any questions about the settings then please contact the github.\nhttps://github.com/hunterjreid/HomeScan")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Ok)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
    #export btn
    def export(self):
        now = datetime.now()
        #grab time
        current_time = now.strftime("%m-%d-%Y--%H-%M-%S")
        #digest exports
        row = 0
        port_x = []
        PID_y = []
        people=psutil.net_connections()
        for person in people:
            port_x.append(person[6])  
            PID_y.append(people[row][3][1])
            row=row+1
        #append to file
        name_of_file = "Settings-" + str(current_time)
        completeName = 'exports/'+ name_of_file + ".log"
        file1 = open(completeName , "w")
        global privateMode, VpnMode, isSound, showMainMessage
        toFile = "Settings\nPrivate mode: " + str(privateMode) + "\nVPN mode: " + str(VpnMode) + "\nSound : " +  str(isSound) + "\nMain message : " + str(showMainMessage)
        #final string
        file1.write(toFile)
        file1.close()
        #show success exports
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText("Thank you for using HomeScan.\n\nYour exports has been saved under Settings-" + str(current_time) + ".log \n\nDo you think there is something wrong with your exports? or have you found a bug? Email support@homescan.com to open a support ticket! Show the github some aswell\nhttps://github.com/hunterjreid/HomeScan")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Ok)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
    #go back
    def goBack(self):
        homeScanMain = HomeScanMain()
        widget.addWidget(homeScanMain)
        widget.setCurrentIndex(widget.currentIndex()+1)
    # TOGGLE FUNCTIONS
    def togglePrivateMode(self):
        global privateMode
        if (privateMode):
            self.pushButton_9.setStyleSheet("background:rgb(171, 0, 0);font-size:13px;padding:10px;color:white;")
            self.pushButton_11.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
            privateMode = not privateMode
        else:
            self.pushButton_9.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
            self.pushButton_11.setStyleSheet("background-color: rgb(0, 139, 0);color:white;font-size:13px;padding:10px;")
            privateMode = not privateMode
    def toggleVpnMode(self):
        global VpnMode
        if (VpnMode):
            self.pushButton_6.setStyleSheet("background:rgb(171, 0, 0);font-size:13px;padding:10px;color:white;")
            self.pushButton_5.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
            VpnMode = not VpnMode
        else:
            self.pushButton_6.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
            self.pushButton_5.setStyleSheet("background-color: rgb(0, 139, 0);color:white;font-size:13px;padding:10px;")
            VpnMode = not VpnMode
    def togglesound(self):
        global isSound
        if (isSound):
            self.pushButton_2.setStyleSheet("background:rgb(171, 0, 0);font-size:13px;padding:10px;color:white;")
            self.pushButton.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
            isSound = not isSound
        else:
            self.pushButton_2.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
            self.pushButton.setStyleSheet("background-color: rgb(0, 139, 0);color:white;font-size:13px;padding:10px;")
            isSound = not isSound
    def togglemainwarnMessageOnExit(self):
        global warnMessageOnExit
        if (warnMessageOnExit):
            self.pushButton_4.setStyleSheet("background:rgb(171, 0, 0);font-size:13px;padding:10px;color:white;")
            self.pushButton_3.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
            warnMessageOnExit = not warnMessageOnExit
        else:
            self.pushButton_4.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
            self.pushButton_3.setStyleSheet("background-color: rgb(0, 139, 0);color:white;font-size:13px;padding:10px;")
            warnMessageOnExit = not warnMessageOnExit
    def togshowMainMessage(self):
        global showMainMessage
        if (showMainMessage):
            self.pushButton_8.setStyleSheet("background:rgb(171, 0, 0);font-size:13px;padding:10px;color:white;")
            self.pushButton_7.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
            showMainMessage = not showMainMessage
        else:
            self.pushButton_8.setStyleSheet("background:grey;color:darkgrey;font-size:13px;padding:10px;")
            self.pushButton_7.setStyleSheet("background-color: rgb(0, 139, 0);color:white;font-size:13px;padding:10px;")
            showMainMessage = not showMainMessage
# Alerts screen (bell icon)
class Alerts(QMainWindow):
    def __init__(self):
        super(Alerts,self).__init__()
        #load ui
        uic.loadUi('router/alert.ui',self)
        #import global
        global alerts
        #load alerts and append them to the alert screen
        for i in alerts:
            print(i)
            self.listWidget_2.addItem(str(i))
        self.back.clicked.connect(self.goBack)
        self.help.clicked.connect(self.goHelp)
    # go help
    def goHelp(self):
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText("This is the alerts screen, here you can read up on your network alerts.")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Ok)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
    # go back
    def goBack(self):
        homeScanMain = HomeScanMain()
        widget.addWidget(homeScanMain)
        widget.setCurrentIndex(widget.currentIndex()+1)
# help screen left bar 3rd down
class Help(QMainWindow):
    def __init__(self):
        super(Help,self).__init__()
        uic.loadUi('router/help_ui.ui',self)
        self.back.clicked.connect(self.goBack)
        self.export_2.clicked.connect(self.goHelp)
    def goHelp(self):
        webbrowser.open('https://github.com/hunterjreid/HomeScan')  
    def goBack(self):
        homeScanMain = HomeScanMain()
        widget.addWidget(homeScanMain)
        widget.setCurrentIndex(widget.currentIndex()+1)
class ARP(QMainWindow):
    def __init__(self):
        super(ARP,self).__init__()
        uic.loadUi('router/top_bar/arp.ui',self)
        self.back.clicked.connect(self.goBack)
        self.help.clicked.connect(self.goHelp)
        self.export_2.clicked.connect(self.export)
        devices = []
        # LOAD ARP 
        for device in os.popen('arp -a'):
            devices.append(device)
            self.listWidget.addItem(str(device))
    def goHelp(self):
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText("Help Iinformation about this screen goes here")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Ok)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
    def export(self):
        now = datetime.now()
        current_time = now.strftime("%m-%d-%Y--%H-%M-%S")
        name_of_file = "ARP-" + str(current_time)
        completeName = 'exports/'+ name_of_file + ".log"
        file1 = open(completeName , "w")
        toFile = "ARP - "
        for device in os.popen('arp -a'):
            toFiles += str(device)
        file1.write(toFile)
        file1.close()
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText("Thank you for using HomeScan.\n\nYour exports has been saved under ARP-" + str(current_time) + ".log \n\nDo you think there is something wrong with your exports? or have you found a bug? Email support@homescan.com to open a support ticket! Show the github some aswell\nhttps://github.com/hunterjreid/HomeScan")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Ok)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
    def goBack(self):
        homeScanMain = HomeScanMain()
        widget.addWidget(homeScanMain)
        widget.setCurrentIndex(widget.currentIndex()+1)
# CONNECTED IP / DEVICES
class Devices(QMainWindow):
    def __init__(self):
        super(Devices,self).__init__()
        uic.loadUi('router/top_bar/error.ui',self)
        self.back.clicked.connect(self.goBack)
        self.help.clicked.connect(self.goHelp)
        self.export_2.clicked.connect(self.export)
        for device in psutil.net_connections(kind='tcp'):
            if str(device.raddr) != "()":
                self.listWidget.addItem(str(device.raddr))
    def export(self):
        now = datetime.now()
        current_time = now.strftime("%m-%d-%Y--%H-%M-%S")
        name_of_file = "Devices-" + str(current_time)
        completeName = 'exports/'+ name_of_file + ".log"
        file1 = open(completeName , "w")
        toFile = self.lineEdit.text()
        file1.write(toFile)
        file1.close()
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText("Thank you for using HomeScan.\n\nYour exports has been saved under Devices-" + str(current_time) + ".log \n\nDo you think there is something wrong with your exports? or have you found a bug? Email support@homescan.com to open a support ticket! Show the github some aswell\nhttps://github.com/hunterjreid/HomeScan")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Ok)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
    def goBack(self):
        homeScanMain = HomeScanMain()
        widget.addWidget(homeScanMain)
        widget.setCurrentIndex(widget.currentIndex()+1)
    def toggleFull(self):
            if self.windowState() & QtCore.Qt.WindowFullScreen:
                self.showNormal()
            else:
                self.showFullScreen()
    def goHelp(self):
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText("Help Iinformation about this screen goes here")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Ok)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
#SCANs ---------------------------------------
#Normal Scan
class Scan(QMainWindow):
    def __init__(self):
        super(Scan,self).__init__()
        uic.loadUi('router/scan_ui/NetworkScan.ui',self)
        self.back.clicked.connect(self.goBack)
        self.help.clicked.connect(self.goHelp)
        self.pushButton.clicked.connect(self.runScan)
        self.pushButton_15.clicked.connect(self.scanIP)
        self.listWidget.itemClicked.connect(self.Clicked)
        global conn
        self.selected_port = 10
    #Check if entered IP is IPv4 or IPv6
    def scanIP(self):
        try: 
            ipaddress.ip_address(self.lineEdit.text())
            self.label_9.setText('IP Valid')
        except: 
            self.label_9.setText('IP is not valid, NOT IPv4 or IPv6 address')
    # click history
    def Clicked(self,item):
        index = self.listWidget.currentRow()
        self.lineEdit.setText(str(conn[index][3][1]))
        self.textEdit.append(str( "You clicked: "+item.text() + " Port :" + str(conn[index][3][1])))
    # run scan btn
    def runScan(self):
        row = 0
        for person in conn:
            self.listWidget.addItem(str(conn[row][3][0]) + ":" + str(conn[row][3][1]) + " | " + str(person[5]))
            if str(person[5]) == "NONE":
                self.listWidget.item(row).setBackground(QColor(113,54,51,0))
            elif str(person[5]) == "CLOSE_WAIT":
                self.listWidget.item(row).setBackground(QColor(255,255,102,122))
            elif str(person[5]) == "ESTABLISHED":
                self.listWidget.item(row).setBackground(QColor(76,154,1,122))
            elif str(person[5]) == "LISTEN":
                self.listWidget.item(row).setBackground(QColor(0,254,251,122))
            elif str(person[5]) == "SYN_SENT":
                self.listWidget.item(row).setBackground(QColor(255,4,111,122))
            elif str(person[5]) == "FIN_WAIT2":
                self.listWidget.item(row).setBackground(QColor(150,145,251,122))
            row=row+1
    def goHelp(self):
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText("Help Iinformation about this screen goes here")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Ok)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
    def goBack(self):
        homeScanMain = HomeScanMain()
        widget.addWidget(homeScanMain)
        widget.setCurrentIndex(widget.currentIndex()+1)
#Quick SCAN
class Qscan(QMainWindow):
    def __init__(self):
        super(Qscan,self).__init__()
        uic.loadUi('router/scan_ui/Quick.ui',self)
        global conn, IPAddr, hostname
        #link btns
        self.QuickScan_Btn.clicked.connect(self.quickscan)
        self.back.clicked.connect(self.goBack)
        self.help.clicked.connect(self.goHelp)
    def quickscan(self):
        #Gets the host IP address
        ip = socket.gethostbyname(socket.gethostname())
        self.label_6.setText(ip)
        #set progress bars
        self.progressBar.setMinimum(0)
        self.progressBar.setMaximum(65535)
        self.progressBar.setValue(0)
        #Shows if there is an internet connection indirectly by checking the sockets rather than a URL
        if ip == "127.0.0.1":
            #If the IP is 12.0.0.1 there is no internet connection
            self.label_5.setText("No Current Internet Connection")
        else:
            self.label_5.setText("Connected to the internet")
        # The range comes from how many ports there are in TCP & UDP = 65535 and how many we are looking at
        for port in range(65535):
            try:
                sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
                """"The arguments passed to socket() are constants used to specify the address family 
                and socket type. AF_INET is the Internet address family for IPv4. SOCK_STREAM is the socket type
                for TCP, the protocol that will be used to transport messages in the network."""
                self.progressBar.setValue(self.progressBar.value() + 1)
                sock.bind((ip,port))
                """The .bind() method is used to associate the socket with a specific network interface and port number: 
                In this case our IP address """
            except:
                self.textEdit.append(str("        ✅      ")+ str("          Open Port Nunber:           ") + str( port))
                self.textEdit.repaint()
    def goHelp(self):
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText("Help Iinformation about this screen goes here")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Ok)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
    def goBack(self):
        homeScanMain = HomeScanMain()
        widget.addWidget(homeScanMain)
        widget.setCurrentIndex(widget.currentIndex()+1)   
#Anrgy SCAN
class AdvancedScreen(QMainWindow):
    def __init__(self):
        super(AdvancedScreen,self).__init__()
        uic.loadUi('router/scan_ui/Angry_scan.ui',self)
        global conn, IPAddr, hostname
        # set headers
        self.textEdit.append(str("Hostname : " + str(hostname)))
        self.textEdit.append(str("Home IP : " + str(IPAddr)))
        self.textEdit.append(str("Scan Init : " + str(datetime.now())))
        #link buttons
        self.pushButton_4.clicked.connect(self.addTxt)
        self.pushButton_3.clicked.connect(self.set_IP)
        self.pushButton.clicked.connect(self.clear_fields)
        self.back.clicked.connect(self.goBack)
        self.help.clicked.connect(self.goHelp)
    def goHelp(self):
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText("Here we explain in great detail what this scan does.")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Ok)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
    def goBack(self):
        homeScanMain = HomeScanMain()
        widget.addWidget(homeScanMain)
        widget.setCurrentIndex(widget.currentIndex()+1)
    #set Ip as your own
    def set_IP(self):
        global conn, IPAddr, hostname
        self.lineEdit_5.setText(IPAddr)
    def addTxt(self):
        if (self.lineEdit_5.text() == ''):
            msgBox = QMessageBox()
            msgBox.setIcon(QMessageBox.Warning)
            msgBox.setText("Please enter a Target")
            msgBox.setWindowTitle("HomeScan")
            msgBox.setStandardButtons(QMessageBox.Ok)
            msgBox.setWindowIcon(QIcon("assets/icon.ico"))
            returnValue = msgBox.exec()
            if returnValue == QMessageBox.Ok:
                return
        elif (self.lineEdit_6.text() == ''):
            msgBox = QMessageBox()
            msgBox.setIcon(QMessageBox.Warning)
            msgBox.setText("Please enter a Starting port")
            msgBox.setWindowTitle("HomeScan")
            msgBox.setStandardButtons(QMessageBox.Yes)
            msgBox.setWindowIcon(QIcon("assets/icon.ico"))
            returnValue = msgBox.exec()
            if returnValue == QMessageBox.Ok:
                return
        # Check if all the values are legit and can be used.
        # Var checking and sanitation below.
        elif (int(self.lineEdit_7.text()) - int(self.lineEdit_6.text()) > 200):
            msgBox = QMessageBox()
            msgBox.setIcon(QMessageBox.Warning)
            msgBox.setText("You are questing to scan more than 200+ ports\n\nAre you sure you want to do this? \n\nIt may crash the application or take along time to complete use low ms")
            msgBox.setWindowTitle("HomeScan")
            msgBox.setStandardButtons(QMessageBox.Ok  | QMessageBox.Abort)
            msgBox.setWindowIcon(QIcon("assets/icon.ico"))
            returnValue = msgBox.exec()
            #KILL with return
            if returnValue == QMessageBox.Abort:
                return
        elif (self.lineEdit_7.text() == ''):
            msgBox = QMessageBox()
            msgBox.setIcon(QMessageBox.Warning)
            msgBox.setText("Please enter a End port")
            msgBox.setWindowTitle("HomeScan")
            msgBox.setStandardButtons(QMessageBox.Ok)
            msgBox.setWindowIcon(QIcon("assets/icon.ico"))
            returnValue = msgBox.exec()
            #KILL with return
            if returnValue == QMessageBox.Ok:
                return
        elif (self.lineEdit_8.text() == ''):
            msgBox = QMessageBox()
            msgBox.setIcon(QMessageBox.Warning)
            msgBox.setText("Please enter a Time delay")
            msgBox.setWindowTitle("HomeScan")
            msgBox.setStandardButtons(QMessageBox.Ok)
            msgBox.setWindowIcon(QIcon("assets/icon.ico"))
            returnValue = msgBox.exec()
            #KILL with return
            if returnValue == QMessageBox.Ok:
                return
        # RUN SCAN !
        target = self.lineEdit_5.text()
        s_port = int(self.lineEdit_6.text())
        e_port = int(self.lineEdit_7.text())
        ms = int(self.lineEdit_8.text())
        #set progress bar
        self.progressBar_2.setMinimum(s_port)
        self.progressBar_2.setMaximum(e_port)
        self.progressBar_2.setValue(s_port)
        #Start Scan
        self.textEdit.append(str("-" * 45))
        self.textEdit.append(str("Scanning Target: " + target))
        self.textEdit.append(str("Scanning Ports: " + self.lineEdit_6.text() + " - " + self.lineEdit_7.text()))
        self.textEdit.append(str("Scanning started at: " + str(datetime.now())))
        self.textEdit.append(str("-" * 45))
        try:
            for port in range(s_port, (e_port+1)):
                print(self.progressBar_2.setValue(self.progressBar_2.value() + 1))
                
                s=socket.socket(socket.AF_INET, socket.SOCK_STREAM)
                socket.setdefaulttimeout(ms/1000)
                res = s.connect_ex((target, port))
                self.textEdit.repaint()
            
                if res == 0:
                    self.textEdit.append(str("Port " + str(port) + " is OPEN✅"))
                else:
                    self.textEdit.append(str("Port " + str(port) + " is closed❌"))
                self.textEdit.repaint()
                s.close()
        except:
            self.textEdit.append("ERROR: Something else went wrong - No IPv4 or IPv6 Found or bad connection")
            msgBox = QMessageBox()
            msgBox.setIcon(QMessageBox.Warning)
            msgBox.setText("ERROR: Something else went wrong, No IPv4 or IPv6 Found or bad connection, Faliure.")
            msgBox.setWindowTitle("HomeScan")
            msgBox.setStandardButtons(QMessageBox.Close)
            msgBox.setWindowIcon(QIcon("assets/icon.ico"))
            msgBox.exec()
    #clear all fields
    def clear_fields(self):
        global clicked
        if (self.lineEdit_5.text() == '' and self.lineEdit_6.text() == '' and self.lineEdit_7.text() == '' and self.lineEdit_8.text() == ''):
            msgBox = QMessageBox()
            msgBox.setIcon(QMessageBox.Warning)
            msgBox.setText("Nothing to clear")
            msgBox.setWindowTitle("HomeScan")
            msgBox.setStandardButtons(QMessageBox.Ok)
            msgBox.setWindowIcon(QIcon("assets/icon.ico"))
            msgBox.exec()
        self.lineEdit_5.setText('')
        self.lineEdit_6.setText('')
        self.lineEdit_7.setText('')
        self.lineEdit_8.setText('')
    #export
    def export(self):
        now = datetime.now()
        current_time = now.strftime("%m-%d-%Y--%H-%M-%S")
        name_of_file = "AngryScan-" + str(current_time)
        completeName = 'exports/'+ name_of_file + ".log"
        file1 = open(completeName , "w")
        toFile = self.textEdit.toPlainText()
        file1.write(toFile)
        file1.close()
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText("Thank you for using HomeScan.\n\nYour exports has been saved under ScanModule-" + str(current_time) + ".log \n\nDo you think there is something wrong with your exports? or have you found a bug? Email support@homescan.com to open a support ticket! Show the github some aswell\nhttps://github.com/hunterjreid/HomeScan")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Ok)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
    #go back
    def goBack(self):
        homeScanMain = HomeScanMain()
        widget.addWidget(homeScanMain)
        widget.setCurrentIndex(widget.currentIndex()+1)
#basic scan
class AScan(QMainWindow):
    def __init__(self):
        super(AScan,self).__init__()
        uic.loadUi('router/scan_ui/BasicScan.ui',self)
        #connect btns
        self.BasicScanBtn.clicked.connect(self.BasicScan)
        self.BasicScanBtn_2.clicked.connect(self.InternetSpeed)
        self.back.clicked.connect(self.goBack)
        self.help.clicked.connect(self.goHelp)
    #Internet speed func
    def InternetSpeed(self):
        st = speedtest.Speedtest(secure=True)
        #init speed test
        down = st.download()
        conDown = (down/1024)/1000 #Converting the internet speed to MBps takes ages tbh
        finalDownSpeed = round(conDown, 2)
        self.DwnLSp.setText(str(finalDownSpeed)+" MBps")
        upload = st.upload()
        conup = (upload/1024)/1000
        finalupSpeed = round(conup, 2)
        self.UpLSpd.setText(str(finalupSpeed) + "MBps")
    #basic scan func
    def BasicScan(self):
        # Gets the host IP address
        ip = socket.gethostbyname(socket.gethostname())
        self.Ipaddy.setText(ip)
        deviceName =  socket.gethostname()
        self.DevName.setText(deviceName)
        self.Os_sys.setText(platform.system())
        # Tests Internet Connection
        #Shows if there is an internet connection indirectly by checking the sockets rather than a URL
        if ip == "127.0.0.1":
         #If the IP is 12.0.0.1 there is no internet connection
            self.Intconn.setText("No Connection")
        else:
            self.Intconn.setText("Connected")
        #Ping test
        ip = socket.gethostbyname(socket.gethostname())
        dot = ip.rfind(".")
        ip = ip[0:dot +1]
        operatorsys = platform.system()
        #append to text edit
        for i in range(1, 5):
            host = ip + str(i)
            if operatorsys == "Windows":
                ping = "ping -n 1 -w 1 "
                response = os.system(ping + host + ">nul")
            else:
                ping = "ping -c 1 -w 1 "
                response = os.system(ping + host + ">/dev/null")
            if response == 0:
                self.textEdit.append(host + " is Active")
            else:
                self.textEdit.append(host + " in Not Active")
    def goHelp(self):
        msgBox = QMessageBox()
        msgBox.setIcon(QMessageBox.Information)
        msgBox.setText("Help Iinformation about this screen goes here")
        msgBox.setWindowTitle("HomeScan")
        msgBox.setStandardButtons(QMessageBox.Ok)
        msgBox.setWindowIcon(QIcon("assets/icon.ico"))
        msgBox.exec()
    def goBack(self):
        homeScanMain = HomeScanMain()
        widget.addWidget(homeScanMain)
        widget.setCurrentIndex(widget.currentIndex()+1)
#SCAN END -------------------------------Scan END
#init app !
if __name__ == '__main__':
    import sys
    #setups
    app = QApplication(sys.argv)
    widget = QtWidgets.QStackedWidget()
    window = HomeScanMain()
    widget.setWindowTitle("HomeScan Pro Edition")
    #Screensizing
    user32 = ctypes.windll.user32
    screensize = user32.GetSystemMetrics(0), user32.GetSystemMetrics(1)
    #fall back
    widget.setFixedSize(920, 550)
    #make dynamic screen size
    if (user32.GetSystemMetrics(1) < 1281):
        widget.setFixedSize(920, 550)
    else:
        widget.setFixedSize(920, 900)
    #icons and metadata
    widget.setWindowFlags(Qt.FramelessWindowHint)
    widget.addWidget(window)
    widget.setWindowIcon(QIcon('assets/icon.ico'))
    #show !!
    widget.show()
    #set start alert
    alerts.append('Notice: App started on ' + str(datetime.now()))
    QFontDatabase.addApplicationFont("assets/Gilroy.ttf")
    font = QFont("Gilroy")
    widget.window().setFont(font)
    QApplication.setFont(font, "QLabel")
    QApplication.setFont(font, "QCheckBox")
    QApplication.setFont(font, "QPushButton")
    QApplication.setFont(font, "QTextEdit")
    QApplication.setFont(font, "QListWidget")
    QApplication.setFont(font, "QLineEdit")
    QApplication.setFont(font, "QPlotWidget")
    QApplication.setFont(font, "QTableWidget")
    QApplication.setFont(font, "QGraphicsWidget")
    #START MAIN THREAD
    sys.exit(app.exec_())
    #eoc