From ad1990cc42e8f35f278d4c9ab45f18bbd02be815 Mon Sep 17 00:00:00 2001
From: rlaphoenix <rlaphoenix@pm.me>
Date: Tue, 28 Feb 2023 17:03:25 +0000
Subject: [PATCH] Print a download cancelled message on CTRL+C

---
 devine/commands/dl.py | 29 ++++++++++++++++-------------
 1 file changed, 16 insertions(+), 13 deletions(-)

diff --git a/devine/commands/dl.py b/devine/commands/dl.py
index 0a7bb55..ecf052c 100644
--- a/devine/commands/dl.py
+++ b/devine/commands/dl.py
@@ -433,16 +433,16 @@ class dl:
             if skip_dl:
                 self.log.info("Skipping Download...")
             else:
-                with Live(
-                    Padding(
-                        download_table,
-                        (1, 5)
-                    ),
-                    console=console,
-                    refresh_per_second=5
-                ):
-                    with ThreadPoolExecutor(workers) as pool:
-                        try:
+                try:
+                    with Live(
+                        Padding(
+                            download_table,
+                            (1, 5)
+                        ),
+                        console=console,
+                        refresh_per_second=5
+                    ):
+                        with ThreadPoolExecutor(workers) as pool:
                             for download in futures.as_completed((
                                 pool.submit(
                                     self.download_track,
@@ -480,9 +480,12 @@ class dl:
                                     self.log.error("Download worker threw an unhandled exception:")
                                     console.print_exception()
                                     return
-                        except KeyboardInterrupt:
-                            self.log.info("Received Keyboard Interrupt, stopping...")
-                            return
+                except KeyboardInterrupt:
+                    console.print(Padding(
+                        ":x: Download Cancelled...",
+                        (0, 5, 1, 5)
+                    ))
+                    return
 
                 video_track_n = 0