mirror of
https://github.com/OCA/pms.git
synced 2025-01-29 00:17:45 +02:00
[IMP]14.0-pms_api_rest: service OCR supplier generic
This commit is contained in:
braisab
Darío Lodeiros
@@ -1,9 +1,337 @@
|
||||
from regula.documentreader.webclient import (
|
||||
DocumentReaderApi,
|
||||
ProcessParams,
|
||||
RecognitionRequest,
|
||||
Result,
|
||||
Scenario,
|
||||
TextFieldType,
|
||||
)
|
||||
from datetime import date, datetime
|
||||
from dateutil.relativedelta import relativedelta
|
||||
|
||||
from odoo import fields, models
|
||||
|
||||
|
||||
|
||||
class PmsProperty(models.Model):
|
||||
_inherit = "pms.property"
|
||||
|
||||
is_used_regula = fields.Boolean(
|
||||
string="Used regula", help="True if this property uses regula's OCR"
|
||||
ocr_checkin_supplier = fields.Selection(
|
||||
selection_add=["regula", "Regula"]
|
||||
)
|
||||
|
||||
def _regula_document_process(self, image_base_64_front, image_base_64_back=False):
|
||||
ocr_regula_url = (
|
||||
self.env["ir.config_parameter"].sudo().get_param("ocr_regula_url")
|
||||
)
|
||||
with DocumentReaderApi(host=ocr_regula_url) as api:
|
||||
params = ProcessParams(
|
||||
scenario=Scenario.FULL_PROCESS,
|
||||
result_type_output=[
|
||||
Result.TEXT,
|
||||
Result.STATUS,
|
||||
Result.VISUAL_TEXT,
|
||||
Result.DOCUMENT_TYPE,
|
||||
],
|
||||
)
|
||||
request = RecognitionRequest(
|
||||
process_params=params, images=[image_base_64_front]
|
||||
)
|
||||
response = api.process(request)
|
||||
if response.text and response.text.field_list:
|
||||
id_country_spain = (
|
||||
self.env["res.country"].search([("code", "=", "ES")]).id
|
||||
)
|
||||
country_id = self._process_nationality(
|
||||
response.text.get_field(TextFieldType.NATIONALITY),
|
||||
response.text.get_field(TextFieldType.NATIONALITY_CODE),
|
||||
response.text.get_field(TextFieldType.NATIONALITY_CODE_NUMERIC),
|
||||
)
|
||||
firstname, lastname, lastname2 = self._process_name(
|
||||
id_country_spain,
|
||||
country_id,
|
||||
response.text.get_field(TextFieldType.GIVEN_NAMES),
|
||||
response.text.get_field(TextFieldType.FIRST_SURNAME),
|
||||
response.text.get_field(TextFieldType.SECOND_SURNAME),
|
||||
response.text.get_field(TextFieldType.SURNAME),
|
||||
response.text.get_field(TextFieldType.SURNAME_AND_GIVEN_NAMES),
|
||||
)
|
||||
pms_ocr_checkin_result = dict()
|
||||
if country_id:
|
||||
pms_ocr_checkin_result['nationality'] = country_id
|
||||
if firstname:
|
||||
pms_ocr_checkin_result['firstname'] = firstname
|
||||
if lastname:
|
||||
pms_ocr_checkin_result['lastname'] = lastname
|
||||
if lastname2:
|
||||
pms_ocr_checkin_result['lastname2'] = lastname2
|
||||
gender = response.text.get_field(TextFieldType.SEX)
|
||||
if gender and gender.value != "":
|
||||
pms_ocr_checkin_result['gender'] = (
|
||||
"male"
|
||||
if gender.value == "M"
|
||||
else "female"
|
||||
if gender.value == "F"
|
||||
else "other"
|
||||
)
|
||||
date_of_birth = response.text.get_field(TextFieldType.DATE_OF_BIRTH)
|
||||
if date_of_birth and date_of_birth.value != "":
|
||||
pms_ocr_checkin_result['birthdate'] = (
|
||||
datetime.strptime(
|
||||
date_of_birth.value.replace("-", "/"), "%Y/%m/%d"
|
||||
)
|
||||
.date()
|
||||
.isoformat()
|
||||
)
|
||||
date_of_expiry = response.text.get_field(TextFieldType.DATE_OF_EXPIRY)
|
||||
age = response.text.get_field(TextFieldType.AGE)
|
||||
document_class_code = response.text.get_field(
|
||||
TextFieldType.DOCUMENT_CLASS_CODE
|
||||
)
|
||||
if (
|
||||
document_class_code
|
||||
and document_class_code.value != ""
|
||||
and document_class_code.value == "P"
|
||||
):
|
||||
pms_ocr_checkin_result['documentType'] = (
|
||||
self.env["res.partner.id_category"]
|
||||
.search([("code", "=", "P")])
|
||||
.id
|
||||
)
|
||||
date_of_issue = response.text.get_field(TextFieldType.DATE_OF_ISSUE)
|
||||
if country_id == id_country_spain and (
|
||||
not date_of_issue or date_of_issue.value == ""
|
||||
):
|
||||
date_of_issue = self._calc_expedition_date(
|
||||
document_class_code,
|
||||
date_of_expiry,
|
||||
age,
|
||||
date_of_birth,
|
||||
)
|
||||
pms_ocr_checkin_result['documentExpeditionDate'] = date_of_issue
|
||||
elif date_of_issue and date_of_issue.value != "":
|
||||
pms_ocr_checkin_result['documentExpeditionDate'] = (
|
||||
date_of_issue.value.replace("-", "/")
|
||||
)
|
||||
support_number, document_number = self._proccess_document_number(
|
||||
id_country_spain,
|
||||
country_id,
|
||||
document_class_code,
|
||||
response.text.get_field(TextFieldType.DOCUMENT_NUMBER),
|
||||
response.text.get_field(TextFieldType.PERSONAL_NUMBER),
|
||||
)
|
||||
if support_number:
|
||||
pms_ocr_checkin_result['documentSupportNumber'] = support_number
|
||||
if document_number:
|
||||
pms_ocr_checkin_result['documentNumber'] = document_number
|
||||
address_street, address_city, address_area = self._process_address(
|
||||
id_country_spain,
|
||||
country_id,
|
||||
response.text.get_field(TextFieldType.ADDRESS_STREET),
|
||||
response.text.get_field(TextFieldType.ADDRESS_CITY),
|
||||
response.text.get_field(TextFieldType.ADDRESS_AREA),
|
||||
response.text.get_field(TextFieldType.ADDRESS),
|
||||
)
|
||||
if address_street:
|
||||
pms_ocr_checkin_result['residenceStreet'] = address_street
|
||||
if address_city:
|
||||
pms_ocr_checkin_result['residenceCity'] = address_city
|
||||
if address_area:
|
||||
pms_ocr_checkin_result['countryState'] = address_area
|
||||
return pms_ocr_checkin_result
|
||||
|
||||
def _process_nationality(
|
||||
self, nationality, nationality_code, nationality_code_numeric
|
||||
):
|
||||
country_id = False
|
||||
country = False
|
||||
if nationality_code_numeric and nationality_code_numeric.value != "":
|
||||
country = self.env["res.country"].search(
|
||||
[("code_numeric", "=", nationality_code_numeric.value)]
|
||||
)
|
||||
elif nationality_code and nationality_code.value != "":
|
||||
country = self.env["res.country"].search(
|
||||
[("code_alpha3", "=", nationality_code.value)]
|
||||
)
|
||||
elif nationality and nationality.value != "":
|
||||
country = self.env["res.country"].search([("name", "=", nationality.value)])
|
||||
|
||||
if country:
|
||||
country_id = country.id
|
||||
|
||||
return country_id
|
||||
|
||||
def _process_address(
|
||||
self,
|
||||
id_country_spain,
|
||||
country_id,
|
||||
address_street,
|
||||
address_city,
|
||||
address_area,
|
||||
address,
|
||||
):
|
||||
res_address_street = False
|
||||
res_address_city = False
|
||||
res_address_area = False
|
||||
state = False
|
||||
if country_id == id_country_spain:
|
||||
if address_street and address_street.value != "":
|
||||
res_address_street = address_street.value
|
||||
if address_city and address_city.value != "":
|
||||
res_address_city = address_city.value
|
||||
if address_area and address_area.value != "":
|
||||
res_address_area = address_area.value
|
||||
if (
|
||||
address
|
||||
and address != ""
|
||||
and not (all([address_street, address_city, address_area]))
|
||||
):
|
||||
address = address.value.replace("^", " ")
|
||||
address_list = address.split(" ")
|
||||
if not res_address_area:
|
||||
res_address_area = address_list[-1]
|
||||
if not res_address_city:
|
||||
res_address_city = address_list[-2]
|
||||
if not res_address_street:
|
||||
res_address_street = address.replace(
|
||||
res_address_area, "", 1
|
||||
).replace(res_address_city, "", 1)
|
||||
if res_address_area:
|
||||
state = self.env["res.country.state"].search(
|
||||
[("name", "ilike", res_address_area)]
|
||||
)
|
||||
if state and len(state) == 1:
|
||||
state = state.id
|
||||
else:
|
||||
if address and address.value != "":
|
||||
res_address_street = address.value.replace("^", " ")
|
||||
return res_address_street, res_address_city, state
|
||||
|
||||
def _process_name(
|
||||
self,
|
||||
id_country_spain,
|
||||
country_id,
|
||||
given_names,
|
||||
first_surname,
|
||||
second_surname,
|
||||
surname,
|
||||
surname_and_given_names,
|
||||
):
|
||||
firstname = False
|
||||
lastname = False
|
||||
lastname2 = False
|
||||
|
||||
if surname_and_given_names.value and surname_and_given_names.value != "":
|
||||
surname_and_given_names = surname_and_given_names.value.replace("^", " ")
|
||||
|
||||
if given_names and given_names.value != "":
|
||||
firstname = given_names.value
|
||||
|
||||
if first_surname and first_surname.value != "":
|
||||
lastname = first_surname.value
|
||||
|
||||
if second_surname and second_surname.value != "":
|
||||
lastname2 = second_surname.value
|
||||
|
||||
if country_id == id_country_spain and not (
|
||||
all([firstname, lastname, lastname2])
|
||||
):
|
||||
if surname and surname.value != "":
|
||||
lastname = lastname if lastname else surname.value.split(" ")[0]
|
||||
lastname2 = lastname2 if lastname2 else surname.value.split(" ")[1:][0]
|
||||
if (
|
||||
surname_and_given_names
|
||||
and surname_and_given_names != ""
|
||||
and not firstname
|
||||
):
|
||||
firstname = surname_and_given_names.replace(
|
||||
lastname, "", 1
|
||||
).replace(lastname2, "", 1)
|
||||
elif surname_and_given_names and surname_and_given_names != "":
|
||||
lastname = (
|
||||
lastname if lastname else surname_and_given_names.split(" ")[0]
|
||||
)
|
||||
lastname2 = (
|
||||
lastname2 if lastname2 else surname_and_given_names.split(" ")[1]
|
||||
)
|
||||
firstname = (
|
||||
firstname
|
||||
if firstname
|
||||
else surname_and_given_names.replace(lastname, "", 1).replace(
|
||||
lastname2, "", 1
|
||||
)
|
||||
)
|
||||
elif (
|
||||
country_id
|
||||
and country_id != id_country_spain
|
||||
and not (all([firstname, lastname]))
|
||||
):
|
||||
if surname and surname.value != "":
|
||||
lastname = lastname if lastname else surname.value
|
||||
if (
|
||||
surname_and_given_names
|
||||
and surname_and_given_names != ""
|
||||
and not firstname
|
||||
):
|
||||
firstname = surname_and_given_names.replace(lastname, "", 1)
|
||||
elif surname_and_given_names and surname_and_given_names != "":
|
||||
lastname = (
|
||||
lastname if lastname else surname_and_given_names.split(" ")[0]
|
||||
)
|
||||
firstname = (
|
||||
firstname
|
||||
if firstname
|
||||
else surname_and_given_names.replace(lastname, "", 1)
|
||||
)
|
||||
return firstname, lastname, lastname2
|
||||
|
||||
def _calc_expedition_date(
|
||||
self, document_class_code, date_of_expiry, age, date_of_birth
|
||||
):
|
||||
result = False
|
||||
person_age = False
|
||||
if age and age.value != "":
|
||||
person_age = int(age.value)
|
||||
elif date_of_birth and date_of_birth.value != "":
|
||||
date_of_birth = datetime.strptime(
|
||||
date_of_birth.value.replace("-", "/"), "%Y/%m/%d"
|
||||
).date()
|
||||
person_age = relativedelta(date.today(), date_of_birth).years
|
||||
if date_of_expiry and date_of_expiry.value != "" and person_age:
|
||||
date_of_expiry = datetime.strptime(
|
||||
date_of_expiry.value.replace("-", "/"), "%Y/%m/%d"
|
||||
).date()
|
||||
if person_age < 30:
|
||||
result = date_of_expiry - relativedelta(years=5)
|
||||
elif (
|
||||
person_age >= 30
|
||||
and document_class_code
|
||||
and document_class_code.value == "P"
|
||||
):
|
||||
result = date_of_expiry - relativedelta(years=10)
|
||||
elif 30 <= person_age < 70:
|
||||
result = date_of_expiry - relativedelta(years=10)
|
||||
return result.isoformat() if result else False
|
||||
|
||||
def _proccess_document_number(
|
||||
self,
|
||||
id_country_spain,
|
||||
country_id,
|
||||
document_class_code,
|
||||
document_number,
|
||||
personal_number,
|
||||
):
|
||||
res_support_number = False
|
||||
res_document_number = False
|
||||
if personal_number and personal_number.value != "":
|
||||
res_document_number = personal_number.value
|
||||
if document_number and document_number.value != "":
|
||||
res_support_number = document_number.value
|
||||
if (
|
||||
country_id == id_country_spain
|
||||
and document_class_code
|
||||
and document_class_code.value != "P"
|
||||
):
|
||||
return res_support_number, res_document_number
|
||||
else:
|
||||
return False, res_support_number
|
||||
|
||||
Reference in New Issue
Block a user