def iter_hosts_dict(page=None): clients_collection = mongo.get_collection("clients") server_collection = mongo.get_collection("servers") response = server_collection.aggregate( [ {"$project": {"hosts": True, "organizations": True}}, {"$unwind": "$hosts"}, {"$unwind": "$organizations"}, {"$group": {"_id": "$hosts", "organizations": {"$addToSet": "$organizations"}}}, ] ) orgs = set() host_orgs = collections.defaultdict(list) for doc in response: orgs = orgs.union(doc["organizations"]) host_orgs[doc["_id"]] = doc["organizations"] org_user_count = organization.get_user_count(orgs) for hst in iter_hosts(page=page): users_online = len(clients_collection.distinct("user_id", {"host_id": hst.id, "type": CERT_CLIENT})) user_count = 0 for org_id in host_orgs[hst.id]: user_count += org_user_count.get(org_id, 0) hst.user_count = user_count hst.users_online = users_online yield hst.dict()
def iter_hosts_dict(page=None): server_collection = mongo.get_collection('servers') response = server_collection.aggregate([ {'$project': { 'host_id': '$instances.host_id', 'client': '$instances.clients', }}, {'$unwind': '$host_id'}, {'$unwind': '$client'}, {'$unwind': '$client'}, {'$match': { 'client.type': CERT_CLIENT, }}, {'$group': { '_id': '$host_id', 'clients': {'$addToSet': '$client.id'}, }}, ]) hosts_clients = {} for doc in response: hosts_clients[doc['_id']] = doc['clients'] response = server_collection.aggregate([ {'$project': { 'hosts': True, 'organizations': True, }}, {'$unwind': '$hosts'}, {'$unwind': '$organizations'}, {'$group': { '_id': '$hosts', 'organizations': {'$addToSet': '$organizations'}, }}, ]) orgs = set() host_orgs = collections.defaultdict(list) for doc in response: orgs = orgs.union(doc['organizations']) host_orgs[doc['_id']] = doc['organizations'] org_user_count = organization.get_user_count(orgs) for hst in iter_hosts(page=page): users_online = len(hosts_clients.get(hst.id, [])) user_count = 0 for org_id in host_orgs[hst.id]: user_count += org_user_count.get(org_id, 0) hst.user_count = user_count hst.users_online = users_online yield hst.dict()
def iter_servers_dict(): server_collection = mongo.get_collection('servers') response = server_collection.aggregate([ {'$project': { 'host_id': '$instances.host_id', 'client': '$instances.clients', }}, {'$unwind': '$host_id'}, {'$unwind': '$client'}, {'$unwind': '$client'}, {'$match': { 'client.type': CERT_CLIENT, }}, {'$group': { '_id': '$host_id', 'clients': {'$addToSet': '$client.id'}, }}, ])['result'] hosts_clients = {} for doc in response: hosts_clients[doc['_id']] = doc['clients'] org_user_count = organization.get_user_count() response = server_collection.aggregate([ {'$project': { 'hosts': True, 'organizations': True, }}, {'$unwind': '$hosts'}, {'$unwind': '$organizations'}, {'$group': { '_id': '$hosts', 'organizations': {'$addToSet': '$organizations'}, }}, ])['result'] host_orgs = collections.defaultdict(list) for doc in response: host_orgs[doc['_id']] = doc['organizations'] for doc in Host.collection.find().sort('name'): hst = Host(doc=doc) users_online = len(hosts_clients.get(hst.id, '')) user_count = 0 for org_id in host_orgs[hst.id]: user_count += org_user_count.get(org_id, 0) hst.user_count = user_count hst.users_online = users_online yield hst.dict()
def iter_servers_dict(): server_collection = mongo.get_collection('servers') response = server_collection.aggregate([ {'$project': { 'host_id': '$instances.host_id', 'client': '$instances.clients', }}, {'$unwind': '$host_id'}, {'$unwind': '$client'}, {'$unwind': '$client'}, {'$match': { 'client.type': CERT_CLIENT, }}, {'$group': { '_id': '$host_id', 'clients': {'$addToSet': '$client.id'}, }}, ])['result'] hosts_clients = {} for doc in response: hosts_clients[doc['_id']] = doc['clients'] org_user_count = organization.get_user_count() response = server_collection.aggregate([ {'$project': { 'hosts': True, 'organizations': True, }}, {'$unwind': '$hosts'}, {'$unwind': '$organizations'}, {'$group': { '_id': '$hosts', 'organizations': {'$addToSet': '$organizations'}, }}, ])['result'] host_orgs = collections.defaultdict(list) for doc in response: host_orgs[doc['_id']] = doc['organizations'] for doc in Host.collection.find().sort('name'): hst = Host(doc=doc) users_online = len(hosts_clients.get(hst.id, [])) user_count = 0 for org_id in host_orgs[hst.id]: user_count += org_user_count.get(org_id, 0) hst.user_count = user_count hst.users_online = users_online yield hst.dict()
def iter_hosts_dict(page=None): server_collection = mongo.get_collection('servers') response = server_collection.aggregate([ {'$project': { 'host_id': '$instances.host_id', 'client': '$instances.clients', }}, {'$unwind': '$host_id'}, {'$unwind': '$client'}, {'$unwind': '$client'}, {'$match': { 'client.type': CERT_CLIENT, }}, {'$group': { '_id': '$host_id', 'clients': {'$addToSet': '$client.id'}, }}, ]) hosts_clients = {} for doc in response: hosts_clients[doc['_id']] = doc['clients'] response = server_collection.aggregate([ {'$project': { 'hosts': True, 'organizations': True, }}, {'$unwind': '$hosts'}, {'$unwind': '$organizations'}, {'$group': { '_id': '$hosts', 'organizations': {'$addToSet': '$organizations'}, }}, ]) orgs = set() host_orgs = collections.defaultdict(list) for doc in response: orgs = orgs.union(doc['organizations']) host_orgs[doc['_id']] = doc['organizations'] org_user_count = organization.get_user_count(orgs) for hst in iter_hosts(page=page): users_online = len(hosts_clients.get(hst.id, [])) user_count = 0 for org_id in host_orgs[hst.id]: user_count += org_user_count.get(org_id, 0) hst.user_count = user_count hst.users_online = users_online yield hst.dict()