def _facility_to_fixture(facility, startdate, enddate): facility_id = facility.get_id facility_element = ElementTree.Element('facility', attrib={ 'id': facility_id, 'name': _(facility.name) }) report_data = {} for report_slug in self.report_slugs: report_data[ report_slug] = FixtureReportResult.by_composite_key( self.domain.name, facility_id, startdate.strftime("%Y-%m-%d"), enddate.strftime("%Y-%m-%d"), report_slug) if report_data[report_slug] is None: name = self.reports[report_slug].name rows = self.reports[report_slug].get_initial_row_data() fixture_result = FixtureReportResult( domain=self.domain.name, location_id=facility_id, start_date=startdate, end_date=enddate, report_slug=report_slug, rows=rows, name=name) report_data[report_slug] = fixture_result facility_element = (_reports_to_fixture(report_data, facility_element)) return facility_element
def _facility_to_fixture(facility, startdate, enddate): facility_id = facility.location_id facility_element = ElementTree.Element('facility', { 'id': facility_id, 'name': _(facility.name) }) report_data = {} m4change_data_source = M4ChangeReportDataSource() report_slugs = m4change_data_source.get_report_slugs() reports = dict((report.slug, report) for report in m4change_data_source.get_reports()) for report_slug in report_slugs: report_data[ report_slug] = FixtureReportResult.by_composite_key( restore_user.domain, facility_id, json_format_date(startdate), json_format_date(enddate), report_slug) if report_data[report_slug] is None: name = reports[report_slug].name rows = reports[report_slug].get_initial_row_data() fixture_result = FixtureReportResult( domain=restore_user.domain, location_id=facility_id, start_date=startdate, end_date=enddate, report_slug=report_slug, rows=rows, name=name) report_data[report_slug] = fixture_result facility_element = (_reports_to_fixture(report_data, facility_element)) return facility_element