From 27a841b6736ce1bb1d81b2a1fc3870b2bb501f6a Mon Sep 17 00:00:00 2001
From: 10se1ucgo <hammaadu1@gmail.com>
Date: Sun, 23 Aug 2015 20:25:43 -0400
Subject: [PATCH] You can revert things seperately now.

Also, pressing "Get privacy" will now revert everything first to prevent any weird stuff happening and then apply the privacy stuff.
---
 run.py | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/run.py b/run.py
index 54b92fe..4dd2045 100644
--- a/run.py
+++ b/run.py
@@ -148,6 +148,10 @@ def about(self, event):
         wx.AboutBox(aboutpg)
 
     def goprivate(self, event):
+        print "Reverting before setting privacy stuff. Ignore everything below"
+        self.revert(event=None)  # Revert first so that the latest settings can be applied and old ones removed.
+        print "Finished reverting. Everything above can be ignored."
+        print "----------------------------------------------------------------------"
         # Disable buttons
         self.okbutton.Disable()
         self.revertbutton.Disable()
@@ -182,10 +186,14 @@ def revert(self, event):
         self.okbutton.Disable()
         self.revertbutton.Disable()
         try:
-            modifyserviceregs(0x0000003)
-            modifytelemetryregs("1")
-            modifyhosts(extra=False, undo=True)
-            blockips(undo=True)
+            if self.servicebox.IsChecked():
+                modifyserviceregs(0x0000003)
+            if self.telemetrybox.IsChecked():
+                modifytelemetryregs("1")
+            if self.hostbox.IsChecked() or self.extrahostbox.IsChecked():
+                modifyhosts(extra=False, undo=True)
+            if self.ipbox.IsChecked():
+                blockips(undo=True)
         finally:
             self.okbutton.Enable()
             self.revertbutton.Enable()