def get_query_report(self, existing_query=True):
url = self.get_query_report_url(existing_query)
report = requests.get(url, stream=True)
if self.kwargs[
"query_param_type"] == "TYPE_REACH_AND_FREQUENCY" and self.kwargs[
"add_date_to_report"]:
start, stop = get_date_start_and_date_stop_from_date_range(
self.kwargs["day_range"])
column_dict = {
"date_start": start.strftime(self.kwargs.get("date_format")),
"date_stop": stop.strftime(self.kwargs.get("date_format")),
}
report_gen = get_report_generator_from_flat_file(
report.iter_lines(), add_column=True, column_dict=column_dict)
return skip_last(report_gen, 1)
else:
report_gen = get_report_generator_from_flat_file(
report.iter_lines())
return skip_last(report_gen, 1)
def test_get_previous_month_dates_if_random_month_of_the_year(self):
self.assertTupleEqual(
get_date_start_and_date_stop_from_date_range("PREVIOUS_MONTH"),
(date(2021, 1, 1), date(2021, 1, 31)))
def test_get_previous_week_dates_if_sunday(self):
self.assertTupleEqual(
get_date_start_and_date_stop_from_date_range("PREVIOUS_WEEK"),
(date(2021, 1, 4), date(2021, 1, 10)))
def test_get_date_start_and_date_stop_from_date_range(
self, date_range, expected):
self.assertTupleEqual(
get_date_start_and_date_stop_from_date_range(date_range), expected)