knowledge/cmis_write/ir_attachment.py

107 lines
4.5 KiB
Python

# -*- encoding: utf-8 -*-
##############################################################################
#
# OpenERP, Open Source Management Solution
# This module copyright (C) 2014 Savoir-faire Linux
# (<http://www.savoirfairelinux.com>).
#
# 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/>.
#
##############################################################################
from openerp.osv import orm, fields
from openerp.addons.connector.session import ConnectorSession
from openerp.addons.connector.queue.job import job
import base64
import logging
_logger = logging.getLogger(__name__)
class ir_attachment(orm.Model):
_inherit = 'ir.attachment'
_columns = {
'id_edm': fields.char('Id of Dms', size=256, help="Id of Edm."),
}
def create(self, cr, uid, values, context=None):
metadata_obj = self.pool.get('metadata')
user_obj = self.pool.get('res.users')
user_login = user_obj.browse(cr, uid, uid, context=context).login
session = ConnectorSession(cr, uid, context=context)
value = {
'datas_fname': values.get('datas_fname'),
'file_type': values.get('file_type') or '',
'datas': values.get('datas'),
'description': values.get('description') or '',
}
metadata_ids = metadata_obj.search(cr, uid, [], context=context)
dict_metadata = {}
list_fields = []
# Get list of metadata
if values.get('res_model'):
for line in metadata_obj.browse(cr, uid, metadata_ids, context=context):
if line.model_id.model == values.get('res_model'):
if line.metadata_list_ids:
for one_field in line.metadata_list_ids:
list_fields.append(one_field.field_id.name)
result = self.pool.get(values.get('res_model')).read(cr, uid, [values.get('res_id')],
list_fields, context=context)[0]
for one_field in list_fields:
dict_metadata['cmis:' + one_field] = result[one_field]
res = super(ir_attachment, self).create(cr, uid, values, context=context)
# Create Job
create_doc_in_edm.delay(session, 'ir.attachment', value, res, dict_metadata, user_login)
return res
@job
def create_doc_in_edm(session, model_name, value, res, dict_metadata, user_login, filters=None):
ir_attach_obj = session.pool.get('ir.attachment')
cmis_backend_obj = session.pool.get('cmis.backend')
if session.context is None:
session.context = {}
#login with the cmis account
client = cmis_backend_obj._auth(session.cr, session.uid, context=session.context)
repo = client.defaultRepository
root = repo.rootFolder
ids = cmis_backend_obj.search(session.cr, session.uid, [])
folder_path = cmis_backend_obj.read(session.cr, session.uid,
ids, ['initial_directory_write'],
context=session.context)[0]['initial_directory_write']
# Document properties
props = {
'cmis:name': value['datas_fname'],
'cmis:description': value['description'],
'cmis:createdBy': user_login,
}
# Add list of metadata in props
if len(dict_metadata):
for k, v in dict_metadata.iteritems():
props[k] = v
if folder_path:
sub1 = repo.getObjectByPath(folder_path)
else:
sub1 = root
someDoc = sub1.createDocumentFromString(value['datas_fname'], props,
contentString=base64.b64decode(value['datas']),
contentType=value['file_type'])
# Updating ir.attachment object with the new id of document generated by DMS
ir_attach_obj.write(session.cr, session.uid, res, {'id_edm': someDoc.getObjectId()}, session.context)
return True
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: