You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

227 lines
9.4 KiB

  1. # -*- encoding: utf-8 -*-
  2. ##############################################################################
  3. #
  4. # OpenERP, Open Source Management Solution
  5. # This module copyright (C) 2013 Therp BV (<http://therp.nl>)
  6. # All Rights Reserved
  7. #
  8. # This program is free software: you can redistribute it and/or modify
  9. # it under the terms of the GNU Affero General Public License as
  10. # published by the Free Software Foundation, either version 3 of the
  11. # License, or (at your option) any later version.
  12. #
  13. # This program is distributed in the hope that it will be useful,
  14. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  16. # GNU Affero General Public License for more details.
  17. #
  18. # You should have received a copy of the GNU Affero General Public License
  19. # along with this program. If not, see <http://www.gnu.org/licenses/>.
  20. #
  21. ##############################################################################
  22. import simplejson
  23. from lxml import etree
  24. from openerp.osv.orm import Model, except_orm, browse_null
  25. from openerp.tools.translate import _
  26. from openerp.osv import fields
  27. from openerp.addons.fetchmail.fetchmail import logger
  28. from openerp.tools.misc import UnquoteEvalContext
  29. from openerp.tools.safe_eval import safe_eval
  30. class fetchmail_server(Model):
  31. _inherit = 'fetchmail.server'
  32. _columns = {
  33. 'folder_ids': fields.one2many(
  34. 'fetchmail.server.folder', 'server_id', 'Folders'),
  35. }
  36. _defaults = {
  37. 'type': 'imap',
  38. }
  39. def __init__(self, pool, cr):
  40. self._columns['object_id'].required = False
  41. return super(fetchmail_server, self).__init__(pool, cr)
  42. def onchange_server_type(
  43. self, cr, uid, ids, server_type=False, ssl=False,
  44. object_id=False):
  45. retval = super(
  46. fetchmail_server, self).onchange_server_type(cr, uid,
  47. ids, server_type, ssl,
  48. object_id)
  49. retval['value']['state'] = 'draft'
  50. return retval
  51. def fetch_mail(self, cr, uid, ids, context=None):
  52. if context is None:
  53. context = {}
  54. check_original = []
  55. for this in self.browse(cr, uid, ids, context):
  56. if this.object_id:
  57. check_original.append(this.id)
  58. context.update(
  59. {
  60. 'fetchmail_server_id': this.id,
  61. 'server_type': this.type
  62. })
  63. connection = this.connect()
  64. for folder in this.folder_ids:
  65. logger.info('start checking for emails in %s server %s',
  66. folder.path, this.name)
  67. matcher = folder.get_algorithm()
  68. if connection.select(folder.path)[0] != 'OK':
  69. logger.error(
  70. 'Could not open mailbox %s on %s' % (
  71. folder.path, this.server))
  72. connection.select()
  73. continue
  74. result, msgids = connection.search(None, 'UNDELETED')
  75. if result != 'OK':
  76. logger.error(
  77. 'Could not search mailbox %s on %s' % (
  78. folder.path, this.server))
  79. continue
  80. for msgid in msgids[0].split():
  81. result, msgdata = connection.fetch(msgid, '(RFC822)')
  82. if result != 'OK':
  83. logger.error(
  84. 'Could not fetch %s in %s on %s' % (
  85. msgid, folder.path, this.server))
  86. continue
  87. mail_message = self.pool.get('mail.message').parse_message(
  88. msgdata[0][1], this.original)
  89. if self.pool.get('mail.message').search(cr, uid, [
  90. ('message_id', '=', mail_message['message-id'])]):
  91. continue
  92. found_ids = matcher.search_matches(
  93. cr, uid, folder,
  94. mail_message, msgdata[0][1])
  95. if found_ids and (len(found_ids) == 1 or
  96. folder.match_first):
  97. try:
  98. matcher.handle_match(
  99. cr, uid, connection,
  100. found_ids[0], folder, mail_message,
  101. msgdata[0][1], msgid, context)
  102. cr.commit()
  103. except Exception, e:
  104. cr.rollback()
  105. logger.exception(
  106. "Failed to fetch mail %s from %s",
  107. msgid, this.name)
  108. elif folder.flag_nonmatching:
  109. connection.store(msgid, '+FLAGS', '\\FLAGGED')
  110. connection.close()
  111. return super(fetchmail_server, self).fetch_mail(
  112. cr, uid, check_original, context)
  113. def attach_mail(
  114. self, cr, uid, ids, connection, object_id, folder,
  115. mail_message, msgid, context=None):
  116. for this in self.browse(cr, uid, ids, context):
  117. partner_id = None
  118. if folder.model_id.model == 'res.partner':
  119. partner_id = object_id
  120. if 'partner_id' in self.pool.get(folder.model_id.model)._columns:
  121. partner_id = self.pool.get(
  122. folder.model_id.model).browse(
  123. cr, uid, object_id, context
  124. ).partner_id.id
  125. self.pool.get('mail.message').create(
  126. cr, uid,
  127. {
  128. 'partner_id': partner_id,
  129. 'model': folder.model_id.model,
  130. 'res_id': object_id,
  131. 'body_text': mail_message.get('body'),
  132. 'body_html': mail_message.get('body_html'),
  133. 'subject': mail_message.get('subject'),
  134. 'email_to': mail_message.get('to'),
  135. 'email_from': mail_message.get('from'),
  136. 'email_cc': mail_message.get('cc'),
  137. 'reply_to': mail_message.get('reply'),
  138. 'date': mail_message.get('date'),
  139. 'message_id': mail_message.get('message-id'),
  140. 'subtype': mail_message.get('subtype'),
  141. 'headers': mail_message.get('headers'),
  142. },
  143. context)
  144. if this.attach:
  145. # TODO: create attachments
  146. pass
  147. if folder.delete_matching:
  148. connection.store(msgid, '+FLAGS', '\\DELETED')
  149. def button_confirm_login(self, cr, uid, ids, context=None):
  150. retval = super(fetchmail_server, self).button_confirm_login(cr, uid,
  151. ids,
  152. context)
  153. for this in self.browse(cr, uid, ids, context):
  154. this.write({'state': 'draft'})
  155. connection = this.connect()
  156. connection.select()
  157. for folder in this.folder_ids:
  158. if connection.select(folder.path)[0] != 'OK':
  159. raise except_orm(
  160. _('Error'), _('Mailbox %s not found!') %
  161. folder.path)
  162. folder.get_algorithm().search_matches(
  163. cr, uid, folder, browse_null(), '')
  164. connection.close()
  165. this.write({'state': 'done'})
  166. return retval
  167. def fields_view_get(self, cr, user, view_id=None, view_type='form',
  168. context=None, toolbar=False, submenu=False):
  169. result = super(fetchmail_server, self).fields_view_get(
  170. cr, user, view_id, view_type, context, toolbar, submenu)
  171. if view_type == 'form':
  172. view = etree.fromstring(
  173. result['fields']['folder_ids']['views']['form']['arch'])
  174. modifiers = {}
  175. docstr = ''
  176. for algorithm in self.pool.get('fetchmail.server.folder')\
  177. ._get_match_algorithms().itervalues():
  178. for modifier in ['required', 'readonly']:
  179. for field in getattr(algorithm, modifier + '_fields'):
  180. modifiers.setdefault(field, {})
  181. modifiers[field].setdefault(modifier, [])
  182. if modifiers[field][modifier]:
  183. modifiers[field][modifier].insert(0, '|')
  184. modifiers[field][modifier].append(
  185. ("match_algorithm", "==", algorithm.__name__))
  186. docstr += _(algorithm.name) + '\n' + _(algorithm.__doc__) + \
  187. '\n\n'
  188. for field in view:
  189. if field.tag == 'field' and field.get('name') in modifiers:
  190. field.set('modifiers', simplejson.dumps(
  191. dict(
  192. eval(field.attrib['modifiers'],
  193. UnquoteEvalContext({})),
  194. **modifiers[field.attrib['name']])))
  195. if (field.tag == 'field' and
  196. field.get('name') == 'match_algorithm'):
  197. field.set('help', docstr)
  198. result['fields']['folder_ids']['views']['form']['arch'] = \
  199. etree.tostring(view)
  200. return result