def find(cls, dataset):
"""
Return the calculations for given *dataset*.
"""
return [mongo_remove_reserved_keys(record) for record in cls.collection.find({
DATASET_ID: dataset[DATASET_ID],
})]
except ParseError, err:
# do not save record, return error
return {ERROR: err}
record = {
DATASET_ID: dataset[DATASET_ID],
cls.FORMULA: formula,
cls.NAME: name,
}
cls.collection.insert(record)
# invalidate summary ALL since we have a new column
stats = dataset.get(STATS)
if stats:
del stats[ALL]
del dataset[STATS]
Dataset.update(dataset, {STATS: stats})
# call remote calculate and pass calculation id
calculate_column.delay(dataset, dframe, formula, name)
return mongo_remove_reserved_keys(record)
@classmethod
def find(cls, dataset):
"""
Return the calculations for given *dataset*.
"""
return [mongo_remove_reserved_keys(record) for record in cls.collection.find({
DATASET_ID: dataset[DATASET_ID],
})]
