添加功能

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'],

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):