From e0c9c355afc5a1f1b4da3bec2c81d812bbaca1b0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=C3=A9dric=20Krier?= <ced@b2ck.com>
Date: Sun, 5 Apr 2009 18:13:07 +0200
Subject: [PATCH] Fix copy of company to call copy of party for issue932

---
 company.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/company.py b/company.py
index 6caf1e9c59..18a9e0e35d 100644
--- a/company.py
+++ b/company.py
@@ -31,6 +31,28 @@ class Company(ModelSQL, ModelView):
             'recursive_companies': 'You can not create recursive companies!',
         })
 
+    def copy(self, cursor, user, ids, default=None, context=None):
+        party_obj = self.pool.get('party.party')
+
+        int_id = False
+        if isinstance(ids, (int, long)):
+            int_id = True
+            ids = [ids]
+        if default is None:
+            default = {}
+        default = default.copy()
+        new_ids = []
+        for company in self.browse(cursor, user, ids, context=context):
+            default['party'] = party_obj.copy(cursor, user, company.party.id,
+                    context=context)
+            new_id = super(Company, self).copy(cursor, user, company.id,
+                    default=default, context=context)
+            new_ids.append(new_id)
+
+        if int_id:
+            return new_ids[0]
+        return new_ids
+
     def write(self, cursor, user, ids, vals, context=None):
         res = super(Company, self).write(cursor, user, ids, vals,
                 context=context)