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