def test_month_date_range(self):
"""Test month_date_range returns correct month range."""
today = datetime.now()
out = utils.month_date_range(today)
start_month = today.replace(day=1, second=1, microsecond=1)
end_month = start_month + relativedelta(months=+1)
timeformat = "%Y%m%d"
expected_string = f"{start_month.strftime(timeformat)}-{end_month.strftime(timeformat)}"
self.assertEqual(out, expected_string)
def test_month_date_range(self):
today = datetime.now()
out = utils.month_date_range(today)
start_month = today.replace(day=1, second=1, microsecond=1)
end_month = start_month + relativedelta(months=+1)
timeformat = '%Y%m%d'
expected_string = '{}-{}'.format(start_month.strftime(timeformat),
end_month.strftime(timeformat))
self.assertEqual(out, expected_string)
def _get_report_path(self, date_time):
"""
Return path of report files.
Args:
date_time (DateTime): The starting datetime object
Returns:
(String): "/prefix/report_name/YYYYMMDD-YYYYMMDD",
example: "/my-prefix/my-report/19701101-19701201"
"""
report_date_range = utils.month_date_range(date_time)
return f"{self.base_path}/{self.report_name}/{report_date_range}"
def _get_report_path(self, date_time):
"""
Return path of report files.
Args:
date_time (DateTime): The starting datetime object
Returns:
(String): "/prefix/report_name/YYYYMMDD-YYYYMMDD",
example: "/my-prefix/my-report/19701101-19701201"
"""
report_date_range = utils.month_date_range(date_time)
return "{}/{}/{}".format(self.report.get("S3Prefix"), self.report_name, report_date_range)
