knowledge/cmis_write/ir_attachment.py
2014-07-18 17:52:20 -04:00

133 lines
5.1 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_dms': fields.char('Id of Dms', size=256, help="Id of Dms."),
}
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 = {
'name': values.get('name'),
'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]
values['datas'] = None
res = super(ir_attachment, self).create(cr, uid, values,
context=context)
# Create Job
# if bool_testdoc in context, we don't need to create
# the doc in the DMS
if not context.get('bool_testdoc'):
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
repo = cmis_backend_obj._auth(session.cr, session.uid,
context=session.context)
root = repo.rootFolder
ids = cmis_backend_obj.search(session.cr, session.uid, [], session.context)
folder_path = cmis_backend_obj.read(
session.cr, session.uid,
ids,
['initial_directory_write'],
context=session.context)[0]['initial_directory_write']
# Document properties
if value['name']:
file_name = value['name']
elif value['datas_fname']:
file_name = value['datas_fname']
else:
file_name = value['datas_fname']
props = {
'cmis:name': file_name,
'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'],
contentString=base64.b64decode(
value['datas']),
contentType=value['file_type'])
# TODO: create custom properties on a document (Alfresco)
#someDoc.getProperties().update(props)
# Updating ir.attachment object with the new id
# of document generated by DMS
ir_attach_obj.write(session.cr, session.uid, res, {
'id_dms': someDoc.getObjectId()}, session.context)
return True
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: