fix minor error

Som-Energia · Aug 16, 2022 · ba02cbf · ba02cbf
1 parent a089442
commit ba02cbf
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/som_generationkwh/som_generationkwh.py b/som_generationkwh/som_generationkwh.py
@@ -530,6 +530,7 @@ def getPriceWithoutGeneration(self, cr, uid, line):
     def getProfit(self, cr, uid, line):
         ai_obj = self.pool.get('account.invoice')
         cfg_obj = self.pool.get('res.config')
+        gffl_obj = self.pool.get('giscedata.facturacio.factura.linia')
 
         if line['quantity'] == 0:
             return 0
@@ -540,9 +541,8 @@ def getProfit(self, cr, uid, line):
             cr, uid, 'end_date_mecanisme_ajust_gas', '2099-12-31')
 
         priceNoGen = float(self.getPriceWithoutGeneration(cr, uid, line)['price_unit'])
-        rmag_lines_ids = self.browse(cr, uid, line).factura_id.get_rmag_lines()
-
-        if rmag_lines_ids and \
+        rmag_lines = gffl_obj.browse(cr, uid, line['id']).factura_id.get_rmag_lines()
+        if rmag_lines and \
                 line['data_desde'] >= start_date_mecanisme_ajust_gas and \
                 line['data_fins'] <= end_date_mecanisme_ajust_gas:
             rmag_line = self.browse(cr, uid, rmag_lines_ids[0])
@@ -564,6 +564,7 @@ def _ff_saving_generation(self, cursor, uid, ids, field_name, arg,
         gffl_obj = self.pool.get('giscedata.facturacio.factura.linia')
 
         res = {k: {} for k in ids}
+
         for gilo_line in self.browse(cursor, uid, ids):
             line = gffl_obj.read(cursor, uid, gilo_line.factura_line_id.id)
             res[gilo_line.id] = self.getProfit(cursor, uid, line)