Make KeyboardInterrupt graceful

KarimPwnz · Dec 12, 2021 · 25f45eb · 25f45eb
1 parent b23301a
commit 25f45eb
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/waybacked/__main__.py b/waybacked/__main__.py
@@ -61,9 +61,12 @@ def main():
                         nargs=None, help="Search query")
     args = parser.parse_args()
     search = WaybackSearch(args.search)
-    for page in search.run():
-        for url in page:
-            print(url)
+    try:
+        for page in search.run():
+            for url in page:
+                print(url)
+    except KeyboardInterrupt:
+        pass
 
 
 if __name__ == "__main__":