server-ux/base_tier_validation/models/res_users.py

46 lines
1.7 KiB
Python
Raw Normal View History

2021-03-23 19:38:33 +00:00
# Copyright 2019 Eficent Business and IT Consulting Services S.L.
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
from flectra import api, fields, models, modules
class Users(models.Model):
_inherit = 'res.users'
review_ids = fields.Many2many(
string="Reviews", comodel_name="tier.review"
)
@api.model
def review_user_count(self):
user_reviews = {}
to_review_docs = {}
for review in self.env.user.review_ids.filtered(
lambda r: r.status == 'pending'):
record = review.env[review.model].browse(review.res_id)
if not user_reviews.get(review['model']):
user_reviews[review.model] = {
'name': record._description,
'model': review.model,
'icon': modules.module.get_module_icon(
self.env[review.model]._original_module),
'pending_count': 0
}
docs = to_review_docs.get(review.model)
if (docs and record not in docs) or not docs:
user_reviews[review.model]['pending_count'] += 1
to_review_docs.setdefault(review.model, []).append(record)
return list(user_reviews.values())
@api.model
def get_reviews(self, data):
review_obj = self.env['tier.review'].with_context(
lang=self.env.user.lang)
res = review_obj.search_read([('id', 'in', data.get('res_ids'))])
for r in res:
# Get the translated status value.
r['display_status'] = dict(
review_obj.fields_get('status')['status']['selection']
).get(r.get('status'))
return res