partner-contact/partner_helper/models/partner.py
2021-05-26 13:31:53 +02:00

49 lines
1.6 KiB
Python

# Copyright (C) 2016 Akretion (http://www.akretion.com)
# Author: Sébastien BEAU <sebastien.beau@akretion.com>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
from flectra import models
def split_char(char, output_number, size):
words = char.split(" ")
result = []
word = words.pop(0)
for index in range(0, output_number):
result.append(word)
word = ""
while len(words) > 0:
word = words.pop(0)
if len(result[index] + " %s" % word) > size:
break
else:
result[index] += " %s" % word
word = ""
return result
class ResPartner(models.Model):
_inherit = "res.partner"
def _get_split_address(self, output_number, max_size):
"""This method allows to get a number of street fields according to
your choice. Default is 2 large fields in Flectra (128 chars).
In some countries you may use 3 or 4 shorter street fields.
example:
res = self.partner_id._get_split_address(3, 35)
street1, street2, street3 = res
"""
self.ensure_one()
street = self.street or ""
street2 = self.street2 or ""
if len(street) <= max_size and len(street2) <= max_size:
result = ["" for i in range(0, output_number)]
result[0] = street
result[1] = street2
return result
elif len(street) <= max_size:
return [street] + split_char(street2, output_number - 1, max_size)
else:
return split_char("{} {}".format(street, street2), output_number, max_size)