auth.py

import json
from secrets import token_urlsafe
from pathlib import Path
import os
import getpass
import sys

if sys.platform == "darwin":
    if not os.path.exists("/Library/Caches/.menousdb"):
        os.mkdir("/Library/Caches/.menousdb")
    if not os.path.exists("/Library/Caches/.menousdb/authdata"):
        os.mkdir("/Library/Caches/.menousdb/authdata")
    path = "/Library/Caches/.menousdb/authdata"

elif sys.platform == "win32":
    if not os.path.exists(os.getenv("APPDATA")+"\\MenoudDb"):
        os.mkdir(os.getenv("APPDATA")+"\\MenoudDb")
    if not os.path.exists(os.getenv("APPDATA")+"\\MenoudDb"+"\\authdata"):
        os.mkdir(os.getenv("APPDATA")+"\\MenoudDb\\authdata")
    path = os.getenv("APPDATA")+"\\MenoudDb"+"\\authdata"


def check_key(key):
    with open(path+'/keys.json') as file:
        data = json.load(file)

    if key in data:
        return True
    else:
        return False

def generate_key():
    with open(path+'/keys.json') as file:
        data = json.load(file)

    key = token_urlsafe(16)

    with open(path+'/keys.json', 'w') as file:
        data.append(key)
        json.dump(data, file, indent=4)

    return key

def login(username, password):
    if not os.path.exists(path + "/login.json"):
        with open(path + "/login.json", 'w') as file:
            file.dump({})
        return 'Sign Up Required! Please contact Administrator'
    with open(path + "/login.json", 'r') as file:
        data = json.load(file)
        for i in data:
            if i == username and data[i][0] == password:
                return True
        return False
    
def getUserKey(username):
    with open(path + "/login.json", 'r') as file:
        data = json.load(file)
        for i in data:
            if i == username:
                return data[i][0]

def signup():
    with open(path + "/login.json", "r") as file:
        data = json.load(file)
    
    while True:
        print("~~~~~~~~~~~~~~~~~~~~~~WELCOME TO MENOUSDB~~~~~~~~~~~~~~~~~~~~~~~~~~~")
        uname = input("ENTER A USERNAME: ")
        pw = getpass.getpass("ENTER PASSWORD: ")
        confirm = getpass.getpass("RE-ENTER PASSWORD: ")
        if pw == confirm:
            data[uname] = [pw, generate_key()]
            with open(path + "/login.json", "w") as file:
                json.dump(data, file, indent=4)
            exit()
        else:
            print("ERROR! PASSWORDS DO NOT MATCH! TRY AGAIN!\n")

def checksignup():
    with open(path + "/login.json", "r") as file:
        data = json.load(file)

    if data == {}:
        signup()

checksignup()