@@ -334,18 +334,30 @@ def migrate_state(self,
334334 (a ['account_uuid' ], a ['unit_uuid' ], a ['balance_type' ]): a ['balance' ] for a in digest_data
335335 }
336336
337- item_data = self .get_item_data (entity_slug = entity_slug , queryset = itemthrough_queryset )
337+ # todo: temporary fix... Inventory accounts must be properly mapped...
338+ item_data = list (self .get_item_data (entity_slug = entity_slug , queryset = itemthrough_queryset ))
339+
340+ for item in item_data :
341+ account_uuid_expense = item .get ('item_model__expense_account__uuid' )
342+ account_uuid_inventory = item .get ('item_model__inventory_account__uuid' )
343+ if account_uuid_expense :
344+ item ['account_uuid' ] = account_uuid_expense
345+ item ['account_balance_type' ] = item .get ('item_model__expense_account__balance_type' )
346+ elif account_uuid_inventory :
347+ item ['account_uuid' ] = account_uuid_inventory
348+ item ['account_balance_type' ] = item .get ('item_model__inventory_account__balance_type' )
349+
338350
339351 if isinstance (self , lazy_loader .get_bill_model ()):
340352 item_data_gb = groupby (item_data ,
341- key = lambda a : (a ['item_model__expense_account__uuid ' ],
353+ key = lambda a : (a ['account_uuid ' ],
342354 a ['entity_unit__uuid' ],
343- a ['item_model__expense_account__balance_type ' ]))
355+ a ['account_balance_type ' ]))
344356 elif isinstance (self , lazy_loader .get_invoice_model ()):
345357 item_data_gb = groupby (item_data ,
346- key = lambda a : (a ['item_model__earnings_account__uuid ' ],
358+ key = lambda a : (a ['account_uuid ' ],
347359 a ['entity_unit__uuid' ],
348- a ['item_model__earnings_account__balance_type ' ]))
360+ a ['account_balance_type ' ]))
349361
350362 progress = self .get_progress ()
351363
0 commit comments