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