Use a common exceptions handler when interacting with a source.

src/pricehist/exceptions.py

@@ -1,3 +1,18 @@
+import logging
+import sys
+from contextlib import contextmanager
+
+
+@contextmanager
+def handler():
+    try:
+        yield
+    except SourceError as e:
+        logging.debug("Critical exception encountered", exc_info=e)
+        logging.critical(str(e))
+        sys.exit(1)
+
+
 class SourceError(Exception):
     """Base exception for errors rased by sources"""
 

src/pricehist/fetch.py

@@ -1,5 +1,4 @@
 import logging
-import sys
 from datetime import date, datetime, timedelta
 
 from pricehist import exceptions
@@ -12,12 +11,8 @@ def fetch(series, source, output, invert: bool, quantize: int, fmt) -> str:
             f"source start date of {source.start()}."
         )
 
-    try:
+    with exceptions.handler():
         series = source.fetch(series)
-    except exceptions.SourceError as e:
-        logging.debug("Critical exception encountered", exc_info=e)
-        logging.critical(str(e))
-        sys.exit(1)
 
     if len(series.prices) == 0:
         logging.warn(f"No data found for the interval [{series.start}--{series.end}].")

src/pricehist/sources/basesource.py

@@ -1,9 +1,11 @@
 import logging
+import sys
 from abc import ABC, abstractmethod
 from textwrap import TextWrapper
 
 import curlify
 
+from pricehist import exceptions
 from pricehist.series import Series
 
 
@@ -56,13 +58,18 @@ class BaseSource(ABC):
         return response
 
     def format_symbols(self) -> str:
+        with exceptions.handler():
             symbols = self.symbols()
+
         width = max([len(sym) for sym, desc in symbols] + [0])
         lines = [sym.ljust(width + 4) + desc + "\n" for sym, desc in symbols]
         return "".join(lines)
 
     def format_search(self, query) -> str:
-        if (symbols := self.search(query)) is None:
+        with exceptions.handler():
+            symbols = self.search(query)
+
+        if symbols is None:
             logging.error(f"Symbol search is not possible for the {self.id()} source.")
             exit(1)
         elif symbols == []:
@@ -75,10 +82,13 @@ class BaseSource(ABC):
 
     def format_info(self, total_width=80) -> str:
         k_width = 11
+        with exceptions.handler():
             parts = [
                 self._fmt_field("ID", self.id(), k_width, total_width),
                 self._fmt_field("Name", self.name(), k_width, total_width),
-            self._fmt_field("Description", self.description(), k_width, total_width),
+                self._fmt_field(
+                    "Description", self.description(), k_width, total_width
+                ),
                 self._fmt_field("URL", self.source_url(), k_width, total_width, False),
                 self._fmt_field("Start", self.start(), k_width, total_width),
                 self._fmt_field("Types", ", ".join(self.types()), k_width, total_width),