From 26d762f1fcd7d045ff6fbc6d4093edd4f0e98424 Mon Sep 17 00:00:00 2001
From: R Nanda Kumar <jerrynandak@gmail.com>
Date: Thu, 23 Dec 2021 12:58:31 +0530
Subject: [PATCH] color and clear temp fix

---
 network_monitor.py | 21 ++++++++++++++++++---
 trace_route.py     |  2 +-
 2 files changed, 19 insertions(+), 4 deletions(-)

diff --git a/network_monitor.py b/network_monitor.py
index 8ce29a9..7774198 100644
--- a/network_monitor.py
+++ b/network_monitor.py
@@ -106,13 +106,14 @@ class NetworkMonitor(QMainWindow):
             self.thread.terminate()
         
     def packetClear(self):
-        self.actionStart.setEnabled(False)
-        self.actionStop.setEnabled(True)
+        
+        #self.actionStart.setEnabled(False)
+        #self.actionStop.setEnabled(True)
         
         self.thread.terminate()
         self.tableWidget.clearContents()
         self.tableWidget.setRowCount(0)
-        self.thread.start()
+        #sself.thread.start()
     
     def interfaceDialog(self):
         self.interfaceDiag = InterfacePick() 
@@ -134,6 +135,20 @@ class NetworkMonitor(QMainWindow):
         self.tableWidget.setItem(rowpos, 5, QTableWidgetItem(str(tableData['length'])))
         self.tableWidget.setItem(rowpos, 6, QTableWidgetItem(tableData['info']))
         
+        if(tableData['Protocol'] == 'TCP'):
+            self.setColortoRow(self.tableWidget, rowpos, QColor(173,191, 255))
+        elif(tableData['Protocol'] == 'UDP'):
+            self.setColortoRow(self.tableWidget, rowpos, QColor(157,240,255))
+        elif(tableData['Protocol'] == 'Other'):
+            self.setColortoRow(self.tableWidget, rowpos, QColor(125,125,146))
+        elif(tableData['Protocol'] == 'ARP'):
+            self.setColortoRow(self.tableWidget, rowpos, QColor(157,240,77))
+
+    
+    def setColortoRow(self, table, rowIndex, color):
+        for j in range(table.columnCount()):
+            table.item(rowIndex, j).setBackground(color)
+            
 class InterfacePick(QDialog):
     def __init__(self, parent=None):
         super().__init__()
diff --git a/trace_route.py b/trace_route.py
index a45a59c..f8aa393 100644
--- a/trace_route.py
+++ b/trace_route.py
@@ -16,7 +16,7 @@ class TraceRoute(QWidget):
         self.startTraceBtn.clicked.connect(self.startTrace)
         
         self.result = QTextEdit()
-        self.result.setEnabled(False)
+        self.result.setReadOnly(True)
         
         self.layoutTrace = QVBoxLayout(self)
         self.layoutTrace.addWidget(self.hostn)