active.py

# _*_ coding: utf8 _*_

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

# link to db
dbClient = MongoClient('mongodb://localhost:27017')
db = dbClient['yangcong-prod25']

# open collections
points = db['points']
users = db['users']
rooms = db['rooms']

# configure daterange

lastWeekStartDate = datetime.datetime(2015,11,01)
lastWeekEndDate   = lastWeekStartDate + datetime.timedelta(days = 7)

startDate = datetime.datetime(2015,11,8)
endDate   = startDate + datetime.timedelta(days = 7)

print("2015-11-8 ~ 2015-11-14")
print("")

print("###### Part.3 周活跃 ######")

def calPcWork(startDate, endDate):
    pipeLine = [
        {"$match": {
            "createdBy": {
                "$gte": startDate,
                "$lt": endDate
            },
            "from": "pc"
        }},
        {
            "$group": {
                "_id": "$user"
            }
        }
    ]
    return list(points.aggregate(pipeLine))

pcActUsers = calPcWork(startDate, endDate)
pcActUsersArray = []
for x in pcActUsers:
    pcActUsersArray.append(x['_id'])
# print("PC本周活动用户:")
# print(len(pcActUsersArray))

def isUsers(startDate, endDate):
    pipeLine = [
        {"$match": {
            "usefulData.registDate": {
                "$gte": startDate,
                "$lt": endDate
            },
            "usefulData.from": "signup"
        }},
        {"$group": {
            "_id": "$_id"
        }}
    ]
    return list(users.aggregate(pipeLine))

def isAllUsers():
    pipeLine = [
        {"$group": {
            "_id": None,
            "users": {"$push": "$_id"}
        }}
    ]
    return list(users.aggregate(pipeLine))[0]['users']


regUsers = isAllUsers()
# print("PC本周注册用户")
# print(len(regUsersArray))

pcOutArray = []
pcOutArray = list(set.intersection(set(regUsers), set(pcActUsersArray)))
print("PC周活跃统计")
print(len(pcOutArray))