imprint.py

Go to the documentation of this file.

import argparse
import pathlib
import sys
import yaml
 
from system_fingerprint import modules
 
 
def main(argv=None):
    parser = argparse.ArgumentParser()
    parser.add_argument('-p', '--print', action='store_true')
    parser.add_argument('output_path', type=pathlib.Path,
                        nargs='?', default='fingerprint.yaml')
    args = parser.parse_args(argv)
 
    D = {}
    for module in modules:
        try:
            D[module.__name__] = module()
        except Exception as e:
            print(f'{repr(e)} occurred when fingerprinting {module.__name__}. Skipping...', file=sys.stderr)
 
    contents = yaml.safe_dump(D)
 
    if args.print:
        print(contents)
 
    print(f'saving fingerprint to {args.output_path}')
    with open(args.output_path, 'w') as f:
        f.write(contents)