server-ux/sequence_range_end/models/ir_sequence.py

78 lines
2.9 KiB
Python
Raw Normal View History

2021-11-28 03:13:50 +00:00
# Copyright 2021 Ecosoft Co., Ltd. (http://ecosoft.co.th)
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
from datetime import datetime
import pytz
from flectra import _, fields, models
from flectra.exceptions import UserError
class IrSequence(models.Model):
_inherit = "ir.sequence"
def _get_prefix_suffix(self, date=None, date_range=None):
if "range_end_" in str(self.prefix) or "range_end_" in str(self.suffix):
return self._get_prefix_suffix_range_end(date_range=date_range)
else:
return super()._get_prefix_suffix(date=date, date_range=date_range)
def _get_prefix_suffix_range_end(self, date=None, date_range=None):
def _interpolate(s, d):
return (s % d) if s else ""
def _interpolation_dict():
now = range_date = range_end_date = effective_date = datetime.now(
pytz.timezone(self._context.get("tz") or "UTC")
)
if date or self._context.get("ir_sequence_date"):
effective_date = fields.Datetime.from_string(
date or self._context.get("ir_sequence_date")
)
if date_range or self._context.get("ir_sequence_date_range"):
range_date = fields.Datetime.from_string(
date_range or self._context.get("ir_sequence_date_range")
)
if date_range or self._context.get("ir_sequence_date_range_end"):
range_end_date = fields.Datetime.from_string(
date_range or self._context.get("ir_sequence_date_range_end")
)
sequences = {
"year": "%Y",
"month": "%m",
"day": "%d",
"y": "%y",
"doy": "%j",
"woy": "%W",
"weekday": "%w",
"h24": "%H",
"h12": "%I",
"min": "%M",
"sec": "%S",
}
res = {}
for key, format in sequences.items():
res[key] = effective_date.strftime(format)
res["current_" + key] = now.strftime(format)
res["range_" + key] = range_date.strftime(format)
res["range_end_" + key] = range_end_date.strftime(format)
return res
self.ensure_one()
d = _interpolation_dict()
try:
interpolated_prefix = _interpolate(self.prefix, d)
interpolated_suffix = _interpolate(self.suffix, d)
except ValueError:
raise UserError(_("Invalid prefix or suffix for sequence '%s'") % self.name)
return interpolated_prefix, interpolated_suffix
class IrSequenceDateRange(models.Model):
_inherit = "ir.sequence.date_range"
def _next(self):
self = self.with_context(ir_sequence_date_range_end=self.date_to)
return super()._next()