def test_get_forms_with_attachments_meta(self):
attachment_file = open('./corehq/ex-submodules/casexml/apps/case/tests/data/attachments/fruity.jpg', 'rb')
attachments = {
'pic.jpg': UploadedFile(attachment_file, 'pic.jpg', content_type='image/jpeg')
}
form_with_pic = create_form_for_test(DOMAIN, attachments=attachments)
plain_form = create_form_for_test(DOMAIN)
forms = FormAccessorSQL.get_forms_with_attachments_meta(
[form_with_pic.form_id, plain_form.form_id], ordered=True
)
self.assertEqual(2, len(forms))
form = forms[0]
self.assertEqual(form_with_pic.form_id, form.form_id)
with self.assertNumQueries(0, using=db_for_read_write(XFormAttachmentSQL)):
expected = {
'form.xml': 'text/xml',
'pic.jpg': 'image/jpeg',
}
attachments = form.get_attachments()
self.assertEqual(2, len(attachments))
self.assertEqual(expected, {att.name: att.content_type for att in attachments})
with self.assertNumQueries(0, using=db_for_read_write(XFormAttachmentSQL)):
expected = {
'form.xml': 'text/xml',
}
attachments = forms[1].get_attachments()
self.assertEqual(1, len(attachments))
self.assertEqual(expected, {att.name: att.content_type for att in attachments})
def test_get_forms_with_attachments_meta(self):
attachment_file = open('./corehq/ex-submodules/casexml/apps/case/tests/data/attachments/fruity.jpg', 'rb')
attachments = {
'pic.jpg': UploadedFile(attachment_file, 'pic.jpg', content_type='image/jpeg')
}
form_with_pic = create_form_for_test(DOMAIN, attachments=attachments)
plain_form = create_form_for_test(DOMAIN)
forms = FormAccessorSQL.get_forms_with_attachments_meta(
[form_with_pic.form_id, plain_form.form_id], ordered=True
)
self.assertEqual(2, len(forms))
form = forms[0]
self.assertEqual(form_with_pic.form_id, form.form_id)
with self.assertNumQueries(0, using=db_for_read_write(XFormAttachmentSQL)):
expected = {
'form.xml': 'text/xml',
'pic.jpg': 'image/jpeg',
}
attachments = form.get_attachments()
self.assertEqual(2, len(attachments))
self.assertEqual(expected, {att.name: att.content_type for att in attachments})
with self.assertNumQueries(0, using=db_for_read_write(XFormAttachmentSQL)):
expected = {
'form.xml': 'text/xml',
}
attachments = forms[1].get_attachments()
self.assertEqual(1, len(attachments))
self.assertEqual(expected, {att.name: att.content_type for att in attachments})
def get_case_transactions(case_id, updated_xforms=None):
transactions = CaseAccessorSQL.get_transactions_for_case_rebuild(case_id)
form_ids = {tx.form_id for tx in transactions}
updated_xforms_map = {
xform.form_id: xform for xform in updated_xforms if not xform.is_deprecated
} if updated_xforms else {}
form_ids_to_fetch = list(form_ids - set(updated_xforms_map.keys()))
xform_map = {form.form_id: form for form in FormAccessorSQL.get_forms_with_attachments_meta(form_ids_to_fetch)}
def get_form(form_id):
if form_id in updated_xforms_map:
return updated_xforms_map[form_id]
try:
return xform_map[form_id]
except KeyError:
raise XFormNotFound
for transaction in transactions:
if transaction.form_id:
try:
transaction.cached_form = get_form(transaction.form_id)
except XFormNotFound:
logging.error('Form not found during rebuild: %s', transaction.form_id)
return transactions
def get_case_transactions(case_id, updated_xforms=None):
"""
This fetches all the transactions required to rebuild the case along
with all the forms for those transactions.
For any forms that have been updated it replaces the old form
with the new one.
:param case_id: ID of case to rebuild
:param updated_xforms: list of forms that have been changed.
:return: list of ``CaseTransaction`` objects with their associated forms attached.
"""
transactions = CaseAccessorSQL.get_transactions_for_case_rebuild(case_id)
form_ids = {tx.form_id for tx in transactions}
updated_xforms_map = {
xform.form_id: xform
for xform in updated_xforms if not xform.is_deprecated
} if updated_xforms else {}
form_ids_to_fetch = list(form_ids - set(updated_xforms_map.keys()))
xform_map = {
form.form_id: form
for form in FormAccessorSQL.get_forms_with_attachments_meta(
form_ids_to_fetch)
}
def get_form(form_id):
if form_id in updated_xforms_map:
return updated_xforms_map[form_id]
try:
return xform_map[form_id]
except KeyError:
raise XFormNotFound
for transaction in transactions:
if transaction.form_id:
try:
transaction.cached_form = get_form(transaction.form_id)
except XFormNotFound:
logging.error('Form not found during rebuild: %s',
transaction.form_id)
return transactions
def get_case_transactions(case_id, updated_xforms=None):
"""
This fetches all the transactions required to rebuild the case along
with all the forms for those transactions.
For any forms that have been updated it replaces the old form
with the new one.
:param case_id: ID of case to rebuild
:param updated_xforms: list of forms that have been changed.
:return: list of ``CaseTransaction`` objects with their associated forms attached.
"""
transactions = CaseAccessorSQL.get_transactions_for_case_rebuild(case_id)
form_ids = {tx.form_id for tx in transactions}
updated_xforms_map = {
xform.form_id: xform for xform in updated_xforms if not xform.is_deprecated
} if updated_xforms else {}
form_ids_to_fetch = list(form_ids - set(updated_xforms_map.keys()))
xform_map = {form.form_id: form for form in FormAccessorSQL.get_forms_with_attachments_meta(form_ids_to_fetch)}
def get_form(form_id):
if form_id in updated_xforms_map:
return updated_xforms_map[form_id]
try:
return xform_map[form_id]
except KeyError:
raise XFormNotFound
for transaction in transactions:
if transaction.form_id:
try:
transaction.cached_form = get_form(transaction.form_id)
except XFormNotFound:
logging.error('Form not found during rebuild: %s', transaction.form_id)
return transactions
def get_case_forms(case_id):
xform_ids = CaseAccessorSQL.get_case_xform_ids(case_id)
return FormAccessorSQL.get_forms_with_attachments_meta(xform_ids)
def get_case_forms(case_id):
xform_ids = CaseAccessorSQL.get_case_xform_ids(case_id)
return FormAccessorSQL.get_forms_with_attachments_meta(xform_ids)
