import os import shutil def save_file(name, data): temp_path = os.path.join(os.path.dirname(__file__), 'temp') with open(os.path.join(temp_path, name), 'wb') as code: code.write(data) def remove_file(path): if len(path) > 0: try: os.remove(path) except: pass def clear_dir(): temp_path = os.path.join(os.path.dirname(__file__), 'temp') flag = True for files in os.listdir(temp_path): path = os.path.join(temp_path, files) try: shutil.rmtree(path) except OSError: os.remove(path) print('Info: temp dir is cleared')