addons/hr_timesheet_sheet/wizard/hr_timesheet_current.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 import time
  22
  23 from openerp.osv import fields, osv
  24 from openerp.tools.translate import _
  25
  26 class hr_timesheet_current_open(osv.osv_memory):
  27     _name = 'hr.timesheet.current.open'
  28     _description = 'hr.timesheet.current.open'
  29
  30     def open_timesheet(self, cr, uid, ids, context=None):
  31         ts = self.pool.get('hr_timesheet_sheet.sheet')
  32         if context is None:
  33             context = {}
  34         view_type = 'form,tree'
  35
  36         user_ids = self.pool.get('hr.employee').search(cr, uid, [('user_id','=',uid)], context=context)
  37         if not len(user_ids):
  38             raise osv.except_osv(_('Error!'), _('Please create an employee and associate it with this user.'))
  39         ids = ts.search(cr, uid, [('user_id','=',uid),('state','in',('draft','new')),('date_from','<=',time.strftime('%Y-%m-%d')), ('date_to','>=',time.strftime('%Y-%m-%d'))], context=context)
  40
  41         if len(ids) > 1:
  42             view_type = 'tree,form'
  43             domain = "[('id','in',["+','.join(map(str, ids))+"]),('user_id', '=', uid)]"
  44         elif len(ids)==1:
  45             domain = "[('user_id', '=', uid)]"
  46         else:
  47             domain = "[('user_id', '=', uid)]"
  48         value = {
  49             'domain': domain,
  50             'name': _('Open Timesheet'),
  51             'view_type': 'form',
  52             'view_mode': view_type,
  53             'res_model': 'hr_timesheet_sheet.sheet',
  54             'view_id': False,
  55             'type': 'ir.actions.act_window'
  56         }
  57         if len(ids) == 1:
  58             value['res_id'] = ids[0]
  59         return value
  60
  61
  62 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: