[FIX] hr_payroll: skip missing structures in get_all_structures for hr.contracts...
[odoo/odoo.git] / addons / hr_recruitment / hr_recruitment.py
1 # -*- coding: utf-8 -*-
2 ##############################################################################
3 #
4 #    OpenERP, Open Source Management Solution
5 #    Copyright (C) 2004-2009 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 time
23 from openerp import tools
24
25 from openerp.addons.base_status.base_stage import base_stage
26 from datetime import datetime
27 from openerp.osv import fields, osv
28 from openerp.tools.translate import _
29 from openerp.tools import html2plaintext
30
31 AVAILABLE_STATES = [
32     ('draft', 'New'),
33     ('cancel', 'Refused'),
34     ('open', 'In Progress'),
35     ('pending', 'Pending'),
36     ('done', 'Hired')
37 ]
38
39 AVAILABLE_PRIORITIES = [
40     ('', ''),
41     ('5', 'Not Good'),
42     ('4', 'On Average'),
43     ('3', 'Good'),
44     ('2', 'Very Good'),
45     ('1', 'Excellent')
46 ]
47
48 class hr_recruitment_source(osv.osv):
49     """ Sources of HR Recruitment """
50     _name = "hr.recruitment.source"
51     _description = "Source of Applicants"
52     _columns = {
53         'name': fields.char('Source Name', size=64, required=True, translate=True),
54     }
55
56 class hr_recruitment_stage(osv.osv):
57     """ Stage of HR Recruitment """
58     _name = "hr.recruitment.stage"
59     _description = "Stage of Recruitment"
60     _order = 'sequence'
61     _columns = {
62         'name': fields.char('Name', size=64, required=True, translate=True),
63         'sequence': fields.integer('Sequence', help="Gives the sequence order when displaying a list of stages."),
64         'department_id':fields.many2one('hr.department', 'Specific to a Department', help="Stages of the recruitment process may be different per department. If this stage is common to all departments, keep this field empty."),
65         'state': fields.selection(AVAILABLE_STATES, 'Status', required=True, help="The related status for the stage. The status of your document will automatically change according to the selected stage. Example, a stage is related to the status 'Close', when your document reach this stage, it will be automatically closed."),
66         'fold': fields.boolean('Hide in views if empty', help="This stage is not visible, for example in status bar or kanban view, when there are no records in that stage to display."),
67         'requirements': fields.text('Requirements'),
68     }
69     _defaults = {
70         'sequence': 1,
71         'state': 'draft',
72         'fold': False,
73     }
74
75 class hr_recruitment_degree(osv.osv):
76     """ Degree of HR Recruitment """
77     _name = "hr.recruitment.degree"
78     _description = "Degree of Recruitment"
79     _columns = {
80         'name': fields.char('Name', size=64, required=True, translate=True),
81         'sequence': fields.integer('Sequence', help="Gives the sequence order when displaying a list of degrees."),
82     }
83     _defaults = {
84         'sequence': 1,
85     }
86     _sql_constraints = [
87         ('name_uniq', 'unique (name)', 'The name of the Degree of Recruitment must be unique!')
88     ]
89
90 class hr_applicant(base_stage, osv.Model):
91     _name = "hr.applicant"
92     _description = "Applicant"
93     _order = "id desc"
94     _inherit = ['mail.thread', 'ir.needaction_mixin']
95     _track = {
96         'state': {
97             'hr_recruitment.mt_applicant_hired': lambda self, cr, uid, obj, ctx=None: obj['state'] == 'done',
98             'hr_recruitment.mt_applicant_refused': lambda self, cr, uid, obj, ctx=None: obj['state'] == 'cancel',
99         },
100         'stage_id': {
101             'hr_recruitment.mt_stage_changed': lambda self, cr, uid, obj, ctx=None: obj['state'] not in ['done', 'cancel'],
102         },
103     }
104
105     def _get_default_department_id(self, cr, uid, context=None):
106         """ Gives default department by checking if present in the context """
107         return (self._resolve_department_id_from_context(cr, uid, context=context) or False)
108
109     def _get_default_stage_id(self, cr, uid, context=None):
110         """ Gives default stage_id """
111         department_id = self._get_default_department_id(cr, uid, context=context)
112         return self.stage_find(cr, uid, [], department_id, [('state', '=', 'draft')], context=context)
113
114     def _resolve_department_id_from_context(self, cr, uid, context=None):
115         """ Returns ID of department based on the value of 'default_department_id'
116             context key, or None if it cannot be resolved to a single
117             department.
118         """
119         if context is None:
120             context = {}
121         if type(context.get('default_department_id')) in (int, long):
122             return context.get('default_department_id')
123         if isinstance(context.get('default_department_id'), basestring):
124             department_name = context['default_department_id']
125             department_ids = self.pool.get('hr.department').name_search(cr, uid, name=department_name, context=context)
126             if len(department_ids) == 1:
127                 return int(department_ids[0][0])
128         return None
129
130     def _read_group_stage_ids(self, cr, uid, ids, domain, read_group_order=None, access_rights_uid=None, context=None):
131         access_rights_uid = access_rights_uid or uid
132         stage_obj = self.pool.get('hr.recruitment.stage')
133         order = stage_obj._order
134         # lame hack to allow reverting search, should just work in the trivial case
135         if read_group_order == 'stage_id desc':
136             order = "%s desc" % order
137         # retrieve section_id from the context and write the domain
138         # - ('id', 'in', 'ids'): add columns that should be present
139         # - OR ('department_id', '=', False), ('fold', '=', False): add default columns that are not folded
140         # - OR ('department_id', 'in', department_id), ('fold', '=', False) if department_id: add department columns that are not folded
141         department_id = self._resolve_department_id_from_context(cr, uid, context=context)
142         search_domain = []
143         if department_id:
144             search_domain += ['|', ('department_id', '=', department_id)]
145         search_domain += ['|', ('id', 'in', ids), ('department_id', '=', False)]
146         stage_ids = stage_obj._search(cr, uid, search_domain, order=order, access_rights_uid=access_rights_uid, context=context)
147         result = stage_obj.name_get(cr, access_rights_uid, stage_ids, context=context)
148         # restore order of the search
149         result.sort(lambda x,y: cmp(stage_ids.index(x[0]), stage_ids.index(y[0])))
150
151         fold = {}
152         for stage in stage_obj.browse(cr, access_rights_uid, stage_ids, context=context):
153             fold[stage.id] = stage.fold or False
154         return result, fold
155
156     def _compute_day(self, cr, uid, ids, fields, args, context=None):
157         """
158         @param cr: the current row, from the database cursor,
159         @param uid: the current user’s ID for security checks,
160         @param ids: List of Openday’s IDs
161         @return: difference between current date and log date
162         @param context: A standard dictionary for contextual values
163         """
164         res = {}
165         for issue in self.browse(cr, uid, ids, context=context):
166             for field in fields:
167                 res[issue.id] = {}
168                 duration = 0
169                 ans = False
170                 hours = 0
171
172                 if field in ['day_open']:
173                     if issue.date_open:
174                         date_create = datetime.strptime(issue.create_date, "%Y-%m-%d %H:%M:%S")
175                         date_open = datetime.strptime(issue.date_open, "%Y-%m-%d %H:%M:%S")
176                         ans = date_open - date_create
177
178                 elif field in ['day_close']:
179                     if issue.date_closed:
180                         date_create = datetime.strptime(issue.create_date, "%Y-%m-%d %H:%M:%S")
181                         date_close = datetime.strptime(issue.date_closed, "%Y-%m-%d %H:%M:%S")
182                         ans = date_close - date_create
183                 if ans:
184                     duration = float(ans.days)
185                     res[issue.id][field] = abs(float(duration))
186         return res
187
188     _columns = {
189         'name': fields.char('Subject', size=128, required=True),
190         'active': fields.boolean('Active', help="If the active field is set to false, it will allow you to hide the case without removing it."),
191         'description': fields.text('Description'),
192         'email_from': fields.char('Email', size=128, help="These people will receive email."),
193         'email_cc': fields.text('Watchers Emails', size=252, help="These email addresses will be added to the CC field of all inbound and outbound emails for this record before being sent. Separate multiple email addresses with a comma"),
194         'probability': fields.float('Probability'),
195         'partner_id': fields.many2one('res.partner', 'Contact'),
196         'create_date': fields.datetime('Creation Date', readonly=True, select=True),
197         'write_date': fields.datetime('Update Date', readonly=True),
198         'stage_id': fields.many2one ('hr.recruitment.stage', 'Stage', track_visibility='onchange',
199                         domain="['&', ('fold', '=', False), '|', ('department_id', '=', department_id), ('department_id', '=', False)]"),
200         'state': fields.related('stage_id', 'state', type="selection", store=True,
201                 selection=AVAILABLE_STATES, string="Status", readonly=True,
202                 help='The status is set to \'Draft\', when a case is created.\
203                       If the case is in progress the status is set to \'Open\'.\
204                       When the case is over, the status is set to \'Done\'.\
205                       If the case needs to be reviewed then the status is \
206                       set to \'Pending\'.'),
207         'categ_ids': fields.many2many('hr.applicant_category', string='Tags'),
208         'company_id': fields.many2one('res.company', 'Company'),
209         'user_id': fields.many2one('res.users', 'Responsible', track_visibility='onchange'),
210         # Applicant Columns
211         'date_closed': fields.datetime('Closed', readonly=True, select=True),
212         'date_open': fields.datetime('Opened', readonly=True, select=True),
213         'date': fields.datetime('Date'),
214         'date_action': fields.date('Next Action Date'),
215         'title_action': fields.char('Next Action', size=64),
216         'priority': fields.selection(AVAILABLE_PRIORITIES, 'Appreciation'),
217         'job_id': fields.many2one('hr.job', 'Applied Job'),
218         'salary_proposed_extra': fields.char('Proposed Salary Extra', size=100, help="Salary Proposed by the Organisation, extra advantages"),
219         'salary_expected_extra': fields.char('Expected Salary Extra', size=100, help="Salary Expected by Applicant, extra advantages"),
220         'salary_proposed': fields.float('Proposed Salary', help="Salary Proposed by the Organisation"),
221         'salary_expected': fields.float('Expected Salary', help="Salary Expected by Applicant"),
222         'availability': fields.integer('Availability'),
223         'partner_name': fields.char("Applicant's Name", size=64),
224         'partner_phone': fields.char('Phone', size=32),
225         'partner_mobile': fields.char('Mobile', size=32),
226         'type_id': fields.many2one('hr.recruitment.degree', 'Degree'),
227         'department_id': fields.many2one('hr.department', 'Department'),
228         'survey': fields.related('job_id', 'survey_id', type='many2one', relation='survey', string='Survey'),
229         'response': fields.integer("Response"),
230         'reference': fields.char('Referred By', size=128),
231         'source_id': fields.many2one('hr.recruitment.source', 'Source'),
232         'day_open': fields.function(_compute_day, string='Days to Open', \
233                                 multi='day_open', type="float", store=True),
234         'day_close': fields.function(_compute_day, string='Days to Close', \
235                                 multi='day_close', type="float", store=True),
236         'color': fields.integer('Color Index'),
237         'emp_id': fields.many2one('hr.employee', 'employee'),
238         'user_email': fields.related('user_id', 'email', type='char', string='User Email', readonly=True),
239     }
240
241     _defaults = {
242         'active': lambda *a: 1,
243         'user_id': lambda s, cr, uid, c: uid,
244         'email_from': lambda s, cr, uid, c: s._get_default_email(cr, uid, c),
245         'stage_id': lambda s, cr, uid, c: s._get_default_stage_id(cr, uid, c),
246         'department_id': lambda s, cr, uid, c: s._get_default_department_id(cr, uid, c),
247         'company_id': lambda s, cr, uid, c: s.pool.get('res.company')._company_default_get(cr, uid, 'hr.applicant', context=c),
248         'color': 0,
249     }
250
251     _group_by_full = {
252         'stage_id': _read_group_stage_ids
253     }
254
255     def onchange_job(self, cr, uid, ids, job, context=None):
256         if job:
257             job_record = self.pool.get('hr.job').browse(cr, uid, job, context=context)
258             if job_record and job_record.department_id:
259                 return {'value': {'department_id': job_record.department_id.id}}
260         return {}
261
262     def onchange_department_id(self, cr, uid, ids, department_id=False, context=None):
263         obj_recru_stage = self.pool.get('hr.recruitment.stage')
264         stage_ids = obj_recru_stage.search(cr, uid, ['|',('department_id','=',department_id),('department_id','=',False)], context=context)
265         stage_id = stage_ids and stage_ids[0] or False
266         return {'value': {'stage_id': stage_id}}
267
268     def onchange_partner_id(self, cr, uid, ids, partner_id, context=None):
269         data = {'partner_phone': False,
270                 'partner_mobile': False,
271                 'email_from': False}
272         if partner_id:
273             addr = self.pool.get('res.partner').browse(cr, uid, partner_id, context)
274             data.update({'partner_phone': addr.phone,
275                         'partner_mobile': addr.mobile,
276                         'email_from': addr.email})
277         return {'value': data}
278
279     def stage_find(self, cr, uid, cases, section_id, domain=[], order='sequence', context=None):
280         """ Override of the base.stage method
281             Parameter of the stage search taken from the lead:
282             - department_id: if set, stages must belong to this section or
283               be a default case
284         """
285         if isinstance(cases, (int, long)):
286             cases = self.browse(cr, uid, cases, context=context)
287         # collect all section_ids
288         department_ids = []
289         if section_id:
290             department_ids.append(section_id)
291         for case in cases:
292             if case.department_id:
293                 department_ids.append(case.department_id.id)
294         # OR all section_ids and OR with case_default
295         search_domain = []
296         if department_ids:
297             search_domain += ['|', ('department_id', 'in', department_ids)]
298         search_domain.append(('department_id', '=', False))
299         # AND with the domain in parameter
300         search_domain += list(domain)
301         # perform search, return the first found
302         stage_ids = self.pool.get('hr.recruitment.stage').search(cr, uid, search_domain, order=order, context=context)
303         if stage_ids:
304             return stage_ids[0]
305         return False
306
307     def action_makeMeeting(self, cr, uid, ids, context=None):
308         """ This opens Meeting's calendar view to schedule meeting on current applicant
309             @return: Dictionary value for created Meeting view
310         """
311         applicant = self.browse(cr, uid, ids[0], context)
312         category = self.pool.get('ir.model.data').get_object(cr, uid, 'hr_recruitment', 'categ_meet_interview', context)
313         res = self.pool.get('ir.actions.act_window').for_xml_id(cr, uid, 'base_calendar', 'action_crm_meeting', context)
314         res['context'] = {
315             'default_partner_ids': applicant.partner_id and [applicant.partner_id.id] or False,
316             'default_user_id': uid,
317             'default_name': applicant.name,
318             'default_categ_ids': category and [category.id] or False,
319         }
320         return res
321
322     def action_print_survey(self, cr, uid, ids, context=None):
323         """
324         If response is available then print this response otherwise print survey form(print template of the survey).
325
326         @param self: The object pointer
327         @param cr: the current row, from the database cursor,
328         @param uid: the current user’s ID for security checks,
329         @param ids: List of Survey IDs
330         @param context: A standard dictionary for contextual values
331         @return: Dictionary value for print survey form.
332         """
333         if context is None:
334             context = {}
335         record = self.browse(cr, uid, ids, context=context)
336         record = record and record[0]
337         context.update({'survey_id': record.survey.id, 'response_id': [record.response], 'response_no': 0, })
338         value = self.pool.get("survey").action_print_survey(cr, uid, ids, context=context)
339         return value
340
341     def message_get_suggested_recipients(self, cr, uid, ids, context=None):
342         recipients = super(hr_applicant, self).message_get_suggested_recipients(cr, uid, ids, context=context)
343         for applicant in self.browse(cr, uid, ids, context=context):
344             if applicant.partner_id:
345                 self._message_add_suggested_recipient(cr, uid, recipients, applicant, partner=applicant.partner_id, reason=_('Contact'))
346             elif applicant.email_from:
347                 self._message_add_suggested_recipient(cr, uid, recipients, applicant, email=applicant.email_from, reason=_('Contact Email'))
348         return recipients
349
350     def message_new(self, cr, uid, msg, custom_values=None, context=None):
351         """ Overrides mail_thread message_new that is called by the mailgateway
352             through message_process.
353             This override updates the document according to the email.
354         """
355         if custom_values is None:
356             custom_values = {}
357         desc = html2plaintext(msg.get('body')) if msg.get('body') else ''
358         defaults = {
359             'name':  msg.get('subject') or _("No Subject"),
360             'description': desc,
361             'email_from': msg.get('from'),
362             'email_cc': msg.get('cc'),
363             'user_id': False,
364             'partner_id': msg.get('author_id', False),
365         }
366         if msg.get('priority'):
367             defaults['priority'] = msg.get('priority')
368         defaults.update(custom_values)
369         return super(hr_applicant, self).message_new(cr, uid, msg, custom_values=defaults, context=context)
370
371     def create(self, cr, uid, vals, context=None):
372         if context is None:
373             context = {}
374         if vals.get('department_id') and not context.get('default_department_id'):
375             context['default_department_id'] = vals.get('department_id')
376
377         obj_id = super(hr_applicant, self).create(cr, uid, vals, context=context)
378         applicant = self.browse(cr, uid, obj_id, context=context)
379         if applicant.job_id:
380             self.pool.get('hr.job').message_post(cr, uid, [applicant.job_id.id], body=_('Applicant <b>created</b>'), subtype="hr_recruitment.mt_job_new_applicant", context=context)
381         return obj_id
382
383     def case_open(self, cr, uid, ids, context=None):
384         """
385             open Request of the applicant for the hr_recruitment
386         """
387         res = super(hr_applicant, self).case_open(cr, uid, ids, context)
388         date = self.read(cr, uid, ids, ['date_open'])[0]
389         if not date['date_open']:
390             self.write(cr, uid, ids, {'date_open': time.strftime('%Y-%m-%d %H:%M:%S'),})
391         return res
392
393     def case_close(self, cr, uid, ids, context=None):
394         res = super(hr_applicant, self).case_close(cr, uid, ids, context)
395         return res
396
397     def case_close_with_emp(self, cr, uid, ids, context=None):
398         if context is None:
399             context = {}
400         hr_employee = self.pool.get('hr.employee')
401         model_data = self.pool.get('ir.model.data')
402         act_window = self.pool.get('ir.actions.act_window')
403         emp_id = False
404         for applicant in self.browse(cr, uid, ids, context=context):
405             address_id = False
406             if applicant.partner_id:
407                 address_id = self.pool.get('res.partner').address_get(cr,uid,[applicant.partner_id.id],['contact'])['contact']
408             if applicant.job_id:
409                 applicant.job_id.write({'no_of_recruitment': applicant.job_id.no_of_recruitment - 1})
410                 emp_id = hr_employee.create(cr,uid,{'name': applicant.partner_name or applicant.name,
411                                                      'job_id': applicant.job_id.id,
412                                                      'address_home_id': address_id,
413                                                      'department_id': applicant.department_id.id
414                                                      })
415                 self.write(cr, uid, [applicant.id], {'emp_id': emp_id}, context=context)
416                 self.case_close(cr, uid, [applicant.id], context)
417             else:
418                 raise osv.except_osv(_('Warning!'), _('You must define Applied Job for this applicant.'))
419
420         action_model, action_id = model_data.get_object_reference(cr, uid, 'hr', 'open_view_employee_list')
421         dict_act_window = act_window.read(cr, uid, action_id, [])
422         if emp_id:
423             dict_act_window['res_id'] = emp_id
424         dict_act_window['view_mode'] = 'form,tree'
425         return dict_act_window
426
427     def case_cancel(self, cr, uid, ids, context=None):
428         """Overrides cancel for crm_case for setting probability
429         """
430         res = super(hr_applicant, self).case_cancel(cr, uid, ids, context)
431         self.write(cr, uid, ids, {'probability': 0.0})
432         return res
433
434     def case_pending(self, cr, uid, ids, context=None):
435         """Marks case as pending"""
436         res = super(hr_applicant, self).case_pending(cr, uid, ids, context)
437         self.write(cr, uid, ids, {'probability': 0.0})
438         return res
439
440     def case_reset(self, cr, uid, ids, context=None):
441         """Resets case as draft
442         """
443         res = super(hr_applicant, self).case_reset(cr, uid, ids, context)
444         self.write(cr, uid, ids, {'date_open': False, 'date_closed': False})
445         return res
446
447     def set_priority(self, cr, uid, ids, priority, *args):
448         """Set applicant priority
449         """
450         return self.write(cr, uid, ids, {'priority': priority})
451
452     def set_high_priority(self, cr, uid, ids, *args):
453         """Set applicant priority to high
454         """
455         return self.set_priority(cr, uid, ids, '1')
456
457     def set_normal_priority(self, cr, uid, ids, *args):
458         """Set applicant priority to normal
459         """
460         return self.set_priority(cr, uid, ids, '3')
461
462
463 class hr_job(osv.osv):
464     _inherit = "hr.job"
465     _name = "hr.job"
466     _inherits = {'mail.alias': 'alias_id'}
467     _columns = {
468         'survey_id': fields.many2one('survey', 'Interview Form', help="Choose an interview form for this job position and you will be able to print/answer this interview from all applicants who apply for this job"),
469         'alias_id': fields.many2one('mail.alias', 'Alias', ondelete="restrict", required=True,
470                                     help="Email alias for this job position. New emails will automatically "
471                                          "create new applicants for this job position."),
472     }
473     _defaults = {
474         'alias_domain': False, # always hide alias during creation
475     }
476
477     def _auto_init(self, cr, context=None):
478         """Installation hook to create aliases for all jobs and avoid constraint errors."""
479         if context is None:
480             context = {}
481         alias_context = dict(context, alias_model_name='hr.applicant')
482         self.pool.get('mail.alias').migrate_to_alias(cr, self._name, self._table, super(hr_job, self)._auto_init,
483             self._columns['alias_id'], 'name', alias_prefix='job+', alias_defaults={'job_id': 'id'}, context=alias_context)
484
485     def create(self, cr, uid, vals, context=None):
486         mail_alias = self.pool.get('mail.alias')
487         if not vals.get('alias_id'):
488             vals.pop('alias_name', None) # prevent errors during copy()
489             alias_id = mail_alias.create_unique_alias(cr, uid,
490                           # Using '+' allows using subaddressing for those who don't
491                           # have a catchall domain setup.
492                           {'alias_name': 'jobs+'+vals['name']},
493                           model_name="hr.applicant",
494                           context=context)
495             vals['alias_id'] = alias_id
496         res = super(hr_job, self).create(cr, uid, vals, context)
497         mail_alias.write(cr, uid, [vals['alias_id']], {"alias_defaults": {'job_id': res}}, context)
498         return res
499
500     def unlink(self, cr, uid, ids, context=None):
501         # Cascade-delete mail aliases as well, as they should not exist without the job position.
502         mail_alias = self.pool.get('mail.alias')
503         alias_ids = [job.alias_id.id for job in self.browse(cr, uid, ids, context=context) if job.alias_id]
504         res = super(hr_job, self).unlink(cr, uid, ids, context=context)
505         mail_alias.unlink(cr, uid, alias_ids, context=context)
506         return res
507
508     def action_print_survey(self, cr, uid, ids, context=None):
509         if context is None:
510             context = {}
511         datas = {}
512         record = self.browse(cr, uid, ids, context=context)[0]
513         if record.survey_id:
514             datas['ids'] = [record.survey_id.id]
515         datas['model'] = 'survey.print'
516         context.update({'response_id': [0], 'response_no': 0,})
517         return {
518             'type': 'ir.actions.report.xml',
519             'report_name': 'survey.form',
520             'datas': datas,
521             'context' : context,
522             'nodestroy':True,
523         }
524
525 class applicant_category(osv.osv):
526     """ Category of applicant """
527     _name = "hr.applicant_category"
528     _description = "Category of applicant"
529     _columns = {
530         'name': fields.char('Name', size=64, required=True, translate=True),
531     }
532
533 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: