addons/base_status/base_state.py

   1 # -*- coding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-today OpenERP SA (<http://www.openerp.com>)
   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 osv import fields, osv
  23 from tools.translate import _
  24
  25 class base_state(object):
  26     """ Base utility mixin class for objects willing to manage their state.
  27         Object subclassing this class should define the following colums:
  28         - ``date_open`` (datetime field)
  29         - ``date_closed`` (datetime field)
  30         - ``user_id`` (many2one to res.users)
  31         - ``partner_id`` (many2one to res.partner)
  32         - ``email_from`` (char field)
  33         - ``state`` (selection field)
  34     """
  35
  36     def _get_default_partner(self, cr, uid, context=None):
  37         """ Gives id of partner for current user
  38             :param context: if portal not in context returns False
  39         """
  40         if context is None:
  41             context = {}
  42         if not context or not context.get('portal'):
  43             return False
  44         user = self.pool.get('res.users').browse(cr, uid, uid, context=context)
  45         if hasattr(user, 'partner_address_id') and user.partner_address_id:
  46             return user.partner_address_id
  47         return user.company_id.partner_id.id
  48
  49     def _get_default_email(self, cr, uid, context=None):
  50         """ Gives default email address for current user
  51             :param context: if portal not in context returns False
  52         """
  53         if context is None:
  54             context = {}
  55         if not context or not context.get('portal'):
  56             return False
  57         user = self.pool.get('res.users').browse(cr, uid, uid, context=context)
  58         return user.email
  59
  60     def _get_default_user(self, cr, uid, context=None):
  61         """ Gives current user id
  62             :param context: if portal not in context returns False
  63         """
  64         if context is None:
  65             context = {}
  66         if not context or not context.get('portal'):
  67             return False
  68         return uid
  69
  70     def onchange_partner_address_id(self, cr, uid, ids, add, email=False):
  71         """ This function returns value of partner email based on Partner Address
  72             :param add: Id of Partner's address
  73             :param email: Partner's email ID
  74         """
  75         data = {'value': {'email_from': False, 'phone':False}}
  76         if add:
  77             address = self.pool.get('res.partner').browse(cr, uid, add)
  78             data['value'] = {'email_from': address and address.email or False ,
  79                              'phone':  address and address.phone or False}
  80         if 'phone' not in self._columns:
  81             del data['value']['phone']
  82         return data
  83
  84     def onchange_partner_id(self, cr, uid, ids, part, email=False):
  85         """ This function returns value of partner address based on partner
  86             :param part: Partner's id
  87             :param email: Partner's email ID
  88         """
  89         data={}
  90         if  part:
  91             addr = self.pool.get('res.partner').address_get(cr, uid, [part], ['contact'])
  92             data.update(self.onchange_partner_address_id(cr, uid, ids, addr['contact'])['value'])
  93         return {'value': data}
  94
  95     def case_escalate(self, cr, uid, ids, context=None):
  96         """ Escalates case to parent level """
  97         cases = self.browse(cr, uid, ids, context=context)
  98         cases[0].state # fill browse record cache, for _action having old and new values
  99         data = {'active': True}
 100         for case in cases:
 101             parent_id = case.section_id.parent_id
 102             if parent_id:
 103                 data['section_id'] = parent_id.id
 104                 if parent_id.change_responsible and parent_id.user_id:
 105                     data['user_id'] = parent_id.user_id.id
 106             else:
 107                 raise osv.except_osv(_('Error !'), _('You can not escalate, you are already at the top level regarding your sales-team category.'))
 108             self.write(cr, uid, [case.id], data, context=context)
 109             case.case_escalate_send_note(parent_id, context=context)
 110         self._action(cr, uid, cases, 'escalate', context=context)
 111         return True
 112
 113     def case_open(self, cr, uid, ids, context=None):
 114         """ Opens case """
 115         cases = self.browse(cr, uid, ids, context=context)
 116         for case in cases:
 117             values = {'active': True}
 118             if case.state == 'draft':
 119                 values['date_open'] = fields.datetime.now()
 120             if not case.user_id:
 121                 values['user_id'] = uid
 122             self.case_set(cr, uid, [case.id], 'open', values, context=context)
 123             self.case_open_send_note(cr, uid, [case.id], context=context)
 124         return True
 125
 126     def case_close(self, cr, uid, ids, context=None):
 127         """ Closes case """
 128         self.case_set(cr, uid, ids, 'done', {'date_closed': fields.datetime.now()}, context=context)
 129         self.case_close_send_note(cr, uid, ids, context=context)
 130         return True
 131
 132     def case_cancel(self, cr, uid, ids, context=None):
 133         """ Cancels case """
 134         self.case_set(cr, uid, ids, 'cancel', {'active': True}, context=context)
 135         self.case_cancel_send_note(cr, uid, ids, context=context)
 136         return True
 137
 138     def case_pending(self, cr, uid, ids, context=None):
 139         """ Sets case as pending """
 140         self.case_set(cr, uid, ids, 'pending', {'active': True}, context=context)
 141         self.case_pending_send_note(cr, uid, ids, context=context)
 142         return True
 143
 144     def case_reset(self, cr, uid, ids, context=None):
 145         """ Resets case as draft """
 146         self.case_set(cr, uid, ids, 'draft', {'active': True}, context=context)
 147         self.case_reset_send_note(cr, uid, ids, context=context)
 148         return True
 149
 150     def case_set(self, cr, uid, ids, state_name, update_values=None, context=None):
 151         """ Generic method for setting case. This methods wraps the update
 152             of the record, as well as call to _action and browse_record
 153             case setting to fill the cache.
 154
 155             :params: state_name: the new value of the state, such as
 156                      'draft' or 'close'.
 157             :params: update_values: values that will be added with the state
 158                      update when writing values to the record.
 159         """
 160         cases = self.browse(cr, uid, ids, context=context)
 161         cases[0].state # fill browse record cache, for _action having old and new values
 162         if update_values is None:
 163             update_values = {}
 164         update_values['state'] = state_name
 165         self.write(cr, uid, ids, update_values, context=context)
 166         self._action(cr, uid, cases, state_name, context=context)
 167
 168     def _action(self, cr, uid, cases, state_to, scrit=None, context=None):
 169         if context is None:
 170             context = {}
 171         context['state_to'] = state_to
 172         rule_obj = self.pool.get('base.action.rule')
 173         model_obj = self.pool.get('ir.model')
 174         model_ids = model_obj.search(cr, uid, [('model','=',self._name)])
 175         rule_ids = rule_obj.search(cr, uid, [('model_id','=',model_ids[0])])
 176         return rule_obj._action(cr, uid, rule_ids, cases, scrit=scrit, context=context)
 177
 178     # ******************************
 179     # Notifications
 180     # ******************************
 181
 182         def case_get_note_msg_prefix(self, cr, uid, id, context=None):
 183                 return ''
 184
 185     def case_open_send_note(self, cr, uid, ids, context=None):
 186         for id in ids:
 187             msg = _('%s has been <b>opened</b>.') % (self.case_get_note_msg_prefix(cr, uid, id, context=context))
 188             self.message_append_note(cr, uid, [id], body=msg, context=context)
 189         return True
 190
 191     def case_escalate_send_note(self, cr, uid, ids, new_section=None, context=None):
 192         for id in ids:
 193             if new_section:
 194                 msg = '%s has been <b>escalated</b> to <b>%s</b>.' % (self.case_get_note_msg_prefix(cr, uid, id, context=context), new_section.name)
 195             else:
 196                 msg = '%s has been <b>escalated</b>.' % (self.case_get_note_msg_prefix(cr, uid, id, context=context))
 197             self.message_append_note(cr, uid, [id], 'System Notification', msg, context=context)
 198         return True
 199
 200     def case_close_send_note(self, cr, uid, ids, context=None):
 201         for id in ids:
 202             msg = _('%s has been <b>closed</b>.') % (self.case_get_note_msg_prefix(cr, uid, id, context=context))
 203             self.message_append_note(cr, uid, [id], body=msg, context=context)
 204         return True
 205
 206     def case_cancel_send_note(self, cr, uid, ids, context=None):
 207         for id in ids:
 208             msg = _('%s has been <b>canceled</b>.') % (self.case_get_note_msg_prefix(cr, uid, id, context=context))
 209             self.message_append_note(cr, uid, [id], body=msg, context=context)
 210         return True
 211
 212     def case_pending_send_note(self, cr, uid, ids, context=None):
 213         for id in ids:
 214             msg = _('%s is now <b>pending</b>.') % (self.case_get_note_msg_prefix(cr, uid, id, context=context))
 215             self.message_append_note(cr, uid, [id], body=msg, context=context)
 216         return True
 217
 218     def case_reset_send_note(self, cr, uid, ids, context=None):
 219         for id in ids:
 220             msg = _('%s has been <b>renewed</b>.') % (self.case_get_note_msg_prefix(cr, uid, id, context=context))
 221             self.message_append_note(cr, uid, [id], body=msg, context=context)
 222         return True