gen_module.py

gen_module.py

import os
MAX_DEPTH = 11

def gen_package(name, levels):
    subpkgs = [name+'a', name+'b']
    if levels == 1:
        subpkgs = []
    for subpkg in subpkgs:
        gen_package(subpkg, levels-1)

    imports = '\n'.join(f'\t"example.com/{x}"' for x in subpkgs)
    expr = '1'
    if subpkgs:
        expr = ' + '.join(f'{x}.V' for x in subpkgs)

    os.mkdir(name)
    with open(name + '/pkg.go', 'w') as f:
        f.write(f'''package {name}

import (
{imports}
)

const V = {expr}
''')
    return name

gen_package('pkg_', MAX_DEPTH)