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):