dataset/seed/postfix-relay/funcs/opendkim.py
2022-10-17 18:28:22 +02:00

31 lines
1 KiB
Python

import dkim.dknewkey as _dknewkey
from os.path import join as _join, isfile as _isfile, isdir as _isdir
from os import makedirs as _makedirs, environ as _environ
from shutil import rmtree as _rmtree
import __main__
def _eprint(*args, **kwargs):
pass
_dknewkey.eprint = _eprint
_HERE = _environ['PWD']
_DKIM_DIR = _join(_HERE, 'pki/dkim')
def get_dkim_key(domain_name_eth0, domain):
dkim_dir = _join(_DKIM_DIR, domain_name_eth0, domain)
# dkim_file_src = _join(dkim_dir, f'{domain}')
dkim_file_key = _join(dkim_dir, f'{domain}.key')
dkim_file = _join(dkim_dir, f'{domain}.dns')
if not _isfile(dkim_file):
if _isdir(dkim_dir):
_rmtree(dkim_dir)
_makedirs(dkim_dir)
priv_key = _dknewkey.GenRSAKeys(dkim_file_key)
_dknewkey.ExtractRSADnsPublicKey(dkim_file_key, dkim_file)
# priv_key = _dknewkey.GenEd25519Keys(dkim_file_key)
# _dknewkey.ExtractEd25519PublicKey(dkim_file_src, priv_key)
with open(dkim_file_key, 'r') as fh:
return fh.read().strip()