import time
from couchbase.bucket import Bucket
from couchbase.n1ql import N1QLQuery
import couchbase.n1ql as N

cb = Bucket('couchbase://localhost/default?fetch_mutation_tokens=true')

NOW_TS = str(time.time())

docs = {}
for x in xrange(10000):
    k = "Key_{}".format(x)
    v = {
        'value': k,
        'time': NOW_TS
    }
    docs[k] = v


rvs = cb.upsert_multi(docs)

ms = N.MutationState()
ms.add_results(*rvs.values())

q = N1QLQuery("SELECT * FROM default WHERE time=$1", NOW_TS)
q.consistent_with(ms)

nrows = 0
for row in cb.n1ql_query(q):
    nrows+=1
    # print row

print "Got {} rows".format(nrows)

cb.remove_multi(docs.keys())