[REF] Refactore prepare data for file document function + clear some code

This commit is contained in:
Florian da Costa 2013-12-17 12:00:25 +01:00
parent 8e691f8a88
commit 1396b72c96
3 changed files with 39 additions and 44 deletions

View File

@ -36,7 +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 ')
'file_document_condition_ids': fields.one2many('file.document.condition', 'server_id', 'File Document ')
}
_defaults = {
@ -50,11 +50,9 @@ 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):

View File

@ -63,12 +63,22 @@ 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_from_basic_condition(self, cr, uid, condition, msg, context=None):
vals = {}
if condition.from_email in msg['from'] and condition.mail_subject == msg['subject']:
for att in msg['attachments']:
if condition.file_extension in att[0]:
vals = {
'name': msg['subject'],
'direction': 'input',
'date': msg['date'],
'ext_id': msg['message_id'],
'datas_fname': att[0],
'datas': base64.b64encode(att[0][1])
}
break
return vals
def _prepare_data_for_file_document(self, cr, uid, msg, context=None):
@ -80,28 +90,16 @@ class file_document(orm.Model):
:rtype: list
"""
res = []
doc_file_condition_obj = self.pool.get('prepare.file.document')
cond_ids = context.get('default_file_document_condition_ids', False)
server_id = context.get('default_fetchmail_server_id', False)
doc_file_condition_obj = self.pool.get('file.document.condition')
cond_ids = doc_file_condition_obj.search(cr, uid, [('server_id', '=', server_id)])
if cond_ids:
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))
vals = self.prepare_data_from_basic_condition(cr, uid, cond, msg, context=context)
else:
vals = eval('self.'+cond.type)(cr, uid, msg, context=context)
if 'datas_fname' in vals:
vals = getattr(self, cond.type)(cr, uid, cond, msg, context=context)
if vals:
res.append(vals)
return res
@ -122,20 +120,19 @@ class file_document(orm.Model):
return None
class prepare_file_document(orm.Model):
_name = "prepare.file.document"
_description = "Prepare File Document"
class file_document_condition(orm.Model):
_name = "file.document.condition"
_description = "File Document Conditions"
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')]
def _get_file_document_condition_type(self, cr, uid, context=None):
return self.get_file_document_condition_type(cr, uid, context=context)
def get_file_document_condition_type(self, cr, uid, context=None):
return [('normal', 'Normal'), ('test', 'TEST')]
_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': fields.selection(_get_file_document_condition_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'),

View File

@ -16,9 +16,9 @@
</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>
<record id="view_file_document_condition_tree" model="ir.ui.view">
<field name="name">view_file_document_condition_tree</field>
<field name="model">file.document.condition</field>
<field name="arch" type="xml">
<tree string="Configuration for File Document">
<field name="from_email"/>
@ -32,7 +32,7 @@
<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="res_model">file.document.condition</field>
<field name="view_type">form</field>
<field name="view_mode">tree,form</field>
</record>