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