from odoo import models, fields, api class ResPartner(models.Model): _inherit = 'res.partner' loyalty_card_ids = fields.One2many( 'loyalty.card', 'partner_id', string='Loyalty Cards' ) is_loyalty_member = fields.Boolean( string='Is Loyalty Member', compute='_compute_is_loyalty_member', ) @api.depends('loyalty_card_ids') def _compute_is_loyalty_member(self): for partner in self: partner.is_loyalty_member = bool(partner.loyalty_card_ids) @api.model def _load_pos_data_domain(self, data, config): domain = super()._load_pos_data_domain(data, config) # OR together the standard POS loaded partners (cashier, active orders) with all individual loyalty members return ['|'] + domain + ['&', ('is_company', '=', False), ('loyalty_card_ids', '!=', False)] @api.model def _load_pos_data_fields(self, config): fields = super()._load_pos_data_fields(config) if 'is_loyalty_member' not in fields: fields.append('is_loyalty_member') return fields @api.model def get_new_partner(self, config_id, domain, offset): # Limit active searches strictly to individual customers who are loyalty members domain.extend([('is_company', '=', False), ('loyalty_card_ids', '!=', False)]) return super().get_new_partner(config_id, domain, offset) @api.model_create_multi def create(self, vals_list): partners = super().create(vals_list) # Check context or stack trace to see if we are creating an internal user or employee import sys is_internal_creation = False frame = sys._getframe() while frame: filename = frame.f_code.co_filename if 'res_users.py' in filename or 'hr_employee.py' in filename: is_internal_creation = True break frame = frame.f_back # Also respect standard Odoo install and import bypass flags if is_internal_creation or self.env.context.get('install_mode') or self.env.context.get('import_file'): return partners for partner in partners: if partner.is_company: continue # Find the lowest membership program level (Membership Silver) lowest_program = self.env['loyalty.program'].sudo().search( [('multi_level_membership', '=', True), ('name', '=ilike', 'Membership Silver')], limit=1 ) if not lowest_program: lowest_program = self.env['loyalty.program'].sudo().search( [('multi_level_membership', '=', True), ('manual_membership', '=', False)], order='minimum_spend asc', limit=1 ) if lowest_program: existing_card = self.env['loyalty.card'].sudo().search([ ('partner_id', '=', partner.id), ('program_id', '=', lowest_program.id) ], limit=1) if not existing_card: self.env['loyalty.card'].sudo().create({ 'partner_id': partner.id, 'program_id': lowest_program.id, 'points': 0, }) if hasattr(partner, 'membership_level_id') and not partner.membership_level_id: partner.sudo().write({'membership_level_id': lowest_program.id}) return partners