audrius/pms
1
0
Fork 1
mirror of https://github.com/OCA/pms.git synced 2025-01-29 00:17:45 +02:00
Code Issues Packages Projects Releases Wiki Activity

[IMP] pms: add logout & connection test

Browse Source
This commit is contained in:
miguelpadin
2021-05-25 19:50:25 +02:00
parent e8f10126ad
commit 7c5d18ee03
3 changed files with 148 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

69
pms_l10n_es/models/pms_property.py
View File

@@ -1,4 +1,9 @@
from odoo import fields, models
import requests
from bs4 import BeautifulSoup as bs
from odoo import fields, models, _
from odoo.exceptions import ValidationError
from odoo.modules import get_module_resource
class PmsProperty(models.Model):
@@ -27,4 +32,64 @@ class PmsProperty(models.Model):
string="Institution password",
help="Password provided by institution to send the data.",
)
sequence_id = fields.Many2one("ir.sequence")
def test_gc_connection(self):
for pms_property in self:
if (
pms_property.institution == "guardia_civil"
and pms_property.institution_property_id
and pms_property.institution_user
and pms_property.institution_password
):
url = "https://hospederias.guardiacivil.es/"
login_route = "/hospederias/login.do"
logout_route = "/hospederias/logout.do"
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,
}
# login
response_login = session.post(
url + login_route,
headers=headers,
data=login_payload,
verify=get_module_resource("pms_l10n_es", "static", "cert.pem"),
)
# logout
session.get(
url + logout_route,
headers=headers,
verify=get_module_resource("pms_l10n_es", "static", "cert.pem"),
)
session.close()
# check if authentication was successful / unsuccessful or the
# resource cannot be accessed
soup = bs(response_login.text, "html.parser")
errors = soup.select("#txterror > ul > li")
if errors:
raise ValidationError(errors[0].text)
else:
login_correct = soup.select(".cabecera2")
if login_correct:
message = {
"type": "ir.actions.client",
"tag": "display_notification",
"params": {
"title": _("Connection Established!"),
"message": _("Connection established succesfully"),
"sticky": False,
},
}
return message
else:
raise ValidationError(_("Connection could not be established"))

32
pms_l10n_es/views/pms_property_views.xml
View File

@@ -8,12 +8,32 @@
<field name="inherit_id" ref="pms.pms_property_views_form" />
<field name="arch" type="xml">
<xpath expr="//page[@name='property_settings']" position="inside">
<group name="property_data" string="Settings of guest information">
<field name="institution" />
<field name="institution_property_id" />
<field name="institution_user" />
<field name="institution_password" />
</group>
<div class="row">
<div class="col-12">
<div class="o_horizontal_separator">
Guest information sending settings
</div>
</div>
<div class="col-6">
<group name="property_data">
<field name="institution" />
<field name="institution_property_id" />
</group>
</div>
<div class="col-6 px-0">
<group name="property_data">
<field name="institution_user" />
<field name="institution_password" password="True" />
</group>
<button
name="test_gc_connection"
class="btn btn-primary btn-sm"
type="object"
string="Test user/password"
/>
</div>
</div>
</xpath>
</field>
</record>

87
pms_l10n_es/wizards/traveller_report.py
View File

@@ -29,40 +29,36 @@ class TravellerReport(models.TransientModel):
# build content
content = self.generate_checkin_list(pms_property.id)
# file creation
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,
if content:
# file creation
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",
}
)
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"),
("arrival", "<=", str(date.today()) + " 23:59:59"),
]
)
== 0
if self.env["pms.checkin.partner"].search_count(
[
("state", "=", "onboard"),
("arrival", ">=", str(date.today()) + " 0:00:00"),
("arrival", "<=", str(date.today()) + " 23:59:59"),
]
):
raise ValidationError(_("There's no guests info to send"))
else:
# get the active property
pms_property = self.env["pms.property"].search([("id", "=", property_id)])
@@ -133,6 +129,7 @@ class TravellerReport(models.TransientModel):
url = "https://hospederias.guardiacivil.es/"
login_route = "/hospederias/login.do"
upload_file_route = "/hospederias/cargaFichero.do"
logout_route = "/hospederias/logout.do"
called_from_user = False
if not pms_property:
called_from_user = True
@@ -146,39 +143,65 @@ class TravellerReport(models.TransientModel):
"Build/MRA58N) AppleWebKit/537.36 (KHTML, like "
"Gecko) Chrome/90.0.4430.93 Mobile Safari/537.36",
}
s = requests.session()
session = requests.session()
login_payload = {
"usuario": pms_property.institution_user,
"pswd": pms_property.institution_password,
}
s.post(
# login
response_login = session.post(
url + login_route,
headers=headers,
data=login_payload,
verify=get_module_resource("pms_l10n_es", "static", "cert.pem"),
)
# 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
pwd = get_module_resource("pms_l10n_es", "wizards", "")
checkin_list_file = open(pwd + pms_property.institution_user + ".999", "w+")
checkin_list_file.write(self.generate_checkin_list(pms_property.id))
checkin_list_file.close()
files = {"fichero": open(pwd + pms_property.institution_user + ".999", "rb")}
response_file_sent = s.post(
# 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"),
)
# remove file locally
os.remove(pwd + pms_property.institution_user + ".999")
s.close()
# 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 > td > a")
if errors:
raise ValidationError(errors[2].text)
else:
if called_from_user:
# TODO: dont show notification when cannot be logged
message = {
"type": "ir.actions.client",
"tag": "display_notification",