borgmatic/tests/end-to-end/commands/fake_lsblk.py

import argparse
import json
import sys


def parse_arguments(*unparsed_arguments):
    parser = argparse.ArgumentParser(add_help=False)

    parser.add_argument('--output', required=True)
    parser.add_argument('--json', action='store_true', required=True)
    parser.add_argument('--list', action='store_true', required=True)

    return parser.parse_args(unparsed_arguments)


BUILTIN_BLOCK_DEVICES = {
   'blockdevices': [
      {
         'name': 'loop0',
         'path': '/dev/loop0',
         'mountpoint': None,
         'type': 'loop'
      },
      {
         'name': 'cryptroot',
         'path': '/dev/mapper/cryptroot',
         'mountpoint': '/',
         'type': 'crypt'
      },{
         'name': 'vgroup-lvolume',
         'path': '/dev/mapper/vgroup-lvolume',
         'mountpoint': '/mnt/lvolume',
         'type': 'lvm'
      },
      {
         'name': 'vgroup-lvolume-real',
         'path': '/dev/mapper/vgroup-lvolume-real',
         'mountpoint': None,
         'type': 'lvm'
      },
   ]
}


def load_snapshots():
    try:
        return json.load(open('/tmp/fake_lvm.json'))
    except FileNotFoundError:
        return []


def print_logical_volumes_json(arguments, snapshots):
    data = dict(BUILTIN_BLOCK_DEVICES)

    for snapshot in snapshots:
        data['blockdevices'].extend(
            {
               'name': snapshot['lv_name'],
               'path': snapshot['lv_path'],
               'mountpoint': None,
               'type': 'lvm'
            }
            for snapshot in snapshots
        )

    print(json.dumps(data))


def main():
    arguments = parse_arguments(*sys.argv[1:])
    snapshots = load_snapshots()

    assert arguments.output == 'name,path,mountpoint,type'

    print_logical_volumes_json(arguments, snapshots)


if __name__ == '__main__':
    main()