Reverted the changes I made as everything I had done is in the code base now.

This commit is contained in:
Greg Doermann
2014-07-30 18:17:28 -06:00
parent 27b0a27d03
commit 6c03b320d1
4 changed files with 13 additions and 34 deletions

View File

@@ -4,7 +4,7 @@ from cassandra import ConsistencyLevel
from cqlengine.columns import *
from cqlengine.functions import *
from cqlengine.models import Model, CounterModel
from cqlengine.models import Model
from cqlengine.query import BatchQuery

View File

@@ -473,11 +473,15 @@ class Float(Column):
class Decimal(Column):
db_type = 'decimal'
class Counter(Column):
#TODO: counter field
def __init__(self, **kwargs):
super(Counter, self).__init__(**kwargs)
raise NotImplementedError
def validate(self, value):
from decimal import Decimal as _Decimal
from decimal import InvalidOperation
val = super(Decimal, self).validate(value)
if val is None: return
try:
return _Decimal(val)
except InvalidOperation:
raise ValidationError("'{}' can't be coerced to decimal".format(val))
def to_python(self, value):
return self.validate(value)

View File

@@ -116,18 +116,3 @@ class Token(BaseQueryFunction):
for i, (col, val) in enumerate(zip(self._columns, self.value)):
ctx[str(self.context_id + i)] = col.to_database(val)
class NotSet(object):
"""
Different from None, so you know when it just wasn't set compared to passing in None
"""
pass
def format_timestamp(timestamp):
if isinstance(timestamp, datetime):
epoch = datetime(1970, 1, 1)
ts = long((timestamp - epoch).total_seconds() * 1000)
else :
ts = long(timestamp)
return ts

View File

@@ -8,7 +8,6 @@ from cqlengine.query import ModelQuerySet, DMLQuery, AbstractQueryableColumn
from cqlengine.query import DoesNotExist as _DoesNotExist
from cqlengine.query import MultipleObjectsReturned as _MultipleObjectsReturned
class ModelDefinitionException(ModelException): pass
@@ -279,6 +278,9 @@ class BaseModel(object):
def __init__(self, **values):
self._values = {}
self._ttl = None
self._timestamp = None
for name, column in self._columns.items():
value = values.get(name, None)
if value is not None or isinstance(column, columns.BaseContainerColumn):
@@ -772,15 +774,3 @@ class Model(BaseModel):
__metaclass__ = ModelMetaClass
class CounterBaseModel(BaseModel):
def _can_update(self):
# INSERT is not allowed for counter column families
return True
class CounterModel(CounterBaseModel):
"""
the db name for the column family can be set as the attribute db_name, or
it will be genertaed from the class name
"""
__metaclass__ = ModelMetaClass