def setUpClass(cls):
super(TestSmsExport, cls).setUpClass()
cls.schema_meta = SMSExportDataSchema.get_latest_export_schema(
domain=cls.domain, include_metadata=True)
cls.schema_no_meta = SMSExportDataSchema.get_latest_export_schema(
domain=cls.domain, include_metadata=False)
cls.export_meta = SMSExportInstance._new_from_schema(cls.schema_meta)
cls.export_no_meta = SMSExportInstance._new_from_schema(
cls.schema_no_meta)
cls.export_meta.export_format = Format.JSON
cls.export_no_meta.export_format = Format.JSON
def get_export(export_type, domain, export_id=None, username=None):
from corehq.apps.export.models import (
FormExportInstance,
CaseExportInstance,
SMSExportInstance,
SMSExportDataSchema
)
if export_type == 'form':
try:
return FormExportInstance.get(export_id)
except ResourceNotFound:
raise Http404()
elif export_type == 'case':
try:
return CaseExportInstance.get(export_id)
except ResourceNotFound:
raise Http404()
elif export_type == 'sms':
if not username:
raise Exception("Username needed to ensure permissions")
include_metadata = MESSAGE_LOG_METADATA.enabled(username)
return SMSExportInstance._new_from_schema(
SMSExportDataSchema.get_latest_export_schema(domain, include_metadata)
)
raise Exception("Unexpected export type received %s" % export_type)
def get_export(export_type, domain, export_id=None, username=None):
from corehq.apps.export.models import (FormExportInstance,
CaseExportInstance,
SMSExportInstance,
SMSExportDataSchema)
if export_type == 'form':
try:
return FormExportInstance.get(export_id)
except ResourceNotFound:
raise Http404()
elif export_type == 'case':
try:
return CaseExportInstance.get(export_id)
except ResourceNotFound:
raise Http404()
elif export_type == 'sms':
if not username:
raise Exception("Username needed to ensure permissions")
include_metadata = MESSAGE_LOG_METADATA.enabled(username)
return SMSExportInstance._new_from_schema(
SMSExportDataSchema.get_latest_export_schema(
domain, include_metadata))
raise Exception("Unexpected export type received %s" % export_type)
def get_export(self, id=None):
include_metadata = MESSAGE_LOG_METADATA.enabled_for_request(
self.request)
return SMSExportInstance._new_from_schema(
SMSExportDataSchema.get_latest_export_schema(
self.domain, include_metadata))