From 8e691f8a881cdc12d4321c7a264a4e9e4b781120 Mon Sep 17 00:00:00 2001
From: Florian da Costa <florian.dacosta@akretion.com.br>
Date: Fri, 13 Dec 2013 23:12:10 +0100
Subject: [PATCH] [IMP] Add possibility to create condition in the fetchmail
 server which will allow to create a file document depending on the file name,
 the email and the subject

---
 fetchmail.py           |  3 ++
 fetchmail_view.xml     |  3 ++
 file_document.py       | 62 ++++++++++++++++++++++++++++++++++++++----
 file_document_view.xml | 26 ++++++++++++++++++
 4 files changed, 89 insertions(+), 5 deletions(-)

diff --git a/fetchmail.py b/fetchmail.py
index 9710338b..be4082bb 100644
--- a/fetchmail.py
+++ b/fetchmail.py
@@ -36,6 +36,7 @@ class fetchmail_server(orm.Model):
         'file_type': fields.selection(_get_file_type, 'File Type',
                 help='The file type will show some special option'),
         'company_id': fields.many2one('res.company', 'Company', required=True),#Why this field do not exist by default?
+        'file_document_condition_ids': fields.one2many('prepare.file.document', 'server_id', 'File Document ')
     }
 
     _defaults = {
@@ -49,9 +50,11 @@ class fetchmail_server(orm.Model):
         else:
             ctx = context.copy()
         ctx['default_file_document_vals'] = {}
+        ctx['default_file_document_condition_ids'] = {}
         server = self.browse(cr, uid, server_id, context=context)
         ctx['default_company_id'] = server.company_id.id
         ctx['default_fetchmail_server_id'] = server_id
+        ctx['default_file_document_condition_ids'] = [cond.id for cond in server.file_document_condition_ids]
         return ctx
 
     def fetch_mail(self, cr, uid, ids, context=None):
diff --git a/fetchmail_view.xml b/fetchmail_view.xml
index f8f91110..9c7b5fdb 100644
--- a/fetchmail_view.xml
+++ b/fetchmail_view.xml
@@ -12,6 +12,9 @@
                 <field name="object_id" position="after">
                     <field name="file_type"/> <!-- TODO attrs="{'invisible': [('object_id','!=', 'file.document')]}"/>-->
                 </field>
+                <field name="action_id" position="after">
+                    <field name="file_document_condition_ids" nolabel="1"/>
+                </field>
             </field>
         </record>
 
diff --git a/file_document.py b/file_document.py
index 7e26412c..86024277 100644
--- a/file_document.py
+++ b/file_document.py
@@ -21,6 +21,7 @@
 ###############################################################################
 
 from openerp.osv import fields, orm
+import base64
 
 
 class file_document(orm.Model):
@@ -30,10 +31,6 @@ class file_document(orm.Model):
         'fetchmail_server_id': fields.many2one('fetchmail.server', 'Email Server'),
     }
 
-    _sql_constraints = [
-        ('fecthmail_server_ext_id_uniq', 'unique(fetchmail_server_id, ext_id)',
-            'The combination of Email Server and External id must be unique !'),
-    ]
 
 
     def message_process(self, cr, uid, model, message, custom_values=None,
@@ -66,6 +63,12 @@ class file_document(orm.Model):
                     content_subtype=content_subtype,
                     **kwargs)
  
+    def custom_data_for_file_document(self, cr, uid, msg, context=None):
+        return {}
+
+
+    def add_more_fields(self, cr, uid, vals, msg, context=None):
+        return {}
 
 
     def _prepare_data_for_file_document(self, cr, uid, msg, context=None):
@@ -76,7 +79,31 @@ class file_document(orm.Model):
         :return: a list of dictionnary that containt the file document data
         :rtype: list
         """
-        return []
+        res = []
+        doc_file_condition_obj = self.pool.get('prepare.file.document')
+        cond_ids = context.get('default_file_document_condition_ids', False)
+        for cond in doc_file_condition_obj.browse(cr, uid, cond_ids):
+            vals = {}
+            if cond.type == 'normal':
+                if cond.from_email in msg['from'] and cond.mail_subject == msg['subject']:
+                    vals = {
+                        'name': msg['subject'],
+                        'direction': 'input',
+                        'date': msg['date'],
+                        'ext_id': msg['message_id'],
+            }
+                    #attachment_names = [att[0] for att in msg['attachments']]
+                    for att in msg['attachments']:
+                        if cond.file_extension in att[0]:
+                            vals['datas_fname'] = att[0]
+                            vals['datas'] = base64.b64encode(att[0][1])
+                            pass
+                    vals.update(self.add_more_fields(cr, uid, vals, msg, context=context)) 
+            else:
+                vals = eval('self.'+cond.type)(cr, uid, msg, context=context)
+            if 'datas_fname' in vals:
+                res.append(vals) 
+        return res
 
     def message_new(self, cr, uid, msg, custom_values, context=None):
         created_ids = []
@@ -94,3 +121,28 @@ class file_document(orm.Model):
             return created_ids[0]
         return None
 
+
+class prepare_file_document(orm.Model):
+    _name = "prepare.file.document"
+    _description = "Prepare File Document"
+
+    def _get_prepare_file_document_type(self, cr, uid, context=None):
+        return self.get_prepare_file_document_type(cr, uid, context=context)
+
+    def get_prepare_file_document_type(self, cr, uid, context=None):
+        return [('normal', 'Normal')]
+
+    _columns = {
+        'from_email': fields.char('Email', size=64),
+        'mail_subject': fields.char('Mail Subject', size=64),
+        'type': fields.selection(_get_prepare_file_document_type,
+               'Type', help="Create your own type if the normal type do not correspond to your need", required=True),
+        'file_extension' : fields.char('File Extension', size=64, help="File extension or file name", required=True),
+        'server_id': fields.many2one('fetchmail.server', 'Server Mail'),
+    }
+
+
+    _defaults = {
+        'type': 'normal'
+    }
+
diff --git a/file_document_view.xml b/file_document_view.xml
index 1529e47d..adce9945 100644
--- a/file_document_view.xml
+++ b/file_document_view.xml
@@ -15,6 +15,32 @@
             </field>
         </record>
 
+        <!-- Views and menus for the new object prepare.file.document -->
+        <record id="view_prepare_file_document_tree" model="ir.ui.view">
+            <field name="name">view_prepare_file_document_tree</field>
+            <field name="model">prepare.file.document</field>
+            <field name="arch" type="xml">
+                <tree string="Configuration for File Document">
+                    <field name="from_email"/>
+                    <field name="mail_subject"/>
+                    <field name="type"/>
+                    <field name="file_extension"/>
+                    <field name="server_id"/>
+                </tree>
+           </field>
+        </record>
+
+        <record id="action_file_document_configuration" model="ir.actions.act_window">
+            <field name="name">Configuration</field>
+            <field name="res_model">prepare.file.document</field>
+            <field name="view_type">form</field>
+            <field name="view_mode">tree,form</field>
+        </record>
+
+        <menuitem id="menu_file_document_configuration"
+                  action="action_file_document_configuration"
+                  parent="file_document.menu_file_exchange_root"
+                  sequence="50" />
 
     </data>
 </openerp>