Cédric Krier
67d5291bfd
By using sql_column, it increases the possibilities of customisation of fields. issue4369 review14711002
62 lines
2.0 KiB
Python
62 lines
2.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.
|
|
from decimal import Decimal
|
|
from sql import Query, Expression, Cast, Literal, Select, CombiningQuery, As
|
|
|
|
from ... import backend
|
|
from .field import SQLType
|
|
from .float import Float
|
|
|
|
|
|
class SQLite_Cast(Cast):
|
|
|
|
def as_(self, output_name):
|
|
# Use PARSE_COLNAMES instead of CAST for final column
|
|
return As(self.expression, '%s [NUMERIC]' % output_name)
|
|
|
|
|
|
class Numeric(Float):
|
|
'''
|
|
Define a numeric field (``decimal``).
|
|
'''
|
|
_type = 'numeric'
|
|
|
|
@staticmethod
|
|
def sql_format(value):
|
|
if isinstance(value, (Query, Expression)):
|
|
return value
|
|
if value is None:
|
|
return None
|
|
if isinstance(value, (int, long)):
|
|
value = Decimal(str(value))
|
|
assert isinstance(value, Decimal)
|
|
return value
|
|
|
|
def sql_type(self):
|
|
db_type = backend.name()
|
|
if db_type == 'mysql':
|
|
return SQLType('DECIMAL', 'DECIMAL(65, 30)')
|
|
return SQLType('NUMERIC', 'NUMERIC')
|
|
|
|
def sql_column(self, table):
|
|
column = super(Numeric, self).sql_column(table)
|
|
db_type = backend.name()
|
|
if db_type == 'sqlite':
|
|
# Must be casted as Decimal is stored as bytes
|
|
column = SQLite_Cast(column, self.sql_type().base)
|
|
return column
|
|
|
|
def _domain_value(self, operator, value):
|
|
value = super(Numeric, self)._domain_value(operator, value)
|
|
db_type = backend.name()
|
|
if db_type == 'sqlite':
|
|
if isinstance(value, (Select, CombiningQuery)):
|
|
return value
|
|
# Must be casted as Decimal is adapted to bytes
|
|
type_ = self.sql_type().base
|
|
if operator in ('in', 'not in'):
|
|
return [Cast(Literal(v), type_) for v in value]
|
|
elif value is not None:
|
|
return Cast(Literal(value), type_)
|
|
return value
|