Function main

scripts/init_profiles_from_roles.py:155–207

()

def main():
    parser = argparse.ArgumentParser(description="Initialize Profiles from roles.json")
    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(
        "--overwrite-profiles",
        action="store_true",
        help="Back up and recreate profiles from roles.json. This deletes only profile rows and keeps papers intact.",
    )
    args = parser.parse_args()

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

    # 初始化数据库
    resolved_db_path = configure_db_path(args.db_path)
    print(f"\nInitializing database: {resolved_db_path}...")
    init_db()

    if args.overwrite_profiles:
        backup_path = backup_profiles(resolved_db_path)
        removed_count = clear_profiles_only(resolved_db_path)
        print("\nOverwriting profiles only.")
        print(f"  Backup: {backup_path}")
        print(f"  Removed profile rows: {removed_count}")
        print("  Papers table was not modified.")

    # 创建 profile
    print("\nCreating profiles...")
    created_count = 0
    existing_count = 0

    for role_name, role_data in sorted(roles.items()):
        profile = build_profile_from_role(role_name, role_data)
        user_id = profile["user_id"]

        profile_id = create_profile(user_id, profile)
        if profile_id:
            created_count += 1
            print(f"  + {user_id} ({role_name}): {len(profile['core_directions'])} directions")
        else:
            existing_count += 1
            print(f"  ~ {user_id} ({role_name}): already exists")

    print()
    print("=" * 60)
    print("Profiles Initialized")
    print("=" * 60)
    print(f"Created: {created_count}")
    print(f"Already exists: {existing_count}")
    print(f"Total: {created_count + existing_count}")


if __name__ == "__main__":