Skip to content

Commit 6521d37

Browse files
authored
Add colours to CLI output (#86)
* Add colours to CLI output * Fix lint
1 parent 246320b commit 6521d37

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

cert_host_scraper/cli.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44

55
import click
66
from requests import RequestException
7+
from rich import box
78
from rich.console import Console
89
from rich.progress import track
910
from rich.table import Table
@@ -87,14 +88,20 @@ def search(search: str, status_code: int, timeout: int, clean: bool, strip: bool
8788
else:
8889
display = result.scraped
8990

90-
table = Table(show_header=True, header_style="bold")
91+
table = Table(show_header=True, header_style="bold", box=box.MINIMAL)
9192
table.add_column("URL")
9293
table.add_column("Status Code")
9394
for url_result in display:
9495
display_code = str(url_result.status_code)
9596
if url_result.status_code == -1:
9697
display_code = "-"
97-
table.add_row(url_result.url, display_code)
98+
99+
url = url_result.url
100+
if url_result.status_code == 200:
101+
display_code = f"[green]{display_code}[/green]"
102+
url = f"[green]{url}[/green]"
103+
104+
table.add_row(url, display_code)
98105

99106
console = Console()
100107
console.print(table)

0 commit comments

Comments
 (0)