From 712eee6e3d87b1d34a5e9c44cc7b496ff0d72153 Mon Sep 17 00:00:00 2001
From: "ajay javiya (OpenERP)" <aja@tinyerp.com>
Date: Tue, 24 Sep 2013 16:22:10 +0530
Subject: [PATCH] [ADD]: Warning message if user try to post a internal comment
 and he/she is not an employee

bzr revid: aja@tinyerp.com-20130924105210-rrfk2z6du7hiblr6
---
 addons/mail/mail_thread.py | 22 ++++++++++++++++------
 1 file changed, 16 insertions(+), 6 deletions(-)

diff --git a/addons/mail/mail_thread.py b/addons/mail/mail_thread.py
index 15123168811..381c74e2f53 100644
--- a/addons/mail/mail_thread.py
+++ b/addons/mail/mail_thread.py
@@ -35,6 +35,7 @@ from openerp.addons.mail.mail_message import decode
 from openerp.osv import fields, osv, orm
 from openerp.tools.safe_eval import safe_eval as eval
 from openerp.tools.translate import _
+from lxml import etree
 
 _logger = logging.getLogger(__name__)
 
@@ -1313,6 +1314,12 @@ class mail_thread(osv.AbstractModel):
                 "Invalid thread_id; should be 0, False, an ID or a list with one ID"
         if isinstance(thread_id, (list, tuple)):
             thread_id = thread_id[0]
+            
+        #check login user is employee
+        #raise error on comment message if user is not employee
+        employee = self._check_user_is_employee(cr, uid, context=context)
+        if not employee and type == 'comment':
+            raise orm.except_orm(_('Access Denied'),_('Restricted to employees.'))
 
         # if we're processing a message directly coming from the gateway, the destination model was
         # set in the context.
@@ -1661,18 +1668,21 @@ class mail_thread(osv.AbstractModel):
             threads.append(data)
         return sorted(threads, key=lambda x: (x['popularity'], x['id']), reverse=True)[:3]
 
-    def fields_view_get(self, cr, uid, view_id=None, view_type='form', context=None, toolbar=False, submenu=False):
-        if context is None: context = {}
+    def _check_user_is_employee(self, cr, uid, context=None):
         flag = 0
-        from lxml import etree
-        group =  self.pool.get('ir.model.data').get_object_reference(cr, uid, 'base', 'group_user')
+        group = self.pool.get('ir.model.data').get_object_reference(cr, uid, 'base', 'group_user')
         if group:
             users = self.pool.get('res.groups').read(cr, uid, group[1], ['users'])['users']
             flag = uid in users and 1 or 0
+        return flag
+
+    def fields_view_get(self, cr, uid, view_id=None, view_type='form', context=None, toolbar=False, submenu=False):
+        if context is None: 
+            context = {}
         res = super(mail_thread, self).fields_view_get(cr, uid, view_id=view_id, view_type=view_type, context=context, toolbar=toolbar, submenu=False)
         doc = etree.XML(res['arch'])
         if view_type == 'form':
             for node in doc.xpath("//field[@name='message_ids']"):
-                node.set('display_log_button',str(flag))
-        res['arch'] = etree.tostring(doc)
+                node.set('display_log_button', str(self._check_user_is_employee(cr, uid, context=context)))
+            res['arch'] = etree.tostring(doc)
         return res