mirror of
https://gitlab.com/flectra-community/server-ux.git
synced 2024-11-15 02:32:06 +00:00
108 lines
3.6 KiB
Python
108 lines
3.6 KiB
Python
|
# Copyright 2016 ACSONE SA/NV (<http://acsone.eu>)
|
||
|
|
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl).
|
||
|
|
|
||
|
|
from flectra import _, api, fields, models
|
||
|
|
from flectra.exceptions import ValidationError
|
||
|
|
|
||
|
|
|
||
|
|
class DateRange(models.Model):
|
||
|
|
_name = "date.range"
|
||
|
|
_description = "Date Range"
|
||
|
|
_order = "type_name, date_start"
|
||
|
|
_check_company_auto = True
|
||
|
|
|
||
|
|
@api.model
|
||
|
|
def _default_company(self):
|
||
|
|
return self.env.company
|
||
|
|
|
||
|
|
name = fields.Char(required=True, translate=True)
|
||
|
|
date_start = fields.Date(string="Start date", required=True)
|
||
|
|
date_end = fields.Date(string="End date", required=True)
|
||
|
|
type_id = fields.Many2one(
|
||
|
|
comodel_name="date.range.type",
|
||
|
|
string="Type",
|
||
|
|
index=1,
|
||
|
|
required=True,
|
||
|
|
ondelete="restrict",
|
||
|
|
domain="['|', ('company_id', '=', company_id), ('company_id', '=', False)]",
|
||
|
|
check_company=True,
|
||
|
|
)
|
||
|
|
type_name = fields.Char(related="type_id.name", store=True, string="Type Name")
|
||
|
|
company_id = fields.Many2one(
|
||
|
|
comodel_name="res.company", string="Company", index=1, default=_default_company
|
||
|
|
)
|
||
|
|
active = fields.Boolean(
|
||
|
|
help="The active field allows you to hide the date range without "
|
||
|
|
"removing it.",
|
||
|
|
compute="_compute_active",
|
||
|
|
readonly=False,
|
||
|
|
store=True,
|
||
|
|
)
|
||
|
|
|
||
|
|
_sql_constraints = [
|
||
|
|
(
|
||
|
|
"date_range_uniq",
|
||
|
|
"unique (name,type_id, company_id)",
|
||
|
|
"A date range must be unique per company !",
|
||
|
|
)
|
||
|
|
]
|
||
|
|
|
||
|
|
@api.depends("type_id.active")
|
||
|
|
def _compute_active(self):
|
||
|
|
for date in self:
|
||
|
|
if date.type_id.active:
|
||
|
|
date.active = True
|
||
|
|
else:
|
||
|
|
date.active = False
|
||
|
|
|
||
|
|
@api.constrains("type_id", "date_start", "date_end", "company_id")
|
||
|
|
def _validate_range(self):
|
||
|
|
for this in self:
|
||
|
|
if this.date_start > this.date_end:
|
||
|
|
raise ValidationError(
|
||
|
|
_("%(name)s is not a valid range (%(date_start)s > %(date_end)s)")
|
||
|
|
% {
|
||
|
|
"name": this.name,
|
||
|
|
"date_start": this.date_start,
|
||
|
|
"date_end": this.date_end,
|
||
|
|
}
|
||
|
|
)
|
||
|
|
if this.type_id.allow_overlap:
|
||
|
|
continue
|
||
|
|
# here we use a plain SQL query to benefit of the daterange
|
||
|
|
# function available in PostgresSQL
|
||
|
|
# (http://www.postgresql.org/docs/current/static/rangetypes.html)
|
||
|
|
SQL = """
|
||
|
|
SELECT
|
||
|
|
id
|
||
|
|
FROM
|
||
|
|
date_range dt
|
||
|
|
WHERE
|
||
|
|
DATERANGE(dt.date_start, dt.date_end, '[]') &&
|
||
|
|
DATERANGE(%s::date, %s::date, '[]')
|
||
|
|
AND dt.id != %s
|
||
|
|
AND dt.active
|
||
|
|
AND dt.company_id = %s
|
||
|
|
AND dt.type_id=%s;"""
|
||
|
|
self.env.cr.execute(
|
||
|
|
SQL,
|
||
|
|
(
|
||
|
|
this.date_start,
|
||
|
|
this.date_end,
|
||
|
|
this.id,
|
||
|
|
this.company_id.id or None,
|
||
|
|
this.type_id.id,
|
||
|
|
),
|
||
|
|
)
|
||
|
|
res = self.env.cr.fetchall()
|
||
|
|
if res:
|
||
|
|
dt = self.browse(res[0][0])
|
||
|
|
raise ValidationError(
|
||
|
|
_("%(thisname)s overlaps %(dtname)s")
|
||
|
|
% {"thisname": this.name, "dtname": dt.name}
|
||
|
|
)
|
||
|
|
|
||
|
|
def get_domain(self, field_name):
|
||
|
|
self.ensure_one()
|
||
|
|
return [(field_name, ">=", self.date_start), (field_name, "<=", self.date_end)]
|