1 files changed, 91 insertions, 0 deletions
diff --git a/addons/project/models/analytic.py b/addons/project/models/analytic.py
new file mode 100644
index 00000000000..65f756cf6a4
--- /dev/null
+++ b/addons/project/models/analytic.py
@@ -0,0 +1,91 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import api, fields, models, _
+from odoo.exceptions import UserError
+
+
+class AccountAnalyticAccount(models.Model):
+    _inherit = 'account.analytic.account'
+    _description = 'Analytic Account'
+
+    company_uom_id = fields.Many2one('product.uom', related='company_id.project_time_mode_id', string="Company UOM")
+    project_ids = fields.One2many('project.project', 'analytic_account_id', string='Projects')
+    project_count = fields.Integer("Project Count", compute='_compute_project_count')
+
+    def _compute_project_count(self):
+        for account in self:
+            account.project_count = len(account.with_context(active_test=False).project_ids)
+
+    @api.model
+    def create(self, vals):
+        analytic_account = super(AccountAnalyticAccount, self).create(vals)
+        analytic_account.project_create(vals)
+        return analytic_account
+
+    @api.multi
+    def write(self, vals):
+        vals_for_project = vals.copy()
+        for account in self:
+            if not vals.get('name'):
+                vals_for_project['name'] = account.name
+            account.project_create(vals_for_project)
+        return super(AccountAnalyticAccount, self).write(vals)
+
+    @api.multi
+    def unlink(self):
+        projects = self.env['project.project'].search([('analytic_account_id', 'in', self.ids)])
+        has_tasks = self.env['project.task'].search_count([('project_id', 'in', projects.ids)])
+        if has_tasks:
+            raise UserError(_('Please remove existing tasks in the project linked to the accounts you want to delete.'))
+        return super(AccountAnalyticAccount, self).unlink()
+
+    @api.model
+    def name_search(self, name, args=None, operator='ilike', limit=100):
+        if args is None:
+            args = []
+        if self.env.context.get('current_model') == 'project.project':
+            return self.search(args + [('name', operator, name)], limit=limit).name_get()
+
+        return super(AccountAnalyticAccount, self).name_search(name, args=args, operator=operator, limit=limit)
+
+    @api.multi
+    def projects_action(self):
+        projects = self.with_context(active_test=False).mapped('project_ids')
+        result = {
+            "type": "ir.actions.act_window",
+            "res_model": "project.project",
+            "views": [[False, "tree"], [False, "form"]],
+            "domain": [["id", "in", projects.ids]],
+            "context": {"create": False},
+            "name": "Projects",
+        }
+        if len(projects) == 1:
+            result['views'] = [(False, "form")]
+            result['res_id'] = projects.id
+        else:
+            result = {'type': 'ir.actions.act_window_close'}
+        return result
+
+    @api.model
+    def _trigger_project_creation(self, vals):
+        """ This function is used to decide if a project needs to be automatically created or not when an analytic
+            account is created. It returns True if it needs to be so, False otherwise.
+        """
+        return 'project_creation_in_progress' not in self.env.context
+
+    @api.multi
+    def project_create(self, vals):
+        """ This function is called at the time of analytic account creation and is used to create a project
+            automatically linked to it if the conditions are meet.
+        """
+        self.ensure_one()
+        Project = self.env['project.project']
+        project = Project.with_context(active_test=False).search([('analytic_account_id', '=', self.id)])
+        if not project and self._trigger_project_creation(vals):
+            project_values = {
+                'name': vals.get('name'),
+                'analytic_account_id': self.id,
+            }
+            return Project.create(project_values).id
+        return False