Skip to content

Commit df0315f

Browse files
mdmintzmdmintz
committed
Make sure to initialize "colorama" before using it
1 parent 808ca14 commit df0315f

File tree

4 files changed

+22
-0
lines changed

4 files changed

+22
-0
lines changed

‎examples/test_chromedriver.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ def test_chromedriver_matches_chrome(self):
2323
chromedriver_version = chrome_dict["chromedriverVersion"]
2424
chromedriver_version = chromedriver_version.split(' ')[0]
2525
major_chromedriver_version = chromedriver_version.split('.')[0]
26+
colorama.init(autoreset=True)
2627
c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
2728
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
2829
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX

‎seleniumbase/console_scripts/run.py‎

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ def show_usage():
3434
sc += ('Type "sbase help [COMMAND]" for specific command info.\n')
3535
sc += ('For info on all commands, type: "seleniumbase --help".\n')
3636
sc += ('* (Use "pytest" for running tests) *\n')
37+
colorama.init(autoreset=True)
3738
c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
3839
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
3940
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
@@ -76,6 +77,7 @@ def show_basic_usage():
7677
sc += (" grid-node [start|stop] --hub=[HUB_IP] [OPTIONS]\n")
7778
sc += (' * (EXAMPLE: "sbase install chromedriver latest") *\n')
7879
sc += ("")
80+
colorama.init(autoreset=True)
7981
c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
8082
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
8183
cr = colorama.Style.RESET_ALL
@@ -85,6 +87,7 @@ def show_basic_usage():
8587

8688

8789
def show_install_usage():
90+
colorama.init(autoreset=True)
8891
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
8992
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
9093
cr = colorama.Style.RESET_ALL
@@ -119,6 +122,7 @@ def show_install_usage():
119122

120123

121124
def show_mkdir_usage():
125+
colorama.init(autoreset=True)
122126
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
123127
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
124128
cr = colorama.Style.RESET_ALL
@@ -140,6 +144,7 @@ def show_mkdir_usage():
140144

141145

142146
def show_mkfile_usage():
147+
colorama.init(autoreset=True)
143148
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
144149
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
145150
cr = colorama.Style.RESET_ALL
@@ -172,6 +177,7 @@ def show_mkfile_usage():
172177

173178

174179
def show_convert_usage():
180+
colorama.init(autoreset=True)
175181
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
176182
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
177183
cr = colorama.Style.RESET_ALL
@@ -191,6 +197,7 @@ def show_convert_usage():
191197

192198

193199
def show_print_usage():
200+
colorama.init(autoreset=True)
194201
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
195202
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
196203
cr = colorama.Style.RESET_ALL
@@ -209,6 +216,7 @@ def show_print_usage():
209216

210217

211218
def show_translate_usage():
219+
colorama.init(autoreset=True)
212220
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
213221
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
214222
cr = colorama.Style.RESET_ALL
@@ -244,6 +252,7 @@ def show_translate_usage():
244252

245253

246254
def show_extract_objects_usage():
255+
colorama.init(autoreset=True)
247256
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
248257
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
249258
cr = colorama.Style.RESET_ALL
@@ -261,6 +270,7 @@ def show_extract_objects_usage():
261270

262271

263272
def show_inject_objects_usage():
273+
colorama.init(autoreset=True)
264274
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
265275
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
266276
cr = colorama.Style.RESET_ALL
@@ -281,6 +291,7 @@ def show_inject_objects_usage():
281291

282292

283293
def show_objectify_usage():
294+
colorama.init(autoreset=True)
284295
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
285296
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
286297
cr = colorama.Style.RESET_ALL
@@ -304,6 +315,7 @@ def show_objectify_usage():
304315

305316

306317
def show_revert_objects_usage():
318+
colorama.init(autoreset=True)
307319
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
308320
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
309321
cr = colorama.Style.RESET_ALL
@@ -325,6 +337,7 @@ def show_revert_objects_usage():
325337

326338

327339
def show_encrypt_usage():
340+
colorama.init(autoreset=True)
328341
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
329342
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
330343
cr = colorama.Style.RESET_ALL
@@ -342,6 +355,7 @@ def show_encrypt_usage():
342355

343356

344357
def show_decrypt_usage():
358+
colorama.init(autoreset=True)
345359
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
346360
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
347361
cr = colorama.Style.RESET_ALL
@@ -359,6 +373,7 @@ def show_decrypt_usage():
359373

360374

361375
def show_download_usage():
376+
colorama.init(autoreset=True)
362377
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
363378
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
364379
cr = colorama.Style.RESET_ALL
@@ -375,6 +390,7 @@ def show_download_usage():
375390

376391

377392
def show_grid_hub_usage():
393+
colorama.init(autoreset=True)
378394
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
379395
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
380396
cr = colorama.Style.RESET_ALL
@@ -402,6 +418,7 @@ def show_grid_hub_usage():
402418

403419

404420
def show_grid_node_usage():
421+
colorama.init(autoreset=True)
405422
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
406423
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
407424
cr = colorama.Style.RESET_ALL
@@ -441,6 +458,7 @@ def show_version_info():
441458

442459

443460
def show_options():
461+
colorama.init(autoreset=True)
444462
c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
445463
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
446464
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
@@ -497,6 +515,7 @@ def show_options():
497515

498516

499517
def show_detailed_help():
518+
colorama.init(autoreset=True)
500519
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
501520
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX
502521
c6 = colorama.Back.CYAN

‎seleniumbase/console_scripts/sb_install.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -114,6 +114,7 @@ def main(override=None):
114114
use_version = ""
115115
new_file = ""
116116
f_name = ""
117+
colorama.init(autoreset=True)
117118
c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
118119
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
119120
c3 = colorama.Fore.BLUE + colorama.Back.LIGHTYELLOW_EX

‎seleniumbase/plugins/pytest_plugin.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ def pytest_addoption(parser):
6262
--visual-baseline (Set the visual baseline for Visual/Layout tests.)
6363
--timeout-multiplier=MULTIPLIER (Multiplies the default timeout values.)
6464
"""
65+
colorama.init(autoreset=True)
6566
c1 = colorama.Fore.BLUE + colorama.Back.LIGHTCYAN_EX
6667
c2 = colorama.Fore.BLUE + colorama.Back.LIGHTGREEN_EX
6768
c3 = colorama.Fore.MAGENTA + colorama.Back.LIGHTYELLOW_EX

0 commit comments

Comments
 (0)