Modification de la dupplication des container

[OpenERP/todolist.git] / todolist.py

diff --git a/todolist.py b/todolist.py
index 12743d6..5d0c666 100644 (file)
--- a/todolist.py
+++ b/todolist.py
@@ -30,16 +30,30 @@ class Container(osv.Model):
         return result
 
 
+    #================================================================================
+    # def copy(self, cr, uid, id, default, context=None):
+    #    container = self.browse(cr, uid, id, context=context)
+    #    new_name =  "Copy of %s" % container.name
+    #    # =like is the original LIKE operator from SQL
+    #    others_count = self.search(cr,  uid, [('name', '=like', new_name+'%')],
+    #                               count=True, context=context)
+    #    if others_count > 0:
+    #        new_name = "%s (%s)" % (new_name, others_count+1)
+    #    default['name'] = new_name
+    #    return osv.Model.copy(self, cr, uid, id, default, context=context)
+    #================================================================================
+
+
     def copy(self, cr, uid, id, default, context=None):
-        container = self.browse(cr, uid, id, context=context)
-        new_name =  "Copy of %s" % container.name
-        # =like is the original LIKE operator from SQL
-        others_count = self.search(cr,  uid, [('name', '=like', new_name+'%')],
-                                   count=True, context=context)
-        if others_count > 0:
-            new_name = "%s (%s)" % (new_name, others_count+1)
-        default['name'] = new_name
-        return osv.Model.copy(self, cr, uid, id, default, context=context)
+        cr.execute("SELECT copierContainer (%s);", (id,))
+        return cr.fetchone()[0]
+
+
+    def _get_manday(self, cr, uid, ids, field, arg, context=None):
+        result={}
+        for container in self.browse(cr, uid, ids, context=context):
+            result[container.id] = sum([t.manday for t in container.tasks if t.state != "done"])
+        return result
 
 
     _name = "todolist.container"
@@ -51,7 +65,7 @@ class Container(osv.Model):
         "description": fields.text(string="Description"),
         "target": fields.date(string="Target", help="Target Date"),
         "milestone": fields.date(string="Milestone", help="Due date"),
-        "manday": fields.float(string="Man-Days", digits=(6, 2)),
+        "manday": fields.function(_get_manday, type="integer", string="Man-Days"),
         "state": fields.selection(_status, string="State", select=True),
         "tasks": fields.one2many("todolist.task", "container_id", string="Tasks"),
         "topics_id": fields.many2many("todolist.topic", "todolist_container_topic_rel", "container_id", "topic_id", string="Topics", domain=[("activated", "=","Active")]),
@@ -114,9 +128,9 @@ class Task(osv.Model):
         "name": fields.char(string="Title", size=64, required=True),
         "description": fields.text(string="Description"),
         "planned": fields.date(string="Planed"),
-        "milestone": fields.date(string="Milestone"),
-        "manday": fields.integer(string="Man-Days"),
-        "priority": fields.selection(_priorities, string="Priority", select=True),
+        "milestone": fields.date(string="Milestone", required=True),
+        "manday": fields.integer(string="Man-Days", required=True),
+        "priority": fields.selection(_priorities, string="Priority", select=True, required=True),
         "state": fields.selection(_states, string="State", select=True),
         "container_id": fields.many2one("todolist.container", string="To do list", required=True),
     }
@@ -135,9 +149,14 @@ class Task(osv.Model):
             "Fields name and description should be different",
         ),
         (
-            "target_before_milestone_constraint",
+            "planned_before_milestone_constraint",
             "CHECK(planned < milestone)",
-            "The target date should be previous milestone date",
+            "The planned date should be previous milestone date",
+        ),
+        (
+            "manday_sup_0_constraint",
+            "CHECK(manday > 0)",
+            "The manday should be positive",
         ),
     ]
 
@@ -162,13 +181,27 @@ class Task(osv.Model):
         self.write(cr, uid, ids, {"state": "done"}, context=context)
         return self
 
+    #chaque utilisateur voit seulement ces taches
     def search(self, cr, user, args=[], offset=0, limit=None, order=None, context=None, count=False):
         args.append(("create_uid", "=", user))
         if len(args) != 1:
             args.insert(0, "&")
         return osv.Model.search(self, cr, user, args, offset, limit, order, context, count)
 
-
+    def write(self, cr, user, ids, vals, context=None):
+        if "milestone" in vals.keys():
+            for task in self.browse(cr, user, ids, context=context):
+                if task.container_id.milestone < vals["milestone"]:
+                    vals["milestone"] = task.container_id.milestone
+        return osv.Model.write(self, cr, user, ids, vals, context)
+
+    def create(self, cr, user, vals, context=None):
+        container_model = self.pool.get("todolist.container")
+        container = container_model.read(cr, user, vals["container_id"], context=context)
+        milestone = container["milestone"]
+        if milestone < vals["milestone"]:
+            vals["milestone"] = milestone
+        return osv.Model.create(self, cr, user, vals, context=context)
 
 
 class Topic(osv.Model):