mirror of
https://github.com/OCA/report-print-send.git
synced 2025-02-16 07:11:31 +02:00
[IMP] rename pingen.task to pingen.document
This commit is contained in:
@
Anna Janiszewska
323
pingen/pingen_document.py
Normal file
323
pingen/pingen_document.py
Normal file
@@ -0,0 +1,323 @@
|
||||
# -*- 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 <http://www.gnu.org/licenses/>.
|
||||
#
|
||||
##############################################################################
|
||||
|
||||
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
|
||||
|
||||
Reference in New Issue
Block a user