#!/usr/bin/python

import sqlite3
import getpass

db = 'users.db'
loggedin = False
connection = sqlite3.connect('users.db')

def menu():
	try:
		option = int(input("1.- Register new user\n2.- Log in\nOption:"))
	except Exception as e:
		print("Please enter a number, little malicious user...")
		menu()
	else:
		if(option == 1):
			register()
		else:
			logIn()
		return option;


def register():
	username = input("Username: ")
	password = getpass.getpass()
	try:
		connection.execute("INSERT INTO users(username, password) VALUES ('" + username + "','" + password + "');")
		connection.commit()
	except Exception as e:
		print("User already exists....")
	else:
		print("User registered successfully...")
	finally:
		menu()

def logIn():
	username = input("Username: ")
	password = getpass.getpass()
	cursor = connection.execute("SELECT username, password FROM users WHERE username = '" + username + "';")
	u_p = cursor.fetchone()
	if(str(u_p) == 'None'):
		print("Wrong password or username...")
	elif(u_p[0] == username and u_p[1] == password):
		print("Logged in successfuly...")
		loggedin = True

menu()

connection.close()