76 lines
3.2 KiB
Python
76 lines
3.2 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
||
|
|
|
||
|
|
from odoo import _, api, fields, models
|
||
|
|
from odoo.exceptions import ValidationError
|
||
|
|
from odoo.osv import expression
|
||
|
|
|
||
|
|
class LoyaltyGenerateWizard(models.TransientModel):
|
||
|
|
_name = 'loyalty.generate.wizard'
|
||
|
|
_description = 'Generate Coupons'
|
||
|
|
|
||
|
|
program_id = fields.Many2one('loyalty.program', required=True, default=lambda self: self.env.context.get('active_id', False) or self.env.context.get('default_program_id', False))
|
||
|
|
program_type = fields.Selection(related='program_id.program_type')
|
||
|
|
|
||
|
|
mode = fields.Selection([
|
||
|
|
('anonymous', 'Anonymous Customers'),
|
||
|
|
('selected', 'Selected Customers')],
|
||
|
|
string='For', required=True, default='anonymous'
|
||
|
|
)
|
||
|
|
|
||
|
|
customer_ids = fields.Many2many('res.partner', string='Customers')
|
||
|
|
customer_tag_ids = fields.Many2many('res.partner.category', string='Customer Tags')
|
||
|
|
|
||
|
|
coupon_qty = fields.Integer('Quantity',
|
||
|
|
compute='_compute_coupon_qty', readonly=False, store=True)
|
||
|
|
points_granted = fields.Float('Grant', required=True, default=1)
|
||
|
|
points_name = fields.Char(related='program_id.portal_point_name', readonly=True)
|
||
|
|
valid_until = fields.Date()
|
||
|
|
will_send_mail = fields.Boolean(compute='_compute_will_send_mail')
|
||
|
|
|
||
|
|
def _get_partners(self):
|
||
|
|
self.ensure_one()
|
||
|
|
if self.mode != 'selected':
|
||
|
|
return self.env['res.partner']
|
||
|
|
domain = []
|
||
|
|
if self.customer_ids:
|
||
|
|
domain = [('id', 'in', self.customer_ids.ids)]
|
||
|
|
if self.customer_tag_ids:
|
||
|
|
domain = expression.OR([domain, [('category_id', 'in', self.customer_tag_ids.ids)]])
|
||
|
|
return self.env['res.partner'].search(domain)
|
||
|
|
|
||
|
|
@api.depends('customer_ids', 'customer_tag_ids', 'mode')
|
||
|
|
def _compute_coupon_qty(self):
|
||
|
|
for wizard in self:
|
||
|
|
if wizard.mode == 'selected':
|
||
|
|
wizard.coupon_qty = len(wizard._get_partners())
|
||
|
|
else:
|
||
|
|
wizard.coupon_qty = wizard.coupon_qty or 0
|
||
|
|
|
||
|
|
@api.depends("mode", "program_id")
|
||
|
|
def _compute_will_send_mail(self):
|
||
|
|
for wizard in self:
|
||
|
|
wizard.will_send_mail = wizard.mode == 'selected' and 'create' in wizard.program_id.mapped('communication_plan_ids.trigger')
|
||
|
|
|
||
|
|
def _get_coupon_values(self, partner):
|
||
|
|
self.ensure_one()
|
||
|
|
return {
|
||
|
|
'program_id': self.program_id.id,
|
||
|
|
'points': self.points_granted,
|
||
|
|
'expiration_date': self.valid_until,
|
||
|
|
'partner_id': partner.id if self.mode == 'selected' else False,
|
||
|
|
}
|
||
|
|
|
||
|
|
def generate_coupons(self):
|
||
|
|
if any(not wizard.program_id for wizard in self):
|
||
|
|
raise ValidationError(_("Can not generate coupon, no program is set."))
|
||
|
|
if any(wizard.coupon_qty <= 0 for wizard in self):
|
||
|
|
raise ValidationError(_("Invalid quantity."))
|
||
|
|
coupon_create_vals = []
|
||
|
|
for wizard in self:
|
||
|
|
customers = wizard._get_partners() or range(wizard.coupon_qty)
|
||
|
|
for partner in customers:
|
||
|
|
coupon_create_vals.append(wizard._get_coupon_values(partner))
|
||
|
|
self.env['loyalty.card'].create(coupon_create_vals)
|
||
|
|
return True
|