mirror of
https://github.com/odoo/documentation.git
synced 2025-12-12 07:29:27 +07:00
a076c63527
- Since the example only works for invoices, references to the other document types supported by the OCR have been removed. - Handle the case where library `requests` isn't available. - Show additional fields detected by the OCR. - Properly set the ID of the JSON-RPC request. - Commit8c93ff7should have adapted the implementation example with the latest API version. X-original-commit:56fa6da441Part-of: odoo/documentation#12284 Signed-off-by: Louis Baudoux (lba) <lba@odoo.com>
95 lines
2.8 KiB
Python
95 lines
2.8 KiB
Python
import base64
|
|
import json
|
|
import sys
|
|
import time
|
|
import uuid
|
|
|
|
try:
|
|
import requests
|
|
except ImportError:
|
|
print("The 'requests' library is required to run this script. More information at https://pypi.org/project/requests.")
|
|
exit()
|
|
|
|
account_token = "integration_token" # Use your token
|
|
domain_name = "https://extract.api.odoo.com"
|
|
path_to_pdf = "/path/to/your/pdf"
|
|
|
|
|
|
def extract_jsonrpc_call(path: str, params: dict):
|
|
payload = {
|
|
'jsonrpc': '2.0',
|
|
'method': 'call',
|
|
'params': params,
|
|
'id': uuid.uuid4().hex, # This should be unique for each call
|
|
}
|
|
response = requests.post(domain_name + path, json=payload, timeout=10)
|
|
response.raise_for_status()
|
|
json_response = response.json()
|
|
return json_response
|
|
|
|
|
|
def send_document_to_extract(doc_path: str):
|
|
with open(doc_path, 'rb') as f:
|
|
encoded_doc = base64.b64encode(f.read()).decode()
|
|
params = {
|
|
'account_token': account_token,
|
|
'version': 123,
|
|
'documents': [encoded_doc],
|
|
}
|
|
response = extract_jsonrpc_call(f"/api/extract/invoice/2/parse", params)
|
|
return response
|
|
|
|
|
|
def get_result_from_extract(document_token: str):
|
|
params = {
|
|
'version': 123,
|
|
'document_token': document_token,
|
|
'account_token': account_token,
|
|
}
|
|
endpoint = f"/api/extract/invoice/2/get_result"
|
|
response = extract_jsonrpc_call(endpoint, params)
|
|
while response['result']['status'] == 'processing':
|
|
print("Still processing... Retrying in 5 seconds")
|
|
time.sleep(5)
|
|
response = extract_jsonrpc_call(endpoint, params)
|
|
return response
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
# Parse the document
|
|
response = send_document_to_extract(path_to_pdf)
|
|
print("/parse call status: ", response['result']['status_msg'])
|
|
|
|
if response['result']['status'] != 'success':
|
|
sys.exit(1)
|
|
|
|
document_token = response['result']['document_token']
|
|
|
|
# Get the results of the parsing
|
|
response = get_result_from_extract(document_token)
|
|
|
|
# Write the response to a file
|
|
output_file = 'response.json'
|
|
with open(output_file, 'w') as f:
|
|
json.dump(response, f, indent=2)
|
|
print("\nResult saved in", output_file)
|
|
|
|
print("/get_results call status: ", response['result']['status_msg'])
|
|
if response['result']['status'] != 'success':
|
|
sys.exit(1)
|
|
|
|
document_results = response['result']['results'][0]
|
|
|
|
|
|
def _get_selected_value(field):
|
|
return document_results.get(field, {}).get('selected_value', {}).get('content', '')
|
|
|
|
|
|
print("\nTotal:", _get_selected_value('total'))
|
|
print("Subtotal:", _get_selected_value('subtotal'))
|
|
print("Reference:", _get_selected_value('invoice_id'))
|
|
print("Date:", _get_selected_value('date'))
|
|
print("Due date:", _get_selected_value('due_date'))
|
|
print("Currency:", _get_selected_value('currency'))
|