Add Duration support to cqlengine
This commit is contained in:
parent
9adf704800
commit
e80f4b6b6b
@ -22,6 +22,7 @@ from cassandra import util
|
||||
from cassandra.cqltypes import SimpleDateType, _cqltypes, UserType
|
||||
from cassandra.cqlengine import ValidationError
|
||||
from cassandra.cqlengine.functions import get_total_seconds
|
||||
from cassandra.util import Duration as _Duration
|
||||
|
||||
log = logging.getLogger(__name__)
|
||||
|
||||
@ -598,6 +599,25 @@ class Time(Column):
|
||||
return value if isinstance(value, util.Time) else util.Time(value)
|
||||
|
||||
|
||||
class Duration(Column):
|
||||
"""
|
||||
Stores a duration (months, days, nanoseconds)
|
||||
|
||||
.. versionadded:: 3.10.0
|
||||
|
||||
requires C* 3.10+ and protocol v4+
|
||||
"""
|
||||
db_type = 'duration'
|
||||
|
||||
def validate(self, value):
|
||||
val = super(Duration, self).validate(value)
|
||||
if val is None:
|
||||
return
|
||||
if not isinstance(val, _Duration):
|
||||
raise TypeError('{0} {1} is not a valid Duration.'.format(self.column_name, value))
|
||||
return val
|
||||
|
||||
|
||||
class UUID(Column):
|
||||
"""
|
||||
Stores a type 1 or 4 UUID
|
||||
|
Loading…
Reference in New Issue
Block a user