Cédric Krier
fc6c6cfdda
* Use python-sql * Add convert_domain method on Fields * Add sql_format and sql_type methods on Fields * Allow to return a full domain for Function.searcher * Replace static backend by dynamic get method * Replace order_field by order_<field name> method review1001002
137 lines
4.4 KiB
Python
137 lines
4.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.
|
|
import datetime
|
|
from sql import Query, Expression
|
|
|
|
from ...config import CONFIG
|
|
from .field import Field, SQLType
|
|
|
|
|
|
class Date(Field):
|
|
'''
|
|
Define a date field (``date``).
|
|
'''
|
|
_type = 'date'
|
|
|
|
@staticmethod
|
|
def sql_format(value):
|
|
if isinstance(value, (Query, Expression)):
|
|
return value
|
|
if value is None:
|
|
return None
|
|
if isinstance(value, basestring):
|
|
year, month, day = map(int, value.split("-", 2))
|
|
return datetime.date(year, month, day)
|
|
|
|
assert(isinstance(value, datetime.date))
|
|
# Allow datetime with min time for XML-RPC
|
|
# datetime must be tested separately because datetime is a
|
|
# subclass of date
|
|
assert(not isinstance(value, datetime.datetime)
|
|
or value.time() == datetime.time())
|
|
return value
|
|
|
|
def sql_type(self):
|
|
return SQLType('DATE', 'DATE')
|
|
|
|
|
|
class DateTime(Field):
|
|
'''
|
|
Define a datetime field (``datetime``).
|
|
'''
|
|
_type = 'datetime'
|
|
|
|
def __init__(self, string='', format='%H:%M:%S', help='', required=False,
|
|
readonly=False, domain=None, states=None, select=False,
|
|
on_change=None, on_change_with=None, depends=None,
|
|
context=None, loading='eager'):
|
|
'''
|
|
:param format: The validation format as used by strftime.
|
|
'''
|
|
super(DateTime, self).__init__(string=string, help=help,
|
|
required=required, readonly=readonly, domain=domain, states=states,
|
|
select=select, on_change=on_change, on_change_with=on_change_with,
|
|
depends=depends, context=context, loading=loading)
|
|
self.format = format
|
|
|
|
__init__.__doc__ += Field.__init__.__doc__
|
|
|
|
@staticmethod
|
|
def sql_format(value):
|
|
if isinstance(value, (Query, Expression)):
|
|
return value
|
|
if not value:
|
|
return None
|
|
if isinstance(value, basestring):
|
|
datepart, timepart = value.split(" ")
|
|
year, month, day = map(int, datepart.split("-", 2))
|
|
hours, minutes, seconds = map(int, timepart.split(":"))
|
|
return datetime.datetime(year, month, day, hours, minutes, seconds)
|
|
assert(isinstance(value, datetime.datetime))
|
|
return value.replace(microsecond=0)
|
|
|
|
def sql_type(self):
|
|
db_type = CONFIG['db_type']
|
|
if db_type == 'sqlite':
|
|
return SQLType('TIMESTAMP', 'TIMESTAMP')
|
|
elif db_type == 'mysql':
|
|
return SQLType('TIMESTAMP', 'TIMESTAMP NULL')
|
|
return SQLType('TIMESTAMP', 'TIMESTAMP(0)')
|
|
|
|
|
|
class Timestamp(Field):
|
|
'''
|
|
Define a timestamp field (``datetime``).
|
|
'''
|
|
_type = 'timestamp'
|
|
|
|
@staticmethod
|
|
def sql_format(value):
|
|
if isinstance(value, (Query, Expression)):
|
|
return value
|
|
if value is None:
|
|
return None
|
|
if isinstance(value, basestring):
|
|
datepart, timepart = value.split(" ")
|
|
year, month, day = map(int, datepart.split("-", 2))
|
|
timepart_full = timepart.split(".", 1)
|
|
hours, minutes, seconds = map(int, timepart_full[0].split(":"))
|
|
if len(timepart_full) == 2:
|
|
microseconds = int(timepart_full[1])
|
|
else:
|
|
microseconds = 0
|
|
return datetime.datetime(year, month, day, hours, minutes, seconds,
|
|
microseconds)
|
|
assert(isinstance(value, datetime.datetime))
|
|
return value
|
|
|
|
def sql_type(self):
|
|
db_type = CONFIG['db_type']
|
|
if db_type == 'sqlite':
|
|
return SQLType('TIMESTAMP', 'TIMESTAMP')
|
|
elif db_type == 'mysql':
|
|
return SQLType('TIMESTAMP', 'TIMESTAMP NULL')
|
|
return SQLType('TIMESTAMP', 'TIMESTAMP(6)')
|
|
|
|
|
|
class Time(DateTime):
|
|
'''
|
|
Define a time field (``time``).
|
|
'''
|
|
_type = 'time'
|
|
|
|
@staticmethod
|
|
def sql_format(value):
|
|
if isinstance(value, (Query, Expression)):
|
|
return value
|
|
if value is None:
|
|
return None
|
|
if isinstance(value, basestring):
|
|
hours, minutes, seconds = map(int, value.split(":"))
|
|
return datetime.time(hours, minutes, seconds)
|
|
assert(isinstance(value, datetime.time))
|
|
return value.replace(microsecond=0)
|
|
|
|
def sql_type(self):
|
|
return SQLType('TIME', 'TIME')
|