feat: print success output

bellingcat · Jul 1, 2022 · c9d36c6 · c9d36c6
1 parent 62da080
commit c9d36c6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/geoclustering/__main__.py b/geoclustering/__main__.py
@@ -54,7 +54,6 @@ def print_debug(s):
         if debug:
             click.secho(s, fg="bright_black")
 
-
     df = io.read_csv_file(filename)
     print_debug(f"Read {len(df)} valid coordinates from {Path(filename).absolute()}")
 
@@ -63,7 +62,7 @@ def print_debug(s):
     )
 
     if not bool(clusters):
-        click.echo("Did not find clusters matching input parameters.")
+        click.secho("Did not find clusters matching input parameters.", fg="yellow")
         return
 
     print_debug(f"Found {len(clusters)} valid clusters using {algorithm}")
@@ -80,6 +79,8 @@ def print_debug(s):
         print_debug(f"Opening visualization in default browser")
         webbrowser.open_new_tab("file://" + str(vis.absolute()))
 
+    click.secho("Clustering completed.", fg="green")
+
 
 if __name__ == "__main__":
     main()