From 70acfc22e7b32e6a7aaaa9b813d5de4b21220226 Mon Sep 17 00:00:00 2001
From: Labrys of Knossos <Labrys.git@gmail.com>
Date: Wed, 19 Dec 2018 19:27:52 -0500
Subject: [PATCH] Add cleanup upon update

---
 core/__init__.py      |  1 +
 core/version_check.py | 18 +++++++++++++++++-
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/core/__init__.py b/core/__init__.py
index 8f8e19fc..50b283dd 100644
--- a/core/__init__.py
+++ b/core/__init__.py
@@ -24,6 +24,7 @@ except ImportError:
         sys.ext('Please install pywin32')
 
 APP_ROOT = libs.util.module_path(parent=True)
+SOURCE_ROOT = libs.util.module_path()
 
 # init preliminaries
 SYS_ARGV = sys.argv[1:]
diff --git a/core/version_check.py b/core/version_check.py
index e8f99a69..37445d16 100644
--- a/core/version_check.py
+++ b/core/version_check.py
@@ -15,6 +15,7 @@ from six.moves.urllib.request import urlretrieve
 
 import core
 from core import github_api as github, logger
+import libs.util
 
 
 class CheckVersion(object):
@@ -79,7 +80,22 @@ class CheckVersion(object):
 
     def update(self):
         if self.updater.need_update():
-            return self.updater.update()
+            result = self.updater.update()
+            self.clean()
+            return result
+
+    @staticmethod
+    def clean():
+        # Clean libs
+        libs.util.git_clean(
+            remove_directories=True,
+            force=True,
+            ignore_rules=True,
+            paths=[
+                libs.LIB_ROOT,
+                core.SOURCE_ROOT,
+            ],
+        )
 
 
 class UpdateManager(object):