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