Python计算n个工作日后的日期

循环和递归

 

import datetime as dt
from datetime import datetime
def main(day,date):
    if day==0:
        return date

    a = datetime.strptime(date, '%Y-%m-%d').date()
    a = (a + dt.timedelta(days=1)).strftime("%Y-%m-%d")
    list1 = a.split('-')
    list1 = list(int(x) for x in list1)
    tup = tuple(list1)
    b = dt.datetime(tup[0], tup[1], tup[2]).weekday()
    if b == 5 or b == 6:
        day = day + 1
    return (main(day-1,a))
a=int(input())

print(main(a,'2023-03-17'))

import datetime as dt
from datetime import datetime
#x = 0
def main(day,date):
    j=day
    i=0
    while i < j:
        #a=dt.date.today()
        a = datetime.strptime(date, '%Y-%m-%d').date()
        a =(a+dt.timedelta(days=i+1)).strftime("%Y-%m-%d")
        list1 = a.split('-')
        list1=list(int(x) for x in list1)
        tup=tuple(list1)
        b=dt.datetime(tup[0],tup[1],tup[2]).weekday()
        if b==5 or b==6 :
            j=j+1
        i=i+1
    res = (dt.datetime.now() + dt.timedelta(days=i)).strftime("%Y-%m-%d")
    return res
print(main(9,'2023-03-17'))