Allow user to provide a single or a list of dataset years to download

okfn-brasil · Jul 7, 2017 · f370b54 · f370b54
1 parent ac8fa2b
commit f370b54
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/serenata_toolbox/chamber_of_deputies/dataset.py b/serenata_toolbox/chamber_of_deputies/dataset.py
@@ -14,7 +14,7 @@ class Dataset:
 
     def __init__(self, path, years=AVAILABLE_YEARS):
         self.path = path
-        self.years = years
+        self.years = years if isinstance(years, list) else [years]
 
     def fetch(self):
         base_url = "http://www.camara.leg.br/cotas/Ano-{}.csv.zip"

diff --git a/serenata_toolbox/federal_senate/dataset.py b/serenata_toolbox/federal_senate/dataset.py
@@ -12,7 +12,7 @@ class Dataset:
 
     def __init__(self, path, years=AVAILABLE_YEARS):
         self.path = path
-        self.years = years
+        self.years = years if isinstance(years, list) else [years]
 
     def fetch(self):
         retrieved_files = []