1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.
from odoo import api, fields, models
from datetime import date
class AccountMove(models.Model):
_inherit = 'account.move'
def button_draft(self):
# OVERRIDE to update the cancel date.
res = super(AccountMove, self).button_draft()
for move in self:
if move.move_type == 'out_invoice':
self.env['membership.membership_line'].search([
('account_invoice_line', 'in', move.mapped('invoice_line_ids').ids)
]).write({'date_cancel': False})
return res
def button_cancel(self):
# OVERRIDE to update the cancel date.
res = super(AccountMove, self).button_cancel()
for move in self:
if move.move_type == 'out_invoice':
self.env['membership.membership_line'].search([
('account_invoice_line', 'in', move.mapped('invoice_line_ids').ids)
]).write({'date_cancel': fields.Date.today()})
return res
def write(self, vals):
# OVERRIDE to write the partner on the membership lines.
res = super(AccountMove, self).write(vals)
if 'partner_id' in vals:
self.env['membership.membership_line'].search([
('account_invoice_line', 'in', self.mapped('invoice_line_ids').ids)
]).write({'partner': vals['partner_id']})
return res
class AccountMoveLine(models.Model):
_inherit = 'account.move.line'
def write(self, vals):
# OVERRIDE
res = super(AccountMoveLine, self).write(vals)
to_process = self.filtered(lambda line: line.move_id.move_type == 'out_invoice' and line.product_id.membership)
# Nothing to process, break.
if not to_process:
return res
existing_memberships = self.env['membership.membership_line'].search([
('account_invoice_line', 'in', to_process.ids)])
to_process = to_process - existing_memberships.mapped('account_invoice_line')
# All memberships already exist, break.
if not to_process:
return res
memberships_vals = []
for line in to_process:
date_from = line.product_id.membership_date_from
date_to = line.product_id.membership_date_to
if (date_from and date_from < (line.move_id.invoice_date or date.min) < (date_to or date.min)):
date_from = line.move_id.invoice_date
memberships_vals.append({
'partner': line.move_id.partner_id.id,
'membership_id': line.product_id.id,
'member_price': line.price_unit,
'date': fields.Date.today(),
'date_from': date_from,
'date_to': date_to,
'account_invoice_line': line.id,
})
self.env['membership.membership_line'].create(memberships_vals)
return res
@api.model_create_multi
def create(self, vals_list):
# OVERRIDE
lines = super(AccountMoveLine, self).create(vals_list)
to_process = lines.filtered(lambda line: line.move_id.move_type == 'out_invoice' and line.product_id.membership)
# Nothing to process, break.
if not to_process:
return lines
existing_memberships = self.env['membership.membership_line'].search([
('account_invoice_line', 'in', to_process.ids)])
to_process = to_process - existing_memberships.mapped('account_invoice_line')
# All memberships already exist, break.
if not to_process:
return lines
memberships_vals = []
for line in to_process:
date_from = line.product_id.membership_date_from
date_to = line.product_id.membership_date_to
if (date_from and date_from < (line.move_id.invoice_date or date.min) < (date_to or date.min)):
date_from = line.move_id.invoice_date
memberships_vals.append({
'partner': line.move_id.partner_id.id,
'membership_id': line.product_id.id,
'member_price': line.price_unit,
'date': fields.Date.today(),
'date_from': date_from,
'date_to': date_to,
'account_invoice_line': line.id,
})
self.env['membership.membership_line'].create(memberships_vals)
return lines
|
