diff --git a/agent/worker/agent.py b/agent/worker/agent.py index 6ca1201..085c6a4 100644 --- a/agent/worker/agent.py +++ b/agent/worker/agent.py @@ -691,9 +691,10 @@ def task_clear_tasks_dir(self): self.logger.info("Start background task: Clearing Tasks data") try: task_dir = constants.AGENT_TASKS_DIR() - shutil.move(task_dir, tmp_dir) - os.makedirs(task_dir) - shutil.rmtree(tmp_dir) + if os.path.exists(task_dir): + shutil.move(task_dir, tmp_dir) + os.makedirs(task_dir) + shutil.rmtree(tmp_dir) except: self.logger.warn("Background task error: Failed to clear tasks data", exc_info=True) else: @@ -718,9 +719,10 @@ def task_clear_pip_cache(self): elif constants.SHOULD_CLEAN_PIP_CACHE(): self.logger.info("Start background task: Clearing pip cache") try: - shutil.move(constants.APPS_PIP_CACHE_DIR(), tmp_dir) - os.makedirs(constants.APPS_PIP_CACHE_DIR()) - shutil.rmtree(tmp_dir) + if os.path.exists(constants.APPS_PIP_CACHE_DIR()): + shutil.move(constants.APPS_PIP_CACHE_DIR(), tmp_dir) + os.makedirs(constants.APPS_PIP_CACHE_DIR()) + shutil.rmtree(tmp_dir) except: self.logger.warn("Background task error: Failed to clear pip cache", exc_info=True) else: @@ -745,9 +747,10 @@ def task_clear_apps_data(self): elif constants.SHOULD_CLEAN_APPS_DATA(): self.logger.info("Start background task: Clearing apps data") try: - shutil.move(constants.AGENT_APPS_CACHE_DIR(), tmp_dir) - os.makedirs(constants.AGENT_APPS_CACHE_DIR()) - shutil.rmtree(tmp_dir) + if os.path.exists(constants.AGENT_APPS_CACHE_DIR()): + shutil.move(constants.AGENT_APPS_CACHE_DIR(), tmp_dir) + os.makedirs(constants.AGENT_APPS_CACHE_DIR()) + shutil.rmtree(tmp_dir) except: self.logger.warn("Background task error: Failed to clear apps data", exc_info=True) else: @@ -770,9 +773,10 @@ def task_clear_apps_data(self): elif constants.SHOULD_CLEAN_APPS_DATA(): self.logger.info("Start background task: Clearing apps data") try: - shutil.move(constants.SUPERVISELY_SYNCED_APP_DATA_CONTAINER(), tmp_dir) - os.makedirs(constants.SUPERVISELY_SYNCED_APP_DATA_CONTAINER()) - shutil.rmtree(tmp_dir) + if os.path.exists(constants.SUPERVISELY_SYNCED_APP_DATA_CONTAINER()): + shutil.move(constants.SUPERVISELY_SYNCED_APP_DATA_CONTAINER(), tmp_dir) + os.makedirs(constants.SUPERVISELY_SYNCED_APP_DATA_CONTAINER()) + shutil.rmtree(tmp_dir) except: self.logger.warn("Background task error: Failed to clear apps data", exc_info=True) else: