|
1 | 1 | # Copyright 2016 Camptocamp SA |
2 | 2 | # License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl.html) |
3 | 3 |
|
| 4 | +import functools |
4 | 5 | from datetime import datetime, timedelta |
5 | 6 |
|
6 | 7 | from odoo import api, fields, models |
@@ -127,15 +128,20 @@ def delay_me_context_key(self): |
127 | 128 | return "ok" |
128 | 129 |
|
129 | 130 | def _register_hook(self): |
130 | | - self._patch_method("delay_me", self._patch_job_auto_delay("delay_me")) |
131 | | - self._patch_method( |
132 | | - "delay_me_options", self._patch_job_auto_delay("delay_me_options") |
| 131 | + patched = self._patch_job_auto_delay("delay_me") |
| 132 | + type(self).delay_me = functools.update_wrapper(patched, type(self).delay_me) |
| 133 | + |
| 134 | + patched = self._patch_job_auto_delay("delay_me_options") |
| 135 | + type(self).delay_me_options = functools.update_wrapper( |
| 136 | + patched, type(self).delay_me_options |
133 | 137 | ) |
134 | | - self._patch_method( |
| 138 | + |
| 139 | + patched = self._patch_job_auto_delay( |
135 | 140 | "delay_me_context_key", |
136 | | - self._patch_job_auto_delay( |
137 | | - "delay_me_context_key", context_key="auto_delay_delay_me_context_key" |
138 | | - ), |
| 141 | + context_key="auto_delay_delay_me_context_key", |
| 142 | + ) |
| 143 | + type(self).delay_me_context_key = functools.update_wrapper( |
| 144 | + patched, type(self).delay_me_context_key |
139 | 145 | ) |
140 | 146 | return super()._register_hook() |
141 | 147 |
|
|
0 commit comments