Handle lookup error on add command

yankeexe · Feb 12, 2022 · 7f6a91b · 7f6a91b
1 parent 8061a3d
commit 7f6a91b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -27,7 +27,7 @@
 
 setuptools.setup(
     name="timezones_cli",
-    version="0.3.0",
+    version="0.3.1",
     author="Yankee Maharjan",
     url="https://github.com/yankeexe/timezones-cli",
     description="Get local datetime from multiple timezones!",

diff --git a/timezones_cli/commands/add.py b/timezones_cli/commands/add.py
@@ -24,7 +24,13 @@ def add(query: str):
     added_timezones = []
     existing_timezones = []
     line_break = "\n"
-    timezones = query_handler(query)
+
+    try:
+        timezones = query_handler(query)
+    except LookupError:
+        return console.print(
+            "Couldn't resolve your query, please try other keywords.:x:"
+        )
 
     if len(timezones) > 1:
         timezones = handle_interaction(timezones)