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 osv import osv, fields
  24 from 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 !'), _('No employee defined for your user !'))
  39         ids = ts.search(cr, uid, [('user_id','=',uid),('state','=','draft'),('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             ts.write(cr, uid, ids, {'date_current': time.strftime('%Y-%m-%d')}, context=context)
  46             domain = "[('user_id', '=', uid)]"
  47         else:
  48             domain = "[('user_id', '=', uid)]"
  49         value = {
  50             'domain': domain,
  51             'name': 'Open Timesheet',
  52             'view_type': 'form',
  53             'view_mode': view_type,
  54             'res_model': 'hr_timesheet_sheet.sheet',
  55             'view_id': False,
  56             'type': 'ir.actions.act_window'
  57         }
  58         if len(ids) == 1:
  59             value['res_id'] = ids[0]
  60         return value
  61
  62 hr_timesheet_current_open()
  63
  64 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: