# -*- coding: utf-8 -*- ############################################################################## # # Author: Guewen Baconnier # Copyright 2012 Camptocamp SA # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Affero General Public License as # published by the Free Software Foundation, either version 3 of the # License, or (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Affero General Public License for more details. # # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see . # ############################################################################## import logging from cStringIO import StringIO from openerp.osv import osv, orm, fields from openerp import tools from openerp.tools.translate import _ from .pingen import Pingen, APIError, ConnectionError, POST_SENDING_STATUS # TODO should be configurable TOKEN = '6bc041af6f02854461ef31c2121ef853' STAGING = True _logger = logging.getLogger(__name__) class pingen_document(orm.Model): """ A pingen document is the state of the synchronization of an attachment with pingen.com It stores the configuration and the current state of the synchronization. It also serves as a queue of documents to push to pingen.com """ _name = 'pingen.document' _inherits = {'ir.attachment': 'attachment_id'} _columns = { 'attachment_id': fields.many2one( 'ir.attachment', 'Document', required=True, readonly=True, ondelete='cascade'), 'state': fields.selection( [('pending', 'Pending'), ('pushed', 'Pushed'), ('sendcenter', 'In Sendcenter'), ('sent', 'Sent'), ('error', 'Connection Error'), ('pingen_error', 'Pingen Error'), ('canceled', 'Canceled')], string='State', readonly=True, required=True), 'date': fields.datetime('Creation Date', readonly=True), 'push_date': fields.datetime('Push Date', readonly=True), # for `error` and `pingen_error` states when we push 'last_error_message': fields.text('Error Message', readonly=True), # pingen IDs 'pingen_id': fields.integer( 'Pingen ID', readonly=True, help="ID of the document in the Pingen Documents"), 'post_id': fields.integer( 'Pingen Post ID', readonly=True, help="ID of the document in the Pingen Sendcenter"), # sendcenter infos 'post_status': fields.char('Post Status', size=128, readonly=True), 'parsed_address': fields.text('Parsed Address', readonly=True), 'cost': fields.float('Cost', readonly=True), 'currency_id': fields.many2one('res.currency', 'Currency', readonly=True), 'country_id': fields.many2one('res.country', 'Country', readonly=True), 'send_date': fields.datetime('Date of sending', readonly=True), 'pages': fields.integer('Pages', readonly=True), } _defaults = { 'state': 'pending', } _sql_constraints = [ ('pingen_document_attachment_uniq', 'unique (attachment_id)', 'Only one Pingen document is allowed per attachment.'), ] def _pingen(self, cr, uid, ids, context=None): """ Return a Pingen instance to work on """ assert not ids, "ids is there by convention, should not be used" # TODO parameterize return Pingen(TOKEN, staging=STAGING) def _push_to_pingen(self, cr, uid, document, context=None): """ Push a document to pingen.com """ attachment_obj = self.pool.get('ir.attachment') decoded_document = attachment_obj._decoded_content( cr, uid, document.attachment_id, context=context) pingen = self._pingen(cr, uid, [], context=context) try: doc_id, post_id, infos = pingen.push_document( document.datas_fname, StringIO(decoded_document), document.pingen_send, document.pingen_speed, document.pingen_color) except ConnectionError as e: _logger.exception( 'Connection Error when pushing Pingen Document %s to %s.' % (document.id, pingen.url)) raise except APIError as e: _logger.error( 'API Error when pushing Pingen Document %s to %s.' % (document.id, pingen.url)) raise document.write( {'last_error_message': False, 'state': 'sendcenter' if post_id else 'pushed', 'push_date': infos['date'], 'pingen_id': doc_id, 'post_id': post_id}, context=context) _logger.info('Pingen Document %s: pushed to %s' % (document.id, pingen.url)) def push_to_pingen(self, cr, uid, ids, context=None): """ Push a document to pingen.com Convert errors to osv.except_osv to be handled by the client. Wrapper method for multiple ids (when triggered from button for instance) for public interface. """ for document in self.browse(cr, uid, ids, context=context): try: self._push_to_pingen(cr, uid, document, context=context) except ConnectionError as e: raise osv.except_osv( _('Pingen Connection Error'), _('Connection Error when asking for sending the document %s to Pingen') % document.name) except APIError as e: raise osv.except_osv( _('Pingen Error'), _('Error when asking Pingen to send the document %s: ' '\n%s') % (document.name, e)) return True def _push_and_send_to_pingen_silent(self, cr, uid, ids, context=None): """ Push a document to pingen.com Instead of raising, store the error in the pingen.document """ if not ids: ids = self.search( cr, uid, # do not retry pingen_error, they should be treated manually [('state', 'in', ['pending', 'pushed', 'error'])], context=context) for document in self.browse(cr, uid, ids, context=context): try: if not document.pingen_id: self._push_to_pingen(cr, uid, document, context=context) if not document.post_id and document.pingen_send: self._ask_pingen_send(cr, uid, document, context=context) except ConnectionError as e: document.write({'last_error_message': e, 'state': 'error'}, context=context) except APIError as e: document.write({'last_error_message': e, 'state': 'pingen_error'}, context=context) return True def _ask_pingen_send(self, cr, uid, document, context=None): """ For a document already pushed to pingen, ask to send it. """ # sending has been explicitely asked so we change the option # for consistency if not document.pingen_send: document.write({'pingen_send': True}, context=context) pingen = self._pingen(cr, uid, [], context=context) try: post_id = pingen.send_document( document.pingen_id, document.pingen_speed, document.pingen_color) except ConnectionError as e: _logger.exception('Connection Error when asking for sending Pingen Document %s to %s.' % (document.id, pingen.url)) raise except APIError as e: _logger.exception('API Error when asking for sending Pingen Document %s to %s.' % (document.id, pingen.url)) raise document.write( {'last_error_message': False, 'state': 'sendcenter', 'post_id': post_id}, context=context) _logger.info('Pingen Document %s: asked for sending to %s' % (document.id, pingen.url)) return True def ask_pingen_send(self, cr, uid, ids, context=None): """ For a document already pushed to pingen, ask to send it. Wrapper method for multiple ids (when triggered from button for instance) for public interface. """ for document in self.browse(cr, uid, ids, context=context): try: self._ask_pingen_send(cr, uid, document, context=context) except ConnectionError as e: raise osv.except_osv( _('Pingen Connection Error'), _('Connection Error when asking for ' 'sending the document %s to Pingen') % document.name) except APIError as e: raise osv.except_osv( _('Pingen Error'), _('Error when asking Pingen to send the document %s: ' '\n%s') % (document.name, e)) return True def _update_post_infos(self, cr, uid, document, context=None): """ Update the informations from pingen of a document in the Sendcenter """ if not document.post_id: return pingen = self._pingen(cr, uid, [], context=context) try: post_infos = pingen.post_infos(document.post_id) except ConnectionError as e: _logger.exception( 'Connection Error when asking for ' 'sending Pingen Document %s to %s.' % (document.id, pingen.url)) raise except APIError as e: _logger.exception( 'API Error when asking for sending Pingen Document %s to %s.' % (document.id, pingen.url)) raise currency_ids = self.pool.get('res.currency').search( cr, uid, [('name', '=', post_infos['currency'])], context=context) country_ids = self.pool.get('res.country').search( cr, uid, [('code', '=', post_infos['country'])], context=context) vals = { 'post_status': POST_SENDING_STATUS[post_infos['status']], 'cost': post_infos['cost'], 'currency_id': currency_ids[0] if currency_ids else False, 'parsed_address': post_infos['address'], 'country_id': country_ids[0] if country_ids else False, 'send_date': post_infos['date'], 'pages': post_infos['pages'], } if pingen.is_posted(post_infos): vals['state'] = 'sent' document.write(vals, context=context) _logger.info('Pingen Document %s: status updated' % document.id) def _update_post_infos_silent(self, cr, uid, ids, context=None): """ Update the informations from pingen of a document in the Sendcenter Do not raise errors, only skip the update of the record. """ if not ids: ids = self.search( cr, uid, [('state', '=', 'sendcenter')], context=context) for document in self.browse(cr, uid, ids, context=context): try: self._update_post_infos(cr, uid, document, context=context) except (ConnectionError, APIError): # Intended silented exception, we can consider that it's not # important if the update not worked, that's # only informative, and it will be retried the next time # In any case, the error has been by _update_post_infos pass return True def update_post_infos(self, cr, uid, ids, context=None): """ Update the informations from pingen of a document in the Sendcenter Wrapper method for multiple ids (when triggered from button for instance) for public interface. """ for document in self.browse(cr, uid, ids, context=context): try: self._update_post_infos(cr, uid, document, context=context) except ConnectionError as e: raise osv.except_osv( _('Pingen Connection Error'), _('Connection Error when updating the status of Document %s' ' from Pingen') % document.name) except APIError as e: raise osv.except_osv( _('Pingen Error'), _('Error when updating the status of Document %s from Pingen: ' '\n%s') % (document.name, e)) return True