applytmpl.py

#!/usr/bin/env python3

import sys
import glob
import json
import yaml
import socket

from jinja2 import Environment, FileSystemLoader
env = Environment(loader=FileSystemLoader('./', encoding='utf8'))

if sys.argv[1] == '--json':
    genJSON = True
    tplfile = sys.argv[2]
else:
    genJSON = False
    tplfile = sys.argv[1]


def getip(host):
    return socket.gethostbyname(host)


def findsk(org):
    dir = 'crypto-config/peerOrganizations/' + \
        org['domain'] + '/users/Admin@' + org['domain'] + '/msp/keystore'
    return glob.glob(dir + '/*_sk')[0]


def readpem(path):
    with open(path) as f:
        pem = '\"'
        for line in f.read().splitlines():
            pem += line + '\\n'
        pem += '\"'
        return pem


env.filters['getip'] = getip
env.filters['findsk'] = findsk
env.filters['readpem'] = readpem

tpl = env.get_template(tplfile)

commonDomain = 'example.com'
commonHostname = 'localhost'

data = {
    'network': 'ccperf',
    'embedKeys': True,
    'client': {
        'org': 'org1'
    },
    'ordererorg': {
        'name': 'orderer',
        'mspid': 'OrdererOrg',
        'domain': commonDomain,
        # 'type': 'etcdraft',
        # 'type': 'kafka',
        'type': 'solo',
        'orderers': [
            {'name': 'orderer1', 'host': commonHostname, 'ports': {
                'requests': 7050, 'pprof': 7060}, 'volume': 'tmpfs'},
        ]
    },
    'orgs': [
        {
            'name': 'org1',
            'mspid': 'PeerOrg1',
            'domain': 'org1.' + commonDomain,
            'useLeaderElection': True,
            'peers': [
                {'name': 'peer1', 'ports': {'requests': 7051, 'pprof': 7061},
                    'host': commonHostname, 'bootstrap': 'peer1', 'volume': 'tmpfs'},
            ]
        }
    ],
    'cadvisors': [
        {'host': 'cadvisor', 'port': 8080},
    ],
    'node_exporters': [
        {'host': 'node_exporter', 'port': 9100},
    ],
    'grafana': {
        'port': 3000
    },
    'prometheus': {
        'port': 9090,
        # 'volume': '/var/tmp/prometheus',
    },
    'pushgateway': {
        'port': 9091
    },
    'proxy': {
        'port': 8888
    }
}

if genJSON:
    data['embedKeys'] = False

out = tpl.render(data)

if genJSON:
    print(json.dumps(yaml.load(out), indent=4))
else:
    print(out)