addons/project_gtd/project_gtd.py

   1 # -*- coding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
   6 #
   7 #    This program is free software: you can redistribute it and/or modify
   8 #    it under the terms of the GNU Affero General Public License as
   9 #    published by the Free Software Foundation, either version 3 of the
  10 #    License, or (at your option) any later version.
  11 #
  12 #    This program is distributed in the hope that it will be useful,
  13 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 #    GNU Affero General Public License for more details.
  16 #
  17 #    You should have received a copy of the GNU Affero General Public License
  18 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 #
  20 ##############################################################################
  21
  22 import sys
  23
  24 from osv import fields, osv
  25 import tools
  26 from tools.translate import _
  27
  28 class project_gtd_context(osv.osv):
  29     _name = "project.gtd.context"
  30     _description = "Context"
  31     _columns = {
  32         'name': fields.char('Context', size=64, required=True, select=1, translate=1),
  33         'sequence': fields.integer('Sequence', help="Gives the sequence order when displaying a list of contexts."),
  34     }
  35     _defaults = {
  36         'sequence': 1
  37     }
  38     _order = "sequence, name"
  39
  40 project_gtd_context()
  41
  42
  43 class project_gtd_timebox(osv.osv):
  44     _name = "project.gtd.timebox"
  45     _order = "sequence"
  46     _columns = {
  47         'name': fields.char('Timebox', size=64, required=True, select=1, translate=1),
  48         'sequence': fields.integer('Sequence', help="Gives the sequence order when displaying a list of timebox."),
  49         'icon': fields.selection(tools.icons, 'Icon', size=64),
  50     }
  51
  52 project_gtd_timebox()
  53
  54 class project_task(osv.osv):
  55     _inherit = "project.task"
  56     _columns = {
  57         'timebox_id': fields.many2one('project.gtd.timebox', "Timebox",help="Time-laps during which task has to be treated"),
  58         'context_id': fields.many2one('project.gtd.context', "Context",help="The context place where user has to treat task"),
  59      }
  60
  61     def copy_data(self, cr, uid, id, default=None, context=None):
  62         if context is None:
  63             context = {}
  64         if not default:
  65             default = {}
  66         default['timebox_id'] = False
  67         default['context_id'] = False
  68         return super(project_task,self).copy_data(cr, uid, id, default, context)
  69
  70     def _get_context(self,cr, uid, context=None):
  71         ids = self.pool.get('project.gtd.context').search(cr, uid, [], context=context)
  72         return ids and ids[0] or False
  73
  74     _defaults = {
  75         'context_id': _get_context
  76     }
  77     def next_timebox(self, cr, uid, ids, *args):
  78         timebox_obj = self.pool.get('project.gtd.timebox')
  79         timebox_ids = timebox_obj.search(cr,uid,[])
  80         if not timebox_ids: return True
  81         for task in self.browse(cr,uid,ids):
  82             timebox = task.timebox_id.id
  83             if not timebox:
  84                 self.write(cr, uid, task.id, {'timebox_id': timebox_ids[0]})
  85             elif timebox_ids.index(timebox) != len(timebox_ids)-1:
  86                 index = timebox_ids.index(timebox)
  87                 self.write(cr, uid, task.id, {'timebox_id': timebox_ids[index+1]})
  88         return True
  89
  90     def prev_timebox(self, cr, uid, ids, *args):
  91         timebox_obj = self.pool.get('project.gtd.timebox')
  92         timebox_ids = timebox_obj.search(cr,uid,[])
  93         for task in self.browse(cr,uid,ids):
  94             timebox = task.timebox_id.id
  95             if timebox:
  96                 if timebox_ids.index(timebox):
  97                     index = timebox_ids.index(timebox)
  98                     self.write(cr, uid, task.id, {'timebox_id': timebox_ids[index - 1]})
  99                 else:
 100                     self.write(cr, uid, task.id, {'timebox_id': False})
 101         return True
 102
 103     def fields_view_get(self, cr, uid, view_id=None, view_type='form', context=None, toolbar=False, submenu=False):
 104         if not context: context = {}
 105         res = super(project_task,self).fields_view_get(cr, uid, view_id, view_type, context, toolbar=toolbar, submenu=submenu)
 106         search_extended = False
 107         timebox_obj = self.pool.get('project.gtd.timebox')
 108         if (res['type'] == 'search') and context.get('gtd', False):
 109             tt = timebox_obj.browse(cr, uid, timebox_obj.search(cr,uid,[]), context=context)
 110             search_extended =''
 111             for time in tt:
 112                 if time.icon:
 113                     icon = time.icon
 114                 else :
 115                     icon=""
 116                 search_extended += '''<filter domain="[('timebox_id','=', ''' + str(time.id) + ''')]" icon="''' + icon + '''" string="''' + time.name + '''" context="{'user_invisible': True}"/>\n'''
 117             search_extended +='''<separator orientation="vertical"/>'''
 118
 119             res['arch'] = tools.ustr(res['arch']).replace('<separator name="gtdsep"/>', search_extended)
 120
 121         return res
 122
 123 project_task()
 124
 125 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: