Bugfix: encryption lockout

This commit is contained in:
Vivek Santayana 2021-12-07 15:15:16 +00:00
parent d76c8a5fed
commit deab85289b
2 changed files with 5 additions and 3 deletions

View File

@ -26,6 +26,8 @@ services:
- ./.env
ports:
- 5000
volumes:
- ./.security:/ref-test/.security
restart: unless-stopped
networks:
- frontend

View File

@ -2,17 +2,17 @@ from os import environ, path
from cryptography.fernet import Fernet
def generate_keyfile():
with open('./common/security/.encryption.key', 'wb') as keyfile:
with open('.security/.encryption.key', 'wb') as keyfile:
key = Fernet.generate_key()
keyfile.write(key)
def load_key():
with open('./common/security/.encryption.key', 'rb') as keyfile:
with open('.security/.encryption.key', 'rb') as keyfile:
key = keyfile.read()
return key
def check_keyfile_exists():
return path.isfile('./common/security/.encryption.key')
return path.isfile('.security/.encryption.key')
def encrypt(input):
if not check_keyfile_exists():