audrius/pms
1
0
Fork 1
mirror of https://github.com/OCA/pms.git synced 2025-01-29 00:17:45 +02:00
Code Issues Packages Projects Releases Wiki Activity

[REF] pms: availability plan wizard massive changes extensibility

Browse Source 
Make the code reuse easier in module extensions
This commit is contained in:
Eric Antones
2021-06-13 12:41:06 +02:00
parent 0892c57f9d
commit 9d96bdec20
1 changed files with 43 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

87
pms/wizards/wizard_massive_changes.py
View File

@@ -378,6 +378,46 @@ class AvailabilityWizard(models.TransientModel):
]
)
def _rules_to_overwrite_by_plans(self, availability_plans):
self.ensure_one()
domain = [
("availability_plan_id", "in", availability_plans.ids),
]
if self.room_type_ids:
domain.append(("room_type_id", "in", self.room_type_ids.ids))
if self.start_date:
domain.append(("date", ">=", self.start_date))
if self.end_date:
domain.append(("date", "<=", self.end_date))
domain_overwrite = self.build_domain_rules()
if len(domain_overwrite):
if len(domain_overwrite) == 1:
domain.append(domain_overwrite[0][0])
else:
domain_overwrite = expression.OR(domain_overwrite)
domain.extend(domain_overwrite)
rules = self.env["pms.availability.plan.rule"]
if self.start_date and self.end_date:
rules = rules.search(domain)
if not self.apply_on_all_week and self.start_date and self.end_date:
week_days_to_apply = (
self.apply_on_monday,
self.apply_on_tuesday,
self.apply_on_wednesday,
self.apply_on_thursday,
self.apply_on_friday,
self.apply_on_saturday,
self.apply_on_sunday,
)
rules = rules.filtered(
lambda x: week_days_to_apply[x.date.timetuple()[6]]
)
return rules
@api.depends(
"start_date",
"end_date",
@@ -420,50 +460,9 @@ class AvailabilityWizard(models.TransientModel):
]
record.massive_changes_on = "availability_plan"
if record.availability_plan_ids:
domain = [
("availability_plan_id", "in", record.availability_plan_ids.ids),
]
if record.room_type_ids:
domain.append(("room_type_id", "in", record.room_type_ids.ids))
if record.start_date:
domain.append(("date", ">=", record.start_date))
if record.end_date:
domain.append(("date", "<=", record.end_date))
domain_overwrite = self.build_domain_rules()
if len(domain_overwrite):
if len(domain_overwrite) == 1:
domain.append(domain_overwrite[0][0])
else:
domain_overwrite = expression.OR(domain_overwrite)
domain.extend(domain_overwrite)
week_days_to_apply = (
record.apply_on_monday,
record.apply_on_tuesday,
record.apply_on_wednesday,
record.apply_on_thursday,
record.apply_on_friday,
record.apply_on_saturday,
record.apply_on_sunday,
)
if record.start_date and record.end_date:
rules = self.env["pms.availability.plan.rule"].search(domain)
if (
not record.apply_on_all_week
and record.start_date
and record.end_date
):
record.rules_to_overwrite = rules.filtered(
lambda x: week_days_to_apply[x.date.timetuple()[6]]
)
else:
record.rules_to_overwrite = rules
else:
record.rules_to_overwrite = False
else:
record.rules_to_overwrite = False
record.rules_to_overwrite = record._rules_to_overwrite_by_plans(
record.availability_plan_ids
)
@api.depends(
"start_date",