added a new dict to handle date fields depending on the endpoint

some date fields doesn't have the expected format to be parsed with to_datetime.
So, it is necessary to pass the format argument to format the field correctly.
Otherwise, it raises an error.
A new dict has been added to check the endpoint before the parsing function is applied.

Author: wilsonfreitas

bcb/__init__.py

@@ -29,19 +29,26 @@ def __call__(self, *args):
 
 
 class EndpointQuery(ODataQuery):
+    _DATE_COLUMN_NAMES_BY_ENDPOINT = {
+        "IfDataCadastro": {"Data": "%Y%m"}
+    }
     _DATE_COLUMN_NAMES = {
         "Data",
         "DataReferencia",
         "dataHoraCotacao",
+        "InicioPeriodo",
+        "FimPeriodo",
+        "DataVigencia",
     }
 
     def collect(self):
         raw_data = super().collect()
         data = pd.DataFrame(raw_data["value"])
         for col in self._DATE_COLUMN_NAMES:
-            if col not in data.columns:
-                continue
-            data[col] = pd.to_datetime(data[col])
+            if self.entity.name in self._DATE_COLUMN_NAMES_BY_ENDPOINT and col in self._DATE_COLUMN_NAMES_BY_ENDPOINT[self.entity.name]:
+                data[col] = pd.to_datetime(data[col], format=self._DATE_COLUMN_NAMES_BY_ENDPOINT[self.entity.name][col])
+            elif col in data.columns:
+                data[col] = pd.to_datetime(data[col])
         return data
 
 
@@ -75,7 +82,7 @@ def get(self, *args, **kwargs):
             _query.show()
         data = _query.collect()
         _query.reset()
-        return pd.DataFrame(data["value"])
+        return data  # pd.DataFrame(data["value"])
 
     def query(self):
         return EndpointQuery(self._entity, self._url)