Skip to content

Commit a73f980

Browse files
committed
[QOL] Get input first before launching browser
1 parent 37a8659 commit a73f980

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

main.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -303,8 +303,6 @@ def getAllLabels():
303303
allChains = '/'.join(chainMap.keys())
304304

305305
if __name__ == "__main__":
306-
driver = webdriver.Chrome(service=ChromeService(
307-
ChromeDriverManager().install()))
308306
targetChain = input('Enter scan site of interest ({}): '.format(allChains))
309307

310308
if (targetChain not in chainMap):
@@ -314,7 +312,10 @@ def getAllLabels():
314312
baseUrl = chainMap[targetChain]['baseUrl']
315313
savePath = chainMap[targetChain]['savePath']
316314

315+
driver = webdriver.Chrome(service=ChromeService(
316+
ChromeDriverManager().install()))
317317
login()
318+
318319
retrievalType = input('Enter retrieval type (single/all): ')
319320
if (retrievalType == 'all'):
320321
getAllLabels()

0 commit comments

Comments
 (0)