Files
tryton/account.py
T
Cédric Krier 2f8a91c2c9 Use SQL CurrentTimestamp to fill numbered_at
This ensure the timestamp stored is coherent with the create and write date of
the records. As the clock of trytond server can be different from the clock of
the database server.

issue6460
issue4285
review285241002
2019-06-26 21:55:46 +02:00

70 lines
2.4 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.
from sql.functions import CurrentTimestamp
from trytond.model import ModelView, Workflow, fields
from trytond.pool import PoolMeta
from trytond.transaction import Transaction
__all__ = ['Invoice']
class Invoice(metaclass=PoolMeta):
__name__ = 'account.invoice'
numbered_at = fields.Timestamp("Numbered At")
@classmethod
def __register__(cls, module_name):
super().__register__(module_name)
table_h = cls.__table_handler__(module_name)
table = cls.__table__()
cursor = Transaction().connection.cursor()
# Migration from 5.2: rename open_date into numbered_at
if table_h.column_exist('open_date'):
cursor.execute(
*table.update(
[table.numbered_at],
[table.open_date]))
table_h.drop_column('open_date')
@classmethod
def __setup__(cls):
super(Invoice, cls).__setup__()
cls.party.datetime_field = 'numbered_at'
if 'numbered_at' not in cls.party.depends:
cls.party.depends.append('numbered_at')
cls.invoice_address.datetime_field = 'numbered_at'
if 'numbered_at' not in cls.invoice_address.depends:
cls.invoice_address.depends.append('numbered_at')
cls.payment_term.datetime_field = 'numbered_at'
if 'numbered_at' not in cls.payment_term.depends:
cls.payment_term.depends.append('numbered_at')
@classmethod
def set_number(cls, invoices):
numbered = [i for i in invoices if not i.number or not i.numbered_at]
super(Invoice, cls).set_number(invoices)
if numbered:
cls.write(numbered, {
'numbered_at': CurrentTimestamp(),
})
@classmethod
@ModelView.button
@Workflow.transition('draft')
def draft(cls, invoices):
super(Invoice, cls).draft(invoices)
cls.write(invoices, {
'numbered_at': None,
})
@classmethod
def copy(cls, invoices, default=None):
if default is None:
default = {}
else:
default = default.copy()
default.setdefault('numbered_at', None)
return super(Invoice, cls).copy(invoices, default=default)