forked from Mapan/odoo17e
80 lines
2.9 KiB
Python
80 lines
2.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
|
|
|
from odoo import api, fields, models, _lt
|
|
from odoo.tools import SQL
|
|
|
|
class Project(models.Model):
|
|
_inherit = 'project.project'
|
|
|
|
assets_count = fields.Integer('# Assets', compute='_compute_assets_count', groups='account.group_account_readonly')
|
|
|
|
@api.depends('analytic_account_id')
|
|
def _compute_assets_count(self):
|
|
if not self.analytic_account_id:
|
|
self.assets_count = 0
|
|
return
|
|
query = self.env['account.asset']._search([])
|
|
query.add_where(
|
|
SQL(
|
|
"%s && %s",
|
|
[str(account_id) for account_id in self.analytic_account_id.ids],
|
|
self.env['account.asset']._query_analytic_accounts(),
|
|
)
|
|
)
|
|
query_string, query_param = query.select(
|
|
r"""DISTINCT id, (regexp_matches(jsonb_object_keys(account_asset.analytic_distribution), '\d+', 'g'))[1]::int as account_id"""
|
|
)
|
|
query_string = f"""
|
|
SELECT account_id, count(id) FROM
|
|
({query_string}) distribution
|
|
GROUP BY account_id
|
|
"""
|
|
self._cr.execute(query_string, query_param)
|
|
data = {res['account_id']: res['count'] for res in self._cr.dictfetchall()}
|
|
for project in self:
|
|
project.assets_count = data.get(project.analytic_account_id.id, 0)
|
|
|
|
# -------------------------------------
|
|
# Actions
|
|
# -------------------------------------
|
|
|
|
def action_open_project_assets(self):
|
|
query = self.env['account.asset']._search([])
|
|
query.add_where(
|
|
SQL(
|
|
"%s && %s",
|
|
[str(self.analytic_account_id.id)],
|
|
self.env['account.asset']._query_analytic_accounts(),
|
|
)
|
|
)
|
|
assets = list(query)
|
|
action = self.env["ir.actions.actions"]._for_xml_id("account_asset.action_account_asset_form")
|
|
action.update({
|
|
'views': [[False, 'tree'], [False, 'form'], [False, 'kanban']],
|
|
'context': {'default_analytic_distribution': {self.analytic_account_id.id: 100}},
|
|
'domain': [('id', 'in', assets)]
|
|
})
|
|
if(len(assets) == 1):
|
|
action["views"] = [[False, 'form']]
|
|
action["res_id"] = assets[0]
|
|
return action
|
|
|
|
# ----------------------------
|
|
# Project Updates
|
|
# ----------------------------
|
|
|
|
def _get_stat_buttons(self):
|
|
buttons = super(Project, self)._get_stat_buttons()
|
|
if self.user_has_groups('account.group_account_readonly'):
|
|
buttons.append({
|
|
'icon': 'pencil-square-o',
|
|
'text': _lt('Assets'),
|
|
'number': self.sudo().assets_count,
|
|
'action_type': 'object',
|
|
'action': 'action_open_project_assets',
|
|
'show': self.sudo().assets_count > 0,
|
|
'sequence': 40,
|
|
})
|
|
return buttons
|