如何通过给出具体的日期来获取周末的日期？

import datetime

def get_weekend_date(date):
    # 将日期转换为日期对象
    date_obj = datetime.datetime.strptime(date, '%Y-%m-%d').date()
    
    # 获取给定日期是星期几（0-6，0表示星期一）
    weekday = date_obj.weekday()
    
    # 计算距离周末的天数
    days_to_weekend = (5 - weekday) % 7
    
    # 计算周末的日期
    weekend_date = date_obj + datetime.timedelta(days=days_to_weekend)
    
    return weekend_date.strftime('%Y-%m-%d')

# 示例用法
date = '2022-01-01'
weekend_date = get_weekend_date(date)
print(weekend_date)

import java.time.DayOfWeek;
import java.time.LocalDate;
import java.time.format.DateTimeFormatter;

public class WeekendDateCalculator {
    public static String getWeekendDate(String date) {
        // 将日期转换为LocalDate对象
        LocalDate localDate = LocalDate.parse(date, DateTimeFormatter.ofPattern("yyyy-MM-dd"));
        
        // 获取给定日期是星期几（1-7，1表示星期一）
        DayOfWeek dayOfWeek = localDate.getDayOfWeek();
        int weekday = dayOfWeek.getValue();
        
        // 计算距离周末的天数
        int daysToWeekend = (6 - weekday) % 7;
        
        // 计算周末的日期
        LocalDate weekendDate = localDate.plusDays(daysToWeekend);
        
        return weekendDate.format(DateTimeFormatter.ofPattern("yyyy-MM-dd"));
    }
    
    // 示例用法
    public static void main(String[] args) {
        String date = "2022-01-01";
        String weekendDate = getWeekendDate(date);
        System.out.println(weekendDate);
    }
}