From cffb46d5531506bc9312498a2e570ac42eac908e Mon Sep 17 00:00:00 2001
From: Hansa Rathod <hansha@techreceptives.com>
Date: Thu, 25 Oct 2018 16:46:20 +0530
Subject: [PATCH] [IMP]: Fixed Brnach issue in current Company

---
 addons/base_branch_company/models/res_branch.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/addons/base_branch_company/models/res_branch.py b/addons/base_branch_company/models/res_branch.py
index 9338b65e..df6d6ca4 100644
--- a/addons/base_branch_company/models/res_branch.py
+++ b/addons/base_branch_company/models/res_branch.py
@@ -149,3 +149,19 @@ class Users(models.Model):
         branches_count = self._branches_count()
         for user in self:
             user.branches_count = branches_count
+
+    @api.model
+    def create(self, vals):
+        res = super(Users, self).create(vals)
+        if 'company_id' in vals:
+            vals.update({
+                'default_branch_id': self.company_id.branch_id.id,
+            })
+        return res
+
+    @api.multi
+    def write(self, vals):
+        res = super(Users, self).write(vals)
+        if 'company_id' in vals:
+            self.default_branch_id = self.company_id.branch_id.id
+        return res