def save_case(case):
transactions_to_save = case.get_tracked_models_to_create(
CaseTransaction)
indices_to_save_or_update = case.get_tracked_models_to_create(
CommCareCaseIndexSQL)
indices_to_save_or_update.extend(
case.get_tracked_models_to_update(CommCareCaseIndexSQL))
index_ids_to_delete = [
index.id for index in case.get_tracked_models_to_delete(
CommCareCaseIndexSQL)
]
attachments_to_save = case.get_tracked_models_to_create(
CaseAttachmentSQL)
attachment_ids_to_delete = [
att.id
for att in case.get_tracked_models_to_delete(CaseAttachmentSQL)
]
for index in indices_to_save_or_update:
index.domain = case.domain # ensure domain is set on indices
# cast arrays that can be empty to appropriate type
query = """SELECT case_pk FROM save_case_and_related_models(
%s, %s, %s, %s::{}[], %s::{}[], %s::INTEGER[], %s::INTEGER[]
)"""
query = query.format(CommCareCaseIndexSQL_DB_TABLE,
CaseAttachmentSQL_DB_TABLE)
with get_cursor(CommCareCaseSQL) as cursor:
try:
cursor.execute(query, [
case.case_id, case, transactions_to_save,
indices_to_save_or_update, attachments_to_save,
index_ids_to_delete, attachment_ids_to_delete
])
result = fetchone_as_namedtuple(cursor)
case.id = result.case_pk
case.clear_tracked_models()
except InternalError as e:
if logging.root.isEnabledFor(logging.DEBUG):
msg = 'save_case_and_related_models called with args: \n{}, {}, {}, {} ,{} ,{}'.format(
case_adapter(case).getquoted(), [
case_transaction_adapter(t).getquoted()
for t in transactions_to_save
], [
case_index_adapter(i).getquoted()
for i in indices_to_save_or_update
], [
case_attachment_adapter(a).getquoted()
for a in attachments_to_save
], index_ids_to_delete, attachment_ids_to_delete)
logging.debug(msg)
raise CaseSaveError(e)
else:
for attachment in case.get_tracked_models_to_delete(
CaseAttachmentSQL):
attachment.delete_content()
def save_case(case):
transactions_to_save = case.get_tracked_models_to_create(CaseTransaction)
indices_to_save_or_update = case.get_tracked_models_to_create(CommCareCaseIndexSQL)
indices_to_save_or_update.extend(case.get_tracked_models_to_update(CommCareCaseIndexSQL))
index_ids_to_delete = [index.id for index in case.get_tracked_models_to_delete(CommCareCaseIndexSQL)]
attachments_to_save = case.get_tracked_models_to_create(CaseAttachmentSQL)
attachment_ids_to_delete = [att.id for att in case.get_tracked_models_to_delete(CaseAttachmentSQL)]
for index in indices_to_save_or_update:
index.domain = case.domain # ensure domain is set on indices
# cast arrays that can be empty to appropriate type
query = """SELECT case_pk FROM save_case_and_related_models(
%s, %s, %s, %s::{}[], %s::{}[], %s::INTEGER[], %s::INTEGER[]
)"""
query = query.format(CommCareCaseIndexSQL_DB_TABLE, CaseAttachmentSQL_DB_TABLE)
with get_cursor(CommCareCaseSQL) as cursor:
try:
cursor.execute(query, [
case.case_id,
case,
transactions_to_save,
indices_to_save_or_update,
attachments_to_save,
index_ids_to_delete,
attachment_ids_to_delete
])
result = fetchone_as_namedtuple(cursor)
case.id = result.case_pk
case.clear_tracked_models()
except InternalError as e:
if logging.root.isEnabledFor(logging.DEBUG):
msg = 'save_case_and_related_models called with args: \n{}, {}, {}, {} ,{} ,{}'.format(
case_adapter(case).getquoted(),
[case_transaction_adapter(t).getquoted() for t in transactions_to_save],
[case_index_adapter(i).getquoted() for i in indices_to_save_or_update],
[case_attachment_adapter(a).getquoted() for a in attachments_to_save],
index_ids_to_delete,
attachment_ids_to_delete
)
logging.debug(msg)
raise CaseSaveError(e)
else:
for attachment in case.get_tracked_models_to_delete(CaseAttachmentSQL):
attachment.delete_content()