viveksantayana/ska-referee-test
viveksantayana/ska-referee-test
1
0
Fork 0
Code Issues 1 Pull Requests Projects Releases 19 Wiki Activity

Bugfix: encryption lockout

Browse Source
This commit is contained in:
Vivek Santayana 2021-12-07 15:15:16 +00:00
parent b87f99e138
commit 866267dc5f
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docker-compose.yml
View File

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

6
ref-test/common/security/__init__.py
View File

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