Add error handling for download failures and extend tests for long series

Issue #19

Author: wilsonfreitas

bcb/sgs/__init__.py

@@ -1,7 +1,8 @@
+import json
 from io import StringIO
 
-import requests
 import pandas as pd
+import requests
 
 from bcb.utils import Date
 
@@ -119,6 +120,12 @@ def get(codes, start=None, end=None, last=0, multi=True, freq=None):
         urd = _get_url_and_payload(code.value, start, end, last)
         res = requests.get(urd["url"], params=urd["payload"])
         if res.status_code != 200:
+            try:
+                res_json = json.loads(res.text)
+            except Exception:
+                res_json = {}
+            if "error" in res_json:
+                raise Exception("Download error: {}".format(res_json["error"]))
             raise Exception("Download error: code = {}".format(code.value))
         df = pd.read_json(StringIO(res.text))
         df = _format_df(df, code, freq)

tests/sgs/test_series.py

@@ -1,5 +1,7 @@
 from datetime import datetime
+
 import pandas as pd
+
 from bcb import sgs
 
 
@@ -87,3 +89,16 @@ def test_get_series():
     assert len(x) == 5
     assert x.index[0] == datetime.strptime("2021-01-18", "%Y-%m-%d")
     assert x.index[-1] == datetime.strptime("2021-01-22", "%Y-%m-%d")
+
+
+def test_get_long_series_error():
+    # Test for error when getting long series
+    try:
+        sgs.get(1, start="2000-01-01", end="2023-01-01")
+    except Exception as e:
+        assert (
+            str(e)
+            == "Download error: O sistema aceita uma janela de consulta de, no máximo, 10 anos em séries de periodicidade diária"
+        )
+    else:
+        assert False, "Expected an exception but none was raised."