import_export.py 文件源码

def import_(ctx, dirs):
    for dir in dirs:
        if not os.path.isdir(dir):
            raise click.UsageError('<%s> must be a directory' % dir)

    def do_import(report_name, instances):
        count = 0
        try:
            for instance in instances:
                import_report_instance(ctx, report_name, instance)
                count += 1
        finally:
            echo_info('Imported %d %s instances' % (count, report_name))

    def instances(dir):
        for filename in glob.glob(os.path.join(dir, '*.json')):
            with open(filename) as f:
                yield json.loads(f.read())

    for dir in dirs:
        report_name = dir.rstrip('/').split('/')[-1]
        do_import(report_name, instances(dir))