[IMP]: base_action_rule: Improvement for computing regex on history and number of...
[odoo/odoo.git] / addons / crm / crm_action_rule.py
1 # -*- coding: utf-8 -*-
2 ##############################################################################
3 #
4 #    OpenERP, Open Source Management Solution
5 #    Copyright (C) 2004-2010 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 import re
24 import os
25 import base64
26 import tools
27 import mx.DateTime
28
29 from tools.translate import _
30 from osv import fields
31 from osv import osv
32 from osv import orm
33 from osv.orm import except_orm
34
35 import crm
36
37 class base_action_rule(osv.osv):
38     """ Base Action Rule """
39     _inherit = 'base.action.rule'
40     _description = 'Action Rules'
41     
42     _columns = {
43         'trg_section_id': fields.many2one('crm.case.section', 'Sales Team'), 
44         'trg_max_history': fields.integer('Maximum Communication History'), 
45         'trg_categ_id':  fields.many2one('crm.case.categ', 'Category'), 
46         'regex_history' : fields.char('Regular Expression on Case History', size=128), 
47         'act_section_id': fields.many2one('crm.case.section', 'Set Team to'), 
48         'act_categ_id': fields.many2one('crm.case.categ', 'Set Category to'), 
49         'act_mail_to_partner': fields.boolean('Mail to Partner', help="Check \
50 this if you want the rule to send an email to the partner."), 
51     }
52     
53
54     def email_send(self, cr, uid, obj, emails, body, emailfrom=tools.config.get('email_from', False), context={}):
55         body = self.format_mail(obj, body)
56         if not emailfrom:
57             if hasattr(obj, 'user_id')  and obj.user_id and obj.user_id.address_id and obj.user_id.address_id.email:
58                 emailfrom = obj.user_id.address_id.email
59             
60         name = '[%d] %s' % (obj.id, tools.ustr(obj.name))
61         emailfrom = tools.ustr(emailfrom)
62         if hasattr(obj, 'section_id') and obj.section_id and obj.section_id.reply_to:
63             reply_to = obj.section_id.reply_to
64         else:
65             reply_to = emailfrom
66         if not emailfrom:
67             raise osv.except_osv(_('Error!'), 
68                     _("No E-Mail ID Found for your Company address!"))
69         return tools.email_send(emailfrom, emails, name, body, reply_to=reply_to, openobject_id=str(obj.id))
70     
71     def do_check(self, cr, uid, action, obj, context={}):
72         """ @param self: The object pointer
73         @param cr: the current row, from the database cursor,
74         @param uid: the current user’s ID for security checks,
75         @param context: A standard dictionary for contextual values"""
76         ok = super(base_action_rule, self).do_check(cr, uid, action, obj, context=context)
77
78         if hasattr(obj, 'section_id'):
79             ok = ok and (not action.trg_section_id or action.trg_section_id.id==obj.section_id.id)
80         if hasattr(obj, 'categ_id'):
81             ok = ok and (not action.trg_categ_id or action.trg_categ_id.id==obj.categ_id.id)
82
83         #Cheking for history 
84         regex = action.regex_history
85         result_history = True
86         if regex:
87             res = False
88             ptrn = re.compile(str(regex))
89             for history in obj.message_ids:
90                 _result = ptrn.search(str(history.name))
91                 if _result:
92                     res = True
93                     break
94             result_history = res
95         ok = ok and (not regex or result_history)
96
97         res_count = True
98         if action.trg_max_history:
99             res_count = False
100             if len(obj.message_ids) >= action.trg_max_history:
101                 res_count = True
102         ok = ok and res_count
103         return ok
104
105     def do_action(self, cr, uid, action, model_obj, obj, context={}):
106         """ @param self: The object pointer
107         @param cr: the current row, from the database cursor,
108         @param uid: the current user’s ID for security checks,
109         @param context: A standard dictionary for contextual values """
110         res = super(base_action_rule, self).do_action(cr, uid, action, model_obj, obj, context=context)
111         write = {}
112         
113         if hasattr(action, 'act_section_id') and action.act_section_id:
114             obj.section_id = action.act_section_id
115             write['section_id'] = action.act_section_id.id
116
117         if hasattr(obj, 'email_cc') and action.act_email_cc:
118             if '@' in (obj.email_cc or ''):
119                 emails = obj.email_cc.split(",")
120                 if  obj.act_email_cc not in emails:# and '<'+str(action.act_email_cc)+">" not in emails:
121                     write['email_cc'] = obj.email_cc+','+obj.act_email_cc
122             else:
123                 write['email_cc'] = obj.act_email_cc
124         
125         model_obj.write(cr, uid, [obj.id], write, context)
126         emails = []
127
128         if hasattr(obj, 'email_from') and action.act_mail_to_partner:
129             emails.append(obj.email_from)
130         emails = filter(None, emails)
131         if len(emails) and action.act_mail_body:
132             emails = list(set(emails))
133             self.email_send(cr, uid, obj, emails, action.act_mail_body)
134         return True
135
136
137     def state_get(self, cr, uid, context={}):
138         """Gets available states for crm
139         @param self: The object pointer
140         @param cr: the current row, from the database cursor,
141         @param uid: the current user’s ID for security checks,
142         @param context: A standard dictionary for contextual values """
143         res = super(base_action_rule, self).state_get(cr, uid, context=context)
144         return res  + crm.AVAILABLE_STATES
145
146     def priority_get(self, cr, uid, context={}):
147         """@param self: The object pointer
148         @param cr: the current row, from the database cursor,
149         @param uid: the current user’s ID for security checks,
150         @param context: A standard dictionary for contextual values """
151         res = super(base_action_rule, self).priority_get(cr, uid, context=context)
152         return res + crm.AVAILABLE_PRIORITIES
153
154 base_action_rule()
155
156 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: