def test_to_mongo(self): record = Record({ 'name': 'bob', '_timestamp': d_tz(2013, 4, 4, 4, 4, 4) }) assert_that(record.to_mongo(), has_key('name')) assert_that(record.to_mongo(), has_key('_timestamp')) assert_that(record.to_mongo(), has_key('_week_start_at'))
def up(db): for name in db.collection_names(): log.info("Migrating collection: {0}".format(name)) collection = db.get_repository(name) query = { "_timestamp": {"$exists": True}, "_week_start_at": {"$exists": False} } for document in collection.find(query): document['_timestamp'] = utc(document['_timestamp']) record = Record(document) collection.save(record.to_mongo())
def up(db): for name in db.collection_names(): log.info("Migrating collection: {0}".format(name)) collection = db[name] query = {"_timestamp": {"$exists": True}, "_month_start_at": {"$exists": False}} for document in collection.find(query): document["_timestamp"] = utc(document["_timestamp"]) if "_week_start_at" in document: document.pop("_week_start_at") if "_updated_at" in document: document.pop("_updated_at") record = Record(document) collection.save(record.to_mongo())
def up(db): for name in db.collection_names(): log.info("Migrating collection: {0}".format(name)) collection = db[name] query = { "_timestamp": { "$exists": True }, "_week_start_at": { "$exists": False } } for document in collection.find(query): document['_timestamp'] = utc(document['_timestamp']) record = Record(document) collection.save(record.to_mongo())
def up(db): for name in db.collection_names(): log.info("Migrating collection: {0}".format(name)) collection = db.get_repository(name) query = { "_timestamp": {"$exists": True}, "_day_start_at": {"$exists": False} } is_capped = name.endswith('_realtime') if not is_capped: for document in collection.find(query): document['_timestamp'] = utc(document['_timestamp']) attrs = ['_updated_at', '_week_start_at', '_month_start_at'] for attr in attrs: if attr in document: document.pop(attr) record = Record(document) collection.save(record.to_mongo())