# Copyright 2016 Nicolas Bessi, Camptocamp SA # Copyright 2018 Aitor Bouzas # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl). from flectra import api, fields, models class ResCityZip(models.Model): """City/locations completion object""" _name = "res.city.zip" _description = __doc__ _order = "name asc" _rec_names_search = ["name", "city_id", "state_id", "country_id"] name = fields.Char("ZIP", required=True) city_id = fields.Many2one( "res.city", "City", required=True, auto_join=True, ondelete="cascade", index=True, ) state_id = fields.Many2one(related="city_id.state_id") country_id = fields.Many2one(related="city_id.country_id") _sql_constraints = [ ( "name_city_uniq", "UNIQUE(name, city_id)", "You already have a zip with that code in the same city. " "The zip code must be unique within it's city", ) ] @api.depends( "name", "city_id", "city_id.name", "city_id.state_id.name", "city_id.state_id", "city_id.country_id", "city_id.country_id.name", ) def _compute_display_name(self): """Get the proper display name formatted as 'ZIP, name, state, country'.""" for rec in self: state_name = ( rec.city_id.state_id.name + ", " if rec.city_id.state_id else "" ) country_name = rec.city_id.country_id.name if rec.city_id.country_id else "" rec.display_name = ( f"{rec.name}, {rec.city_id.name}, " f"{state_name}{country_name}" )