partner-contact/partner_external_map/models/res_users.py
2021-09-12 02:12:03 +00:00

67 lines
2.2 KiB
Python

# Copyright 2015 Akretion (Alexis de Lattre <alexis.delattre@akretion.com>)
# Copyright 2016 Akretion (Alexis de Lattre <alexis.delattre@akretion.com>)
# Copyright 2016 Pedro M. Baeza <pedro.baeza@tecnativa.com>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
import logging
from flectra import api, fields, models
logger = logging.getLogger(__name__)
class ResUsers(models.Model):
_inherit = "res.users"
@api.model
def _default_map_website(self):
return self.env["map.website"].search(
["|", ("address_url", "!=", False), ("lat_lon_url", "!=", False)], limit=1
)
@api.model
def _default_route_map_website(self):
return self.env["map.website"].search(
[
"|",
("route_address_url", "!=", False),
("route_lat_lon_url", "!=", False),
],
limit=1,
)
# begin with context_ to allow user to change it by himself
context_map_website_id = fields.Many2one(
"map.website",
string="Map Website",
default=lambda self: self._default_map_website(),
domain=["|", ("address_url", "!=", False), ("lat_lon_url", "!=", False)],
)
# We want to give the possibility to the user to have one map provider for
# regular maps and another one for routing
context_route_map_website_id = fields.Many2one(
"map.website",
string="Route Map Website",
domain=[
"|",
("route_address_url", "!=", False),
("route_lat_lon_url", "!=", False),
],
default=lambda self: self._default_route_map_website(),
help="Map provided used when you click on the car icon on the partner "
"form to display an itinerary.",
)
context_route_start_partner_id = fields.Many2one(
"res.partner", string="Start Address for Route Map"
)
@api.model
def create(self, vals):
"""On creation, if no starting partner is provided, assign the current
created one.
"""
user = super(ResUsers, self).create(vals)
if not vals.get("context_route_start_partner_id"):
user.write({"context_route_start_partner_id": user.partner_id.id})
return user