Ajout d'un champ charge_reel dans palier

[OpenERP/cmmi.git] / axes.py

diff --git a/axes.py b/axes.py
index aa2e4ac..aa5a83f 100644 (file)
--- a/axes.py
+++ b/axes.py
@@ -292,6 +292,13 @@ class Palier(osv.Model):
         return result
 
 
+    def _get_charge_reel(self, cr, uid, ids, field, arg, context=None):
+        result = {}
+        for p in self.browse(cr, uid, ids, context=context):
+            result[p.id] = sum([e.charge_reel for e in p.evolutions])
+        return result
+
+
     _columns = {
         "type_palier": fields.selection(_types_palier, string="Type"),
         "projet_id": fields.many2one("cmmi.projet",
@@ -309,6 +316,9 @@ class Palier(osv.Model):
         "charge_plan": fields.function(_get_charge_plan,
                                        type="integer",
                                        string="Charge plannifiée"),
+        "charge_reel": fields.function(_get_charge_reel,
+                                       type="integer",
+                                       string="Charge réelle"),
     }
 
     _defaults = {
@@ -442,6 +452,13 @@ class Chantier(osv.Model):
         return result
 
 
+    def _get_charge_reel(self, cr, uid, ids, field, arg, context=None):
+        result = {}
+        for c in self.browse(cr, uid, ids, context=context):
+            result[c.id] = sum([e.charge_reel for e in c.evolutions])
+        return result
+
+
     _columns = {
         "projet_id": fields.many2one("cmmi.projet",
                                      string="Projet",
@@ -460,4 +477,7 @@ class Chantier(osv.Model):
         "charge_plan": fields.function(_get_charge_plan,
                                        type="integer",
                                        string="Charge plannifiée"),
+        "charge_reel": fields.function(_get_charge_reel,
+                                       type="integer",
+                                       string="Charge réelle"),
     }