reset to master, keep working on the dirty-fields-patch in another branch

This commit is contained in:
Florian Schlachter 2010-01-31 18:00:01 +01:00
parent 963a223e7e
commit 73aff806f3
3 changed files with 6 additions and 36 deletions

View File

@ -15,14 +15,13 @@ class BaseField(object):
_index_with_types = True _index_with_types = True
def __init__(self, name=None, required=False, default=None, unique=False, def __init__(self, name=None, required=False, default=None, unique=False,
unique_with=None, primary_key=False, modified=False): unique_with=None, primary_key=False):
self.name = name if not primary_key else '_id' self.name = name if not primary_key else '_id'
self.required = required or primary_key self.required = required or primary_key
self.default = default self.default = default
self.unique = bool(unique or unique_with) self.unique = bool(unique or unique_with)
self.unique_with = unique_with self.unique_with = unique_with
self.primary_key = primary_key self.primary_key = primary_key
self.modified = modified
def __get__(self, instance, owner): def __get__(self, instance, owner):
"""Descriptor for retrieving a value from a field in a document. Do """Descriptor for retrieving a value from a field in a document. Do
@ -45,7 +44,6 @@ class BaseField(object):
"""Descriptor for assigning a value to a field in a document. """Descriptor for assigning a value to a field in a document.
""" """
instance._data[self.name] = value instance._data[self.name] = value
self.modified = True
def to_python(self, value): def to_python(self, value):
"""Convert a MongoDB-compatible type to a Python type. """Convert a MongoDB-compatible type to a Python type.
@ -254,11 +252,8 @@ class BaseDocument(object):
def __init__(self, **values): def __init__(self, **values):
self._data = {} self._data = {}
modified = 'id' in values.keys()
# Assign initial values to instance # Assign initial values to instance
for attr_name, attr_value in self._fields.items(): for attr_name, attr_value in self._fields.items():
attr_value.modified = modified
if attr_name in values: if attr_name in values:
setattr(self, attr_name, values.pop(attr_name)) setattr(self, attr_name, values.pop(attr_name))
else: else:
@ -382,9 +377,9 @@ class BaseDocument(object):
# that has been queried to return this SON # that has been queried to return this SON
return None return None
cls = subclasses[class_name] cls = subclasses[class_name]
for field_name, field in cls._fields.items(): for field_name, field in cls._fields.items():
if field.name in data: if field.name in data:
data[field_name] = field.to_python(data[field.name]) data[field_name] = field.to_python(data[field.name])
return cls(**data) return cls(**data)

View File

@ -1,9 +1,8 @@
from base import (DocumentMetaclass, TopLevelDocumentMetaclass, BaseDocument, from base import (DocumentMetaclass, TopLevelDocumentMetaclass, BaseDocument,
ValidationError) ValidationError)
from queryset import OperationError, QuerySet from queryset import OperationError
from connection import _get_db from connection import _get_db
import pymongo import pymongo
@ -76,34 +75,12 @@ class Document(BaseDocument):
if force_insert: if force_insert:
object_id = collection.insert(doc, safe=safe) object_id = collection.insert(doc, safe=safe)
else: else:
if getattr(self, 'id', None) == None: object_id = collection.save(doc, safe=safe)
# new document
object_id = collection.save(doc, safe=safe)
else:
# update document
modified_fields = map(lambda obj: obj[0], filter(lambda obj: obj[1].modified, self._fields.items()))
modified_doc = dict(filter(lambda k: k[0] in modified_fields, doc.items()))
try:
#
# WORK IN PROGRESS
# - EmbeddedDocuments still aren't tracked
#
id_field = self._meta['id_field']
idObj = self._fields[id_field].to_mongo(self['id'])
collection.update({'_id': idObj}, {'$set': modified_doc}, safe=safe)
except pymongo.errors.OperationFailure, err:
if str(err) == 'multi not coded yet':
raise OperationError('update() method requires MongoDB 1.1.3+')
raise OperationError('Update failed (%s)' % str(err))
object_id = self['id']
for field in self._fields.values(): field.modified = False
except pymongo.errors.OperationFailure, err: except pymongo.errors.OperationFailure, err:
message = 'Could not save document (%s)' message = 'Could not save document (%s)'
if 'duplicate key' in str(err): if 'duplicate key' in str(err):
message = 'Tried to save duplicate unique keys (%s)' message = 'Tried to save duplicate unique keys (%s)'
raise OperationError(message % str(err)) raise OperationError(message % str(err))
id_field = self._meta['id_field'] id_field = self._meta['id_field']
self[id_field] = self._fields[id_field].to_python(object_id) self[id_field] = self._fields[id_field].to_python(object_id)
@ -129,7 +106,6 @@ class Document(BaseDocument):
obj = self.__class__.objects(**{id_field: self[id_field]}).first() obj = self.__class__.objects(**{id_field: self[id_field]}).first()
for field in self._fields: for field in self._fields:
setattr(self, field, obj[field]) setattr(self, field, obj[field])
obj.modified = False
@classmethod @classmethod
def drop_collection(cls): def drop_collection(cls):

View File

@ -71,7 +71,6 @@ class FloatField(BaseField):
return float(value) return float(value)
def validate(self, value): def validate(self, value):
if isinstance(value, int): value = float(value)
assert isinstance(value, float) assert isinstance(value, float)
if self.min_value is not None and value < self.min_value: if self.min_value is not None and value < self.min_value: