mongoFunctions.py

import pymongo

#I will change the username/password of the mongohq once the app is actually in production
connection = pymongo.Connection('mongodb://stumbler:pennapps@linus.mongohq.com:10061/Stumblr')
db = connection.Stumblr.Users

def insert_user(userID, tags):

    tagdict = {}
    for tag in tags:
        tagdict[tag] = 1
    if tagdict == {}:
        tagdict['Hacking'] = 10

    #Inserts a user's information into the collection.
    post = {"userid": userID,
            "tags": tagdict,
            "urls": {},
            "recently_visited": [],
            "favorites": {}
            }
    #only insert if the user is not already in the collection.
    if db.find({"userid": userID}).count() == 0:
        db.insert(post)

def hastags(userid):
    val = db.find_one({'userid': userid})
    tags = val['tags']
    return not tags

def recently_visited(userid):
    val = db.find_one({'userid': userid})
    if len(val['recently_visited']) > 100:
        val['recently_visited'] = []
    db.update({'userid': userid}, val)
    return val['recently_visited']

def add_to_recently_visited(userid, site):
    val = db.find_one({'userid': userid})
    recentSites = val['recently_visited']
    recentSites.append(site)
    db.update({'userid': userid}, val)

def add_to_favorites(userid, url, title):
    val = db.find_one({'userid': userid})
    favorites = val['favorites']
    favorites[title] = url
    db.update({'userid': userid}, val)

def replace_tags(userid, tags):
    """
    In case I want to remove a tag from the db
    This takes a list of tags, an replaces the tags in the db with this new list
    """

    val = db.find_one({'userid': userid})
    val['tags'] = tags
    db.update({'userid': userid}, val)

def update_tags(userID, tags, num):
    val = db.find_one({'userid': userID})
    tagVals = val['tags']

    for t in tags:
        if t == "invalid_tag":
            del(tagVals[t])
            continue
        if t in tagVals.keys():
            tagVals[t] = tagVals[t] + num
        else:
            tagVals[t] = 1
        if tagVals[t] <= 0:
            tagVals[t] = 1
    db.update({'userid':userID},val)

def get_tags(userid):
    return db.find_one({'userid': userid})['tags']

def get_favorites(userid):
    return db.find_one({"userid": userid})['favorites']

def exists(userid):
    return userid in db.find({"userid": userid})['userid']

def get_tags(userid):
    return db.find_one({'userid': userid})['tags']