sm_mong_objs.py

def mongo_objs(conn_str, database, tests=[lambda x: x.document_count({})]):
    client = pymongo.MongoClient(conn_str)
    db = client[database]
    sslog = db.sslog
    cycle = db.cycle
    sslog_results = {}
    cycle_results = {}
    for test in tests:
        try:
            sslog_results[test.__name__] = test(sslog)
        except TypeError as e:
            sslog_results[test.__name__] = e
        
        try:
            cycle_results[test.__name__] = test(cycle)
        except TypeError as e:
            cycle_results[test.__name__] = e
    
    return client, db, sslog, cycle, sslog_results, cycle_results