diff --git a/couchpotato/core/plugins/updater/main.py b/couchpotato/core/plugins/updater/main.py index edd5e3f0..16185e1b 100644 --- a/couchpotato/core/plugins/updater/main.py +++ b/couchpotato/core/plugins/updater/main.py @@ -5,7 +5,9 @@ from couchpotato.core.logger import CPLog from couchpotato.core.plugins.base import Plugin from couchpotato.environment import Env from git.repository import LocalRepository +import os import time +import traceback log = CPLog(__name__) @@ -83,6 +85,10 @@ class Updater(Plugin): try: log.info('Updating to latest version'); self.repo.pull() + + # Delete leftover .pyc files + self.deletePyc() + return True except Exception, e: log.error('Failed updating via GIT: %s' % e) @@ -91,5 +97,21 @@ class Updater(Plugin): return False + def deletePyc(self): + + for root, dirs, files in os.walk(Env.get('app_dir')): + + pyc_files = filter(lambda filename: filename.endswith(".pyc"), files) + py_files = set(filter(lambda filename: filename.endswith(".py"), files)) + excess_pyc_files = filter(lambda pyc_filename: pyc_filename[:-1] not in py_files, pyc_files) + + for excess_pyc_file in excess_pyc_files: + full_path = os.path.join(root, excess_pyc_file) + log.debug("Removing old PYC file:", full_path) + try: + os.remove(full_path) + except: + log.error('Couldn\'t remove %s: %s' % (full_path, traceback.format_exc())) + def isEnabled(self): return Plugin.isEnabled(self) and Env.get('uses_git')