app-MAIL-temp/app/pgp_utils.py

27 lines
641 B
Python
Raw Normal View History

2020-03-08 12:51:33 +01:00
import gnupg
from app.config import GNUPGHOME
gpg = gnupg.GPG(gnupghome=GNUPGHOME)
2020-03-08 12:51:33 +01:00
class PGPException(Exception):
pass
2020-03-08 12:51:33 +01:00
def load_public_key(public_key: str) -> str:
"""Load a public key into keyring and return the fingerprint. If error, raise Exception"""
import_result = gpg.import_keys(public_key)
try:
return import_result.fingerprints[0]
except Exception as e:
raise PGPException("Cannot load key") from e
def encrypt(data: str, fingerprint: str) -> str:
r = gpg.encrypt(data, fingerprint, always_trust=True)
if not r.ok:
raise PGPException("Cannot encrypt")
return str(r)