mirror of
https://github.com/OCA/pms.git
synced 2025-01-29 00:17:45 +02:00
589 lines
23 KiB
Python
589 lines
23 KiB
Python
from datetime import datetime
|
|
|
|
from odoo.addons.base_rest import restapi
|
|
from odoo.addons.base_rest_datamodel.restapi import Datamodel
|
|
from odoo.addons.component.core import Component
|
|
|
|
|
|
class PmsPartnerService(Component):
|
|
_inherit = "base.rest.service"
|
|
_name = "pms.partner.service"
|
|
_usage = "partners"
|
|
_collection = "pms.services"
|
|
|
|
@restapi.method(
|
|
[
|
|
(
|
|
[
|
|
"/",
|
|
],
|
|
"GET",
|
|
)
|
|
],
|
|
input_param=Datamodel("pms.partner.search.param", is_list=False),
|
|
output_param=Datamodel("pms.partner.info", is_list=True),
|
|
auth="jwt_api_pms",
|
|
)
|
|
def get_partners(self, pms_partner_search_params):
|
|
result_partners = []
|
|
domain = []
|
|
PmsPartnerInfo = self.env.datamodels["pms.partner.info"]
|
|
for partner in self.env["res.partner"].search(domain):
|
|
checkouts = (
|
|
self.env["pms.checkin.partner"]
|
|
.search([("partner_id.id", "=", partner.id)])
|
|
.mapped("checkout")
|
|
)
|
|
result_partners.append(
|
|
PmsPartnerInfo(
|
|
id=partner.id,
|
|
name=partner.name if partner.name else None,
|
|
firstname=partner.firstname if partner.firstname else None,
|
|
lastname=partner.lastname if partner.lastname else None,
|
|
lastname2=partner.lastname2 if partner.lastname2 else None,
|
|
email=partner.email if partner.email else None,
|
|
phone=partner.phone if partner.phone else None,
|
|
gender=partner.gender if partner.gender else None,
|
|
birthdate=datetime.combine(
|
|
partner.birthdate_date, datetime.min.time()
|
|
).isoformat()
|
|
if partner.birthdate_date
|
|
else None,
|
|
age=partner.age if partner.age else None,
|
|
residenceStreet=partner.residence_street
|
|
if partner.residence_street
|
|
else None,
|
|
residenceStreet2=partner.residence_street2
|
|
if partner.residence_street2
|
|
else None,
|
|
residenceZip=partner.residence_zip
|
|
if partner.residence_zip
|
|
else None,
|
|
residenceCity=partner.residence_city
|
|
if partner.residence_city
|
|
else None,
|
|
nationality=partner.nationality_id.id
|
|
if partner.nationality_id
|
|
else None,
|
|
residenceStateId=partner.residence_state_id.id
|
|
if partner.residence_state_id
|
|
else None,
|
|
street=partner.street if partner.street else None,
|
|
street2=partner.street2 if partner.street2 else None,
|
|
zip=partner.zip if partner.zip else None,
|
|
countryId=partner.country_id.id if partner.country_id else None,
|
|
stateId=partner.state_id.id if partner.state_id else None,
|
|
city=partner.city if partner.city else None,
|
|
isAgency=partner.is_agency,
|
|
isCompany=partner.is_company,
|
|
residenceCountryId=partner.residence_country_id.id
|
|
if partner.residence_country_id
|
|
else None,
|
|
vatNumber=partner.vat if partner.vat else None,
|
|
vatDocumentType=partner.vat_document_type
|
|
if partner.vat_document_type
|
|
else None,
|
|
comment=partner.comment if partner.comment else None,
|
|
language=partner.lang if partner.lang else None,
|
|
userId=partner.user_id if partner.user_id else None,
|
|
paymentTerms=partner.property_payment_term_id
|
|
if partner.property_payment_term_id
|
|
else None,
|
|
pricelistId=partner.property_product_pricelist
|
|
if partner.property_product_pricelist
|
|
else None,
|
|
salesReference=partner.ref if partner.ref else None,
|
|
saleChannelId=partner.sale_channel_id
|
|
if partner.sale_channel_id
|
|
else None,
|
|
commission=partner.default_commission
|
|
if partner.default_commission
|
|
else None,
|
|
invoicingPolicy=partner.invoicing_policy
|
|
if partner.invoicing_policy
|
|
else None,
|
|
daysAutoInvoice=partner.margin_days_autoinvoice
|
|
if partner.margin_days_autoinvoice
|
|
else None,
|
|
invoicingMonthDay=partner.invoicing_month_day
|
|
if partner.invoicing_month_day
|
|
else None,
|
|
invoiceToAgency=partner.invoice_to_agency
|
|
if partner.invoice_to_agency
|
|
else None,
|
|
agencyStateId=partner.state_id.id if partner.state_id else None,
|
|
agencyCity=partner.city if partner.city else None,
|
|
tagIds=partner.category_id.ids if partner.category_id else [],
|
|
lastStay=max(checkouts).strftime("%d/%m/%Y") if checkouts else "",
|
|
)
|
|
)
|
|
return result_partners
|
|
|
|
@restapi.method(
|
|
[
|
|
(
|
|
[
|
|
"/",
|
|
],
|
|
"POST",
|
|
)
|
|
],
|
|
input_param=Datamodel("pms.partner.info", is_list=False),
|
|
auth="jwt_api_pms",
|
|
)
|
|
def create_partner(self, partner_info):
|
|
vals = self.mapping_partner_values(partner_info)
|
|
self.env["res.partner"].create(vals)
|
|
|
|
|
|
@restapi.method(
|
|
[
|
|
(
|
|
[ "/<int:partner_id>/hosted-reservations",
|
|
],
|
|
"GET",
|
|
)
|
|
],
|
|
output_param=Datamodel("pms.reservation.short.info", is_list=True),
|
|
auth="jwt_api_pms",
|
|
)
|
|
def get_partner_as_host(self, partner_id):
|
|
checkins = self.env["pms.checkin.partner"].search(
|
|
[("partner_id", "=", partner_id)]
|
|
)
|
|
PmsReservationShortInfo = self.env.datamodels["pms.reservation.short.info"]
|
|
reservations = []
|
|
if checkins:
|
|
for checkin in checkins:
|
|
reservation = self.env["pms.reservation"].search(
|
|
[("id", "=", checkin.reservation_id.id)]
|
|
)
|
|
folio = self.env["pms.folio"].search(
|
|
[("id", "=", reservation.folio_id.id)]
|
|
)
|
|
reservations.append(
|
|
PmsReservationShortInfo(
|
|
id=reservation.id,
|
|
checkin=reservation.checkin.strftime("%d/%m/%Y"),
|
|
checkout=reservation.checkout.strftime("%d/%m/%Y"),
|
|
adults=reservation.adults,
|
|
priceTotal=round(reservation.price_room_services_set, 2),
|
|
stateDescription=dict(
|
|
reservation.fields_get(["state"])["state"]["selection"]
|
|
)[reservation.state],
|
|
paymentStateDescription=dict(
|
|
folio.fields_get(["payment_state"])["payment_state"][
|
|
"selection"
|
|
]
|
|
)[folio.payment_state],
|
|
)
|
|
)
|
|
return reservations
|
|
|
|
@restapi.method(
|
|
[
|
|
(
|
|
[
|
|
"/<int:partner_id>",
|
|
],
|
|
"PATCH",
|
|
)
|
|
],
|
|
input_param=Datamodel("pms.partner.info", is_list=False),
|
|
auth="jwt_api_pms",
|
|
)
|
|
def write_partner(self, partner_id, partner_info):
|
|
partner = self.env["res.partner"].browse(partner_id)
|
|
if partner:
|
|
partner.write(self.mapping_partner_values(partner_info))
|
|
|
|
# REVIEW: analyze in which service file this method should be
|
|
|
|
|
|
@restapi.method(
|
|
[
|
|
(
|
|
[ "/<int:partner_id>/customer-reservations",
|
|
],
|
|
"GET",
|
|
)
|
|
],
|
|
output_param=Datamodel("pms.reservation.short.info", is_list=True),
|
|
auth="jwt_api_pms",
|
|
)
|
|
def get_partner_as_customer(self, partner_id):
|
|
partnerReservations = self.env["pms.reservation"].search(
|
|
[("partner_id", "=", partner_id)]
|
|
)
|
|
PmsReservationShortInfo = self.env.datamodels["pms.reservation.short.info"]
|
|
reservations = []
|
|
for reservation in partnerReservations:
|
|
folio = self.env["pms.folio"].search([("id", "=", reservation.folio_id.id)])
|
|
reservations.append(
|
|
PmsReservationShortInfo(
|
|
checkin=datetime.combine(
|
|
reservation.checkin, datetime.min.time()
|
|
).isoformat(),
|
|
checkout=datetime.combine(
|
|
reservation.checkout, datetime.min.time()
|
|
).isoformat(),
|
|
adults=reservation.adults,
|
|
priceTotal=round(reservation.price_room_services_set, 2),
|
|
stateDescription=dict(
|
|
reservation.fields_get(["state"])["state"]["selection"]
|
|
)[reservation.state],
|
|
paymentStateDescription=dict(
|
|
folio.fields_get(["payment_state"])["payment_state"][
|
|
"selection"
|
|
]
|
|
)[folio.payment_state],
|
|
)
|
|
)
|
|
return reservations
|
|
|
|
@restapi.method(
|
|
[
|
|
(
|
|
[
|
|
"/<int:partner_id>/payments",
|
|
],
|
|
"GET",
|
|
)
|
|
],
|
|
output_param=Datamodel("pms.payment.info", is_list=True),
|
|
auth="jwt_api_pms",
|
|
)
|
|
def get_partner_payments(self, partner_id):
|
|
partnerPayments = self.env["account.payment"].search(
|
|
[("partner_id", "=", partner_id)]
|
|
)
|
|
PmsPaymentInfo = self.env.datamodels["pms.payment.info"]
|
|
payments = []
|
|
for payment in partnerPayments:
|
|
payments.append(
|
|
PmsPaymentInfo(
|
|
id=payment.id,
|
|
amount=round(payment.amount, 2),
|
|
journalId=payment.journal_id.id,
|
|
date=payment.date.strftime("%d/%m/%Y"),
|
|
memo=payment.ref,
|
|
)
|
|
)
|
|
return payments
|
|
|
|
@restapi.method(
|
|
[
|
|
(
|
|
[
|
|
"/<int:partner_id>/invoices",
|
|
],
|
|
"GET",
|
|
)
|
|
],
|
|
output_param=Datamodel("pms.account.move.info", is_list=True),
|
|
auth="jwt_api_pms",
|
|
)
|
|
def get_partner_invoices(self, partner_id):
|
|
partnerInvoices = self.env["account.move"].search(
|
|
[("partner_id", "=", partner_id)]
|
|
)
|
|
PmsAcoountMoveInfo = self.env.datamodels["pms.account.move.info"]
|
|
invoices = []
|
|
for invoice in partnerInvoices:
|
|
invoices.append(
|
|
PmsAcoountMoveInfo(
|
|
id=invoice.id,
|
|
name=invoice.name,
|
|
amount=round(invoice.amount_total, 2),
|
|
date=invoice.date.strftime("%d/%m/%Y"),
|
|
state=invoice.state,
|
|
paymentState=invoice.payment_state
|
|
if invoice.payment_state
|
|
else None,
|
|
)
|
|
)
|
|
return invoices
|
|
|
|
@restapi.method(
|
|
[
|
|
(
|
|
[
|
|
"/<string:documentType>/<string:documentNumber>",
|
|
],
|
|
"GET",
|
|
)
|
|
],
|
|
output_param=Datamodel("pms.partner.info", is_list=True),
|
|
auth="jwt_api_pms",
|
|
)
|
|
def get_partner_by_doc_number(self, document_type, document_number):
|
|
doc_number = self.env["res.partner.id_number"].search(
|
|
[("name", "=", document_number), ("category_id", "=", int(document_type))]
|
|
)
|
|
partners = []
|
|
PmsPartnerInfo = self.env.datamodels["pms.partner.info"]
|
|
if not doc_number:
|
|
pass
|
|
else:
|
|
if doc_number.valid_from:
|
|
document_expedition_date = doc_number.valid_from.strftime("%d/%m/%Y")
|
|
if doc_number.partner_id.birthdate_date:
|
|
birthdate_date = doc_number.partner_id.birthdate_date.strftime(
|
|
"%d/%m/%Y"
|
|
)
|
|
partners.append(
|
|
PmsPartnerInfo(
|
|
# id=doc_number.partner_id.id,
|
|
name=doc_number.partner_id.name
|
|
if doc_number.partner_id.name
|
|
else None,
|
|
firstname=doc_number.partner_id.firstname
|
|
if doc_number.partner_id.firstname
|
|
else None,
|
|
lastname=doc_number.partner_id.lastname
|
|
if doc_number.partner_id.lastname
|
|
else None,
|
|
lastname2=doc_number.partner_id.lastname2
|
|
if doc_number.partner_id.lastname2
|
|
else None,
|
|
email=doc_number.partner_id.email
|
|
if doc_number.partner_id.email
|
|
else None,
|
|
mobile=doc_number.partner_id.mobile
|
|
if doc_number.partner_id.mobile
|
|
else None,
|
|
documentType=int(document_type),
|
|
documentNumber=doc_number.name,
|
|
documentExpeditionDate=document_expedition_date
|
|
if doc_number.valid_from
|
|
else None,
|
|
documentSupportNumber=doc_number.support_number
|
|
if doc_number.support_number
|
|
else None,
|
|
gender=doc_number.partner_id.gender
|
|
if doc_number.partner_id.gender
|
|
else None,
|
|
birthdate=birthdate_date
|
|
if doc_number.partner_id.birthdate_date
|
|
else None,
|
|
residenceStreet=doc_number.partner_id.residence_street
|
|
if doc_number.partner_id.residence_street
|
|
else None,
|
|
zip=doc_number.partner_id.residence_zip
|
|
if doc_number.partner_id.residence_zip
|
|
else None,
|
|
residenceCity=doc_number.partner_id.residence_city
|
|
if doc_number.partner_id.residence_city
|
|
else None,
|
|
nationality=doc_number.partner_id.nationality_id.id
|
|
if doc_number.partner_id.nationality_id
|
|
else None,
|
|
countryState=doc_number.partner_id.residence_state_id.id
|
|
if doc_number.partner_id.residence_state_id
|
|
else None,
|
|
)
|
|
)
|
|
return partners
|
|
|
|
@restapi.method(
|
|
[
|
|
(
|
|
[
|
|
"/<int:partner_id>/deactivate",
|
|
],
|
|
"PATCH",
|
|
)
|
|
],
|
|
auth="jwt_api_pms",
|
|
)
|
|
def deactivate_partner(self, partner_id):
|
|
self.env["res.partner"].browse(partner_id).active = False
|
|
|
|
@restapi.method(
|
|
[
|
|
(
|
|
[
|
|
"/<int:partner_id>/activate",
|
|
],
|
|
"PATCH",
|
|
)
|
|
],
|
|
auth="jwt_api_pms",
|
|
)
|
|
def activate_partner(self, partner_id):
|
|
self.env["res.partner"].browse(partner_id).active = True
|
|
|
|
@restapi.method(
|
|
[
|
|
(
|
|
[
|
|
"/partner",
|
|
],
|
|
"GET",
|
|
)
|
|
],
|
|
input_param=Datamodel("pms.partner.search.param", is_list=False),
|
|
output_param=Datamodel("pms.partner.info", is_list=False),
|
|
auth="jwt_api_pms",
|
|
)
|
|
def get_partner(self, pms_partner_search_params):
|
|
domain = []
|
|
PmsPartnerInfo = self.env.datamodels["pms.partner.info"]
|
|
if pms_partner_search_params.vatNumber:
|
|
domain = [
|
|
"|",
|
|
("vat", "=", pms_partner_search_params.vatNumber),
|
|
("aeat_identification", "=", pms_partner_search_params.vatNumber),
|
|
]
|
|
partner = self.env["res.partner"].search(domain)
|
|
if not partner:
|
|
return PmsPartnerInfo()
|
|
else:
|
|
return PmsPartnerInfo(
|
|
id=partner.id,
|
|
name=partner.name if partner.name else None,
|
|
firstname=partner.firstname if partner.firstname else None,
|
|
lastname=partner.lastname if partner.lastname else None,
|
|
lastname2=partner.lastname2 if partner.lastname2 else None,
|
|
email=partner.email if partner.email else None,
|
|
mobile=partner.mobile if partner.mobile else None,
|
|
phone=partner.phone if partner.phone else None,
|
|
gender=partner.gender if partner.gender else None,
|
|
birthdate=datetime.combine(
|
|
partner.birthdate_date, datetime.min.time()
|
|
).isoformat()
|
|
if partner.birthdate_date
|
|
else None,
|
|
age=partner.age if partner.age else None,
|
|
residenceStreet=partner.residence_street
|
|
if partner.residence_street
|
|
else None,
|
|
residenceStreet2=partner.residence_street2
|
|
if partner.residence_street2
|
|
else None,
|
|
residenceZip=partner.residence_zip if partner.residence_zip else None,
|
|
residenceCity=partner.residence_city
|
|
if partner.residence_city
|
|
else None,
|
|
nationality=partner.nationality_id.id
|
|
if partner.nationality_id
|
|
else None,
|
|
residenceStateId=partner.residence_state_id.id
|
|
if partner.residence_state_id
|
|
else None,
|
|
street=partner.street if partner.street else None,
|
|
street2=partner.street2 if partner.street2 else None,
|
|
zip=partner.zip if partner.zip else None,
|
|
countryId=partner.country_id.id if partner.country_id else None,
|
|
stateId=partner.state_id.id if partner.state_id else None,
|
|
city=partner.city if partner.city else None,
|
|
isAgency=partner.is_agency,
|
|
isCompany=partner.is_company,
|
|
residenceCountryId=partner.residence_country_id.id
|
|
if partner.residence_country_id
|
|
else None,
|
|
vatNumber=partner.vat if partner.vat else None,
|
|
vatDocumentType=partner.vat_document_type
|
|
if partner.vat_document_type
|
|
else None,
|
|
comment=partner.comment if partner.comment else None,
|
|
language=partner.lang if partner.lang else None,
|
|
userId=partner.user_id if partner.user_id else None,
|
|
paymentTerms=partner.property_payment_term_id
|
|
if partner.property_payment_term_id
|
|
else None,
|
|
pricelistId=partner.property_product_pricelist
|
|
if partner.property_product_pricelist
|
|
else None,
|
|
salesReference=partner.ref if partner.ref else None,
|
|
saleChannelId=partner.sale_channel_id
|
|
if partner.sale_channel_id
|
|
else None,
|
|
commission=partner.default_commission
|
|
if partner.default_commission
|
|
else None,
|
|
invoicingPolicy=partner.invoicing_policy
|
|
if partner.invoicing_policy
|
|
else None,
|
|
daysAutoInvoice=partner.margin_days_autoinvoice
|
|
if partner.margin_days_autoinvoice
|
|
else None,
|
|
invoicingMonthDay=partner.invoicing_month_day
|
|
if partner.invoicing_month_day
|
|
else None,
|
|
invoiceToAgency=partner.invoice_to_agency
|
|
if partner.invoice_to_agency
|
|
else None,
|
|
)
|
|
|
|
def mapping_partner_values(self, pms_partner_info):
|
|
vals = dict()
|
|
partner_fields = {
|
|
"name": pms_partner_info.name,
|
|
"firstname": pms_partner_info.firstname,
|
|
"lastname": pms_partner_info.lastname,
|
|
"lastname2": pms_partner_info.lastname2,
|
|
"email": pms_partner_info.email,
|
|
"mobile": pms_partner_info.mobile,
|
|
"phone": pms_partner_info.phone,
|
|
"gender": pms_partner_info.gender,
|
|
"residence_street": pms_partner_info.residenceStreet,
|
|
"residence_street2": pms_partner_info.residenceStreet2,
|
|
"nationality_id": pms_partner_info.nationality,
|
|
"residence_zip": pms_partner_info.zip,
|
|
"residence_city": pms_partner_info.residenceCity,
|
|
"residence_state_id": pms_partner_info.residenceStateId,
|
|
"residence_country_id": pms_partner_info.residenceCountryId,
|
|
"is_agency": pms_partner_info.isAgency,
|
|
"is_company": pms_partner_info.isCompany,
|
|
"street": pms_partner_info.street,
|
|
"street2": pms_partner_info.street2,
|
|
"zip": pms_partner_info.zip,
|
|
"city": pms_partner_info.city,
|
|
"state_id": pms_partner_info.stateId,
|
|
"country_id": pms_partner_info.countryId,
|
|
"user_id": pms_partner_info.userId,
|
|
"lang": pms_partner_info.language,
|
|
"comment": pms_partner_info.comment,
|
|
"property_payment_term_id": pms_partner_info.paymentTerms,
|
|
"property_product_pricelist": pms_partner_info.pricelistId,
|
|
"ref": pms_partner_info.salesReference,
|
|
"sale_channel_id": pms_partner_info.saleChannelId,
|
|
"default_commission": pms_partner_info.commission,
|
|
"invoicing_policy": pms_partner_info.invoicingPolicy,
|
|
"margin_days_autoinvoice": pms_partner_info.daysAutoInvoice,
|
|
"invoicing_month_day": pms_partner_info.invoicingMonthDay,
|
|
"invoice_to_agency": pms_partner_info.invoiceToAgency,
|
|
}
|
|
|
|
if (
|
|
pms_partner_info.isAgency
|
|
or pms_partner_info.isCompany
|
|
or (
|
|
pms_partner_info.vatDocumentType == "02"
|
|
or pms_partner_info.vatDocumentType == "04"
|
|
)
|
|
):
|
|
partner_fields.update(
|
|
{
|
|
"vat": pms_partner_info.vatNumber,
|
|
"vat_document_type": "vat",
|
|
}
|
|
)
|
|
else:
|
|
partner_fields.update(
|
|
{
|
|
"aeat_identification_type": pms_partner_info.vatDocumentType,
|
|
"aeat_identification": pms_partner_info.vatNumber,
|
|
}
|
|
)
|
|
|
|
if pms_partner_info.birthdate:
|
|
birthdate = datetime.strptime(pms_partner_info.birthdate, "%d/%m/%Y")
|
|
birthdate = birthdate.strftime("%Y-%m-%d")
|
|
vals.update({"birthdate_date": birthdate})
|
|
for k, v in partner_fields.items():
|
|
if v:
|
|
vals.update({k: v})
|
|
return vals
|