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()