force predicted values to None or empty base on doc type

This commit is contained in:
PhanThanhTrung 2024-03-11 16:42:51 +07:00
parent b8e37068f6
commit c3d177dc0c

View File

@ -210,13 +210,20 @@ def _update_subscription_rq_file(request_id, index_in_request, doc_type, result)
purchase_date = elem['value']
else:
imei_number = elem['value']
_predict_result = {
"retailername": retailer_name,
"sold_to_party": sold_to_party,
"purchase_date": purchase_date,
"imei_number": imei_number
}
if doc_type=='invoice':
_predict_result = {
"retailername": retailer_name,
"sold_to_party": sold_to_party,
"purchase_date": purchase_date,
"imei_number": []
}
else:
_predict_result = {
"retailername": None,
"sold_to_party": None,
"purchase_date": [],
"imei_number": imei_number
}
image.predict_result = _predict_result
image.save()