Rewrite! Closes #153, #147, #146, #124

Changes:
Everything rewritten! Many things borrowed from my other program, pyjam.
Split Windows Defender and WifiSense into two options
GUI should now scale properly (hopefully)
Everything is less terrible!
Code is somewhat cleaner!
Fixed DiagTrack service

Split everything into 3 files:
- dwt.py: Entry file, contains GUI and logging stuff
- dwt_util.py: Contains functions for disabling the tracking
- dwt_about.py: Contains the about and licenses dialog.

Have fun!

1 files changed, 426 insertions, 0 deletions

diff --git a/dwt.py b/dwt.py
new file mode 100644
index 0000000..1963672
--- /dev/null
+++ b/dwt.py
@@ -0,0 +1,426 @@
+# Copyright (C) 10se1ucgo 2015-2016
+#
+# This file is part of DisableWinTracking.
+#
+# DisableWinTracking is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# DisableWinTracking is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with DisableWinTracking.  If not, see <http://www.gnu.org/licenses/>.
+import os
+import logging
+import sys
+import io
+import platform
+import traceback
+from ctypes import windll
+
+import wx
+from wx.lib.itemspicker import ItemsPicker, IP_SORT_SELECTED, IP_SORT_CHOICES, IP_REMOVE_FROM_CHOICES
+
+import dwt_about
+import dwt_util
+
+
+class RedirectText(io.StringIO):
+    def __init__(self, console, old_stdout):
+        super(RedirectText, self).__init__()
+
+        self.out = console
+        self.old_out = old_stdout
+
+    def write(self, string):
+        # Oh my god this is the DUMBEST THING I've ever done. (Keeping a reference to the old stdout)
+        self.old_out.write(string)
+        self.out.WriteText(string)
+
+
+class ConsoleDialog(wx.Dialog):
+    def __init__(self, old_stdout):
+        wx.Dialog.__init__(self, parent=wx.GetApp().TopWindow, title="Console Output", size=(500, 200),
+                           style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER)
+
+        console_box = wx.TextCtrl(self, style=wx.TE_MULTILINE | wx.TE_READONLY)
+        sys.stdout = RedirectText(console_box, old_stdout)
+
+        top_sizer = wx.BoxSizer(wx.VERTICAL)
+        console_sizer = wx.BoxSizer(wx.VERTICAL)
+        button_sizer = self.CreateButtonSizer(wx.OK | wx.CANCEL)
+
+        report_button = wx.FindWindowById(wx.ID_CANCEL, self)
+        report_button.SetLabel("Report an issue")
+
+        console_sizer.Add(console_box, 1, wx.LEFT | wx.RIGHT | wx.EXPAND | wx.ALIGN_TOP, 5)
+
+        top_sizer.Add(console_sizer, 1, wx.ALL | wx.EXPAND, 5)
+        top_sizer.Add(button_sizer, 0, wx.ALL | wx.ALIGN_LEFT, 5)
+
+        self.Bind(wx.EVT_CLOSE, handler=lambda x: sys.exit(0))
+        self.Bind(wx.EVT_BUTTON, handler=lambda x: sys.exit(0), id=wx.ID_OK)
+        self.Bind(wx.EVT_BUTTON, source=report_button, handler=self.submit_issue)
+        self.SetSizer(top_sizer)
+
+    def submit_issue(self, event):
+        os.startfile("https://github.com/10se1ucgo/DisableWinTracking/issues/new")
+
+
+class MainFrame(wx.Frame):
+    def __init__(self):
+        super(MainFrame, self).__init__(parent=wx.GetApp().GetTopWindow(), title="Disable Windows 10 Tracking",
+                                        size=(415, 450))
+        self.SetMinSize(self.GetSize())
+        panel = MainPanel(self)
+
+        file_menu = wx.Menu()
+        settings = file_menu.Append(wx.ID_SETUP, "&Settings", "DWT settings")
+
+        help_menu = wx.Menu()
+        about = help_menu.Append(wx.ID_ABOUT, "&About", "About DWT")
+        licenses = help_menu.Append(wx.ID_ANY, "&Licenses", "Open-source licenses")
+
+        menu_bar = wx.MenuBar()
+        menu_bar.Append(file_menu, "&File")
+        menu_bar.Append(help_menu, "&Help")
+        self.SetMenuBar(menu_bar)
+
+        if bool(windll.advpack.IsNTAdmin(0, None)):
+            warn = wx.MessageDialog(parent=self,
+                                    message="Program requires elevation, please run it as an administrator.",
+                                    caption="ERROR!", style=wx.OK | wx.ICON_WARNING)
+            warn.ShowModal()
+            warn.Destroy()
+            sys.exit(1)
+
+        self.SetIcon(wx.Icon(sys.executable, wx.BITMAP_TYPE_ICO))
+        self.Bind(wx.EVT_MENU, lambda x: dwt_about.about_dialog(self), about)
+        self.Bind(wx.EVT_MENU, panel.settings, settings)
+        self.Bind(wx.EVT_MENU, lambda x: dwt_about.Licenses(self), licenses)
+
+
+class MainPanel(wx.Panel):
+    def __init__(self, parent):
+        super(MainPanel, self).__init__(parent)
+
+        self.parent = parent
+        self.picked_normal = []
+        self.picked_extra = []
+        self.picked_ips = []
+
+        self.service_check = wx.CheckBox(self, label="Services")
+        self.service_check.SetToolTip("Disables or deletes tracking services. Choose option in 'Services Method'")
+
+        self.diagtrack_check = wx.CheckBox(self, label="Clear DiagTrack log")
+        self.diagtrack_check.SetToolTip("Clears Dianostic Tracking log and prevents modification to it. "
+                                        "Cannot be undone automatically.")
+
+        # Telemetry checkbox
+        self.telemetry_check = wx.CheckBox(self, label="Telemetry")
+        self.telemetry_check.SetToolTip("Sets 'AllowTelemetry' to 0. "
+                                        "On non-Enterprise OS editions, requires HOSTS file modification.")
+
+        # HOSTS file checkbox
+        self.host_check = wx.CheckBox(self, label="Block tracking domains")
+        self.host_check.SetToolTip("Adds known tracking domains to HOSTS file. Required to disable Telemetry")
+
+        # Extra HOSTS checkbox
+        self.extra_host_check = wx.CheckBox(self, label="Block even more tracking domains")
+        self.extra_host_check.SetToolTip("For the paranoid. Adds extra domains to the HOSTS file.\n"
+                                         "May cause issues with Skype, Dr. Watson, Hotmail and/or Error Reporting.")
+
+        # IP block checkbox
+        self.ip_check = wx.CheckBox(self, label="Block tracking IP addresses")
+        self.ip_check.SetToolTip("Blocks known tracking IP addresses with Windows Firewall.")
+
+        # Windows Privacy Regs (Policy Manager)
+        self.defender_check = wx.CheckBox(self, label="Windows Defender collection")
+        self.defender_check.SetToolTip("Modifies registry to prevent Defender collection")
+
+        # WifiSense checkbox
+        self.wifisense_check = wx.CheckBox(self, label="WifiSense")
+
+        # OneDrive uninstall checkbox
+        self.onedrive_check = wx.CheckBox(self, label="Uninstall OneDrive")
+        self.onedrive_check.SetToolTip("Uninstalls OneDrive from your computer and removes it from Explorer.")
+
+        self.service_rad = wx.RadioBox(self, label="Service Method", choices=("Disable", "Delete"))
+        self.service_rad.SetItemToolTip(item=0, text="Simply disables the services. This can be undone.")
+        self.service_rad.SetItemToolTip(item=1, text="Deletes the services completely. This can't be undone.")
+
+        self.mode_rad = wx.RadioBox(self, label="Mode", choices=("Privacy", "Revert"))
+        self.mode_rad.SetItemToolTip(item=0, text="Applies the selected settings.")
+        self.mode_rad.SetItemToolTip(item=1, text="Reverts the selected settings.")
+
+        go_button = wx.Button(self, label="Go!")
+
+        self.app_box = wx.StaticBoxSizer(wx.VERTICAL, self, "Built-in Apps")
+        stat_box = self.app_box.GetStaticBox
+        top_app_sizer = wx.BoxSizer(wx.HORIZONTAL)
+        button_app_sizer = wx.BoxSizer(wx.HORIZONTAL)
+        left_app_sizer = wx.BoxSizer(wx.VERTICAL)
+        middle_app_sizer = wx.BoxSizer(wx.VERTICAL)
+        right_app_sizer = wx.BoxSizer(wx.VERTICAL)
+
+        # wx.CheckBox(app_box.GetStaticBox(), label="Name", name="search_name")
+        wx.CheckBox(stat_box(), label="3D Builder", name="3dbuilder")
+        wx.CheckBox(stat_box(), label="Calender and Mail", name="windowscommunicationsapps")
+        wx.CheckBox(stat_box(), label="Camera", name="windowscamera")
+        wx.CheckBox(stat_box(), label="Get Office App", name="officehub")
+        wx.CheckBox(stat_box(), label="Get Skype App", name="skypeapp")
+        wx.CheckBox(stat_box(), label="Get Started App", name="getstarted")
+        wx.CheckBox(stat_box(), label="Groove Music", name="zunemusic")
+        wx.CheckBox(stat_box(), label="Maps", name="windowsmaps")
+        wx.CheckBox(stat_box(), label="Solitaire Collection", name="solitairecollection")
+        wx.CheckBox(stat_box(), label="Money", name="bingfinance")
+        wx.CheckBox(stat_box(), label="Movies && TV", name="zunevideo")
+        wx.CheckBox(stat_box(), label="News", name="bingnews")
+        wx.CheckBox(stat_box(), label="OneNote App", name="onenote")
+        wx.CheckBox(stat_box(), label="People", name="people")
+        wx.CheckBox(stat_box(), label="Phone Companion", name="windowsphone")
+        wx.CheckBox(stat_box(), label="Photos", name="photos")
+        wx.CheckBox(stat_box(), label="Sports", name="bingsports")
+        wx.CheckBox(stat_box(), label="Voice Recorder", name="soundrecorder")
+        wx.CheckBox(stat_box(), label="Weather", name="bingweather")
+        wx.CheckBox(stat_box(), label="Xbox", name="xboxapp")
+        remove_app_button = wx.Button(stat_box(), label="Remove selected apps")
+        select_all_check = wx.CheckBox(stat_box(), label="Select all")
+
+        sorted_list = sorted(stat_box().GetChildren(), key=lambda x: x.GetLabel())
+        for index, item in enumerate([x for x in sorted_list if isinstance(x, wx.CheckBox) and x != select_all_check]):
+            n = len(sorted_list) // 3
+            if index <= n:
+                left_app_sizer.Add(item, 1, wx.ALL, 1)
+            elif index <= n * 2:
+                middle_app_sizer.Add(item, 1, wx.ALL, 1)
+            else:
+                right_app_sizer.Add(item, 1, wx.ALL, 1)
+
+        top_app_sizer.Add(left_app_sizer, 1, wx.ALL, 1)
+        top_app_sizer.Add(middle_app_sizer, 1, wx.ALL, 1)
+        top_app_sizer.Add(right_app_sizer, 1, wx.ALL, 1)
+        button_app_sizer.Add(remove_app_button, 0, wx.ALL, 1)
+        button_app_sizer.Add(select_all_check, 1, wx.ALL, 5)
+        self.app_box.Add(top_app_sizer)
+        self.app_box.Add(button_app_sizer)
+
+        top_sizer = wx.BoxSizer(wx.VERTICAL)
+        top_row_sizer = wx.BoxSizer(wx.HORIZONTAL)
+        check_sizer = wx.BoxSizer(wx.VERTICAL)
+        rad_sizer = wx.BoxSizer(wx.VERTICAL)
+
+        top_sizer.Add(top_row_sizer, 0, wx.ALL, 5)
+        top_sizer.Add(self.app_box, 0, wx.ALL, 5)
+        top_row_sizer.Add(check_sizer, 0, wx.ALL)
+        top_row_sizer.Add(rad_sizer, 0, wx.ALL)
+        rad_sizer.Add(self.service_rad, 0, wx.ALL, 10)
+        rad_sizer.Add(go_button, 0, wx.ALL ^ wx.BOTTOM | wx.ALIGN_CENTER, 10)
+        rad_sizer.Add(self.mode_rad, 0, wx.ALL, 10)
+        check_sizer.Add(self.service_check, 0, wx.ALL, 1)
+        check_sizer.Add(self.diagtrack_check, 0, wx.ALL, 1)
+        check_sizer.Add(self.telemetry_check, 0, wx.ALL, 1)
+        check_sizer.Add(self.host_check, 0, wx.ALL, 1)
+        check_sizer.Add(self.extra_host_check, 0, wx.ALL, 1)
+        check_sizer.Add(self.ip_check, 0, wx.ALL, 1)
+        check_sizer.Add(self.defender_check, 0, wx.ALL, 1)
+        check_sizer.Add(self.wifisense_check, 0, wx.ALL, 1)
+        check_sizer.Add(self.onedrive_check, 0, wx.ALL, 1)
+
+        self.Bind(wx.EVT_CHECKBOX, handler=self.select_all_apps, source=select_all_check)
+        self.Bind(wx.EVT_BUTTON, handler=self.remove_apps, source=remove_app_button)
+        self.Bind(wx.EVT_BUTTON, handler=self.go, source=go_button)
+
+        self.SetSizer(top_sizer)
+        print(self.GetSize())
+
+    def select_all_apps(self, event):
+        # Iters through all children of the wxStaticBox of the wxStaticBoxSizer and checks/un checks all wxCheckBoxes.
+        for child in self.app_box.GetStaticBox().GetChildren():
+            if isinstance(child, wx.CheckBox):
+                child.SetValue(event.IsChecked())
+
+    def go(self, event):
+        if not all((self.picked_ips, self.picked_extra, self.picked_normal)):
+            self.settings(event=None)
+
+        undo = bool(self.mode_rad.GetSelection())
+
+        if self.ip_check.IsChecked():
+            dwt_util.ip_block(self.picked_ips, undo=undo)
+        if self.diagtrack_check.IsChecked():
+            dwt_util.clear_diagtrack()
+        if self.service_check.IsChecked():
+            if self.service_rad.GetSize():
+                dwt_util.delete_service("dmwappushsvc")
+                dwt_util.delete_service("DiagTrack")
+            else:
+                dwt_util.disable_service("dmwappushsvc")
+                dwt_util.disable_service("DiagTrack")
+            dwt_util.services(undo=undo)
+        if self.telemetry_check.IsChecked():
+            dwt_util.telemetry(undo=undo)
+        if self.host_check.IsChecked():
+            dwt_util.host_file(self.picked_normal, undo=undo)
+        if self.extra_host_check.IsChecked():
+            dwt_util.host_file(self.picked_extra, undo=undo)
+        if self.defender_check.IsChecked():
+            dwt_util.defender(undo=undo)
+        if self.wifisense_check.IsChecked():
+            dwt_util.wifisense(undo=undo)
+        if self.onedrive_check.IsChecked():
+            dwt_util.onedrive(undo=undo)
+        logger.info("Done. It's recommended that you reboot as soon as possible for the full effect.")
+        logger.info(("If you feel something didn't work properly, please press the 'Report an issue"
+                      "button and follow the directions"))
+        console.Center()
+        console.Show()
+
+    def remove_apps(self, event):
+        children = [child for child in self.app_box.GetStaticBox().GetChildren() if child.GetName() != "check"]
+        app_list = [child.GetName() for child in children if isinstance(child, wx.CheckBox) and child.IsChecked()]
+        dwt_util.app_manager(app_list, undo=False)
+
+    def settings(self, event):
+        dialog = wx.Dialog(parent=self, title="Settings", style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER)
+        sizer = wx.BoxSizer(wx.VERTICAL)
+
+        normal_domains = (
+            'a-0001.a-msedge.net', 'a-0002.a-msedge.net', 'a-0003.a-msedge.net', 'a-0004.a-msedge.net',
+            'a-0005.a-msedge.net', 'a-0006.a-msedge.net', 'a-0007.a-msedge.net', 'a-0008.a-msedge.net',
+            'a-0009.a-msedge.net', 'a-msedge.net', 'a.ads1.msn.com', 'a.ads2.msads.net', 'a.ads2.msn.com',
+            'a.rad.msn.com', 'ac3.msn.com', 'ad.doubleclick.net', 'adnexus.net', 'adnxs.com', 'ads.msn.com',
+            'ads1.msads.net', 'ads1.msn.com', 'aidps.atdmt.com', 'aka-cdn-ns.adtech.de',
+            'az361816.vo.msecnd.net', 'az512334.vo.msecnd.net', 'b.ads1.msn.com', 'b.ads2.msads.net',
+            'b.rad.msn.com', 'bs.serving-sys.com', 'c.atdmt.com', 'c.msn.com', 'cdn.atdmt.com',
+            'cds26.ams9.msecn.net', 'choice.microsoft.com', 'choice.microsoft.com.nsatc.net',
+            'compatexchange.cloudapp.net', 'corp.sts.microsoft.com', 'corpext.msitadfs.glbdns2.microsoft.com',
+            'cs1.wpc.v0cdn.net', 'db3aqu.atdmt.com', 'df.telemetry.microsoft.com',
+            'diagnostics.support.microsoft.com', 'ec.atdmt.com', 'feedback.microsoft-hohm.com',
+            'feedback.search.microsoft.com', 'feedback.windows.com', 'flex.msn.com', 'g.msn.com', 'h1.msn.com',
+            'i1.services.social.microsoft.com', 'i1.services.social.microsoft.com.nsatc.net',
+            'lb1.www.ms.akadns.net', 'live.rads.msn.com', 'm.adnxs.com', 'msedge.net', 'msftncsi.com',
+            'msnbot-65-55-108-23.search.msn.com', 'msntest.serving-sys.com', 'oca.telemetry.microsoft.com',
+            'oca.telemetry.microsoft.com.nsatc.net', 'pre.footprintpredict.com', 'preview.msn.com',
+            'rad.live.com', 'rad.msn.com', 'redir.metaservices.microsoft.com',
+            'reports.wes.df.telemetry.microsoft.com', 'schemas.microsoft.akadns.net',
+            'secure.adnxs.com', 'secure.flashtalking.com', 'services.wes.df.telemetry.microsoft.com',
+            'settings-sandbox.data.microsoft.com', 'settings-win.data.microsoft.com',
+            'sls.update.microsoft.com.akadns.net', 'sqm.df.telemetry.microsoft.com',
+            'sqm.telemetry.microsoft.com', 'sqm.telemetry.microsoft.com.nsatc.net', 'ssw.live.com',
+            'static.2mdn.net', 'statsfe1.ws.microsoft.com', 'statsfe2.ws.microsoft.com',
+            'telecommand.telemetry.microsoft.com', 'telecommand.telemetry.microsoft.com.nsatc.net',
+            'telemetry.appex.bing.net', 'telemetry.microsoft.com', 'telemetry.urs.microsoft.com',
+            'v10.vortex-win.data.microsoft.com', 'vortex-bn2.metron.live.com.nsatc.net',
+            'vortex-cy2.metron.live.com.nsatc.net', 'vortex-sandbox.data.microsoft.com',
+            'vortex-win.data.metron.live.com.nsatc.net', 'vortex-win.data.microsoft.com',
+            'vortex.data.glbdns2.microsoft.com', 'vortex.data.microsoft.com', 'watson.live.com',
+            'web.vortex.data.microsoft.com', 'www.msftncsi.com'
+        )
+
+        extra_domains = (
+            'fe2.update.microsoft.com.akadns.net', 's0.2mdn.net', 'statsfe2.update.microsoft.com.akadns.net',
+            'survey.watson.microsoft.com', 'view.atdmt.com', 'watson.microsoft.com',
+            'watson.ppe.telemetry.microsoft.com', 'watson.telemetry.microsoft.com',
+            'watson.telemetry.microsoft.com.nsatc.net', 'wes.df.telemetry.microsoft.com', 'ui.skype.com',
+            'pricelist.skype.com', 'apps.skype.com', 'm.hotmail.com', 's.gateway.messenger.live.com'
+        )
+
+        ip_addresses = (
+            '2.22.61.43', '2.22.61.66', '65.39.117.230', '65.55.108.23', '23.218.212.69', '134.170.30.202',
+            '137.116.81.24', '157.56.106.189', '204.79.197.200', '65.52.108.33', '64.4.54.254'
+        )
+
+        normal_domain_picker = ItemsPicker(dialog, choices=[], selectedLabel="Domains to be blocked",
+                                           ipStyle=IP_SORT_SELECTED | IP_SORT_CHOICES | IP_REMOVE_FROM_CHOICES)
+        if self.picked_normal:
+            normal_domain_picker.SetSelections(self.picked_normal)
+            normal_domain_picker.SetItems([domain for domain in normal_domains if domain not in self.picked_normal])
+        else:
+            normal_domain_picker.SetSelections(normal_domains)
+
+        extra_domain_picker = ItemsPicker(dialog, choices=[], selectedLabel="Extra domains to be blocked",
+                                          ipStyle=IP_SORT_SELECTED | IP_SORT_CHOICES | IP_REMOVE_FROM_CHOICES)
+        if self.picked_normal:
+            extra_domain_picker.SetSelections(self.picked_extra)
+            extra_domain_picker.SetItems([domain for domain in extra_domains if domain not in self.picked_extra])
+        else:
+            extra_domain_picker.SetSelections(extra_domains)
+
+        ip_picker = ItemsPicker(dialog, choices=[], selectedLabel="IP addresses to be blocked",
+                                ipStyle=IP_SORT_SELECTED | IP_SORT_CHOICES | IP_REMOVE_FROM_CHOICES)
+        if self.picked_ips:
+            ip_picker.SetSelections(self.picked_ips)
+            ip_picker.SetItems([ip for ip in ip_addresses if ip not in self.picked_ips])
+        else:
+            ip_picker.SetSelections(ip_addresses)
+
+        sizer.Add(normal_domain_picker, 0, wx.EXPAND)
+        sizer.Add(extra_domain_picker, 0, wx.EXPAND)
+        sizer.Add(ip_picker, 0, wx.EXPAND)
+        if event is not None:
+            dialog.SetSizerAndFit(sizer)
+            dialog.Center()
+            dialog.ShowModal()
+        dialog.Destroy()
+        self.picked_normal = normal_domain_picker.GetSelections()
+        self.picked_extra = extra_domain_picker.GetSelections()
+        self.picked_ips = ip_picker.GetSelections()
+
+
+def setup_logging():
+    global logger
+    logger = logging.getLogger('dwt')
+    logger.setLevel(logging.DEBUG)
+
+    formatter = logging.Formatter(fmt='%(asctime)s %(levelname)s: %(message)s', datefmt='%H:%M:%S')
+
+    stdout_log = logging.StreamHandler(sys.stdout)
+    stdout_log.setLevel(logging.DEBUG)
+    stdout_log.setFormatter(formatter)
+    logger.addHandler(stdout_log)
+
+    try:
+        file_log = logging.FileHandler(filename='dwt.log')
+        file_log.setLevel(logging.DEBUG)
+        file_log.setFormatter(formatter)
+        logger.addHandler(file_log)
+    except (OSError, IOError):
+        error_dialog = wx.MessageDialog(parent=wx.GetApp().GetTopWindow(),
+                                        message="Could not create log file, errors will not be recorded!",
+                                        caption="ERROR!", style=wx.OK | wx.ICON_ERROR)
+        error_dialog.ShowModal()
+        error_dialog.Destroy()
+        logger.exception("Could not create log file.")
+
+    logger.info("Python {version} on {platform}".format(version=sys.version, platform=sys.platform))
+    logger.info(platform.uname())
+    logger.info("DisableWinTracking version {v}".format(v=dwt_about.__version__))
+
+
+def exception_hook(error, value, trace):
+    error_message = ''.join(traceback.format_exception(error, value, trace))
+    logger.critical(error_message)
+    error_dialog = wx.MessageDialog(parent=wx.GetApp().GetTopWindow(),
+                                    message="An error has occured!\n\n" + error_message,
+                                    caption="ERROR!", style=wx.OK | wx.CANCEL | wx.ICON_ERROR)
+    error_dialog.SetOKCancelLabels("Ignore", "Quit")
+    if error_dialog.ShowModal() == wx.ID_OK:
+        error_dialog.Destroy()
+    else:
+        error_dialog.Destroy()
+        sys.exit(1)
+
+
+if __name__ == '__main__':
+    wx_app = wx.App()
+    frame = MainFrame()
+    console = ConsoleDialog(sys.stdout)
+    setup_logging()
+    sys.excepthook = exception_hook
+    frame.Show()
+    wx_app.MainLoop()