|
| 1 | +import ast |
| 2 | + |
| 3 | +from odoo import Command, _, api, fields, models |
| 4 | + |
| 5 | + |
| 6 | +class DisableRemainingDaysRule(models.Model): |
| 7 | + _name = "disable.remaining.days.rule" |
| 8 | + _description = "Disable Remaining Days Rule" |
| 9 | + |
| 10 | + res_model_id = fields.Many2one( |
| 11 | + "ir.model", |
| 12 | + string="Model", |
| 13 | + required=True, |
| 14 | + ondelete="cascade", |
| 15 | + copy=False, |
| 16 | + help="Select the model where you want to disable the remaining days widget.", |
| 17 | + ) |
| 18 | + res_model_name = fields.Char( |
| 19 | + related="res_model_id.model", |
| 20 | + string="Technical model name", |
| 21 | + store=True, |
| 22 | + ) |
| 23 | + active = fields.Boolean( |
| 24 | + string="Active record", |
| 25 | + default=True |
| 26 | + ) |
| 27 | + company_id = fields.Many2one( |
| 28 | + "res.company", |
| 29 | + required=True, |
| 30 | + default=lambda self: self.env.company, |
| 31 | + string="Company", |
| 32 | + ) |
| 33 | + diseable_view_types = fields.Char( |
| 34 | + string="Disable View Types", |
| 35 | + compute="_compute_diseable_view_types", |
| 36 | + store=True, |
| 37 | + help="Name of the view types where the remaining days widget will be disabled.", |
| 38 | + ) |
| 39 | + diseable_view_types_json = fields.Char( |
| 40 | + string="Disable View Types JSON", |
| 41 | + readonly=True, |
| 42 | + help="JSON representation of the view types where the remaining days" |
| 43 | + " widget will be disabled. {'list': List, 'form': Form}", |
| 44 | + ) |
| 45 | + date_type_fields_ids = fields.Many2many( |
| 46 | + "ir.model.fields", |
| 47 | + string="Date/Datetime Fields", |
| 48 | + help="Select the date/datetime fields of the model where you want to disable" |
| 49 | + " the remaining days widget.", |
| 50 | + domain="[('model_id', '=', res_model_id)," |
| 51 | + " ('ttype', 'in', ['date', 'datetime'])]", |
| 52 | + copy=False, |
| 53 | + ) |
| 54 | + |
| 55 | + @api.depends("diseable_view_types_json") |
| 56 | + def _compute_diseable_view_types(self): |
| 57 | + """ |
| 58 | + Get Values of diseable_view_types_json |
| 59 | + """ |
| 60 | + for record in self: |
| 61 | + # Convert chart to json array |
| 62 | + diseable_view_types_json = ast.literal_eval( |
| 63 | + record.diseable_view_types_json or "{}" |
| 64 | + ) |
| 65 | + # Get values of dict |
| 66 | + view_types = [] |
| 67 | + for value in diseable_view_types_json.values(): |
| 68 | + if value: |
| 69 | + view_types.append(value) |
| 70 | + record.diseable_view_types = ", ".join(view_types) |
| 71 | + |
| 72 | + @api.constrains("res_model_id") |
| 73 | + def _constrains_res_model_id(self): |
| 74 | + """Ensure that there is only one rule per model""" |
| 75 | + for record in self: |
| 76 | + domain = [ |
| 77 | + ("res_model_id", "=", record.res_model_id.id), |
| 78 | + ("id", "!=", record.id), |
| 79 | + ("company_id", "=", record.company_id.id), |
| 80 | + ] |
| 81 | + if self.search_count(domain) > 0: |
| 82 | + raise models.ValidationError( |
| 83 | + self.env._( |
| 84 | + "There is already a rule for the model '%s'." |
| 85 | + " You cannot create two rules for the same model.", |
| 86 | + record.res_model_id.name, |
| 87 | + ) |
| 88 | + ) |
| 89 | + |
| 90 | + @api.onchange("res_model_id") |
| 91 | + def _onchange_res_model_id(self): |
| 92 | + """ " On change of the model, reset the date field""" |
| 93 | + for record in self: |
| 94 | + record.date_type_fields_ids = False |
| 95 | + |
| 96 | + def action_open_set_disable_remaining_days_rule_wizard(self): |
| 97 | + """ |
| 98 | + Open the wizard to set the disable remaining days rule |
| 99 | + """ |
| 100 | + self.ensure_one() |
| 101 | + model_id = self.res_model_id |
| 102 | + view_ids = model_id.view_ids |
| 103 | + if not view_ids: |
| 104 | + raise models.ValidationError( |
| 105 | + _( |
| 106 | + "The model '%s' does not have any views." |
| 107 | + " You cannot set the disable remaining days rule by view type" |
| 108 | + " for this model." |
| 109 | + ) |
| 110 | + % model_id.name |
| 111 | + ) |
| 112 | + IrUiView = self.env["ir.ui.view"] |
| 113 | + # Get selection from ir.ui.view. |
| 114 | + selection = IrUiView._fields["type"].selection |
| 115 | + view_types = [] |
| 116 | + for technical_name, name in selection: |
| 117 | + selected = technical_name in ast.literal_eval( |
| 118 | + self.diseable_view_types_json or "{}" |
| 119 | + ) |
| 120 | + view_types.append((name, technical_name, selected)) |
| 121 | + wizard_id = ( |
| 122 | + self.env["disable.remaining.days.rule.wizard"] |
| 123 | + .sudo() |
| 124 | + .create( |
| 125 | + { |
| 126 | + "disable_remaining_days_rule_id": self.id, |
| 127 | + "line_ids": [ |
| 128 | + Command.create( |
| 129 | + { |
| 130 | + "name": name, |
| 131 | + "technical_name": technical_name, |
| 132 | + "selected": selected, |
| 133 | + }, |
| 134 | + ) |
| 135 | + for name, technical_name, selected in view_types |
| 136 | + ], |
| 137 | + } |
| 138 | + ) |
| 139 | + ) |
| 140 | + # Return wizard created |
| 141 | + return { |
| 142 | + "type": "ir.actions.act_window", |
| 143 | + "view_mode": "form", |
| 144 | + "res_model": "disable.remaining.days.rule.wizard", |
| 145 | + "target": "new", |
| 146 | + "res_id": wizard_id.id, |
| 147 | + } |
| 148 | + |
| 149 | + @api.model |
| 150 | + def get_all_rules(self): |
| 151 | + """ |
| 152 | + Get all rules like {'model_name': active} |
| 153 | + """ |
| 154 | + # Return one rule per model if some company has it active |
| 155 | + rules_ids = self.search([]) |
| 156 | + if self.get_disable_all_models(): |
| 157 | + # If all models are disabled, return True for all |
| 158 | + return True |
| 159 | + rules = {} |
| 160 | + for rule in rules_ids: |
| 161 | + view_types = rule.get_data_by_view_type() |
| 162 | + fields = rule.get_data_by_field() |
| 163 | + rules[rule.res_model_id.model] = { |
| 164 | + "model": rule.get_data_by_model() |
| 165 | + if not view_types and not fields |
| 166 | + else False, |
| 167 | + "view_types": view_types, |
| 168 | + "fields": fields, |
| 169 | + } |
| 170 | + return rules |
| 171 | + |
| 172 | + def get_data_by_model(self): |
| 173 | + """ |
| 174 | + Get rule data for a specific model |
| 175 | + """ |
| 176 | + self.ensure_one() |
| 177 | + return self.active |
| 178 | + |
| 179 | + def get_data_by_view_type(self): |
| 180 | + """ |
| 181 | + Get rule data for a specific model and view type |
| 182 | + """ |
| 183 | + self.ensure_one() |
| 184 | + diseable_view_types_json = ast.literal_eval( |
| 185 | + self.diseable_view_types_json or "{}" |
| 186 | + ) |
| 187 | + key_list = list(diseable_view_types_json.keys()) |
| 188 | + if not key_list: |
| 189 | + return [] |
| 190 | + return key_list |
| 191 | + |
| 192 | + def get_data_by_field(self): |
| 193 | + """ |
| 194 | + Get rule data for a specific model and field |
| 195 | + """ |
| 196 | + self.ensure_one() |
| 197 | + field_ids = self.date_type_fields_ids |
| 198 | + if not field_ids: |
| 199 | + return [] |
| 200 | + return field_ids.mapped("name") |
| 201 | + |
| 202 | + @api.model |
| 203 | + def get_disable_all_models(self): |
| 204 | + """ |
| 205 | + Get all models with the remaining days disabled |
| 206 | + :return: list of model names |
| 207 | + """ |
| 208 | + disable_remaining_days = self.env.company.disable_remaining_days |
| 209 | + if disable_remaining_days: |
| 210 | + return True |
| 211 | + return False |
0 commit comments