timed_objectid.py

>>> import contextlib
>>> import struct
>>> import bson
>>> import mock
>>>
>>> @contextlib.contextmanager
... def timed_oid(time):
...     increment = [0]
...     RealObjectId = bson.ObjectId
...     def TimedObjectId(oid=None):
...         if oid:
...             return RealObjectId(oid)
...         increment[0] += 1
...         return TimedObjectId(struct.pack(">2I2H", int(time), 0, 0, increment[0]))
...     with mock.patch('bson.ObjectId', side_effect=TimedObjectId):
...         yield
...
>>> with timed_oid(6666):
...     print bson.ObjectId()
...     print bson.ObjectId()
...     print bson.ObjectId()
...     print bson.ObjectId("58dac45151c356651605250a")
...     print bson.ObjectId()
...
00001a0a0000000000000001
00001a0a0000000000000002
00001a0a0000000000000003
58dac45151c356651605250a
00001a0a0000000000000004
>>> with timed_oid(7777):
...     print bson.ObjectId()
...     print bson.ObjectId()
...     print bson.ObjectId()
...     print bson.ObjectId()
...
00001e610000000000000001
00001e610000000000000002
00001e610000000000000003
00001e610000000000000004