Add [Y/n] prompt to suggestion (#143)

src/norwegianblue/__init__.py

@@ -7,6 +7,7 @@
 import datetime as dt
 import json
 import logging
+from functools import lru_cache
 
 from dateutil.relativedelta import relativedelta
 from termcolor import colored
@@ -66,8 +67,9 @@ def norwegianblue(
 
         logging.info("HTTP status code: %d", r.status_code)
         if r.status_code == 404:
-            suggestion = _suggest_product(product)
-            raise ValueError(ERROR_404_TEXT.format(product, suggestion))
+            suggestion = suggest_product(product)
+            msg = ERROR_404_TEXT.format(product, suggestion)
+            raise ValueError(msg)
 
         # Raise if we made a bad request
         # (4XX client error or 5XX server error response)
@@ -98,7 +100,8 @@ def norwegianblue(
     return output
 
 
-def _suggest_product(product: str) -> str:
+@lru_cache(maxsize=None)
+def suggest_product(product: str) -> str:
     import warnings
 
     with warnings.catch_warnings():

src/norwegianblue/cli.py

@@ -19,6 +19,8 @@
 import platform
 import sys
 
+from termcolor import colored
+
 import norwegianblue
 from norwegianblue import _cache
 
@@ -103,7 +105,19 @@ def main() -> None:
                 show_title=multiple_products,
             )
         except ValueError as e:
-            sys.exit(e)
+            prompt = f"{e} [Y/n] "
+            if args.color != "no":
+                prompt = colored(prompt, "yellow")
+            answer = input(prompt)
+            if answer not in ("", "y", "Y"):
+                sys.exit()
+            suggestion = norwegianblue.suggest_product(product)
+            output = norwegianblue.norwegianblue(
+                product=suggestion,
+                format=args.format,
+                color=args.color,
+                show_title=multiple_products,
+            )
         print(output)
         print()
         if args.web: