AjmalCybro
1 year ago
6 changed files with 101 additions and 9 deletions
@ -0,0 +1,82 @@ |
|||
# -*- coding: utf-8 -*- |
|||
############################################################################# |
|||
# |
|||
# Cybrosys Technologies Pvt. Ltd. |
|||
# |
|||
# Copyright (C) 2024-TODAY Cybrosys Technologies(<https://www.cybrosys.com>) |
|||
# Author: Cybrosys Techno Solutions(<https://www.cybrosys.com>) |
|||
# |
|||
# You can modify it under the terms of the GNU LESSER |
|||
# GENERAL PUBLIC LICENSE (LGPL v3), Version 3. |
|||
# |
|||
# This program is distributed in the hope that it will be useful, |
|||
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
|||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|||
# GNU LESSER GENERAL PUBLIC LICENSE (LGPL v3) for more details. |
|||
# |
|||
# You should have received a copy of the GNU LESSER GENERAL PUBLIC LICENSE |
|||
# (LGPL v3) along with this program. |
|||
# If not, see <http://www.gnu.org/licenses/>. |
|||
# |
|||
############################################################################# |
|||
from collections import defaultdict |
|||
|
|||
from odoo import api, modules, _ |
|||
from odoo.addons.mail.models.res_users import Users |
|||
|
|||
|
|||
@api.model |
|||
def systray_get_activities(self): |
|||
""" |
|||
Overridden, to also consider the newly added states in mail activity. |
|||
""" |
|||
activities = self.env["mail.activity"].search( |
|||
[("user_id", "=", self.env.uid)]) |
|||
activities_by_record_by_model_name = defaultdict( |
|||
lambda: defaultdict(lambda: self.env["mail.activity"])) |
|||
for activity in activities: |
|||
record = self.env[activity.res_model].browse(activity.res_id) |
|||
activities_by_record_by_model_name[activity.res_model][ |
|||
record] += activity |
|||
model_ids = list({self.env["ir.model"]._get(name).id for name in |
|||
activities_by_record_by_model_name.keys()}) |
|||
user_activities = {} |
|||
for model_name, activities_by_record in activities_by_record_by_model_name.items(): |
|||
domain = [ |
|||
("id", "in", list({r.id for r in activities_by_record.keys()}))] |
|||
allowed_records = self.env[model_name].search(domain) |
|||
if not allowed_records: |
|||
continue |
|||
module = self.env[model_name]._original_module |
|||
icon = module and modules.module.get_module_icon(module) |
|||
model = self.env["ir.model"]._get(model_name).with_prefetch(model_ids) |
|||
user_activities[model_name] = { |
|||
"id": model.id, |
|||
"name": model.name, |
|||
"model": model_name, |
|||
"type": "activity", |
|||
"icon": icon, |
|||
"total_count": 0, |
|||
"today_count": 0, |
|||
"overdue_count": 0, |
|||
"planned_count": 0, |
|||
"done_count": 0, |
|||
"cancel_count": 0, |
|||
"actions": [ |
|||
{ |
|||
"icon": "fa-clock-o", |
|||
"name": "Summary", |
|||
} |
|||
], |
|||
} |
|||
for record, activities in activities_by_record.items(): |
|||
if record not in allowed_records: |
|||
continue |
|||
for activity in activities: |
|||
user_activities[model_name]["%s_count" % activity.state] += 1 |
|||
if activity.state in ("today", "overdue"): |
|||
user_activities[model_name]["total_count"] += 1 |
|||
return list(user_activities.values()) |
|||
|
|||
|
|||
Users.systray_get_activities = systray_get_activities |
|||
Loading…
Reference in new issue