Function main

scripts/init_user_metadata.py:80–124

()

def main():
    parser = argparse.ArgumentParser(description="Initialize User Metadata")
    parser.add_argument("--roles-path", type=str, default="data/roles.json", help="roles.json 路径")
    parser.add_argument("--db-path", type=str, default="data/paperflow.db", help="数据库路径")
    parser.add_argument("--output-dir", type=str, default="data/simulation_output", help="输出目录")
    args = parser.parse_args()

    # 加载数据
    print(f"Loading roles from {args.roles_path}...")
    roles = load_roles(args.roles_path)
    print(f"  Found {len(roles)} roles")

    print(f"Loading profiles from {args.db_path}...")
    profiles = load_profiles(args.db_path)
    print(f"  Found {len(profiles)} profiles")

    # 初始化用户元数据
    users = init_user_metadata(roles, profiles)

    # 输出
    output_dir = Path(args.output_dir)
    if not output_dir.is_absolute():
        output_dir = PROJECT_ROOT / output_dir

    output_dir.mkdir(parents=True, exist_ok=True)

    output_path = output_dir / "users.json"

    with output_path.open("w", encoding="utf-8") as f:
        json.dump({"users": users}, f, ensure_ascii=False, indent=2)

    print()
    print("=" * 60)
    print("User Metadata Initialized")
    print("=" * 60)
    print(f"Users: {len(users)}")
    print(f"Output: {output_path}")

    # 打印预览
    print()
    print("Preview:")
    for user in users[:5]:
        print(f"  - {user['user_id']} ({user['role_name']}): {len(user['seed_directions'])} directions")
    if len(users) > 5:
        print(f"  ... and {len(users) - 5} more")


if __name__ == "__main__":