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.'))