9df648b4c7
We have introduced a second version of the Extract API that adds the IAP
account token to the `get_result` and `validate` routes.
There are also other changes in this commit:
- The deprecated stuff has been removed as it's been deprecated for a
while and won't be supported anymore in the future.
- The `status` table of the `get_result` route has been added as it was
missing for some reason.
- The documentation regarding the `/validate` routes have been removed
for security reasons.
closes odoo/documentation#6277
X-original-commit: a8c6272899
Signed-off-by: Antoine Vandevenne (anv) <anv@odoo.com>
Signed-off-by: Louis Baudoux (lba) <lba@odoo.com>
91 lines
2.7 KiB
Python
91 lines
2.7 KiB
Python
import base64
|
|
import json
|
|
import sys
|
|
import time
|
|
|
|
import requests
|
|
|
|
account_token = "integration_token" # Use your token
|
|
domain_name = "https://extract.api.odoo.com"
|
|
path_to_pdf = "/path/to/your/pdf"
|
|
doc_type = "invoice" # invoice, expense or applicant
|
|
|
|
# Do not change
|
|
API_VERSION = {
|
|
'invoice': 122,
|
|
'expense': 132,
|
|
'applicant': 102,
|
|
}
|
|
|
|
def extract_jsonrpc_call(path: str, params: dict):
|
|
payload = {
|
|
'jsonrpc': '2.0',
|
|
'method': 'call',
|
|
'params': params,
|
|
'id': 0, # 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': API_VERSION[doc_type],
|
|
'documents': [encoded_doc],
|
|
}
|
|
response = extract_jsonrpc_call(f"/api/extract/{doc_type}/2/parse", params)
|
|
return response
|
|
|
|
|
|
def get_result_from_extract(document_token: str):
|
|
params = {
|
|
'version': API_VERSION[doc_type],
|
|
'document_token': document_token,
|
|
'account_token': account_token,
|
|
}
|
|
endpoint = f"/api/extract/{doc_type}/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]
|
|
|
|
print("\nTotal:", document_results['total']['selected_value']['content'])
|
|
print("Subtotal:", document_results['subtotal']['selected_value']['content'])
|
|
print("Invoice id:", document_results['invoice_id']['selected_value']['content'])
|
|
print("Date:", document_results['date']['selected_value']['content'])
|
|
print("...\n")
|