109 lines
3 KiB
Python
109 lines
3 KiB
Python
# -*- coding: utf-8 -*-
|
|
# This file is part of the cashbook-module from m-ds for Tryton.
|
|
# The COPYRIGHT file at the top level of this repository contains the
|
|
# full copyright notices and license terms.
|
|
|
|
from trytond.model import MultiValueMixin, ValueMixin, fields, Unique
|
|
from trytond.transaction import Transaction
|
|
from trytond.pool import Pool
|
|
from sql import With, Literal
|
|
from sql.functions import Function
|
|
|
|
|
|
class ArrayApppend(Function):
|
|
""" sql: array_append
|
|
"""
|
|
__slots__ = ()
|
|
_function = 'ARRAY_APPEND'
|
|
|
|
# end ArrayApppend
|
|
|
|
|
|
class ArrayToString(Function):
|
|
""" sql: array_to_string
|
|
"""
|
|
__slots__ = ()
|
|
_function = 'ARRAY_TO_STRING'
|
|
|
|
# end ArrayToString
|
|
|
|
|
|
class Array(Function):
|
|
""" sql: array-type
|
|
"""
|
|
__slots__ = ()
|
|
_function = 'ARRAY'
|
|
|
|
def __str__(self):
|
|
return self._function + '[' + ', '.join(
|
|
map(self._format, self.args)) + ']'
|
|
|
|
# end Array
|
|
|
|
|
|
def order_name_hierarchical(model_name, tables):
|
|
""" order by pos
|
|
a recursive sorting
|
|
"""
|
|
Model2 = Pool().get(model_name)
|
|
tab_mod = Model2.__table__()
|
|
tab_mod2 = Model2.__table__()
|
|
table, _ = tables[None]
|
|
|
|
lines = With('id', 'name', 'name_path', recursive=True)
|
|
lines.query = tab_mod.select(
|
|
tab_mod.id, tab_mod.name, Array(tab_mod.name),
|
|
where = tab_mod.parent==None,
|
|
)
|
|
lines.query |= tab_mod2.join(lines,
|
|
condition=lines.id==tab_mod2.parent,
|
|
).select(
|
|
tab_mod2.id, tab_mod2.name, ArrayApppend(lines.name_path, tab_mod2.name),
|
|
)
|
|
lines.query.all_ = True
|
|
|
|
query = lines.select(
|
|
ArrayToString(lines.name_path, '/').as_('rec_name'),
|
|
where = table.id==lines.id,
|
|
with_ = [lines])
|
|
return [query]
|
|
|
|
|
|
class UserValueMixin(ValueMixin):
|
|
iduser = fields.Many2One(model_name='res.user', string="User",
|
|
select=True, ondelete='CASCADE', required=True)
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(UserValueMixin, cls).__setup__()
|
|
tab_val = cls.__table__()
|
|
cls._sql_constraints.extend([
|
|
('val_uniq',
|
|
Unique(tab_val, tab_val.iduser),
|
|
'cashbook.msg_setting_already_exists'),
|
|
])
|
|
|
|
# end UserValueMixin
|
|
|
|
|
|
class UserMultiValueMixin(MultiValueMixin):
|
|
|
|
def updt_multivalue_pattern(self, pattern):
|
|
""" add values to pattern
|
|
"""
|
|
pattern.setdefault('iduser', Transaction().user)
|
|
return pattern
|
|
|
|
def get_multivalue(self, name, **pattern):
|
|
Value = self.multivalue_model(name)
|
|
if issubclass(Value, UserValueMixin):
|
|
pattern = self.updt_multivalue_pattern(pattern)
|
|
return super(UserMultiValueMixin, self).get_multivalue(name, **pattern)
|
|
|
|
def set_multivalue(self, name, value, **pattern):
|
|
Value = self.multivalue_model(name)
|
|
if issubclass(Value, UserValueMixin):
|
|
pattern = self.updt_multivalue_pattern(pattern)
|
|
return super(UserMultiValueMixin, self).set_multivalue(name, value, **pattern)
|
|
|
|
# end UserMultiValueMixin
|