addons/event_project/event_project.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 from osv import fields, osv
  23
  24 class one2many_mod_task(fields.one2many):
  25
  26     def get(self, cr, obj, ids, name, user=None, offset=0, context=None, values=None):
  27         if not values:
  28                 values = {}
  29         res = {}
  30         for id in ids:
  31             res[id] = []
  32         for id in ids:
  33             query = "select project_id from event_event where id = %s"
  34             cr.execute(query, (id,))
  35             project_ids = [ x[0] for x in cr.fetchall()]
  36             ids2 = obj.pool.get(self._obj).search(cr, user, [(self._fields_id, 'in', project_ids), ('state', '<>', 'done')], limit=self._limit)
  37             for r in obj.pool.get(self._obj)._read_flat(cr, user, ids2, [self._fields_id], context=context, load='_classic_write'):
  38                 res[id].append( r['id'] )
  39         return res
  40
  41 class event(osv.osv):
  42     _inherit = 'event.event'
  43
  44     def write(self, cr, uid, ids, vals, *args, **kwargs):
  45         if 'date_begin' in vals and vals['date_begin']:
  46             for eve in self.browse(cr, uid, ids):
  47                 if eve.project_id:
  48                     self.pool.get('project.project').write(cr, uid, [eve.project_id.id], {'date_end': eve.date_begin[:10]})
  49         return super(event,self).write(cr, uid, ids, vals, *args, **kwargs)
  50
  51     _columns = {
  52         'project_id': fields.many2one('project.project', 'Project', readonly=True),
  53         'task_ids': one2many_mod_task('project.task', 'project_id', "Project tasks", readonly=True, domain="[('state', '&lt;&gt;', 'done')]"),
  54     }
  55
  56 event()
  57
  58 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: