def load_data(settings=None, clear=None):
if settings is None:
settings = global_settings
global_settings.update(json.load(open(sys.argv[1])))
db = connect(settings)
if clear:
db.users.remove()
with open(users_filename) as users_file:
users = json.load(users_file)
for user in users:
db.users.insert(user)
def load_data(filename, settings=None, clear=None):
if settings is None:
settings = global_settings
global_settings.update(json.load(file(sys.argv[1])))
db = connect(settings)
obj_name = os.path.basename(filename).split('.')[0]
collection = getattr(db, obj_name)
if clear:
collection.remove()
with file(filename) as the_file:
objs = json.load(the_file)
for obj in objs:
collection.insert(obj)
def __init__(self, settings):
self.db = connect(settings)
def __init__(self, db_settings):
self.db = connect(db_settings)
def __init__(self, settings):
""" Connect to the given database and activate session management """
self.db = connect(settings)
cherrypy.config.update({'tools.sessions.on': True})
"""
from __future__ import print_function
from collections import defaultdict
import csv
import json
import sys
import bson
from bson.objectid import ObjectId
from cr.db.loader import load_dataset_to_dict
from cr.db.store import global_settings as settings
from cr.db.store import connect
settings.update({"url": "mongodb://localhost:27017/test_crunch_fitness"})
db = connect(settings)
def get_dataset(dataset_id=None):
"""
Get a dataset from the Mongo database
dataset_id:
None to pick a dataset at random.
Otherwise, the hex document ID of the dataset
Return the dataset document.
Raise IndexError if no matching dataset found.
"""
if dataset_id is None:
dataset_query = {}
else:
dataset_query = {'_id': ObjectId(dataset_id)}
