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)