weekly-new-users.py

__author__ = 'ronfe'

from pymongo import MongoClient
import datetime
from bson.objectid import ObjectId

db = MongoClient('mongodb://10.8.8.8:27017')['matrix-yangcong-prod25']

start_date = datetime.datetime(2015, 10, 26)
mid_date = datetime.datetime(2015, 11, 1)
end_date = datetime.datetime(2015, 11, 7)

points = db['points']
pipeLine = [
    {"$match": {"usefulData.registDate": {"$gte": start_date, "$lt": end_date}}},
    {"$group": {"_id": None, "users": {"$push": "$_id"}}}
]

users = db['users']
two_weeks_users = set(list(users.aggregate(pipeLine))[0]['users'])
print 'done A'

# Get all the points user
pipeLine = [
    {"$match": {"createdBy": {"$gte": mid_date, "$lt": end_date}}},
    {"$group": {"_id": None, "users": {"$addToSet": "$user"}}}
]
one_week_activates = set(list(points.aggregate(pipeLine))[0]['users'])

result = two_weeks_users.intersection(one_week_activates)
print len(result)