addons/hr_timesheet/wizard/hr_timesheet_print_employee.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 datetime
  22
  23 from osv import osv, fields
  24 from tools.translate import _
  25
  26 class analytical_timesheet_employee(osv.osv_memory):
  27     _name = 'hr.analytical.timesheet.employee'
  28     _description = 'Print Employee Timesheet & Print My Timesheet'
  29     _columns = {
  30         'month': fields.selection([(x, datetime.date(2000, x, 1).strftime('%B')) for x in range(1, 13)],
  31                                   'Month', required=True),
  32         'year': fields.integer('Year', required=True),
  33         'employee_id': fields.many2one('hr.employee', 'Employee', required=True)
  34
  35                 }
  36
  37     def _get_user(self, cr, uid, context=None):
  38
  39         emp_obj = self.pool.get('hr.employee')
  40         emp_id = emp_obj.search(cr, uid, [('user_id', '=', uid)], context=context)
  41         if not emp_id:
  42             raise osv.except_osv(_("Warning"), _("No employee defined for this user"))
  43         return emp_id and emp_id[0] or False
  44
  45     _defaults = {
  46          'month': lambda *a: datetime.date.today().month,
  47          'year': lambda *a: datetime.date.today().year,
  48          'employee_id': _get_user
  49              }
  50
  51     def print_report(self, cr, uid, ids, context=None):
  52         data = self.read(cr, uid, ids, context=context)[0]
  53         if isinstance(data['employee_id'], tuple):
  54             data['employee_id'] = data['employee_id'][0]
  55         datas = {
  56              'ids': [],
  57              'model': 'hr.employee',
  58              'form': data
  59                  }
  60         return {
  61             'type': 'ir.actions.report.xml',
  62             'report_name': 'hr.analytical.timesheet',
  63             'datas': datas,
  64             }
  65 analytical_timesheet_employee()
  66
  67 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: