fooocus / modules /auth.py
fazzebo's picture
all
5353f6a
import json
import hashlib
import modules.constants as constants
from os.path import exists
def auth_list_to_dict(auth_list):
auth_dict = {}
for auth_data in auth_list:
if 'user' in auth_data:
if 'hash' in auth_data:
auth_dict |= {auth_data['user']: auth_data['hash']}
elif 'pass' in auth_data:
auth_dict |= {auth_data['user']: hashlib.sha256(bytes(auth_data['pass'], encoding='utf-8')).hexdigest()}
return auth_dict
def load_auth_data(filename=None):
auth_dict = None
if filename != None and exists(filename):
with open(filename, encoding='utf-8') as auth_file:
try:
auth_obj = json.load(auth_file)
if isinstance(auth_obj, list) and len(auth_obj) > 0:
auth_dict = auth_list_to_dict(auth_obj)
except Exception as e:
print('load_auth_data, e: ' + str(e))
return auth_dict
auth_dict = load_auth_data(constants.AUTH_FILENAME)
auth_enabled = auth_dict != None
def check_auth(user, password):
if user not in auth_dict:
return False
else:
return hashlib.sha256(bytes(password, encoding='utf-8')).hexdigest() == auth_dict[user]