@@ -108,13 +108,15 @@ def write_files(cleanup_files):
108108 path .write_text ("\n " .join (getattr (objects , obj )) + "\n " )
109109
110110
111- def prepare_pkgcheck ():
111+ @contextmanager
112+ def create_generated_files ():
112113 cleanup_files = []
113114 try :
114115 write_verinfo (cleanup_files )
115116 write_const (cleanup_files )
116117 write_objects (cleanup_files )
117118 write_files (cleanup_files )
119+ yield
118120 finally :
119121 for path in cleanup_files :
120122 try :
@@ -125,14 +127,14 @@ def prepare_pkgcheck():
125127
126128def build_wheel (wheel_directory , config_settings = None , metadata_directory = None ):
127129 """Builds a wheel, places it in wheel_directory"""
128- prepare_pkgcheck ()
129- return buildapi .build_wheel (wheel_directory , config_settings , metadata_directory )
130+ with create_generated_files ():
131+ return buildapi .build_wheel (wheel_directory , config_settings , metadata_directory )
130132
131133
132134build_editable = buildapi .build_editable
133135
134136
135137def build_sdist (sdist_directory , config_settings = True ):
136138 """Builds an sdist, places it in sdist_directory"""
137- prepare_pkgcheck ()
138- return buildapi .build_sdist (sdist_directory , config_settings )
139+ with create_generated_files ():
140+ return buildapi .build_sdist (sdist_directory , config_settings )
0 commit comments