mirror of
https://gitlab.com/flectra-community/server-ux.git
synced 2024-11-15 02:32:06 +00:00
59 lines
2.5 KiB
Python
59 lines
2.5 KiB
Python
|
# Copyright (C) 2017 Creu Blanca
|
||
|
|
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl.html).
|
||
|
|
|
||
|
|
from flectra import fields, models
|
||
|
|
from datetime import datetime, timedelta, date as datetime_date
|
||
|
|
from dateutil.relativedelta import relativedelta
|
||
|
|
|
||
|
|
|
||
|
|
class IrSequence(models.Model):
|
||
|
|
_inherit = "ir.sequence"
|
||
|
|
|
||
|
|
range_reset = fields.Selection([
|
||
|
|
('daily', 'Daily'),
|
||
|
|
('weekly', 'Weekly'),
|
||
|
|
('monthly', 'Monthly'),
|
||
|
|
('yearly', 'Yearly')
|
||
|
|
])
|
||
|
|
|
||
|
|
def _compute_date_from_to(self, date):
|
||
|
|
self.ensure_one()
|
||
|
|
date_from = date_to = fields.Date.from_string(date)
|
||
|
|
if self.range_reset == 'weekly':
|
||
|
|
date_from = date_from - timedelta(days=date_from.weekday())
|
||
|
|
date_to = date_from + timedelta(days=6)
|
||
|
|
elif self.range_reset == 'monthly':
|
||
|
|
date_from = datetime_date(date_from.year, date_from.month, 1)
|
||
|
|
date_to = date_from + relativedelta(months=1)
|
||
|
|
date_to += relativedelta(days=-1)
|
||
|
|
elif self.range_reset == 'yearly':
|
||
|
|
date_from = datetime_date(date_from.year, 1, 1)
|
||
|
|
date_to = datetime_date(date_from.year, 12, 31)
|
||
|
|
return date_from.strftime('%Y-%m-%d'), date_to.strftime('%Y-%m-%d')
|
||
|
|
|
||
|
|
def _create_date_range_seq(self, date):
|
||
|
|
self.ensure_one()
|
||
|
|
if not self.range_reset:
|
||
|
|
return super()._create_date_range_seq(date)
|
||
|
|
date_from, date_to = self._compute_date_from_to(date)
|
||
|
|
date_range = self.env['ir.sequence.date_range'].search(
|
||
|
|
[('sequence_id', '=', self.id), ('date_from', '>=', date),
|
||
|
|
('date_from', '<=', date_to)], order='date_from desc', limit=1)
|
||
|
|
if date_range:
|
||
|
|
date_to = datetime.strptime(date_range.date_from,
|
||
|
|
'%Y-%m-%d') + timedelta(days=-1)
|
||
|
|
date_to = date_to.strftime('%Y-%m-%d')
|
||
|
|
date_range = self.env['ir.sequence.date_range'].search(
|
||
|
|
[('sequence_id', '=', self.id), ('date_to', '>=', date_from),
|
||
|
|
('date_to', '<=', date)], order='date_to desc', limit=1)
|
||
|
|
if date_range:
|
||
|
|
date_from = datetime.strptime(date_range.date_to,
|
||
|
|
'%Y-%m-%d') + timedelta(days=1)
|
||
|
|
date_from = date_from.strftime('%Y-%m-%d')
|
||
|
|
seq_date_range = self.env['ir.sequence.date_range'].sudo().create({
|
||
|
|
'date_from': date_from,
|
||
|
|
'date_to': date_to,
|
||
|
|
'sequence_id': self.id,
|
||
|
|
})
|
||
|
|
return seq_date_range
|