diff --git a/party.py b/party.py index 1694c99766..f006fa672b 100644 --- a/party.py +++ b/party.py @@ -152,17 +152,17 @@ class Party(ModelSQL, ModelView): return '' @classmethod - def create(cls, values): + def create(cls, vlist): Sequence = Pool().get('ir.sequence') Configuration = Pool().get('party.configuration') - values = values.copy() - if not values.get('code'): - config = Configuration(1) - values['code'] = Sequence.get_id(config.party_sequence.id) - - values['code_length'] = len(values['code']) - return super(Party, cls).create(values) + vlist = [x.copy() for x in vlist] + for values in vlist: + if not values.get('code'): + config = Configuration(1) + values['code'] = Sequence.get_id(config.party_sequence.id) + values['code_length'] = len(values['code']) + return super(Party, cls).create(vlist) @classmethod def write(cls, parties, vals): diff --git a/tests/test_party.py b/tests/test_party.py index 87adf7f688..2ac5e86210 100644 --- a/tests/test_party.py +++ b/tests/test_party.py @@ -44,9 +44,9 @@ class PartyTestCase(unittest.TestCase): ''' with Transaction().start(DB_NAME, USER, context=CONTEXT) as transaction: - category1 = self.category.create({ - 'name': 'Category 1', - }) + category1, = self.category.create([{ + 'name': 'Category 1', + }]) self.assert_(category1.id) transaction.cursor.commit() @@ -59,10 +59,10 @@ class PartyTestCase(unittest.TestCase): ('name', '=', 'Category 1'), ], limit=1) - category2 = self.category.create({ - 'name': 'Category 2', - 'parent': category1.id, - }) + category2, = self.category.create([{ + 'name': 'Category 2', + 'parent': category1.id, + }]) self.assert_(category2.id) self.assertRaises(Exception, self.category.write, @@ -76,9 +76,9 @@ class PartyTestCase(unittest.TestCase): ''' with Transaction().start(DB_NAME, USER, context=CONTEXT) as transaction: - party1 = self.party.create({ - 'name': 'Party 1', - }) + party1, = self.party.create([{ + 'name': 'Party 1', + }]) self.assert_(party1.id) transaction.cursor.commit() @@ -91,9 +91,9 @@ class PartyTestCase(unittest.TestCase): code = party1.code - party2 = self.party.create({ - 'name': 'Party 2', - }) + party2, = self.party.create([{ + 'name': 'Party 2', + }]) self.assertRaises(Exception, self.party.write, [party2], { @@ -107,11 +107,11 @@ class PartyTestCase(unittest.TestCase): with Transaction().start(DB_NAME, USER, context=CONTEXT): party1, = self.party.search([], limit=1) - address = self.address.create({ - 'party': party1.id, - 'street': 'St sample, 15', - 'city': 'City', - }) + address, = self.address.create([{ + 'party': party1.id, + 'street': 'St sample, 15', + 'city': 'City', + }]) self.assert_(address.id)