151 lines
6.0 KiB
Python
151 lines
6.0 KiB
Python
#This file is part of Tryton. The COPYRIGHT file at the top level of this repository contains the full copyright notices and license terms.
|
|
"Attachment"
|
|
import os
|
|
import md5
|
|
import base64
|
|
from trytond.osv import fields, OSV
|
|
from trytond.config import CONFIG
|
|
|
|
class Attachment(OSV):
|
|
"Attachment"
|
|
_name = 'ir.attachment'
|
|
_description = __doc__
|
|
name = fields.Char('Attachment Name', required=True)
|
|
datas = fields.Function('get_datas', fnct_inv='set_datas',
|
|
type='binary', string='Datas')
|
|
description = fields.Text('Description')
|
|
res_model = fields.Char('Resource Model', readonly=True)
|
|
res_id = fields.Integer('Resource ID', readonly=True)
|
|
link = fields.Char('Link')
|
|
digest = fields.Char('Digest', size=32)
|
|
collision = fields.Integer('Collision')
|
|
datas_size = fields.Function('get_datas', type='integer',
|
|
string='Datas size')
|
|
|
|
def __init__(self):
|
|
super(Attachment, self).__init__()
|
|
self._sql_constraints += [
|
|
('res_model_res_id_name',
|
|
'UNIQUE (res_model, res_id, name)',
|
|
'Error! You can not create attachment with the same name!'),
|
|
]
|
|
|
|
def default_collision(self, cursor, user, context=None):
|
|
return 0
|
|
|
|
def get_datas(self, cursor, user, ids, name, arg, context=None):
|
|
res = {}
|
|
db_name = cursor.dbname
|
|
for attachment in self.browse(cursor, user, ids, context=context):
|
|
value = False
|
|
if name == 'datas_size':
|
|
value = 0
|
|
if attachment.digest:
|
|
filename = attachment.digest
|
|
if attachment.collision:
|
|
filename = filename + '-' + str(attachment.collision)
|
|
filename = os.path.join(CONFIG['data_path'], db_name,
|
|
filename[0:2], filename[2:4], filename)
|
|
if name == 'datas_size':
|
|
statinfo = os.stat(filename)
|
|
value = statinfo.st_size
|
|
else:
|
|
file_p = file(filename, 'rb')
|
|
value = base64.encodestring(file_p.read())
|
|
file_p.close()
|
|
res[attachment.id] = value
|
|
return res
|
|
|
|
def set_datas(self, cursor, user, obj_id, name, value, args, context=None):
|
|
if not value:
|
|
return
|
|
db_name = cursor.dbname
|
|
directory = os.path.join(CONFIG['data_path'], db_name)
|
|
if not os.path.isdir(directory):
|
|
os.makedirs(directory, 0770)
|
|
data = base64.decodestring(value)
|
|
digest = md5.new(data).hexdigest()
|
|
directory = os.path.join(directory, digest[0:2], digest[2:4])
|
|
if not os.path.isdir(directory):
|
|
os.makedirs(directory, 0770)
|
|
filename = os.path.join(directory, digest)
|
|
collision = 0
|
|
if os.path.isfile(filename):
|
|
file_p = file(filename, 'rb')
|
|
data2 = file_p.read()
|
|
file_p.close()
|
|
if data != data2:
|
|
cursor.execute('SELECT DISTINCT(collision) FROM ir_attachment ' \
|
|
'WHERE digest = %s ' \
|
|
'AND collision != 0 ' \
|
|
'ORDER BY collision', (digest,))
|
|
collision2 = 0
|
|
for row in cursor.fetchall():
|
|
collision2 = row[0]
|
|
filename = os.path.join(directory,
|
|
digest + '-' + str(collision2))
|
|
if os.path.isfile(filename):
|
|
file_p = file(filename, 'rb')
|
|
data2 = file_p.read()
|
|
file_p.close()
|
|
if data == data2:
|
|
collision = collision2
|
|
break
|
|
if collision == 0:
|
|
collision = collision2 + 1
|
|
filename = os.path.join(directory,
|
|
digest + '-' + str(collision))
|
|
file_p = file(filename, 'wb')
|
|
file_p.write(data)
|
|
file_p.close()
|
|
else:
|
|
file_p = file(filename, 'wb')
|
|
file_p.write(data)
|
|
file_p.close()
|
|
cursor.execute('UPDATE ir_attachment ' \
|
|
'SET digest = %s, ' \
|
|
'collision = %s ' \
|
|
'WHERE id = %s', (digest, collision, obj_id))
|
|
|
|
def check_access(self, cursor, user, ids, mode='read', context=None):
|
|
model_access_obj = self.pool.get('ir.model.access')
|
|
if user == 0:
|
|
return
|
|
if not ids:
|
|
return
|
|
if isinstance(ids, (int, long)):
|
|
ids = [ids]
|
|
model_names = set()
|
|
for attachment in self.browse(cursor, 0, ids, context=context):
|
|
if attachment.res_model:
|
|
model_names.add(attachment.res_model)
|
|
for model_name in model_names:
|
|
model_access_obj.check(cursor, user, model_name, mode=mode,
|
|
context=context)
|
|
|
|
def read(self, cursor, user, ids, fields_names=None, context=None,
|
|
load='_classic_read'):
|
|
self.check_access(cursor, user, ids, mode='read', context=context)
|
|
return super(Attachment, self).read(cursor, user, ids,
|
|
fields_names=fields_names, context=context, load=load)
|
|
|
|
def delete(self, cursor, user, ids, context=None):
|
|
self.check_access(cursor, user, ids, mode='delete', context=context)
|
|
return super(Attachment, self).delete(cursor, user, ids,
|
|
context=context)
|
|
|
|
def write(self, cursor, user, ids, vals, context=None):
|
|
self.check_access(cursor, user, ids, mode='write', context=context)
|
|
res = super(Attachment, self).write(cursor, user, ids, vals,
|
|
context=context)
|
|
self.check_access(cursor, user, ids, mode='write', context=context)
|
|
return res
|
|
|
|
def create(self, cursor, user, vals, context=None):
|
|
res = super(Attachment, self).create(cursor, user, vals,
|
|
context=context)
|
|
self.check_access(cursor, user, res, mode='create', context=context)
|
|
return res
|
|
|
|
Attachment()
|