X-Git-Url: http://git.inspyration.org/?a=blobdiff_plain;f=todolist.py;h=0209a9b6c99d64a353e219cc30738307972d6b59;hb=42942e622b82142baa538b11c855afea238fd487;hp=c25f457df5dafc45f8ed0bda7d1998d1a3586cf7;hpb=bec1bdb7bc229a8765d3b8323abf17b5d474b14e;p=OpenERP%2Ftodolist.git

diff --git a/todolist.py b/todolist.py
index c25f457..0209a9b 100644
--- a/todolist.py
+++ b/todolist.py
@@ -8,14 +8,40 @@ from openerp.osv import osv, fields
 class Container(osv.Model):
     """TODO List : Tasks container"""
 
+    def _get_nb_tasks(self, cr, uid, ids, field, arg, context=None):
+        result = {}
+        for container in self.browse(cr, uid, ids, context=context):
+            result[container.id] = len(container.tasks)
+        return result
+#       OR : return dict((c.id, len(c.tasks)) for c in self.browse(cr, uid, ids, context=context))
+
+
+    def _get_nb_tasks_done(self, cr, uid, ids, field, arg, context=None):
+        result={}
+        for c in self.browse(cr, uid, ids, context=context):
+            result[c.id] = len([t for t in c.tasks if t.state == "done"])
+        return result
+
 
-    def _get_nb_task(self, cr, uid, ids, field, arg, context=None):
+    def _tasks_progress(self, cr, uid, ids, field, arg, context=None):
         result = {}
-        for tasks in self.browse(cr, uid, ids, context=context):
-            result[tasks.id] = len(tasks.tasks)
+        for c in self.browse(cr, uid, ids, context=context):
+            result[c.id] = c.number_tasks and c.number_tasks_done*100./c.number_tasks or 0.
         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)
+
+
     _name = "todolist.container"
 
     _status = [("draft", "Draft"), ("pending", "Pending"), ("done", "Done")]
@@ -29,7 +55,9 @@ class Container(osv.Model):
         "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")]),
-        "number_tasks": fields.function(_get_nb_task, type="integer", string="Number of tasks"),
+        "number_tasks": fields.function(_get_nb_tasks, type="integer", string="Number of tasks"),
+        "number_tasks_done": fields.function(_get_nb_tasks_done, type="integer", string="Number of tasks done"),
+        "progress_tasks": fields.function(_tasks_progress, type="float", string="Progression"),
     }
 
     _defaults = {
@@ -49,6 +77,8 @@ class Container(osv.Model):
         ),
     ]
 
+    _order = "name"
+
     def action_start(self, cr, uid, ids, context=None):
         self.write(cr, uid, ids, {"state": "pending"}, context=context)
         return self
@@ -64,6 +94,11 @@ class Container(osv.Model):
         return self
 
 
+    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)
 
 
 class Task(osv.Model):
@@ -71,7 +106,7 @@ class Task(osv.Model):
 
     _name = "todolist.task"
 
-    _priorities = [("useful", "Useful"), ("necessary", "Necessary"), ("essential", "Essential")]
+    _priorities = [("Useful", "Useful"), ("Necessary", "Necessary"), ("Essential", "Essential")]
 
     _states = [("draft", "Draft"), ("proposal", "Proposal"), ("approved", "Approved"), ("started", "Started"), ("done", "Done")]
 
@@ -87,9 +122,21 @@ class Task(osv.Model):
     }
 
     _defaults = {
-        "state": "draft"
+        "state": "draft",
+        "priority": "useful"
     }
 
+    _order = "planned"
+
+    _sql_constraints = [
+        (
+            "name_different_from_description_constraint",
+            "CHECK(name <> description)",
+            "Fields name and description should be different",
+        ),
+    ]
+
+
     def action_draft(self, cr, uid, ids, context=None):
         self.write(cr, uid, ids, {"state": "draft"}, context=context)
         return self
@@ -110,12 +157,46 @@ class Task(osv.Model):
         self.write(cr, uid, ids, {"state": "done"}, context=context)
         return self
 
+    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)
+
+
 
-# ------------------- class Theme ------------------- #
 
 class Topic(osv.Model):
     """TODO List : Container"s Topic"""
 
+    def _get_nb_lists(self, cr, uid, ids, field, arg, context=None):
+        result = {}
+        for topic in self.browse(cr, uid, ids, context=context):
+            result[topic.id] = len(topic.todolist_ids)
+        return result
+
+
+    def _get_number_tasks(self, cr, uid, ids, field, arg, context=None):
+        result = {}
+        for topic in self.browse(cr, uid, ids, context=context):
+            result[topic.id] = sum([t.number_tasks for t in topic.todolist_ids])
+        return result
+
+
+    def _get_number_tasks_done(self, cr, uid, ids, field, arg, context=None):
+        result = {}
+        for topic in self.browse(cr, uid, ids, context=context):
+            result[topic.id] = sum([t.number_tasks_done for t in topic.todolist_ids])
+        return result
+
+
+    def _progress_tasks(self, cr, uid, ids, field, arg, context=None):
+        result = {}
+        for t in self.browse(cr, uid, ids, context=context):
+            result[t.id] = t.number_tasks and t.number_tasks_done*100./t.number_tasks or 0.
+        return result
+
+
     _name = "todolist.topic"
 
     _states = [("Active", "Active"), ("Inactive", "Inactive")]
@@ -125,8 +206,21 @@ class Topic(osv.Model):
         "description": fields.text(string="Description"),
         "activated": fields.selection(_states, string="State", select=True),
         "todolist_ids": fields.many2many("todolist.container", "todolist_container_topic_rel", "topic_id", "Container_id", string="TO DO Lists"),
+        "nb_lists": fields.function(_get_nb_lists, type="integer", string="Number of lists"),
+        "number_tasks": fields.function(_get_number_tasks, type="integer", string="Number of tasks"),
+        "number_tasks_done": fields.function(_get_number_tasks_done, type="integer", string="Number of tasks done"),
+        "progress_tasks": fields.function(_progress_tasks, type="float", string="Number of lists"),
     }
 
     _defaults = {
         "activated": "Active",
     }
+
+
+    _sql_constraints = [
+        (
+            "name_different_from_description_constraint",
+            "CHECK(name <> description)",
+            "Fields name and description should be different",
+        ),
+    ]