diff --git a/soong_workflow/models/soong_workflow_record.py b/soong_workflow/models/soong_workflow_record.py index 143728279..dc1a3107e 100644 --- a/soong_workflow/models/soong_workflow_record.py +++ b/soong_workflow/models/soong_workflow_record.py @@ -22,6 +22,13 @@ class SoongWorkflowRecord(models.Model): ] return self.search_count(domain) + def get_workflow_circulation(self): + domain = [ + ('is_circulation', '=', True), + ] + return self.search_count(domain) + + def get_is_logs2_count(self): domain = [ ('is_log2', '=', True), diff --git a/sun_badge_menu/models/ir_ui_menu.py b/sun_badge_menu/models/ir_ui_menu.py index cd1d406a2..74a0f272f 100644 --- a/sun_badge_menu/models/ir_ui_menu.py +++ b/sun_badge_menu/models/ir_ui_menu.py @@ -177,6 +177,8 @@ class IrUiMenu(models.Model): badge_count = self.env[res_model].get_no_read_count() elif hasattr(self.env[res_model], 'get_workflow_count'): badge_count = self.env[res_model].get_workflow_count() + elif hasattr(self.env[res_model], 'get_workflow_count'): + badge_count = self.env[res_model].get_workflow_circulation() elif hasattr(res_model, 'read_number'): badge_count = self.env[res_model].read_number() elif hasattr(res_model, 'read_coming_count'):