pos_loyalty_member_custom/models/res_partner.py

107 lines
4.3 KiB
Python

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)
# Limit initial load of extra loyalty members to 30 to ensure fast POS startup (1-2 seconds)
loyalty_members = self.search(
[('is_company', '=', False), ('loyalty_card_ids', '!=', False)],
limit=30,
order='write_date desc'
)
return ['|'] + domain + [('id', 'in', loyalty_members.ids)]
@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)
# Skip auto-Silver assignment for system/internal partner creations.
# Internal callers (res_users, hr_employee, etc.) should pass
# context key 'no_loyalty_auto_assign=True' to suppress this logic.
if (
self.env.context.get('install_mode')
or self.env.context.get('import_file')
or self.env.context.get('no_loyalty_auto_assign')
):
return partners
for partner in partners:
if partner.is_company:
continue
# If membership_level_id was explicitly set in the create vals
# (e.g. by the XMLRPC migration script), skip auto-Silver entirely.
# The caller is responsible for creating the correct loyalty card.
if partner.membership_level_id:
continue
# Find Membership Silver first, then fall back to lowest spend program
lowest_program = self.env['loyalty.program'].sudo().search(
[('multi_level_membership', '=', True), ('manual_membership', '=', False), ('name', '=ilike', 'Membership Silver')],
limit=1
)
if not lowest_program:
# Fallback: pick the non-manual multi-level program with the lowest
# minimum_spend. We require minimum_spend > 0 to exclude programs like
# "Membership Direksi" that have no spend requirement (by-invitation only)
# and would otherwise be picked first due to minimum_spend defaulting to 0.
lowest_program = self.env['loyalty.program'].sudo().search(
[
('multi_level_membership', '=', True),
('manual_membership', '=', False),
('minimum_spend', '>', 0),
],
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,
})
# Always write membership_level_id if not already set.
# pos_loyalty_multi_level is a hard dependency so the field always exists.
if not partner.membership_level_id:
partner.sudo().write({'membership_level_id': lowest_program.id})
return partners