FileManager.py

from pymongo import MongoClient
import bcrypt
import Login

usersMockup = {'Tommy':'pop', 'Brendan':'ih8slackbot', 'Nate':'kingProjectManager', 'Justin':'whyPop?', 'Terrence':'What?', 'John':'mongoDBisDaBest', 'Simran':'pyth0n'}

DATABASE_ADDRESS = 'ec2-54-173-26-10.compute-1.amazonaws.com'
#DATABASE_ADDRESS = 'localhost'
DATABASE_PORT = 27017
DATABASE_NAME = 'SaveYourDB'

def connectToDB():
	global client
	global db
	global users
	client = MongoClient()
	db = client[DATABASE_NAME]
	users = db.SaveYourDB


def getFieldMockup(username, field):
	""" MOCKUP FOR NOW"""
	if (field != 'password'):
		return 'ACCESS DENIED'

	try:
		return usersMockup[username]
	except KeyError:
		return 'NOT FOUND'

def getField(username, field):
	data = users.find_one({"usr": username},{field:True})
	if data == None:
		return None
	try:
		return data[field]
		
	except KeyError:
		return None

def setField(username, field, data):
	users.update({'usr': username}, {'$set': {field: data}})

def deleteField(username, field):
	users.update({'usr': username}, {'$unset': {field:''}})

def createUser(username, password):
	foundUsers = users.find({'usr':username})
	if foundUsers.count() > 0:
		return False
	hashedPW = bcrypt.hashpw(password, bcrypt.gensalt())
	users.insert({'usr': username, 'pwd':hashedPW, 'data':''})

	return True

def removeUser(username):
	users.remove({'usr': username})

connectToDB()