2222
2323from .common import CHUNK_SIZE , ClientError
2424from .merginproject import MerginProject
25- from .utils import save_to_file
25+ from .utils import cleanup_tmp_dir , save_to_file
2626
2727
2828# status = download_project_async(...)
@@ -145,7 +145,7 @@ def download_project_async(mc, project_path, directory, project_version=None):
145145 mp .log .info ("--- version: " + mc .user_agent_info ())
146146 mp .log .info (f"--- start download { project_path } " )
147147
148- tmp_dir = tempfile .TemporaryDirectory (prefix = "python-api-client-" , ignore_cleanup_errors = True , delete = True )
148+ tmp_dir = tempfile .TemporaryDirectory (prefix = "python-api-client-" )
149149
150150 try :
151151 # check whether we download the latest version or not
@@ -250,7 +250,7 @@ def download_project_finalize(job):
250250 # final update of project metadata
251251 job .mp .update_metadata (job .project_info )
252252
253- job .tmp_dir . cleanup ( )
253+ cleanup_tmp_dir ( job .mp , job . tmp_dir )
254254
255255
256256def download_project_cancel (job ):
@@ -263,6 +263,7 @@ def download_project_cancel(job):
263263 job .is_cancelled = True
264264 job .executor .shutdown (wait = True )
265265 job .mp .log .info ("--- download cancelled" )
266+ cleanup_tmp_dir (job .mp , job .tmp_dir )
266267
267268
268269class UpdateTask :
@@ -424,7 +425,7 @@ def pull_project_async(mc, directory):
424425 # then we just download the whole file
425426 _pulling_file_with_diffs = lambda f : "diffs" in f and len (f ["diffs" ]) != 0
426427
427- tmp_dir = tempfile .TemporaryDirectory (prefix = "mm-pull-" , ignore_cleanup_errors = True , delete = True )
428+ tmp_dir = tempfile .TemporaryDirectory (prefix = "mm-pull-" )
428429 pull_changes = mp .get_pull_changes (server_info ["files" ])
429430 mp .log .debug ("pull changes:\n " + pprint .pformat (pull_changes ))
430431 fetch_files = []
@@ -550,6 +551,7 @@ def pull_project_cancel(job):
550551 job .is_cancelled = True
551552 job .executor .shutdown (wait = True )
552553 job .mp .log .info ("--- pull cancelled" )
554+ cleanup_tmp_dir (job .mp , job .tmp_dir ) # delete our temporary dir and all its content
553555
554556
555557class FileToMerge :
@@ -637,6 +639,7 @@ def pull_project_finalize(job: PullJob):
637639 except Exception as e :
638640 job .mp .log .error ("Failed to apply pull changes: " + str (e ))
639641 job .mp .log .info ("--- pull aborted" )
642+ cleanup_tmp_dir (job .mp , job .tmp_dir ) # delete our temporary dir and all its content
640643 raise ClientError ("Failed to apply pull changes: " + str (e ))
641644
642645 job .mp .update_metadata (job .project_info )
@@ -646,7 +649,7 @@ def pull_project_finalize(job: PullJob):
646649 else :
647650 job .mp .log .info ("--- pull finished -- at version " + job .mp .version ())
648651
649- job .tmp_dir . cleanup ( ) # delete our temporary dir and all its content
652+ cleanup_tmp_dir ( job .mp , job . tmp_dir ) # delete our temporary dir and all its content
650653 return conflicts
651654
652655
0 commit comments