odoo 11 / Python 3: как определить дату окончания подписки, если исключены некоторые будние дни

Question

Я работаю в системе управления подписками.У меня проблема с расчетом срока действия подписки.Сценарий таков: если человек подписывается на 1 месяц, скажем, с 11 января по 30 ноября 2017 года [ДД / ММ / ГГГГ] всего 30 дней, но он хочет исключить пятницу и субботу из каждой недели по 30 дней.Итак, как мне рассчитать дату истечения срока действия?

Логика такова: скажи дата окончания = дата истечения, а затем найдите пт / сб с 11.01.2008 по 30.11.2017, который выходит5 пт и 4 сб = 9 дней.Добавить до истечения срока, который будет 09/12/2018.Теперь ищите Fur & Sat между датой окончания и датой истечения срока, которая выходит 1 Fri & 2 Sat = 3 days.Дата окончания срока действия = Дата окончания и Дата окончания срока действия + 3 дня = 12/12/2018.Поиск между датой окончания и датой истечения для пятницы и субботы, который равен 0, поэтому дата истечения срока действия равна 12/12/2018; возвращаемое значение << </p>

Следующий код делает это, но метод возвращает 09/12/ 2018 вместо 12/12/2018.Что не так в этом ??

@api.one
def get_expiry_date(self, start, end, day_selection):
    print("i am in 2nd", start, end, day_selection)
    dayformat = '%Y-%m-%d'
    current_date = datetime.now().date()
    if start:
        start = datetime.strptime(str(start), dayformat).date()
    if end:    
        end = datetime.strptime(str(end), dayformat).date()        
    if day_selection:
        selected_days = day_selection        
    if start < current_date:
        start = datetime.strptime(str(start), dayformat).date()
    weekdays = self.weekday_count(start,end)
    print("days for start and end date",start,end, day_selection) 
    adddays = 0
    if weekdays:
        for i in range(len(day_selection)):                          
            for item in weekdays[0]:
                weekdays_dict = item
                print("dict", type(weekdays), type(weekdays[0]), weekdays_dict)
                print("compare", selected_days[i], weekdays_dict, selected_days[i] == weekdays_dict)
                if selected_days[i] == item:
                    adddays = adddays + weekdays[0].get(item)
        new_start = end
        end = datetime.strptime(str(end), dayformat).date() + timedelta(days=adddays)
        start = new_start
        print("New Expiry Date", start, end, adddays)
        if adddays > 0:               
            self.get_expiry_date(start, end, day_selection)
            print("type of end is ", type(end))
            print("selected days are", selected_days[i],weekdays[0], weekdays[0].get(item), adddays)  
            print("last returned values is",end)

return end

Answer 1

Логика, которую вы пытаетесь сделать, сложна

Насколько я понимаю, вы используете систему подписки, основанную на количестве дней, а не дат, поэтому для вычисления даты окончания вам просто нужно это:

1. дата начала (например, 01/11/2018)
2. количество дней (например, 30 дней)
3. исключенные дни (например, пятница, суббота)

# day are like this
#     Fri
#     Sat
#     Sun
#     Mon
#     Tue
#     Wed
#     Thu

# start day
def get_expiry_date(start_date, number_of_days, excluded_days = None):
    """ compute end date of subcription period """
    if excluded_days is None:
        excluded_days = []
    # check params
    start_date = str(start_date)
    if number_of_days < 1:
        raise exception.UserError(_('Number of days should be > 0!!'))  # import the translate "_"  method 
    if len(excluded_days) > 5:
        raise exception.UserError(_('To much excluded days!!'))  # import the translate "_"  method 

    date_format = '%Y-%m-%d'
    end_date = datetime.strptime(start_date, date_format)

    # compute end date
    # keeping adding one day until you finish your days
    add_one_day = timedelta(days=1)
    while number_of_days > 1:
        end_date += add_one_day
        if end_date.strftime('%a') not in excluded_days:
            # day is not excluded compute it
            number_of_days += -1

    return end_date.strftime(date_format)

И это тестовое значение, которое вы можете проверить:

        print get_expiry_date('2018-11-01', 30, ['Fri', 'Sat'])  # 2018-12-12
        print get_expiry_date('2018-11-01', 30, ['Fri']) # 2018-12-05
        print get_expiry_date('2018-11-01', 30, []) # 2018-11-30
        print get_expiry_date('2018-11-01', 90, []) # 2019-01-29
        print get_expiry_date('2018-11-01', 30, ['Mon', 'Thu']) # 2018-12-11

Если у вас есть все готовые работающие системы, вы можете использовать это:

def get_expiry_date(start_date, end_date, excluded_days = None):
    if excluded_days is None:
        excluded_days = []
    start_date = str(start_date)
    end_date = str(end_date)
    date_format = '%Y-%m-%d'
    # compute number of days
    number_of_days = (datetime.strptime(end_date, date_format) - datetime.strptime(start_date, date_format)).days
    expiry_date = datetime.strptime(start_date, date_format)

    if len(excluded_days) > 5:
        raise Exception('To much excluded days!!')

    # keeping adding one day until you finish your days
    one_day = timedelta(days=1)
    while number_of_days > 0:
        expiry_date += one_day
        # if day is not excluded reduce the number of left days
        if expiry_date.strftime('%a') not in excluded_days:
            number_of_days += -1

    return expiry_date.strftime(date_format)

print get_expiry_date('2018-11-01', '2018-11-30', ['Fri', 'Sat'])  # 2018-12-12

Answer 2

В вашем вопросе нет определения ни day_selection, ни weekday_count, поэтому трудно понять, что происходит.Вероятно, проблема в рекурсии, которая не нужна.

Если day_selection определено как дни, выбранные для исключения (список strftime('%a'), например, ['Mon', 'Tue']), то:

from datetime import datetime, timedelta

def get_expiry_date(start, end, day_selection):
    dayformat = '%Y-%m-%d'
    weekdays= {}
    start = datetime.strptime(str(start), dayformat).date()
    end = datetime.strptime(str(end), dayformat).date()
    # Count weekdays
    for i in range((end - start).days+1):
        weekday = (start + timedelta(i)).strftime('%a')
        weekdays[weekday] = 1 + weekdays.setdefault(weekday, 0)
    # Count subscription days to add
    sub_days_to_add = 0
    for selected in day_selection:
        sub_days_to_add += weekdays.setdefault(selected, 0)
    # Count calender days to add
    cal_days_extension = 0
    while sub_days_to_add > 0:
        if (end + timedelta(days=cal_days_extension + 1)).strftime('%a') not in day_selection:
            sub_days_to_add -= 1
        cal_days_extension += 1
    # Add to end day
    return end + timedelta(days=cal_days_extension)

Тесты:

print (get_expiry_date('2018-11-01', '2018-11-30', ['Fri', 'Sat']))
# --->  2018-12-12
print (get_expiry_date('2018-11-01', '2018-11-30', []))
# --->  2018-11-30

Также было бы безопаснее использовать from odoo.tools import DEFAULT_SERVER_DATE_FORMAT, чем dayformat = '%Y-%m-%d'.А для таких параметров, как start & end, если они являются обязательными, можно сделать эти поля обязательными в Odoo.