fastapi-dls/app/util.py

29 lines
840 B
Python
Raw Permalink Normal View History

2022-12-28 10:53:56 +00:00
def load_file(filename) -> bytes:
with open(filename, 'rb') as file:
content = file.read()
return content
2022-12-29 19:40:42 +00:00
def load_key(filename) -> "RsaKey":
try:
# Crypto | Cryptodome on Debian
from Crypto.PublicKey import RSA
from Crypto.PublicKey.RSA import RsaKey
except ModuleNotFoundError:
from Cryptodome.PublicKey import RSA
from Cryptodome.PublicKey.RSA import RsaKey
2022-12-28 10:53:56 +00:00
return RSA.import_key(extern_key=load_file(filename), passphrase=None)
2022-12-29 19:40:42 +00:00
def generate_key() -> "RsaKey":
try:
# Crypto | Cryptodome on Debian
from Crypto.PublicKey import RSA
from Crypto.PublicKey.RSA import RsaKey
except ModuleNotFoundError:
from Cryptodome.PublicKey import RSA
from Cryptodome.PublicKey.RSA import RsaKey
return RSA.generate(bits=2048)