From f370b5495f4c3d1aa87a0613431af6d1c55121bf Mon Sep 17 00:00:00 2001 From: Irio Musskopf Date: Fri, 7 Jul 2017 12:08:03 +0300 Subject: [PATCH] Allow user to provide a single or a list of dataset years to download --- serenata_toolbox/chamber_of_deputies/dataset.py | 2 +- serenata_toolbox/federal_senate/dataset.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/serenata_toolbox/chamber_of_deputies/dataset.py b/serenata_toolbox/chamber_of_deputies/dataset.py index 6875cb5..552228c 100644 --- 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 index 100148e..a8d3d84 100644 --- 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 = []