From 8fce057d08ba236dcbae418d8f684b99c51dbdf4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E9=B9=8F=E5=AE=87?= <9664676+pengyuthon@user.noreply.gitee.com> Date: Mon, 23 Feb 2026 20:33:48 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=8A=9F=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- sun_badge_menu/models/ir_ui_menu.py | 2 ++ yuthon_hr_employee/models/yuthon_hr_employee.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/sun_badge_menu/models/ir_ui_menu.py b/sun_badge_menu/models/ir_ui_menu.py index b8ef731a0..cd1d406a2 100644 --- a/sun_badge_menu/models/ir_ui_menu.py +++ b/sun_badge_menu/models/ir_ui_menu.py @@ -185,6 +185,8 @@ class IrUiMenu(models.Model): badge_count = self.env[res_model].read_due_count() elif hasattr(res_model, 'get_is_logs2_count'): badge_count = self.env[res_model].get_is_logs2_count() + elif hasattr(res_model, 'get_contract_count'): + badge_count = self.env[res_model].get_contract_count() web_menus[menu['id']] = { "id": menu['id'], diff --git a/yuthon_hr_employee/models/yuthon_hr_employee.py b/yuthon_hr_employee/models/yuthon_hr_employee.py index a29ad6644..c302d0e47 100644 --- a/yuthon_hr_employee/models/yuthon_hr_employee.py +++ b/yuthon_hr_employee/models/yuthon_hr_employee.py @@ -253,7 +253,7 @@ class YuthonHrEmployee(models.Model): ('divorced', 'Divorced') ], string='Marital Status', groups="hr.group_hr_user", default='single', tracking=True) - def read_number(self): + def get_contract_count(self): return self.env['hr.employee'].search_count([('end_comparison', 'in', ['red', 'yellow', 'blue'])]) def update_role_group_ids(self):