@@ -136,50 +136,6 @@ def _create_vuforia_resource_names() -> tuple[str, str, str, str]:
136136 )
137137
138138
139- def _create_shared_vumark_resources (
140- email_address : str ,
141- password : str ,
142- ) -> tuple ["VuMarkDatabaseDict" , str ]:
143- """Create shared VuMark resources used by all generated secrets
144- files.
145- """
146- shared_vumark_details : VuMarkDatabaseDict | None = None
147- shared_vumark_target_id : str | None = None
148-
149- while shared_vumark_details is None or shared_vumark_target_id is None :
150- vumark_driver = vws_web_tools .create_chrome_driver ()
151- (
152- _license_name ,
153- _database_name ,
154- vumark_database_name ,
155- vumark_template_name ,
156- ) = _create_vuforia_resource_names ()
157- try :
158- vws_web_tools .log_in (
159- driver = vumark_driver ,
160- email_address = email_address ,
161- password = password ,
162- )
163- vws_web_tools .wait_for_logged_in (driver = vumark_driver )
164- shared_vumark_details = _create_and_get_vumark_details (
165- driver = vumark_driver ,
166- vumark_database_name = vumark_database_name ,
167- )
168- shared_vumark_target_id = _create_and_get_vumark_target_id (
169- driver = vumark_driver ,
170- vumark_database_name = vumark_database_name ,
171- vumark_template_name = vumark_template_name ,
172- )
173- except TimeoutException :
174- sys .stderr .write (
175- "Timed out waiting for shared VuMark setup/details after "
176- "retries\n "
177- )
178- vumark_driver .quit ()
179-
180- return shared_vumark_details , shared_vumark_target_id
181-
182-
183139def main () -> None :
184140 """Create secrets files."""
185141 email_address = os .environ ["VWS_EMAIL_ADDRESS" ]
@@ -209,18 +165,8 @@ def main() -> None:
209165 for i in range (num_databases )
210166 ]
211167 files_to_create = [file for file in required_files if not file .exists ()]
212- if not files_to_create :
213- sys .stdout .write ("No secrets files need to be created.\n " )
214- return
215-
216- shared_vumark_details , shared_vumark_target_id = (
217- _create_shared_vumark_resources (
218- email_address = email_address ,
219- password = password ,
220- )
221- )
222-
223168 driver : WebDriver | None = None
169+
224170 while files_to_create :
225171 if driver is None :
226172 driver = vws_web_tools .create_chrome_driver ()
@@ -229,8 +175,8 @@ def main() -> None:
229175 (
230176 license_name ,
231177 database_name ,
232- _vumark_database_name ,
233- _vumark_template_name ,
178+ vumark_database_name ,
179+ vumark_template_name ,
234180 ) = _create_vuforia_resource_names ()
235181
236182 try :
@@ -249,14 +195,41 @@ def main() -> None:
249195 driver = None
250196 continue
251197
198+ try :
199+ vumark_details = _create_and_get_vumark_details (
200+ driver = driver ,
201+ vumark_database_name = vumark_database_name ,
202+ )
203+ except TimeoutException :
204+ sys .stderr .write (
205+ "Timed out waiting for VuMark setup/details after retries\n "
206+ )
207+ driver .quit ()
208+ driver = None
209+ continue
210+
211+ try :
212+ vumark_target_id = _create_and_get_vumark_target_id (
213+ driver = driver ,
214+ vumark_database_name = vumark_database_name ,
215+ vumark_template_name = vumark_template_name ,
216+ )
217+ except TimeoutException :
218+ sys .stderr .write (
219+ "Timed out waiting for VuMark template upload after retries\n "
220+ )
221+ driver .quit ()
222+ driver = None
223+ continue
224+
252225 driver .quit ()
253226 driver = None
254227
255228 file_contents = _generate_secrets_file_content (
256229 database_details = database_details ,
257- vumark_details = shared_vumark_details ,
230+ vumark_details = vumark_details ,
258231 inactive_database_details = inactive_database_details ,
259- vumark_target_id = shared_vumark_target_id ,
232+ vumark_target_id = vumark_target_id ,
260233 )
261234 file .write_text (data = file_contents )
262235 sys .stdout .write (f"Created database { file .name } \n " )
0 commit comments