add script to make datamatrix key backup
This commit is contained in:
parent
db51f5af7d
commit
73723e5a9f
1 changed files with 63 additions and 0 deletions
63
bin/gpg-key-to-datamatrix
Executable file
63
bin/gpg-key-to-datamatrix
Executable file
|
@ -0,0 +1,63 @@
|
|||
#! /usr/bin/env nix-shell
|
||||
#! nix-shell -p typst dmtx-utils -i python
|
||||
|
||||
import subprocess
|
||||
import argparse
|
||||
import shlex
|
||||
import tempfile
|
||||
from pathlib import Path
|
||||
from math import ceil
|
||||
|
||||
|
||||
def sub(cmd: str, *args, **kwargs):
|
||||
return subprocess.check_output(shlex.split(cmd), *args, **kwargs)
|
||||
|
||||
|
||||
def typst_file_tmpl(files):
|
||||
local_paths = [f.relative_to(Path.cwd()) for f in files]
|
||||
return "\n".join(f'#image("{f}")' for f in local_paths)
|
||||
|
||||
|
||||
def chunk_key(key):
|
||||
"""
|
||||
split byte string into even chunks
|
||||
chunks will be <= 1555 bytes in length since
|
||||
1555 is the maxiumum byte size for datamatrix
|
||||
"""
|
||||
n = ceil(len(key) / ceil(len(key) / 1555))
|
||||
return [key[i : i + n] for i in range(0, len(key), n)]
|
||||
|
||||
|
||||
def generate_pdf(byte_chunks, svgpath, outpath):
|
||||
for f, data in zip(
|
||||
files := [svgpath / f"dmtx-{i}.svg" for i, _ in enumerate(byte_chunks)],
|
||||
byte_chunks,
|
||||
):
|
||||
sub(f"dmtxwrite -e 8 -o {f}", input=data)
|
||||
sub(f"typst compile - {outpath}", input=typst_file_tmpl(files), text=True)
|
||||
|
||||
|
||||
def get_key_data(key):
|
||||
suffix = key if key else ""
|
||||
return chunk_key(
|
||||
sub(f"gpg --export-secret-key --export-options export-minimal {suffix}")
|
||||
)
|
||||
|
||||
|
||||
def main():
|
||||
p = argparse.ArgumentParser()
|
||||
p.add_argument("--key", help="key id")
|
||||
p.add_argument(
|
||||
"--output", help="pdf to write matrices to", default=Path.cwd() / "key.pdf"
|
||||
)
|
||||
|
||||
args = p.parse_args()
|
||||
byte_chunks = get_key_data(args.key)
|
||||
|
||||
with tempfile.TemporaryDirectory(dir=Path.cwd()) as tmpdirname:
|
||||
tmpdir = Path(tmpdirname)
|
||||
generate_pdf(byte_chunks, tmpdir, args.output)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
Loading…
Reference in a new issue