Files
pms/pms_api_rest/services/pms_partner_service.py

362 lines
15 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):
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,
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,
tagIds=partner.category_id.ids if partner.category_id else [],
vatNumber=partner.vat if partner.vat else None,
vatDocumentType=partner.vat_document_type
if partner.vat_document_type
else None,
website=partner.website if partner.website 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,
)
)
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)
partner = self.env["res.partner"].create(vals)
if partner_info.documentNumber:
doc_type_id = (
self.env["res.partner.id_category"].search([("name", "=", "DNI")]).id
)
self.env["res.partner.id_number"].create(
{
"partner_id": partner.id,
"category_id": doc_type_id,
"name": partner_info.documentNumber,
}
)
@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))
if partner_info.documentNumber:
doc_type_id = (
self.env["res.partner.id_category"].search([("name", "=", "DNI")]).id
)
doc_number = self.env["res.partner.id_number"].search(
[
("partner_id", "=", partner_id),
("name", "=", partner_info.documentNumber),
("category_id", "=", doc_type_id),
]
)
if not doc_number:
self.env["res.partner.id_number"].create(
{
"category_id": doc_type_id,
"name": partner_info.documentNumber,
}
)
else:
doc_number.write(
{
"name": partner_info.documentNumber,
}
)
@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.checkin.partner.info", is_list=False),
auth="jwt_api_pms",
)
def get_partner(self, pms_partner_search_params):
domain = []
# if pms_partner_search_params.documentType
# and pms_partner_search_params.documentNumber:
# doc_number = self.env["res.partner.id_number"].search(
# [
# ("name", "=", pms_partner_search_params.documentNumber),
# ("category_id", "=", pms_partner_search_params.documentType)
# ]
# )
# if doc_number.valid_from:
# document_expedition_date = doc_number.valid_from.strftime("%d/%m/%Y")
# document_type = pms_partner_search_params.documentType
# document_number = pms_partner_search_params.documentNumber
# domain.append(('id_numbers', 'in', doc_number))
PmsPartnerInfo = self.env.datamodels["pms.partner.info"]
if pms_partner_search_params.vatNumber:
domain.append(("vat", "=", pms_partner_search_params.vatNumber))
partner = self.env["res.partner"].search(domain)
if len(partner) > 1:
partner = partner.filtered("is_company")
if partner:
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,
tagIds=partner.category_id.ids if partner.category_id else [],
vatNumber=partner.vat if partner.vat else None,
vatDocumentType=partner.vat_document_type
if partner.vat_document_type
else None,
website=partner.website if partner.website 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,
)
else:
return []
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,
"vat": pms_partner_info.vatNumber,
"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,
}
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