1154 lines
40 KiB
Python
1154 lines
40 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.
|
|
import re
|
|
import heapq
|
|
|
|
from sql import Null
|
|
from sql.aggregate import Max
|
|
from sql.conditionals import Case
|
|
from collections import defaultdict
|
|
try:
|
|
import simplejson as json
|
|
except ImportError:
|
|
import json
|
|
|
|
from ..model import ModelView, ModelSQL, Workflow, fields, Unique
|
|
from ..report import Report
|
|
from ..wizard import Wizard, StateView, StateAction, Button
|
|
from ..transaction import Transaction
|
|
from ..cache import Cache
|
|
from ..pool import Pool
|
|
from ..pyson import Bool, Eval
|
|
from ..rpc import RPC
|
|
from .. import backend
|
|
from ..protocols.jsonrpc import JSONDecoder, JSONEncoder
|
|
from ..tools import is_instance_method, cursor_dict
|
|
try:
|
|
from ..tools.StringMatcher import StringMatcher
|
|
except ImportError:
|
|
from difflib import SequenceMatcher as StringMatcher
|
|
|
|
__all__ = [
|
|
'Model', 'ModelField', 'ModelAccess', 'ModelFieldAccess', 'ModelButton',
|
|
'ModelData', 'PrintModelGraphStart', 'PrintModelGraph', 'ModelGraph',
|
|
'ModelWorkflowGraph',
|
|
]
|
|
|
|
IDENTIFIER = re.compile(r'^[a-zA-z_][a-zA-Z0-9_]*$')
|
|
|
|
|
|
class Model(ModelSQL, ModelView):
|
|
"Model"
|
|
__name__ = 'ir.model'
|
|
_order_name = 'model'
|
|
name = fields.Char('Model Description', translate=True, loading='lazy',
|
|
states={
|
|
'readonly': Bool(Eval('module')),
|
|
},
|
|
depends=['module'])
|
|
model = fields.Char('Model Name', required=True,
|
|
states={
|
|
'readonly': Bool(Eval('module')),
|
|
},
|
|
depends=['module'])
|
|
info = fields.Text('Information',
|
|
states={
|
|
'readonly': Bool(Eval('module')),
|
|
},
|
|
depends=['module'])
|
|
module = fields.Char('Module',
|
|
help="Module in which this model is defined", readonly=True)
|
|
global_search_p = fields.Boolean('Global Search')
|
|
fields = fields.One2Many('ir.model.field', 'model', 'Fields',
|
|
required=True)
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(Model, cls).__setup__()
|
|
table = cls.__table__()
|
|
cls._sql_constraints += [
|
|
('model_uniq', Unique(table, table.model),
|
|
'The model must be unique!'),
|
|
]
|
|
cls._error_messages.update({
|
|
'invalid_module': ('Module name "%s" is not a valid python '
|
|
'identifier.'),
|
|
})
|
|
cls._order.insert(0, ('model', 'ASC'))
|
|
cls.__rpc__.update({
|
|
'list_models': RPC(),
|
|
'list_history': RPC(),
|
|
'global_search': RPC(),
|
|
})
|
|
|
|
@classmethod
|
|
def register(cls, model, module_name):
|
|
pool = Pool()
|
|
Property = pool.get('ir.property')
|
|
cursor = Transaction().connection.cursor()
|
|
|
|
ir_model = cls.__table__()
|
|
cursor.execute(*ir_model.select(ir_model.id,
|
|
where=ir_model.model == model.__name__))
|
|
model_id = None
|
|
if cursor.rowcount == -1 or cursor.rowcount is None:
|
|
data = cursor.fetchone()
|
|
if data:
|
|
model_id, = data
|
|
elif cursor.rowcount != 0:
|
|
model_id, = cursor.fetchone()
|
|
if not model_id:
|
|
cursor.execute(*ir_model.insert(
|
|
[ir_model.model, ir_model.name, ir_model.info,
|
|
ir_model.module],
|
|
[[model.__name__, model._get_name(), model.__doc__,
|
|
module_name]]))
|
|
Property._models_get_cache.clear()
|
|
cursor.execute(*ir_model.select(ir_model.id,
|
|
where=ir_model.model == model.__name__))
|
|
(model_id,) = cursor.fetchone()
|
|
elif model.__doc__:
|
|
cursor.execute(*ir_model.update(
|
|
[ir_model.name, ir_model.info],
|
|
[model._get_name(), model.__doc__],
|
|
where=ir_model.id == model_id))
|
|
return model_id
|
|
|
|
@classmethod
|
|
def validate(cls, models):
|
|
super(Model, cls).validate(models)
|
|
cls.check_module(models)
|
|
|
|
@classmethod
|
|
def check_module(cls, models):
|
|
'''
|
|
Check module
|
|
'''
|
|
for model in models:
|
|
if model.module and not IDENTIFIER.match(model.module):
|
|
cls.raise_user_error('invalid_module', (model.rec_name,))
|
|
|
|
@classmethod
|
|
def list_models(cls):
|
|
'Return a list of all models names'
|
|
models = cls.search([], order=[
|
|
('module', 'ASC'), # Optimization assumption
|
|
('model', 'ASC'),
|
|
('id', 'ASC'),
|
|
])
|
|
return [m.model for m in models]
|
|
|
|
@classmethod
|
|
def list_history(cls):
|
|
'Return a list of all models with history'
|
|
return [name for name, model in Pool().iterobject()
|
|
if getattr(model, '_history', False)]
|
|
|
|
@classmethod
|
|
def create(cls, vlist):
|
|
pool = Pool()
|
|
Property = pool.get('ir.property')
|
|
res = super(Model, cls).create(vlist)
|
|
# Restart the cache of models_get
|
|
Property._models_get_cache.clear()
|
|
return res
|
|
|
|
@classmethod
|
|
def write(cls, models, values, *args):
|
|
pool = Pool()
|
|
Property = pool.get('ir.property')
|
|
super(Model, cls).write(models, values, *args)
|
|
# Restart the cache of models_get
|
|
Property._models_get_cache.clear()
|
|
|
|
@classmethod
|
|
def delete(cls, models):
|
|
pool = Pool()
|
|
Property = pool.get('ir.property')
|
|
super(Model, cls).delete(models)
|
|
# Restart the cache of models_get
|
|
Property._models_get_cache.clear()
|
|
|
|
@classmethod
|
|
def global_search(cls, text, limit, menu='ir.ui.menu'):
|
|
"""
|
|
Search on models for text including menu
|
|
Returns a list of tuple (ratio, model, model_name, id, name, icon)
|
|
The size of the list is limited to limit
|
|
"""
|
|
pool = Pool()
|
|
ModelAccess = pool.get('ir.model.access')
|
|
|
|
if not limit > 0:
|
|
raise ValueError('limit must be > 0: %r' % (limit,))
|
|
|
|
models = cls.search(['OR',
|
|
('global_search_p', '=', True),
|
|
('model', '=', menu),
|
|
])
|
|
access = ModelAccess.get_access([m.model for m in models])
|
|
s = StringMatcher()
|
|
if isinstance(text, str):
|
|
text = text.decode('utf-8')
|
|
s.set_seq2(text)
|
|
|
|
def generate():
|
|
for model in models:
|
|
if not access[model.model]['read']:
|
|
continue
|
|
Model = pool.get(model.model)
|
|
if not hasattr(Model, 'search_global'):
|
|
continue
|
|
for record, name, icon in Model.search_global(text):
|
|
if isinstance(name, str):
|
|
name = name.decode('utf-8')
|
|
s.set_seq1(name)
|
|
yield (s.ratio(), model.model, model.rec_name,
|
|
record.id, name, icon)
|
|
return heapq.nlargest(int(limit), generate())
|
|
|
|
|
|
class ModelField(ModelSQL, ModelView):
|
|
"Model field"
|
|
__name__ = 'ir.model.field'
|
|
name = fields.Char('Name', required=True,
|
|
states={
|
|
'readonly': Bool(Eval('module')),
|
|
},
|
|
depends=['module'])
|
|
relation = fields.Char('Model Relation',
|
|
states={
|
|
'readonly': Bool(Eval('module')),
|
|
},
|
|
depends=['module'])
|
|
model = fields.Many2One('ir.model', 'Model', required=True,
|
|
select=True, ondelete='CASCADE',
|
|
states={
|
|
'readonly': Bool(Eval('module')),
|
|
},
|
|
depends=['module'])
|
|
field_description = fields.Char('Field Description', translate=True,
|
|
loading='lazy',
|
|
states={
|
|
'readonly': Bool(Eval('module')),
|
|
},
|
|
depends=['module'])
|
|
ttype = fields.Char('Field Type',
|
|
states={
|
|
'readonly': Bool(Eval('module')),
|
|
},
|
|
depends=['module'])
|
|
groups = fields.Many2Many('ir.model.field-res.group', 'field',
|
|
'group', 'Groups')
|
|
help = fields.Text('Help', translate=True, loading='lazy',
|
|
states={
|
|
'readonly': Bool(Eval('module')),
|
|
},
|
|
depends=['module'])
|
|
module = fields.Char('Module',
|
|
help="Module in which this field is defined")
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(ModelField, cls).__setup__()
|
|
table = cls.__table__()
|
|
cls._sql_constraints += [
|
|
('name_model_uniq', Unique(table, table.name, table.model),
|
|
'The field name in model must be unique!'),
|
|
]
|
|
cls._error_messages.update({
|
|
'invalid_name': ('Model Field name "%s" is not a valid python '
|
|
'identifier.'),
|
|
})
|
|
cls._order.insert(0, ('name', 'ASC'))
|
|
|
|
@classmethod
|
|
def register(cls, model, module_name, model_id):
|
|
pool = Pool()
|
|
Model = pool.get('ir.model')
|
|
cursor = Transaction().connection.cursor()
|
|
|
|
ir_model_field = cls.__table__()
|
|
ir_model = Model.__table__()
|
|
|
|
cursor.execute(*ir_model_field.join(ir_model,
|
|
condition=ir_model_field.model == ir_model.id
|
|
).select(ir_model_field.id.as_('id'),
|
|
ir_model_field.name.as_('name'),
|
|
ir_model_field.field_description.as_('field_description'),
|
|
ir_model_field.ttype.as_('ttype'),
|
|
ir_model_field.relation.as_('relation'),
|
|
ir_model_field.module.as_('module'),
|
|
ir_model_field.help.as_('help'),
|
|
where=ir_model.model == model.__name__))
|
|
model_fields = {f['name']: f for f in cursor_dict(cursor)}
|
|
|
|
for field_name, field in model._fields.iteritems():
|
|
if hasattr(field, 'model_name'):
|
|
relation = field.model_name
|
|
elif hasattr(field, 'relation_name'):
|
|
relation = field.relation_name
|
|
else:
|
|
relation = None
|
|
|
|
if field_name not in model_fields:
|
|
cursor.execute(*ir_model_field.insert(
|
|
[ir_model_field.model, ir_model_field.name,
|
|
ir_model_field.field_description,
|
|
ir_model_field.ttype, ir_model_field.relation,
|
|
ir_model_field.help, ir_model_field.module],
|
|
[[model_id, field_name, field.string, field._type,
|
|
relation, field.help, module_name]]))
|
|
elif (model_fields[field_name]['field_description'] != field.string
|
|
or model_fields[field_name]['ttype'] != field._type
|
|
or model_fields[field_name]['relation'] != relation
|
|
or model_fields[field_name]['help'] != field.help):
|
|
cursor.execute(*ir_model_field.update(
|
|
[ir_model_field.field_description,
|
|
ir_model_field.ttype, ir_model_field.relation,
|
|
ir_model_field.help],
|
|
[field.string, field._type, relation, field.help],
|
|
where=ir_model_field.id ==
|
|
model_fields[field_name]['id']))
|
|
|
|
# Clean ir_model_field from field that are no more existing.
|
|
for field_name in model_fields:
|
|
if model_fields[field_name]['module'] == module_name \
|
|
and field_name not in model._fields:
|
|
# XXX This delete field even when it is defined later
|
|
# in the module
|
|
cursor.execute(*ir_model_field.delete(
|
|
where=ir_model_field.id ==
|
|
model_fields[field_name]['id']))
|
|
|
|
@staticmethod
|
|
def default_name():
|
|
return 'No Name'
|
|
|
|
@staticmethod
|
|
def default_field_description():
|
|
return 'No description available'
|
|
|
|
@classmethod
|
|
def validate(cls, fields):
|
|
super(ModelField, cls).validate(fields)
|
|
cls.check_name(fields)
|
|
|
|
@classmethod
|
|
def check_name(cls, fields):
|
|
'''
|
|
Check name
|
|
'''
|
|
for field in fields:
|
|
if not IDENTIFIER.match(field.name):
|
|
cls.raise_user_error('invalid_name', (field.name,))
|
|
|
|
@classmethod
|
|
def read(cls, ids, fields_names=None):
|
|
pool = Pool()
|
|
Translation = pool.get('ir.translation')
|
|
Model = pool.get('ir.model')
|
|
|
|
to_delete = []
|
|
if Transaction().context.get('language'):
|
|
if fields_names is None:
|
|
fields_names = cls._fields.keys()
|
|
|
|
if 'field_description' in fields_names \
|
|
or 'help' in fields_names:
|
|
if 'model' not in fields_names:
|
|
fields_names.append('model')
|
|
to_delete.append('model')
|
|
if 'name' not in fields_names:
|
|
fields_names.append('name')
|
|
to_delete.append('name')
|
|
|
|
res = super(ModelField, cls).read(ids, fields_names=fields_names)
|
|
|
|
if (Transaction().context.get('language')
|
|
and ('field_description' in fields_names
|
|
or 'help' in fields_names)):
|
|
model_ids = set()
|
|
for rec in res:
|
|
if isinstance(rec['model'], (list, tuple)):
|
|
model_ids.add(rec['model'][0])
|
|
else:
|
|
model_ids.add(rec['model'])
|
|
model_ids = list(model_ids)
|
|
cursor = Transaction().connection.cursor()
|
|
model = Model.__table__()
|
|
cursor.execute(*model.select(model.id, model.model,
|
|
where=model.id.in_(model_ids)))
|
|
id2model = dict(cursor.fetchall())
|
|
|
|
trans_args = []
|
|
for rec in res:
|
|
if isinstance(rec['model'], (list, tuple)):
|
|
model_id = rec['model'][0]
|
|
else:
|
|
model_id = rec['model']
|
|
if 'field_description' in fields_names:
|
|
trans_args.append((id2model[model_id] + ',' + rec['name'],
|
|
'field', Transaction().language, None))
|
|
if 'help' in fields_names:
|
|
trans_args.append((id2model[model_id] + ',' + rec['name'],
|
|
'help', Transaction().language, None))
|
|
Translation.get_sources(trans_args)
|
|
for rec in res:
|
|
if isinstance(rec['model'], (list, tuple)):
|
|
model_id = rec['model'][0]
|
|
else:
|
|
model_id = rec['model']
|
|
if 'field_description' in fields_names:
|
|
res_trans = Translation.get_source(
|
|
id2model[model_id] + ',' + rec['name'],
|
|
'field', Transaction().language)
|
|
if res_trans:
|
|
rec['field_description'] = res_trans
|
|
if 'help' in fields_names:
|
|
res_trans = Translation.get_source(
|
|
id2model[model_id] + ',' + rec['name'],
|
|
'help', Transaction().language)
|
|
if res_trans:
|
|
rec['help'] = res_trans
|
|
|
|
if to_delete:
|
|
for rec in res:
|
|
for field in to_delete:
|
|
del rec[field]
|
|
return res
|
|
|
|
|
|
class ModelAccess(ModelSQL, ModelView):
|
|
"Model access"
|
|
__name__ = 'ir.model.access'
|
|
_rec_name = 'model'
|
|
model = fields.Many2One('ir.model', 'Model', required=True,
|
|
ondelete="CASCADE")
|
|
group = fields.Many2One('res.group', 'Group',
|
|
ondelete="CASCADE")
|
|
perm_read = fields.Boolean('Read Access')
|
|
perm_write = fields.Boolean('Write Access')
|
|
perm_create = fields.Boolean('Create Access')
|
|
perm_delete = fields.Boolean('Delete Access')
|
|
description = fields.Text('Description')
|
|
_get_access_cache = Cache('ir_model_access.get_access', context=False)
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(ModelAccess, cls).__setup__()
|
|
cls._error_messages.update({
|
|
'read': 'You can not read this document! (%s)',
|
|
'write': 'You can not write in this document! (%s)',
|
|
'create': 'You can not create this kind of document! (%s)',
|
|
'delete': 'You can not delete this document! (%s)',
|
|
})
|
|
cls.__rpc__.update({
|
|
'get_access': RPC(),
|
|
})
|
|
|
|
@classmethod
|
|
def __register__(cls, module_name):
|
|
TableHandler = backend.get('TableHandler')
|
|
|
|
super(ModelAccess, cls).__register__(module_name)
|
|
|
|
table = TableHandler(cls, module_name)
|
|
|
|
# Migration from 2.6 (model, group) no more unique
|
|
table.drop_constraint('model_group_uniq')
|
|
|
|
@staticmethod
|
|
def check_xml_record(accesses, values):
|
|
return True
|
|
|
|
@staticmethod
|
|
def default_perm_read():
|
|
return False
|
|
|
|
@staticmethod
|
|
def default_perm_write():
|
|
return False
|
|
|
|
@staticmethod
|
|
def default_perm_create():
|
|
return False
|
|
|
|
@staticmethod
|
|
def default_perm_delete():
|
|
return False
|
|
|
|
@classmethod
|
|
def get_access(cls, models):
|
|
'Return access for models'
|
|
# root user above constraint
|
|
if Transaction().user == 0:
|
|
return defaultdict(lambda: defaultdict(lambda: True))
|
|
|
|
pool = Pool()
|
|
Model = pool.get('ir.model')
|
|
UserGroup = pool.get('res.user-res.group')
|
|
cursor = Transaction().connection.cursor()
|
|
user = Transaction().user
|
|
model_access = cls.__table__()
|
|
ir_model = Model.__table__()
|
|
user_group = UserGroup.__table__()
|
|
|
|
access = {}
|
|
for model in models:
|
|
maccess = cls._get_access_cache.get((user, model), default=-1)
|
|
if maccess == -1:
|
|
break
|
|
access[model] = maccess
|
|
else:
|
|
return access
|
|
|
|
default = {'read': True, 'write': True, 'create': True, 'delete': True}
|
|
access = dict((m, default) for m in models)
|
|
cursor.execute(*model_access.join(ir_model, 'LEFT',
|
|
condition=model_access.model == ir_model.id
|
|
).join(user_group, 'LEFT',
|
|
condition=user_group.group == model_access.group
|
|
).select(
|
|
ir_model.model,
|
|
Max(Case((model_access.perm_read == True, 1), else_=0)),
|
|
Max(Case((model_access.perm_write == True, 1), else_=0)),
|
|
Max(Case((model_access.perm_create == True, 1), else_=0)),
|
|
Max(Case((model_access.perm_delete == True, 1), else_=0)),
|
|
where=ir_model.model.in_(models)
|
|
& ((user_group.user == user) | (model_access.group == Null)),
|
|
group_by=ir_model.model))
|
|
access.update(dict(
|
|
(m, {'read': r, 'write': w, 'create': c, 'delete': d})
|
|
for m, r, w, c, d in cursor.fetchall()))
|
|
for model, maccess in access.iteritems():
|
|
cls._get_access_cache.set((user, model), maccess)
|
|
return access
|
|
|
|
@classmethod
|
|
def check(cls, model_name, mode='read', raise_exception=True):
|
|
'Check access for model_name and mode'
|
|
assert mode in ['read', 'write', 'create', 'delete'], \
|
|
'Invalid access mode for security'
|
|
if ((Transaction().user == 0)
|
|
or (raise_exception
|
|
and not Transaction().context.get('_check_access'))):
|
|
return True
|
|
|
|
access = cls.get_access([model_name])[model_name][mode]
|
|
if not access and access is not None:
|
|
if raise_exception:
|
|
cls.raise_user_error(mode, model_name)
|
|
else:
|
|
return False
|
|
return True
|
|
|
|
@classmethod
|
|
def check_relation(cls, model_name, field_name, mode='read'):
|
|
'Check access to relation field for model_name and mode'
|
|
pool = Pool()
|
|
Model = pool.get(model_name)
|
|
field = getattr(Model, field_name)
|
|
if field._type in ('one2many', 'many2one'):
|
|
return cls.check(field.model_name, mode=mode,
|
|
raise_exception=False)
|
|
elif field._type in ('many2many', 'one2one'):
|
|
if (field.target
|
|
and not cls.check(field.target, mode=mode,
|
|
raise_exception=False)):
|
|
return False
|
|
elif (field.relation_name
|
|
and not cls.check(field.relation_name, mode=mode,
|
|
raise_exception=False)):
|
|
return False
|
|
else:
|
|
return True
|
|
elif field._type == 'reference':
|
|
selection = field.selection
|
|
if isinstance(selection, basestring):
|
|
sel_func = getattr(Model, field.selection)
|
|
if not is_instance_method(Model, field.selection):
|
|
selection = sel_func()
|
|
else:
|
|
# XXX Can not check access right on instance method
|
|
selection = []
|
|
for model_name, _ in selection:
|
|
if not cls.check(model_name, mode=mode,
|
|
raise_exception=False):
|
|
return False
|
|
return True
|
|
else:
|
|
return True
|
|
|
|
@classmethod
|
|
def write(cls, accesses, values, *args):
|
|
super(ModelAccess, cls).write(accesses, values, *args)
|
|
# Restart the cache
|
|
cls._get_access_cache.clear()
|
|
ModelView._fields_view_get_cache.clear()
|
|
|
|
@classmethod
|
|
def create(cls, vlist):
|
|
res = super(ModelAccess, cls).create(vlist)
|
|
# Restart the cache
|
|
cls._get_access_cache.clear()
|
|
ModelView._fields_view_get_cache.clear()
|
|
return res
|
|
|
|
@classmethod
|
|
def delete(cls, accesses):
|
|
super(ModelAccess, cls).delete(accesses)
|
|
# Restart the cache
|
|
cls._get_access_cache.clear()
|
|
ModelView._fields_view_get_cache.clear()
|
|
|
|
|
|
class ModelFieldAccess(ModelSQL, ModelView):
|
|
"Model Field Access"
|
|
__name__ = 'ir.model.field.access'
|
|
_rec_name = 'field'
|
|
field = fields.Many2One('ir.model.field', 'Field', required=True,
|
|
ondelete='CASCADE')
|
|
group = fields.Many2One('res.group', 'Group', ondelete='CASCADE')
|
|
perm_read = fields.Boolean('Read Access')
|
|
perm_write = fields.Boolean('Write Access')
|
|
perm_create = fields.Boolean('Create Access')
|
|
perm_delete = fields.Boolean('Delete Access')
|
|
description = fields.Text('Description')
|
|
_get_access_cache = Cache('ir_model_field_access.check', context=False)
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(ModelFieldAccess, cls).__setup__()
|
|
cls._error_messages.update({
|
|
'read': 'You can not read the field! (%s.%s)',
|
|
'write': 'You can not write on the field! (%s.%s)',
|
|
})
|
|
|
|
@classmethod
|
|
def __register__(cls, module_name):
|
|
TableHandler = backend.get('TableHandler')
|
|
|
|
super(ModelFieldAccess, cls).__register__(module_name)
|
|
|
|
table = TableHandler(cls, module_name)
|
|
|
|
# Migration from 2.6 (field, group) no more unique
|
|
table.drop_constraint('field_group_uniq')
|
|
|
|
@staticmethod
|
|
def check_xml_record(field_accesses, values):
|
|
return True
|
|
|
|
@staticmethod
|
|
def default_perm_read():
|
|
return False
|
|
|
|
@staticmethod
|
|
def default_perm_write():
|
|
return False
|
|
|
|
@staticmethod
|
|
def default_perm_create():
|
|
return True
|
|
|
|
@staticmethod
|
|
def default_perm_delete():
|
|
return True
|
|
|
|
@classmethod
|
|
def get_access(cls, models):
|
|
'Return fields access for models'
|
|
# root user above constraint
|
|
if Transaction().user == 0:
|
|
return defaultdict(lambda: defaultdict(
|
|
lambda: defaultdict(lambda: True)))
|
|
|
|
pool = Pool()
|
|
Model = pool.get('ir.model')
|
|
ModelField = pool.get('ir.model.field')
|
|
UserGroup = pool.get('res.user-res.group')
|
|
user = Transaction().user
|
|
field_access = cls.__table__()
|
|
ir_model = Model.__table__()
|
|
model_field = ModelField.__table__()
|
|
user_group = UserGroup.__table__()
|
|
|
|
accesses = {}
|
|
for model in models:
|
|
maccesses = cls._get_access_cache.get((user, model))
|
|
if maccesses is None:
|
|
break
|
|
accesses[model] = maccesses
|
|
else:
|
|
return accesses
|
|
|
|
default = {}
|
|
accesses = dict((m, default) for m in models)
|
|
cursor = Transaction().connection.cursor()
|
|
cursor.execute(*field_access.join(model_field,
|
|
condition=field_access.field == model_field.id
|
|
).join(ir_model,
|
|
condition=model_field.model == ir_model.id
|
|
).join(user_group, 'LEFT',
|
|
condition=user_group.group == field_access.group
|
|
).select(
|
|
ir_model.model,
|
|
model_field.name,
|
|
Max(Case((field_access.perm_read == True , 1), else_=0)),
|
|
Max(Case((field_access.perm_write == True, 1), else_=0)),
|
|
Max(Case((field_access.perm_create == True, 1), else_=0)),
|
|
Max(Case((field_access.perm_delete == True, 1), else_=0)),
|
|
where=ir_model.model.in_(models)
|
|
& ((user_group.user == user) | (field_access.group == Null)),
|
|
group_by=[ir_model.model, model_field.name]))
|
|
for m, f, r, w, c, d in cursor.fetchall():
|
|
accesses[m][f] = {'read': r, 'write': w, 'create': c, 'delete': d}
|
|
for model, maccesses in accesses.iteritems():
|
|
cls._get_access_cache.set((user, model), maccesses)
|
|
return accesses
|
|
|
|
@classmethod
|
|
def check(cls, model_name, fields, mode='read', raise_exception=True,
|
|
access=False):
|
|
'''
|
|
Check access for fields on model_name.
|
|
'''
|
|
assert mode in ('read', 'write', 'create', 'delete'), \
|
|
'Invalid access mode'
|
|
if ((Transaction().user == 0)
|
|
or (raise_exception
|
|
and not Transaction().context.get('_check_access'))):
|
|
if access:
|
|
return dict((x, True) for x in fields)
|
|
return True
|
|
|
|
accesses = dict((f, a[mode])
|
|
for f, a in cls.get_access([model_name])[model_name].iteritems())
|
|
if access:
|
|
return accesses
|
|
for field in fields:
|
|
if not accesses.get(field, True):
|
|
if raise_exception:
|
|
cls.raise_user_error(mode, (model_name, field))
|
|
else:
|
|
return False
|
|
return True
|
|
|
|
@classmethod
|
|
def write(cls, field_accesses, values, *args):
|
|
super(ModelFieldAccess, cls).write(field_accesses, values, *args)
|
|
# Restart the cache
|
|
cls._get_access_cache.clear()
|
|
ModelView._fields_view_get_cache.clear()
|
|
|
|
@classmethod
|
|
def create(cls, vlist):
|
|
res = super(ModelFieldAccess, cls).create(vlist)
|
|
# Restart the cache
|
|
cls._get_access_cache.clear()
|
|
ModelView._fields_view_get_cache.clear()
|
|
return res
|
|
|
|
@classmethod
|
|
def delete(cls, field_accesses):
|
|
super(ModelFieldAccess, cls).delete(field_accesses)
|
|
# Restart the cache
|
|
cls._get_access_cache.clear()
|
|
ModelView._fields_view_get_cache.clear()
|
|
|
|
|
|
class ModelButton(ModelSQL, ModelView):
|
|
"Model Button"
|
|
__name__ = 'ir.model.button'
|
|
name = fields.Char('Name', required=True, readonly=True)
|
|
model = fields.Many2One('ir.model', 'Model', required=True, readonly=True,
|
|
ondelete='CASCADE', select=True)
|
|
groups = fields.Many2Many('ir.model.button-res.group', 'button', 'group',
|
|
'Groups')
|
|
_groups_cache = Cache('ir.model.button.groups')
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(ModelButton, cls).__setup__()
|
|
table = cls.__table__()
|
|
cls._sql_constraints += [
|
|
('name_model_uniq', Unique(table, table.name, table.model),
|
|
'The button name in model must be unique!'),
|
|
]
|
|
cls._order.insert(0, ('model', 'ASC'))
|
|
|
|
@classmethod
|
|
def create(cls, vlist):
|
|
result = super(ModelButton, cls).create(vlist)
|
|
# Restart the cache for get_groups
|
|
cls._groups_cache.clear()
|
|
return result
|
|
|
|
@classmethod
|
|
def write(cls, buttons, values, *args):
|
|
super(ModelButton, cls).write(buttons, values, *args)
|
|
# Restart the cache for get_groups
|
|
cls._groups_cache.clear()
|
|
|
|
@classmethod
|
|
def delete(cls, buttons):
|
|
super(ModelButton, cls).delete(buttons)
|
|
# Restart the cache for get_groups
|
|
cls._groups_cache.clear()
|
|
|
|
@classmethod
|
|
def get_groups(cls, model, name):
|
|
'''
|
|
Return a set of group ids for the named button on the model.
|
|
'''
|
|
key = (model, name)
|
|
groups = cls._groups_cache.get(key)
|
|
if groups is not None:
|
|
return groups
|
|
buttons = cls.search([
|
|
('model.model', '=', model),
|
|
('name', '=', name),
|
|
])
|
|
if not buttons:
|
|
groups = set()
|
|
else:
|
|
button, = buttons
|
|
groups = set(g.id for g in button.groups)
|
|
cls._groups_cache.set(key, groups)
|
|
return groups
|
|
|
|
|
|
class ModelData(ModelSQL, ModelView):
|
|
"Model data"
|
|
__name__ = 'ir.model.data'
|
|
fs_id = fields.Char('Identifier on File System', required=True,
|
|
help="The id of the record as known on the file system.",
|
|
select=True)
|
|
model = fields.Char('Model', required=True, select=True)
|
|
module = fields.Char('Module', required=True, select=True)
|
|
db_id = fields.Integer('Resource ID',
|
|
help="The id of the record in the database.", select=True,
|
|
required=True)
|
|
values = fields.Text('Values')
|
|
fs_values = fields.Text('Values on File System')
|
|
noupdate = fields.Boolean('No Update')
|
|
out_of_sync = fields.Function(fields.Boolean('Out of Sync'),
|
|
'get_out_of_sync', searcher='search_out_of_sync')
|
|
_get_id_cache = Cache('ir_model_data.get_id', context=False)
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(ModelData, cls).__setup__()
|
|
table = cls.__table__()
|
|
cls._sql_constraints = [
|
|
('fs_id_module_model_uniq',
|
|
Unique(table, table.fs_id, table.module, table.model),
|
|
'The triple (fs_id, module, model) must be unique!'),
|
|
]
|
|
cls._buttons.update({
|
|
'sync': {
|
|
'invisible': ~Eval('out_of_sync'),
|
|
},
|
|
})
|
|
|
|
@classmethod
|
|
def __register__(cls, module_name):
|
|
TableHandler = backend.get('TableHandler')
|
|
cursor = Transaction().connection.cursor()
|
|
model_data = cls.__table__()
|
|
|
|
super(ModelData, cls).__register__(module_name)
|
|
|
|
table = TableHandler(cls, module_name)
|
|
|
|
# Migration from 2.6: remove inherit
|
|
if table.column_exist('inherit'):
|
|
cursor.execute(*model_data.delete(
|
|
where=model_data.inherit == True))
|
|
table.drop_column('inherit', True)
|
|
|
|
@staticmethod
|
|
def default_noupdate():
|
|
return False
|
|
|
|
def get_out_of_sync(self, name):
|
|
return self.values != self.fs_values and self.fs_values is not None
|
|
|
|
@classmethod
|
|
def search_out_of_sync(cls, name, clause):
|
|
table = cls.__table__()
|
|
name, operator, value = clause
|
|
Operator = fields.SQL_OPERATORS[operator]
|
|
query = table.select(table.id,
|
|
where=Operator(
|
|
(table.fs_values != table.values) & (table.fs_values != Null),
|
|
value))
|
|
return [('id', 'in', query)]
|
|
|
|
@classmethod
|
|
def write(cls, data, values, *args):
|
|
super(ModelData, cls).write(data, values, *args)
|
|
# Restart the cache for get_id
|
|
cls._get_id_cache.clear()
|
|
|
|
@classmethod
|
|
def get_id(cls, module, fs_id):
|
|
"""
|
|
Return for an fs_id the corresponding db_id.
|
|
"""
|
|
key = (module, fs_id)
|
|
id_ = cls._get_id_cache.get(key)
|
|
if id_ is not None:
|
|
return id_
|
|
data = cls.search([
|
|
('module', '=', module),
|
|
('fs_id', '=', fs_id),
|
|
], limit=1)
|
|
if not data:
|
|
raise Exception("Reference to %s not found"
|
|
% ".".join([module, fs_id]))
|
|
id_ = cls.read([d.id for d in data], ['db_id'])[0]['db_id']
|
|
cls._get_id_cache.set(key, id_)
|
|
return id_
|
|
|
|
@classmethod
|
|
def dump_values(cls, values):
|
|
return json.dumps(sorted(values.iteritems()), cls=JSONEncoder)
|
|
|
|
@classmethod
|
|
def load_values(cls, values):
|
|
try:
|
|
return dict(json.loads(values, object_hook=JSONDecoder()))
|
|
except ValueError:
|
|
# Migration from 3.2
|
|
from decimal import Decimal
|
|
import datetime
|
|
return eval(values, {
|
|
'Decimal': Decimal,
|
|
'datetime': datetime,
|
|
})
|
|
|
|
@classmethod
|
|
@ModelView.button
|
|
def sync(cls, records):
|
|
pool = Pool()
|
|
to_write = []
|
|
for data in records:
|
|
Model = pool.get(data.model)
|
|
values = cls.load_values(data.values)
|
|
fs_values = cls.load_values(data.fs_values)
|
|
# values could be the same once loaded
|
|
# if they come from version < 3.2
|
|
if values != fs_values:
|
|
record = Model(data.db_id)
|
|
Model.write([record], fs_values)
|
|
values = fs_values
|
|
to_write.extend([[data], {
|
|
'values': cls.dump_values(values),
|
|
}])
|
|
if to_write:
|
|
cls.write(*to_write)
|
|
|
|
|
|
class PrintModelGraphStart(ModelView):
|
|
'Print Model Graph'
|
|
__name__ = 'ir.model.print_model_graph.start'
|
|
level = fields.Integer('Level', required=True)
|
|
filter = fields.Text('Filter', help="Entering a Python "
|
|
"Regular Expression will exclude matching models from the graph.")
|
|
|
|
@staticmethod
|
|
def default_level():
|
|
return 1
|
|
|
|
|
|
class PrintModelGraph(Wizard):
|
|
__name__ = 'ir.model.print_model_graph'
|
|
|
|
start = StateView('ir.model.print_model_graph.start',
|
|
'ir.print_model_graph_start_view_form', [
|
|
Button('Cancel', 'end', 'tryton-cancel'),
|
|
Button('Print', 'print_', 'tryton-ok', default=True),
|
|
])
|
|
print_ = StateAction('ir.report_model_graph')
|
|
|
|
def transition_print_(self):
|
|
return 'end'
|
|
|
|
def do_print_(self, action):
|
|
return action, {
|
|
'id': Transaction().context.get('active_id'),
|
|
'ids': Transaction().context.get('active_ids'),
|
|
'level': self.start.level,
|
|
'filter': self.start.filter,
|
|
}
|
|
|
|
|
|
class ModelGraph(Report):
|
|
__name__ = 'ir.model.graph'
|
|
|
|
@classmethod
|
|
def execute(cls, ids, data):
|
|
import pydot
|
|
pool = Pool()
|
|
Model = pool.get('ir.model')
|
|
ActionReport = pool.get('ir.action.report')
|
|
|
|
if not data['filter']:
|
|
filter = None
|
|
else:
|
|
filter = re.compile(data['filter'], re.VERBOSE)
|
|
action_report_ids = ActionReport.search([
|
|
('report_name', '=', cls.__name__)
|
|
])
|
|
if not action_report_ids:
|
|
raise Exception('Error', 'Report (%s) not find!' % cls.__name__)
|
|
action_report = ActionReport(action_report_ids[0])
|
|
|
|
models = Model.browse(ids)
|
|
|
|
graph = pydot.Dot(fontsize="8")
|
|
graph.set('center', '1')
|
|
graph.set('ratio', 'auto')
|
|
cls.fill_graph(models, graph, level=data['level'], filter=filter)
|
|
data = graph.create(prog='dot', format='png')
|
|
return ('png', fields.Binary.cast(data), False, action_report.name)
|
|
|
|
@classmethod
|
|
def fill_graph(cls, models, graph, level=1, filter=None):
|
|
'''
|
|
Fills a pydot graph with a models structure.
|
|
'''
|
|
import pydot
|
|
pool = Pool()
|
|
Model = pool.get('ir.model')
|
|
|
|
sub_models = set()
|
|
if level > 0:
|
|
for model in models:
|
|
for field in model.fields:
|
|
if field.name in ('create_uid', 'write_uid'):
|
|
continue
|
|
if field.relation and not graph.get_node(field.relation):
|
|
sub_models.add(field.relation)
|
|
if sub_models:
|
|
model_ids = Model.search([
|
|
('model', 'in', list(sub_models)),
|
|
])
|
|
sub_models = Model.browse(model_ids)
|
|
if set(sub_models) != set(models):
|
|
cls.fill_graph(sub_models, graph, level=level - 1,
|
|
filter=filter)
|
|
|
|
for model in models:
|
|
if filter and re.search(filter, model.model):
|
|
continue
|
|
label = '"{' + model.model + '\\n'
|
|
if model.fields:
|
|
label += '|'
|
|
for field in model.fields:
|
|
if field.name in ('create_uid', 'write_uid',
|
|
'create_date', 'write_date', 'id'):
|
|
continue
|
|
label += '+ ' + field.name + ': ' + field.ttype
|
|
if field.relation:
|
|
label += ' ' + field.relation
|
|
label += '\l'
|
|
label += '}"'
|
|
node_name = '"%s"' % model.model
|
|
node = pydot.Node(node_name, shape='record', label=label)
|
|
graph.add_node(node)
|
|
|
|
for field in model.fields:
|
|
if field.name in ('create_uid', 'write_uid'):
|
|
continue
|
|
if field.relation:
|
|
node_name = '"%s"' % field.relation
|
|
if not graph.get_node(node_name):
|
|
continue
|
|
args = {}
|
|
tail = model.model
|
|
head = field.relation
|
|
edge_model_name = '"%s"' % model.model
|
|
edge_relation_name = '"%s"' % field.relation
|
|
if field.ttype == 'many2one':
|
|
edge = graph.get_edge(edge_model_name,
|
|
edge_relation_name)
|
|
if edge:
|
|
continue
|
|
args['arrowhead'] = "normal"
|
|
elif field.ttype == 'one2many':
|
|
edge = graph.get_edge(edge_relation_name,
|
|
edge_model_name)
|
|
if edge:
|
|
continue
|
|
args['arrowhead'] = "normal"
|
|
tail = field.relation
|
|
head = model.model
|
|
elif field.ttype == 'many2many':
|
|
if graph.get_edge(edge_model_name, edge_relation_name):
|
|
continue
|
|
if graph.get_edge(edge_relation_name, edge_model_name):
|
|
continue
|
|
args['arrowtail'] = "inv"
|
|
args['arrowhead'] = "inv"
|
|
|
|
edge = pydot.Edge(str(tail), str(head), **args)
|
|
graph.add_edge(edge)
|
|
|
|
|
|
class ModelWorkflowGraph(Report):
|
|
__name__ = 'ir.model.workflow_graph'
|
|
|
|
@classmethod
|
|
def execute(cls, ids, data):
|
|
import pydot
|
|
pool = Pool()
|
|
Model = pool.get('ir.model')
|
|
ActionReport = pool.get('ir.action.report')
|
|
|
|
action_report_ids = ActionReport.search([
|
|
('report_name', '=', cls.__name__)
|
|
])
|
|
if not action_report_ids:
|
|
raise Exception('Error', 'Report (%s) not find!' % cls.__name__)
|
|
action_report = ActionReport(action_report_ids[0])
|
|
|
|
models = Model.browse(ids)
|
|
|
|
graph = pydot.Dot()
|
|
graph.set('center', '1')
|
|
graph.set('ratio', 'auto')
|
|
direction = Transaction().context.get('language_direction', 'ltr')
|
|
graph.set('rankdir', {'ltr': 'LR', 'rtl': 'RL'}[direction])
|
|
cls.fill_graph(models, graph)
|
|
data = graph.create(prog='dot', format='png')
|
|
return ('png', fields.Binary.cast(data), False, action_report.name)
|
|
|
|
@classmethod
|
|
def fill_graph(cls, models, graph):
|
|
'Fills pydot graph with models wizard.'
|
|
import pydot
|
|
pool = Pool()
|
|
|
|
for record in models:
|
|
Model = pool.get(record.model)
|
|
|
|
if not issubclass(Model, Workflow):
|
|
continue
|
|
|
|
subgraph = pydot.Cluster('%s' % record.id, label=record.model)
|
|
graph.add_subgraph(subgraph)
|
|
|
|
state_field = getattr(Model, Model._transition_state)
|
|
for state, _ in state_field.selection:
|
|
node = pydot.Node(
|
|
'"%s"' % state, shape='octagon', label=state)
|
|
subgraph.add_node(node)
|
|
|
|
for from_, to in Model._transitions:
|
|
edge = pydot.Edge('"%s"' % from_, '"%s"' % to,
|
|
arrowhead='normal')
|
|
subgraph.add_edge(edge)
|