如何使用特定列表从开始日期生成结束日期

Question

shift_lst = ["D","D","D","D","N","N","W","N","N","D","D","D","D","W","N","N","N"]

current_date = date(2021, 4, 1)

for shift in (shift_lst):
    shift_dict=dict()
    if shift == 'D':
       shift_type = 'S1'
       while shift_type == 'S1':
       # start_date = current_date
       # enddate = start_date + timedelta(days=count)
       count = count + 1

    elif shift == 'N':
       shift='S2'
       start_date = current_date
       # enddate = start_date + timedelta(days=count)
       count = count + 1

Output:

{shift: S1,'start_date':2021,4,1,'end_date':2021,4,4},

{shift: S2,'start_date':2021,4,5,'end_date':2021,4,9},

{shift: S1,'start_date':2021,4,11,'end_date':2021,4,13}

I have list of Shift In that list i have D (Day shift) then date should start from 1st april to enddate until N not come then when N Start so start date also start from enddate of "D" and create dictionary of shift

Answer 1

Hope this would be what you are looking for. There is no need for while loop inside. You just increment your current_date when it is D or W .

current_date = date(2021, 4, 1)

print("start date: ", current_date)

for shift in (shift_lst):
    if shift == "D":
        current_date = current_date + dt.timedelta(days=1)
    elif shift == "W":
        current_date = current_date + dt.timedelta(weeks=1)
    elif shift == "N":
        current_date = current_date
    print(current_date)

end_date = current_date

print("end date: ", end_date)

output:

start date:  2021-04-01
2021-04-02
2021-04-03
2021-04-04
2021-04-05
2021-04-05
2021-04-05
2021-04-12
2021-04-12
2021-04-12
2021-04-13
2021-04-14
2021-04-15
2021-04-16
2021-04-23
2021-04-23
2021-04-23
2021-04-23
end date:  2021-04-23

Answer 2

simply what i want if shift == D then i want start date and end date of D, for example in first how many times D appear it 4 times so start date =2021-04-01 and end date = 2021-04-04 that's it same to N also from when N start and end i want just start date end date and create new dictionary for shift_type

current_date = date(2021, 4, 1)

print("start date: ", current_date)

for shift in (shift_lst):
    shift_dict=dict()
    if shift == "D":
        shift_type='S1'
        current_date = current_date + dt.timedelta(days=1)
    elif shift == "W":
        current_date = current_date + dt.timedelta(weeks=1)
    elif shift == "N":
        shift_type='S2'
        current_date = current_date
    print(current_date)

end_date = current_date
shift_dict['shift_type']=shift
shift_dict['start_date']=current_date
shift_dict['end_date']=enddate

Output:

{shift_type:S1,"start_date":"2021,4,1","end_date":"2021,4,4"} {shift_type:S2,"start_date":"2021,4,5","end_date":"2021,4,9"}