def _wrap_data_source_by_doc_type(doc, allow_deleted=False): from corehq.apps.userreports.models import ( DataSourceConfiguration, RegistryDataSourceConfiguration, ) if is_deleted(doc) and not allow_deleted: raise DataSourceConfigurationNotFoundError() doc_type = remove_deleted_doc_type_suffix(doc["doc_type"]) return { "DataSourceConfiguration": DataSourceConfiguration, "RegistryDataSourceConfiguration": RegistryDataSourceConfiguration, }[doc_type].wrap(doc)
def _raise_not_found(): raise DataSourceConfigurationNotFoundError( _('The data source referenced by this report could not be found.'))