analysis.py

import csv
from pymongo import MongoClient
from datetime import datetime
import json

client = MongoClient()

incidents = client.acab_db.incidents

def incident_types(year):
    return incidents.aggregate([
        { '$match' : {'ZIP':'94110', 'Date': {'$gte':datetime(year,1,1),'$lte':datetime(year,12,31)} } },
        { '$group' : {"_id":"$Resolution", "count":{"$sum":1} } },
        { '$sort' : {'count':-1} }
    ])


def danger(year):
    return incidents.find({'Date': {'$gte':datetime(year,1,1),'$lte':datetime(year,12,31)}, 'Descript':'DANGER OF LEADING IMMORAL LIFE', 'Resolution':'JUVENILE BOOKED' })

def juveniles(year):
    return incidents.aggregate([
        { '$match' : {'Date': {'$gte':datetime(year,1,1),'$lte':datetime(year,12,31)}, 'Resolution':'JUVENILE BOOKED' } },
        { '$group' : {"_id":"$Descript", "count":{"$sum":1} } },
        { '$sort' : {'count':-1} }
    ])

def incidentnums(year):
    thing = incidents.aggregate([
        { '$match' : {'Date': {'$gte':datetime(year,1,1),'$lte':datetime(year,12,31)} } },
        { '$group' : {"_id":"$IncidntNum", "count":{"$sum":1} } },
        { '$sort' : {'count':-1} }
    ])
    return len(thing['result'])

def resolution(year):
    return incidents.aggregate([
        { '$match' : {'Date': {'$gte':datetime(year,1,1),'$lte':datetime(year,12,31)} } },
        { '$group' : {"_id":"$Resolution", "count":{"$sum":1} } },
        { '$sort' : {'count':-1} }
    ])

def write_to_file(filename,data):
    outfile = open('analysis/%s.json' % filename,'w+')
    outjson = json.dumps(data)
    outfile.write(outjson)
    outfile.close()

def do_stuff(func,filename):
    outdata = {}
    for year in range(2003,2015):
        outdata[year] = globals()[func](year)['result']
    write_to_file(filename, outdata)
    

"""
Quality of life crimes
    -Drunk in 
Number of officers
"""