from pymongo import MongoClient
import sys
import os
import time
from passlib.hash import sha256_crypt
import pprint

client = MongoClient()


def login():
    loginUsername = input('Please input username to be signed in.')
    loginPassword = input('Please input password to be signed in.')


def registerAccount():
    registerUsername = input('Please input a username to be registered into our database.')
    registerPassword = input('Please input a password to be entered into our database.')
    passwordHash = sha256_crypt.hash(registerPassword)

    regDetails = {
        "username": registerUsername,
        "password": registerPassword,
        "passwordhash": passwordHash
    }

    db = client.users
    db.users.insert(regDetails)
    for a in db.users.find():
        pprint.pprint(a)


def checkRegistered():
    print('Coming soon')


def quit():
    os.system('clear')
    print('Closing...')
    time.sleep(3)
    sys.exit()


def menu():
    print('''
    1. Login
    2. Register new user
    3. Check number of people registered in the database
    4. Quit

    Select an option.
            ''')
    options = {'1': login, '2': registerAccount, '3': checkRegistered, '4': quit}
    options[input()]()


while True:
    menu()