407 lines
14 KiB
Python
407 lines
14 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 time
|
|
from string import Template
|
|
|
|
from trytond import backend
|
|
from trytond.exceptions import UserError
|
|
from trytond.i18n import gettext
|
|
from trytond.model import Check, DeactivableMixin, ModelSQL, ModelView, fields
|
|
from trytond.model.exceptions import AccessError, ValidationError
|
|
from trytond.pool import Pool
|
|
from trytond.pyson import And, Eval
|
|
from trytond.transaction import Transaction
|
|
|
|
sql_sequence = backend.Database.has_sequence()
|
|
|
|
|
|
class AffixError(ValidationError):
|
|
pass
|
|
|
|
|
|
class MissingError(UserError):
|
|
pass
|
|
|
|
|
|
class LastTimestampError(ValidationError):
|
|
pass
|
|
|
|
|
|
class SQLSequenceError(ValidationError):
|
|
pass
|
|
|
|
|
|
class SequenceType(ModelSQL, ModelView):
|
|
"Sequence type"
|
|
__name__ = 'ir.sequence.type'
|
|
|
|
name = fields.Char('Sequence Name', required=True, translate=True)
|
|
|
|
@classmethod
|
|
def __register__(cls, module):
|
|
super().__register__(module)
|
|
table_h = cls.__table_handler__(module)
|
|
|
|
# Migration from 5.8: remove code
|
|
# We keep the column until ir.sequence has been migrated
|
|
table_h.not_null_action('code', action='remove')
|
|
|
|
|
|
class Sequence(DeactivableMixin, ModelSQL, ModelView):
|
|
"Sequence"
|
|
__name__ = 'ir.sequence'
|
|
|
|
_strict = False
|
|
name = fields.Char('Sequence Name', required=True, translate=True)
|
|
sequence_type = fields.Many2One(
|
|
'ir.sequence.type', "Sequence Type",
|
|
required=True, ondelete='RESTRICT',
|
|
states={
|
|
'readonly': Eval('id', -1) >= 0,
|
|
},
|
|
depends=['id'])
|
|
prefix = fields.Char('Prefix', strip='leading')
|
|
suffix = fields.Char('Suffix', strip='trailing')
|
|
type = fields.Selection([
|
|
('incremental', 'Incremental'),
|
|
('decimal timestamp', 'Decimal Timestamp'),
|
|
('hexadecimal timestamp', 'Hexadecimal Timestamp'),
|
|
], 'Type')
|
|
number_next_internal = fields.Integer('Next Number',
|
|
states={
|
|
'invisible': ~Eval('type').in_(['incremental']),
|
|
'required': And(Eval('type').in_(['incremental']),
|
|
not sql_sequence),
|
|
}, depends=['type'])
|
|
number_next = fields.Function(number_next_internal, 'get_number_next',
|
|
'set_number_next')
|
|
number_increment = fields.Integer('Increment Number',
|
|
states={
|
|
'invisible': ~Eval('type').in_(['incremental']),
|
|
'required': Eval('type').in_(['incremental']),
|
|
}, depends=['type'])
|
|
padding = fields.Integer('Number padding',
|
|
states={
|
|
'invisible': ~Eval('type').in_(['incremental']),
|
|
'required': Eval('type').in_(['incremental']),
|
|
}, depends=['type'])
|
|
timestamp_rounding = fields.Float('Timestamp Rounding', required=True,
|
|
states={
|
|
'invisible': ~Eval('type').in_(
|
|
['decimal timestamp', 'hexadecimal timestamp']),
|
|
}, depends=['type'])
|
|
timestamp_offset = fields.Float('Timestamp Offset', required=True,
|
|
states={
|
|
'invisible': ~Eval('type').in_(
|
|
['decimal timestamp', 'hexadecimal timestamp']),
|
|
}, depends=['type'])
|
|
last_timestamp = fields.Integer('Last Timestamp',
|
|
states={
|
|
'invisible': ~Eval('type').in_(
|
|
['decimal timestamp', 'hexadecimal timestamp']),
|
|
'required': Eval('type').in_(
|
|
['decimal timestamp', 'hexadecimal timestamp']),
|
|
}, depends=['type'])
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(Sequence, cls).__setup__()
|
|
table = cls.__table__()
|
|
cls._sql_constraints += [
|
|
('check_timestamp_rounding',
|
|
Check(table, table.timestamp_rounding > 0),
|
|
'Timestamp rounding should be greater than 0'),
|
|
]
|
|
|
|
@classmethod
|
|
def __register__(cls, module):
|
|
pool = Pool()
|
|
SequenceType = pool.get('ir.sequence.type')
|
|
cursor = Transaction().connection.cursor()
|
|
table = cls.__table__()
|
|
sequence_type = SequenceType.__table__()
|
|
|
|
super().__register__(module)
|
|
|
|
table_h = cls.__table_handler__(module)
|
|
|
|
# Migration from 5.8: replace code by sequence_type
|
|
if table_h.column_exist('code'):
|
|
cursor.execute(*table.update(
|
|
[table.sequence_type],
|
|
sequence_type.select(
|
|
sequence_type.id,
|
|
where=sequence_type.code == table.code)))
|
|
table_h.drop_column('code')
|
|
|
|
@staticmethod
|
|
def default_type():
|
|
return 'incremental'
|
|
|
|
@staticmethod
|
|
def default_number_increment():
|
|
return 1
|
|
|
|
@staticmethod
|
|
def default_number_next():
|
|
return 1
|
|
|
|
@staticmethod
|
|
def default_padding():
|
|
return 0
|
|
|
|
@staticmethod
|
|
def default_timestamp_rounding():
|
|
return 1.0
|
|
|
|
@staticmethod
|
|
def default_timestamp_offset():
|
|
return 946681200.0 # Offset for 2000-01-01
|
|
|
|
@staticmethod
|
|
def default_last_timestamp():
|
|
return 0
|
|
|
|
def get_number_next(self, name):
|
|
if self.type != 'incremental':
|
|
return
|
|
|
|
transaction = Transaction()
|
|
if sql_sequence and not self._strict:
|
|
return transaction.database.sequence_next_number(
|
|
transaction.connection, self._sql_sequence_name)
|
|
else:
|
|
return self.number_next_internal
|
|
|
|
@classmethod
|
|
def set_number_next(cls, sequences, name, value):
|
|
super(Sequence, cls).write(sequences, {
|
|
'number_next_internal': value,
|
|
})
|
|
|
|
@classmethod
|
|
def view_attributes(cls):
|
|
return [
|
|
('//group[@id="incremental"]', 'states', {
|
|
'invisible': ~Eval('type').in_(['incremental']),
|
|
}),
|
|
('//group[@id="timestamp"]', 'states', {
|
|
'invisible': ~Eval('type').in_(
|
|
['decimal timestamp', 'hexadecimal timestamp']),
|
|
}),
|
|
]
|
|
|
|
@classmethod
|
|
def create(cls, vlist):
|
|
sequences = super(Sequence, cls).create(vlist)
|
|
for sequence, values in zip(sequences, vlist):
|
|
if sql_sequence and not cls._strict:
|
|
sequence.update_sql_sequence(values.get('number_next',
|
|
cls.default_number_next()))
|
|
return sequences
|
|
|
|
@classmethod
|
|
def write(cls, *args):
|
|
transaction = Transaction()
|
|
if (transaction.user != 0
|
|
and transaction.context.get('_check_access')):
|
|
for values in args[1::2]:
|
|
if 'sequence_type' in values:
|
|
raise AccessError(gettext(
|
|
'ir.msg_sequence_change_sequence_type'))
|
|
super().write(*args)
|
|
if sql_sequence and not cls._strict:
|
|
actions = iter(args)
|
|
for sequences, values in zip(actions, actions):
|
|
for sequence in sequences:
|
|
sequence.update_sql_sequence(values.get('number_next'))
|
|
|
|
@classmethod
|
|
def delete(cls, sequences):
|
|
if sql_sequence and not cls._strict:
|
|
for sequence in sequences:
|
|
sequence.delete_sql_sequence()
|
|
return super(Sequence, cls).delete(sequences)
|
|
|
|
@classmethod
|
|
def validate(cls, sequences):
|
|
super().validate(sequences)
|
|
cls.check_last_timestamp(sequences)
|
|
|
|
@classmethod
|
|
def validate_fields(cls, sequences, field_names):
|
|
super().validate_fields(sequences, field_names)
|
|
cls.check_affixes(sequences, field_names)
|
|
|
|
@classmethod
|
|
def check_affixes(cls, sequences, field_names=None):
|
|
"Check prefix and suffix"
|
|
if field_names and not (field_names & {'prefix', 'suffix'}):
|
|
return
|
|
for sequence in sequences:
|
|
for affix, error_message in [
|
|
(sequence.prefix, 'msg_sequence_invalid_prefix'),
|
|
(sequence.suffix, 'msg_sequence_invalid_suffix')]:
|
|
try:
|
|
cls._process(affix)
|
|
except (TypeError, ValueError) as exc:
|
|
raise AffixError(gettext('ir.%s' % error_message,
|
|
affix=affix,
|
|
sequence=sequence.rec_name)) from exc
|
|
|
|
@classmethod
|
|
def check_last_timestamp(cls, sequences):
|
|
"Check last_timestamp"
|
|
|
|
for sequence in sequences:
|
|
next_timestamp = cls._timestamp(sequence)
|
|
if (sequence.last_timestamp is not None
|
|
and sequence.last_timestamp > next_timestamp):
|
|
raise LastTimestampError(
|
|
gettext('ir.msg_sequence_last_timestamp_future'))
|
|
|
|
@property
|
|
def _sql_sequence_name(self):
|
|
'Return SQL sequence name'
|
|
return '%s_%s' % (self._table, self.id)
|
|
|
|
def create_sql_sequence(self, number_next=None):
|
|
'Create the SQL sequence'
|
|
transaction = Transaction()
|
|
|
|
if self.type != 'incremental':
|
|
return
|
|
if number_next is None:
|
|
number_next = self.number_next
|
|
try:
|
|
transaction.database.sequence_create(
|
|
transaction.connection, self._sql_sequence_name,
|
|
self.number_increment, number_next)
|
|
except Exception as exception:
|
|
raise SQLSequenceError(
|
|
gettext('ir.msg_sequence_invalid_number_increment_next',
|
|
number_increment=self.number_increment,
|
|
number_next=number_next,
|
|
exception=exception)) from exception
|
|
|
|
def update_sql_sequence(self, number_next=None):
|
|
'Update the SQL sequence'
|
|
transaction = Transaction()
|
|
|
|
exist = transaction.database.sequence_exist(
|
|
transaction.connection, self._sql_sequence_name)
|
|
if self.type != 'incremental':
|
|
if exist:
|
|
self.delete_sql_sequence()
|
|
return
|
|
if not exist:
|
|
self.create_sql_sequence(number_next)
|
|
return
|
|
if number_next is None:
|
|
number_next = self.number_next
|
|
try:
|
|
transaction.database.sequence_update(
|
|
transaction.connection, self._sql_sequence_name,
|
|
self.number_increment, number_next)
|
|
except Exception as exception:
|
|
raise SQLSequenceError(
|
|
gettext('ir.msg_sequence_invalid_number_increment_next',
|
|
number_increment=self.number_increment,
|
|
number_next=number_next,
|
|
exception=exception)) from exception
|
|
|
|
def delete_sql_sequence(self):
|
|
'Delete the SQL sequence'
|
|
transaction = Transaction()
|
|
if self.type != 'incremental':
|
|
return
|
|
transaction.database.sequence_delete(
|
|
transaction.connection, self._sql_sequence_name)
|
|
|
|
@classmethod
|
|
def _process(cls, string, date=None):
|
|
return Template(string or '').substitute(
|
|
**cls._get_substitutions(date))
|
|
|
|
@classmethod
|
|
def _get_substitutions(cls, date):
|
|
'''
|
|
Returns a dictionary with the keys and values of the substitutions
|
|
available to format the sequence
|
|
'''
|
|
pool = Pool()
|
|
Date = pool.get('ir.date')
|
|
if not date:
|
|
date = Date.today()
|
|
return {
|
|
'year': date.strftime('%Y'),
|
|
'month': date.strftime('%m'),
|
|
'day': date.strftime('%d'),
|
|
}
|
|
|
|
@staticmethod
|
|
def _timestamp(sequence):
|
|
return int((time.time() - sequence.timestamp_offset)
|
|
/ sequence.timestamp_rounding)
|
|
|
|
@classmethod
|
|
def _get_sequence(cls, sequence):
|
|
if sequence.type == 'incremental':
|
|
if sql_sequence and not cls._strict:
|
|
cursor = Transaction().connection.cursor()
|
|
cursor.execute('SELECT nextval(\'"%s"\')'
|
|
% sequence._sql_sequence_name)
|
|
number_next, = cursor.fetchone()
|
|
else:
|
|
# Pre-fetch number_next
|
|
number_next = sequence.number_next_internal
|
|
|
|
cls.write([sequence], {
|
|
'number_next_internal': (number_next
|
|
+ sequence.number_increment),
|
|
})
|
|
return '%%0%sd' % sequence.padding % number_next
|
|
elif sequence.type in ('decimal timestamp', 'hexadecimal timestamp'):
|
|
timestamp = sequence.last_timestamp
|
|
while timestamp == sequence.last_timestamp:
|
|
timestamp = cls._timestamp(sequence)
|
|
cls.write([sequence], {
|
|
'last_timestamp': timestamp,
|
|
})
|
|
if sequence.type == 'decimal timestamp':
|
|
return '%d' % timestamp
|
|
else:
|
|
return hex(timestamp)[2:].upper()
|
|
return ''
|
|
|
|
def get(self, _lock=False):
|
|
'''
|
|
Return the next sequence value
|
|
'''
|
|
cls = self.__class__
|
|
# bypass rules on sequences
|
|
with Transaction().set_context(user=False, _check_access=False):
|
|
with Transaction().set_user(0):
|
|
try:
|
|
sequence = cls(self.id)
|
|
except TypeError:
|
|
raise MissingError(gettext('ir.msg_sequence_missing'))
|
|
if _lock:
|
|
self.lock()
|
|
date = Transaction().context.get('date')
|
|
return '%s%s%s' % (
|
|
cls._process(sequence.prefix, date=date),
|
|
cls._get_sequence(sequence),
|
|
cls._process(sequence.suffix, date=date),
|
|
)
|
|
|
|
|
|
class SequenceStrict(Sequence):
|
|
"Sequence Strict"
|
|
__name__ = 'ir.sequence.strict'
|
|
_table = None # Needed to reset Sequence._table
|
|
_strict = True
|
|
|
|
def get(self, _lock=True):
|
|
return super().get(_lock=True)
|