@@ -284,8 +284,9 @@ def compute_vpd(
284284 logger .info (msg )
285285 filename = get_diagnostic_filename (basename , cfg , extension = "nc" )
286286 iris .save (vpd , filename )
287- with ProvenanceLogger (cfg ) as provenance_logger :
288- provenance_logger .log (filename , provenance )
287+ if not cfg ["remove_vpd_files" ]:
288+ with ProvenanceLogger (cfg ) as provenance_logger :
289+ provenance_logger .log (filename , provenance )
289290 return filename
290291
291292
@@ -433,14 +434,14 @@ def main(cfg: dict) -> None:
433434 if cfg ["remove_vpd_files" ]:
434435 logger .info ("Removing VPD files in %s" , cfg ["work_dir" ])
435436 f_not_removed = []
436- for f in list ( Path (f"{ cfg ['work_dir' ]} /" ).glob ("*vpd*.*" ) ):
437- logger .info ("Removing %s" , f .split ( "/" )[ - 1 ] )
437+ for f in Path (f"{ cfg ['work_dir' ]} /" ).glob ("*vpd*.*" ):
438+ logger .info ("Removing %s" , f .name )
438439 try :
439440 Path (f ).unlink ()
440- logger .info ("Removed %s" , f .split ( "/" )[ - 1 ] )
441+ logger .info ("Removed %s" , f .name )
441442 except OSError as e :
442- logger .debug ("Error removing %s: %s" , f .split ( "/" )[ - 1 ] , e )
443- f_not_removed .append (f .split ( "/" )[ - 1 ] )
443+ logger .debug ("Error removing %s: %s" , f .name , e )
444+ f_not_removed .append (f .name )
444445 logger .info ("Files not removed: %s" , f_not_removed )
445446
446447 # Remove or not ConFire files after diagnostic run
@@ -450,16 +451,14 @@ def main(cfg: dict) -> None:
450451 cfg ["work_dir" ],
451452 )
452453 f_not_removed = []
453- for f in list (
454- Path (f"{ cfg ['work_dir' ]} /ConFire_outputs/" ).glob ("*.nc" ),
455- ):
456- logger .info ("Removing %s" , f .split ("/" )[- 1 ])
454+ for f in Path (f"{ cfg ['work_dir' ]} /ConFire_outputs/" ).glob ("*.nc" ):
455+ logger .info ("Removing %s" , f .name )
457456 try :
458457 Path (f ).unlink ()
459- logger .info ("Removed %s" , f .split ( "/" )[ - 1 ] )
458+ logger .info ("Removed %s" , f .name )
460459 except OSError as e :
461- logger .debug ("Error removing %s: %s" , f .split ( "/" )[ - 1 ] , e )
462- f_not_removed .append (f .split ( "/" )[ - 1 ] )
460+ logger .debug ("Error removing %s: %s" , f .name , e )
461+ f_not_removed .append (f .name )
463462 logger .info ("Files not removed: %s" , f_not_removed )
464463 if len (f_not_removed ) == 0 :
465464 Path .rmdir (f"{ cfg ['work_dir' ]} /ConFire_outputs" )
0 commit comments