db.py

import os
import pymongo
# for secure passwords
import bcrypt

### Database setup ###
mongo_pass = os.getenv('mongoaccess')
mongo_uri = "mongodb+srv://HexHalf:{}@hexhax-3ipx7.mongodb.net/HexHax?retryWrites=true&w=majority".format(mongo_pass)
client = pymongo.MongoClient(mongo_uri)
db = client.main
users = db.users

### Utility functions ###
# Basic db structure of a user
def user_template():
	return {
		"name": "",
		"pwrd": "",
		"tags": [],
		"stories": {}
	}

def get_user_by(field, value):
	# Get user by field (username, password, id, etc)
	return users.find_one({field: value}, {"_id":0, "pwrd": 0})

def get_user_by_full(field, value):
	# Get user by field (username, password, id, etc)
	return users.find_one({field: value}, {"_id":0})

def create_user(name, pwrd):
	# Prevent creating a user if name already exists
	if not get_user_by("name", name):
		# Create user and password
		base = user_template()
		base["name"] = name
		base["pwrd"] = bcrypt.hashpw(pwrd.encode("UTF-8"), bcrypt.gensalt())
		users.insert_one(base)
		return base

def verify_user(name, pwrd):
	user = get_user_by_full("name", name)
	if user:
		# Make sure the user exists.
		return bcrypt.checkpw(pwrd.encode("UTF-8"), get_user_by_full("name", name)["pwrd"])
	return False

def delete_user(name):
	return users.delete_one({"name": name})

def change_bio(name, bio):
	user = get_user_by("name", name)
	if user:
		users.update_one({'name': name}, {"$set": {"bio": bio}})

def change_tags(name, tags):
	user = get_user_by("name", name)
	if user:
		users.update_one({'name': name}, {"$set": {"tags": tags}})

def friend(name, target):
	user = get_user_by("name", name)
	if user:
		users.update_one({'name': name}, {"$addToSet": {"friends": target}})

def unfriend(name, target):
	user = get_user_by("name", name)
	if user:
		users.update_one({'name': name}, {"$pull": {"friends": target}})

def new_story(name, title, story):
	user = get_user_by("name", name)
	if user:
		users.update_one({'name': name}, {"$set": {"stories."+title: story}})

def comment(author, name, title, comment):
	author_obj = get_user_by("name", author)
	if author_obj:
		#print({"stories."+title+".comments": [name, comment]})
		users.update_one({'name': author}, {"$push": {"stories."+title+".comments": [name, comment]}})

def delete_story(name, title):
	user = get_user_by("name", name)
	if user:
		users.update_one({'name': name}, {"$unset": {"stories."+title: ""}})

### DANGER ###
def delete_all_users():
	return users.delete_many({})