mirror of
https://github.com/OCA/pms.git
synced 2025-01-29 00:17:45 +02:00
473 lines
18 KiB
Python
473 lines
18 KiB
Python
import base64
|
|
import datetime
|
|
import io
|
|
import json
|
|
import time
|
|
from datetime import date
|
|
|
|
import PyPDF2
|
|
import requests
|
|
from bs4 import BeautifulSoup as bs
|
|
|
|
from odoo import _, api, fields, models
|
|
from odoo.exceptions import MissingError, ValidationError
|
|
from odoo.modules.module import get_module_resource
|
|
|
|
|
|
class TravellerReport(models.TransientModel):
|
|
_name = "traveller.report.wizard"
|
|
_description = "Traveller Report"
|
|
|
|
txt_filename = fields.Text()
|
|
txt_binary = fields.Binary(string="File Download")
|
|
txt_message = fields.Char(string="File Preview")
|
|
|
|
def generate_file_from_user_action(self):
|
|
|
|
# get the active property
|
|
pms_property = self.env["pms.property"].search(
|
|
[("id", "=", self.env.user.get_active_property_ids()[0])]
|
|
)
|
|
# check if there's institution settings properly established
|
|
if (
|
|
not pms_property
|
|
or not pms_property.institution_property_id
|
|
or not pms_property.institution_user
|
|
or not pms_property.institution_password
|
|
):
|
|
raise ValidationError(
|
|
_("The guest information sending settings is not property updated.")
|
|
)
|
|
|
|
# build content
|
|
content = self.generate_checkin_list(pms_property.id)
|
|
|
|
if content:
|
|
txt_binary = self.env["traveller.report.wizard"].create(
|
|
{
|
|
"txt_filename": pms_property.institution_property_id + ".999",
|
|
"txt_binary": base64.b64encode(str.encode(content)),
|
|
"txt_message": content,
|
|
}
|
|
)
|
|
return {
|
|
"name": _("Traveller Report"),
|
|
"res_id": txt_binary.id,
|
|
"res_model": "traveller.report.wizard",
|
|
"target": "new",
|
|
"type": "ir.actions.act_window",
|
|
"view_id": self.env.ref("pms_l10n_es.traveller_report_wizard").id,
|
|
"view_mode": "form",
|
|
"view_type": "form",
|
|
}
|
|
|
|
def generate_checkin_list(self, property_id):
|
|
|
|
# check if there's guests info pending to send
|
|
if self.env["pms.checkin.partner"].search_count(
|
|
[
|
|
("state", "=", "onboard"),
|
|
("arrival", ">=", str(date.today()) + " 0:00:00"),
|
|
]
|
|
):
|
|
|
|
# get the active property
|
|
pms_property = self.env["pms.property"].search([("id", "=", property_id)])
|
|
# get checkin partners info to send
|
|
lines = self.env["pms.checkin.partner"].search(
|
|
[
|
|
("state", "=", "onboard"),
|
|
("arrival", ">=", str(date.today()) + " 0:00:00"),
|
|
("arrival", "<=", str(date.today()) + " 23:59:59"),
|
|
]
|
|
)
|
|
# build the property info record
|
|
# 1 | property id | property name | date | nº of checkin partners
|
|
content = (
|
|
"1|"
|
|
+ pms_property.institution_property_id.upper()
|
|
+ "|"
|
|
+ pms_property.name.upper()
|
|
+ "|"
|
|
+ datetime.datetime.now().strftime("%Y%m%d|%H%M")
|
|
+ "|"
|
|
+ str(len(lines))
|
|
+ "\n"
|
|
)
|
|
# build each checkin partner line's record
|
|
# 2|DNI nº|Doc.number|doc.type|exp.date|lastname|lastname2|name|...
|
|
# ...gender|birthdate|nation.|checkin
|
|
|
|
for line in lines:
|
|
content += "2"
|
|
# [P|N|..]
|
|
if line.document_type != "D":
|
|
content += "||" + line.document_number.upper() + "|"
|
|
else:
|
|
content += "|" + line.document_number.upper() + "||"
|
|
content += line.document_type + "|"
|
|
content += line.document_expedition_date.strftime("%Y%m%d") + "|"
|
|
content += line.lastname.upper() + "|"
|
|
if line.lastname2:
|
|
content += line.lastname2.upper()
|
|
content += "|" + line.firstname.upper() + "|"
|
|
if line.gender == "female":
|
|
content += "F|"
|
|
else:
|
|
content += "M|"
|
|
content += line.birthdate_date.strftime("%Y%m%d") + "|"
|
|
content += line.nationality_id.name.upper() + "|"
|
|
content += line.arrival.strftime("%Y%m%d") + "\n"
|
|
|
|
return content
|
|
|
|
def send_file_gc(self, file_content, called_from_user, pms_property):
|
|
url = "https://hospederias.guardiacivil.es/"
|
|
login_route = "/hospederias/login.do"
|
|
upload_file_route = "/hospederias/cargaFichero.do"
|
|
logout_route = "/hospederias/logout.do"
|
|
|
|
if file_content:
|
|
headers = {
|
|
"User-Agent": "Mozilla/5.0 (Linux; Android 6.0; Nexus 5 "
|
|
"Build/MRA58N) AppleWebKit/537.36 (KHTML, like "
|
|
"Gecko) Chrome/90.0.4430.93 Mobile Safari/537.36",
|
|
}
|
|
session = requests.session()
|
|
login_payload = {
|
|
"usuario": pms_property.institution_user,
|
|
"pswd": pms_property.institution_password,
|
|
}
|
|
response_login = session.post(
|
|
url + login_route,
|
|
headers=headers,
|
|
data=login_payload,
|
|
verify=get_module_resource("pms_l10n_es", "static", "cert.pem"),
|
|
)
|
|
time.sleep(0.1)
|
|
|
|
# check if authentication was successful / unsuccessful or the
|
|
# resource cannot be accessed
|
|
soup = bs(response_login.text, "html.parser")
|
|
errors_login = soup.select("#txterror > ul > li")
|
|
if errors_login:
|
|
raise ValidationError(errors_login[0].text)
|
|
else:
|
|
login_correct = soup.select(".cabecera2")
|
|
if not login_correct:
|
|
session.close()
|
|
raise ValidationError(_("Connection could not be established"))
|
|
|
|
# build the file to send
|
|
files = {"fichero": (pms_property.institution_user + ".999", file_content)}
|
|
|
|
# send file
|
|
response_file_sent = session.post(
|
|
url + upload_file_route,
|
|
data={"autoSeq": "on"},
|
|
files=files,
|
|
verify=get_module_resource("pms_l10n_es", "static", "cert.pem"),
|
|
)
|
|
time.sleep(0.1)
|
|
|
|
# logout & close connection
|
|
session.get(
|
|
url + logout_route,
|
|
headers=headers,
|
|
verify=get_module_resource("pms_l10n_es", "static", "cert.pem"),
|
|
)
|
|
session.close()
|
|
|
|
# check if the file send has been correct
|
|
soup = bs(response_file_sent.text, "html.parser")
|
|
errors = soup.select("#errores > tbody > tr")
|
|
if errors:
|
|
msg = "Errores en el fichero:\n"
|
|
for e in errors:
|
|
msg += "Error en línea " + e.select("a")[0].text + ": "
|
|
msg += e.select("a")[2].text + "\n"
|
|
self.env["pms.log.institution.traveller.report"].create(
|
|
{
|
|
"error_sending_data": True,
|
|
"txt_incidencies_from_institution": msg,
|
|
}
|
|
)
|
|
raise ValidationError(msg)
|
|
else:
|
|
self.env["pms.log.institution.traveller.report"].create(
|
|
{
|
|
"error_sending_data": False,
|
|
"txt_message": _("Successful file sending"),
|
|
}
|
|
)
|
|
if called_from_user:
|
|
message = {
|
|
"type": "ir.actions.client",
|
|
"tag": "display_notification",
|
|
"params": {
|
|
"title": _("Sent succesfully!"),
|
|
"message": _("Successful file sending"),
|
|
"sticky": False,
|
|
},
|
|
}
|
|
return message
|
|
|
|
def send_file_pn(self, file_content, called_from_user, pms_property):
|
|
base_url = "https://webpol.policia.es"
|
|
headers = {
|
|
"User-Agent": "Mozilla/5.0 (Linux; Android 6.0; Nexus 5 "
|
|
"Build/MRA58N) AppleWebKit/537.36 (KHTML, like "
|
|
"Gecko) Chrome/90.0.4430.93 Mobile Safari/537.36",
|
|
}
|
|
pre_login_route = "/e-hotel/login"
|
|
login_route = "/e-hotel/execute_login"
|
|
next_file_name_route = "/e-hotel/hospederia/ficheros/vista/envioFicheros"
|
|
upload_file_route = "/e-hotel/hospederia/ficheros/subirFichero"
|
|
pre_get_list_files_sent_route = (
|
|
"/e-hotel/hospederia/vista/seguimientoHospederia"
|
|
)
|
|
files_sent_list_route = "/e-hotel/hospederia/listar/ficherosHospederia"
|
|
last_file_errors_route = "/e-hotel/hospederia/report/erroresFicheroHospederia"
|
|
logout_route = "/e-hotel/execute_logout"
|
|
|
|
session = requests.session()
|
|
|
|
# retrieve token
|
|
response_pre_login = session.post(
|
|
base_url + pre_login_route,
|
|
headers=headers,
|
|
verify=False,
|
|
)
|
|
time.sleep(0.1)
|
|
token = bs(response_pre_login.text, "html.parser").select(
|
|
"input[name='_csrf']"
|
|
)[0]["value"]
|
|
|
|
if not token:
|
|
raise MissingError(_("Could not get token login."))
|
|
|
|
# do login
|
|
session.post(
|
|
base_url + login_route,
|
|
headers=headers,
|
|
data={
|
|
"username": pms_property.institution_user,
|
|
"password": pms_property.institution_password,
|
|
"_csrf": token,
|
|
},
|
|
verify=False,
|
|
)
|
|
time.sleep(0.1)
|
|
headers["x-csrf-token"] = token
|
|
# retrieve file name to send
|
|
response_name_file_route = session.post(
|
|
base_url + next_file_name_route,
|
|
headers=headers,
|
|
verify=False,
|
|
)
|
|
time.sleep(0.1)
|
|
soup = bs(response_name_file_route.text, "html.parser")
|
|
file_name = soup.select("#msjNombreFichero > b > u")[0].text
|
|
|
|
if not file_name:
|
|
raise MissingError(_("Could not get next file name to send."))
|
|
|
|
# send file
|
|
upload_result = session.post(
|
|
base_url + upload_file_route,
|
|
headers=headers,
|
|
verify=False,
|
|
data={
|
|
"jsonHiddenComunes": "",
|
|
"ficheroJ": "",
|
|
"_csrf": token,
|
|
},
|
|
files={"fichero": (file_name, file_content)},
|
|
)
|
|
|
|
if upload_result.status_code != 200:
|
|
raise MissingError(_("Could not upload file."))
|
|
|
|
time.sleep(0.1)
|
|
|
|
# retrieve property data
|
|
response_pre_files_sent_list_route = session.post(
|
|
base_url + pre_get_list_files_sent_route,
|
|
headers=headers,
|
|
verify=False,
|
|
data={
|
|
"jsonHiddenComunes": "",
|
|
"ficheroJ": "",
|
|
"_csrf": token,
|
|
},
|
|
)
|
|
if response_pre_files_sent_list_route.status_code != 200:
|
|
raise MissingError(_("Could not get property_info."))
|
|
|
|
time.sleep(0.1)
|
|
|
|
soup = bs(response_pre_files_sent_list_route.text, "html.parser")
|
|
property_specific_data = {
|
|
"codigoHospederia": soup.select("#codigoHospederia")[0]["value"],
|
|
"nombreHospederia": soup.select("#nombreHospederia")[0]["value"],
|
|
"direccionCompleta": soup.select("#direccionCompleta")[0]["value"],
|
|
"telefono": soup.select("#telefono")[0]["value"],
|
|
"tieneAgrup": soup.select("#tieneAgrup")[0]["value"],
|
|
}
|
|
common_file_data = {
|
|
"jsonHiddenComunes": "",
|
|
"fechaDesde": (
|
|
datetime.date.today() + datetime.timedelta(days=-1)
|
|
).strftime("%d/%m/%Y"),
|
|
"fechaHasta": datetime.date.today().strftime("%d/%m/%Y"),
|
|
"_csrf": token,
|
|
"_search": False,
|
|
"nd": str(int(time.time() * 1000)),
|
|
"rows": 10,
|
|
"page": 1,
|
|
"sidx": "",
|
|
"sord": "dat_fich.fecha_alta desc",
|
|
}
|
|
# retrieve list of sent files
|
|
file_data = dict()
|
|
for _attempt in range(1, 10):
|
|
response_files_sent_list_route = session.post(
|
|
base_url + files_sent_list_route,
|
|
headers=headers,
|
|
verify=False,
|
|
data={
|
|
**property_specific_data,
|
|
**common_file_data,
|
|
"primeraConsulta": True,
|
|
},
|
|
)
|
|
file_list = json.loads(
|
|
str(bs(response_files_sent_list_route.text, "html.parser"))
|
|
)["rows"]
|
|
|
|
file_data = list(
|
|
filter(lambda x: x["nombreFichero"] == file_name, file_list)
|
|
)
|
|
if file_data:
|
|
file_data = file_data[0]
|
|
break
|
|
else:
|
|
time.sleep(1)
|
|
|
|
if not file_data:
|
|
raise ValidationError(_("Could not get last file sent"))
|
|
else:
|
|
response_last_file_errors_route = session.post(
|
|
base_url + last_file_errors_route,
|
|
headers=headers,
|
|
verify=False,
|
|
data={
|
|
"idFichero": file_data["idFichero"],
|
|
"numErroresHuespedes": file_data["numErroresHuespedes"],
|
|
"numAvisosHuespedes": file_data["numAvisosHuespedes"],
|
|
"nombreFichero": file_data["nombreFichero"],
|
|
"fechaAlta": file_data["fechaAlta"],
|
|
"envioDesdeAgrupacion": file_data["envioDesdeAgrupacion"],
|
|
"envioDesdeAgrupacionImg": "",
|
|
"estadoProceso": file_data["estadoProceso"],
|
|
"numTotalErrores": file_data["numTotalErrores"],
|
|
"numTotalAvisos": file_data["numTotalAvisos"],
|
|
"separadorTabla": "",
|
|
"numHuespedesInformados": file_data["numHuespedesInformados"],
|
|
"numHuespedes": file_data["numHuespedes"],
|
|
"primeraConsulta": False,
|
|
**property_specific_data,
|
|
**common_file_data,
|
|
"datosServidor": False,
|
|
},
|
|
)
|
|
|
|
if response_last_file_errors_route.status_code != 200:
|
|
raise ValidationError(_("Could last files sent"))
|
|
|
|
time.sleep(0.1)
|
|
soup = bs(response_last_file_errors_route.text, "html.parser")
|
|
# get file sent pdf report
|
|
response_last_file_errors_route = session.get(
|
|
base_url + soup.select("#iframePdf")[0].attrs["src"],
|
|
headers=headers,
|
|
verify=False,
|
|
)
|
|
|
|
if response_last_file_errors_route.status_code != 200:
|
|
raise ValidationError(_("Could last files sent"))
|
|
|
|
time.sleep(0.1)
|
|
pdfReader = PyPDF2.PdfFileReader(
|
|
io.BytesIO(response_last_file_errors_route.content)
|
|
)
|
|
|
|
if (
|
|
pdfReader.getPage(0).extractText().find("ERRORES Y AVISOS HUESPEDES")
|
|
== -1
|
|
):
|
|
message = _("Successful file sending")
|
|
error = False
|
|
else:
|
|
message = _("Errors (check the pdf file).")
|
|
error = True
|
|
|
|
self.env["pms.log.institution.traveller.report"].create(
|
|
{
|
|
"error_sending_data": error,
|
|
"txt_incidencies_from_institution": message,
|
|
"file_incidencies_from_institution": base64.b64encode(
|
|
response_last_file_errors_route.content
|
|
),
|
|
"txt_filename": file_name + ".pdf",
|
|
}
|
|
)
|
|
|
|
# do logout
|
|
session.post(
|
|
base_url + logout_route,
|
|
headers=headers,
|
|
verify=False,
|
|
data={"_csrf": token},
|
|
)
|
|
|
|
session.close()
|
|
if called_from_user:
|
|
message = {
|
|
"type": "ir.actions.client",
|
|
"tag": "display_notification",
|
|
"params": {
|
|
"title": _("Sent succesfully!"),
|
|
"message": _("Successful file sending"),
|
|
"sticky": False,
|
|
},
|
|
}
|
|
return message
|
|
|
|
def send_file_institution(self, pms_property=False):
|
|
called_from_user = False
|
|
if not pms_property:
|
|
called_from_user = True
|
|
pms_property = self.env["pms.property"].search(
|
|
[("id", "=", self.env.user.get_active_property_ids()[0])]
|
|
)
|
|
if (
|
|
not pms_property
|
|
or not pms_property.institution_property_id
|
|
or not pms_property.institution_user
|
|
or not pms_property.institution_password
|
|
):
|
|
raise ValidationError(
|
|
_("The guest information sending settings is not complete.")
|
|
)
|
|
file_content = self.generate_checkin_list(pms_property.id)
|
|
if pms_property.institution == "policia_nacional":
|
|
return self.send_file_pn(file_content, called_from_user, pms_property)
|
|
elif pms_property.institution == "guardia_civil":
|
|
return self.send_file_gc(file_content, called_from_user, pms_property)
|
|
|
|
@api.model
|
|
def send_file_institution_async(self):
|
|
for prop in self.env["pms.property"].search([]):
|
|
self.with_delay().send_file_institution(prop)
|