generator.py 文件源码

def generate(filename, data_file, output_dir, template_file, skip_first_row):
    if not os.path.exists(filename):
        raise click.UsageError("Layout not found: %s\n" % filename)
    try:
        module = _import_file(filename)
    except (ImportError, ValueError) as e:
        raise click.UsageError("Unable to load %r: %s\n" % (filename, e))
    layout_classes = list(_iter_layout_classes(module))
    if not layout_classes:
        raise click.UsageError("No layout found in file: %s\n" % filename)
    layout_cls = layout_classes.pop()
    click.secho('Generating documents...', fg='white')
    layout = layout_cls(data_file, output_dir, template_file, skip_first_row)

    # Check files/paths
    layout._check_paths()

    data = csv.reader(open(layout.data_file))
    if layout.skip_first_row:
        next(data)

    # TODO: Show info about rows count
    for row in data:
        layout.generate_document(row)