convert_ymls.py

import argparse, yaml, os, os.path

parser = argparse.ArgumentParser()
parser.add_argument("-i", "--input")
parser.add_argument("-o", "--output")
parser.add_argument("--tm2", "--tm", action="store_const", const="tm2", dest="action")
parser.add_argument("--tm2source", "--tmsource", action="store_const", const="tm2source", dest="action")

parser.add_argument("--zoom", default="14", help="Last zoom in the vector tile, default 14")

parser.add_argument("--source", action="store_true", dest="source")
parser.add_argument("--no-source", action="store_false", dest="source")

parser.add_argument("--only-shapefiles", action="store_true")
parser.add_argument("--only-postgis", action="store_true")

args = parser.parse_args()

cwd = os.getcwd()

with open(args.input) as fp:
    projectfile = yaml.load(fp)

if args.action == 'tm2':
    new_layers = []
    for layer in projectfile['Layer']:
        new_layers.append({'id': layer['id']})
    projectfile['Layer'] = new_layers

    if args.source:
        projectfile['source'] = "tmsource://{}/osm-carto.tm2source/".format(cwd)

elif args.action == 'tm2source':
    del projectfile['source']
    del projectfile['Stylesheet']

    zoom = int(args.zoom)
    projectfile['maxzoom'] = zoom
    for layer in projectfile['Layer']:
        # If the maxzoom is less than the minzoom, don't do anything
        # This can happen for a generic land polygon shapefile, which has
        # maxzoom: 9. If you include a minzoom: 14 in that layer, then that
        # layer won't show up from 0-9, i.e. it won't show up at all.
        if layer['properties'].get('minzoom', 22) > zoom and layer['properties'].get('maxzoom', 22) >= zoom:
            layer['properties']['minzoom'] = zoom

    if args.only_shapefiles:
        projectfile['Layer'] = [l for l in projectfile['Layer'] if l['Datasource']['type'] == 'shape']
    elif args.only_postgis:
        projectfile['Layer'] = [l for l in projectfile['Layer'] if l['Datasource']['type'] == 'postgis']

else:
    raise NotImplementedError()

with open(args.output, 'w') as fp:
    yaml.safe_dump(projectfile, fp)