1

我一直在研究这个函数几个小时，我似乎无法弄清楚为什么它没有给我我期望的结果。我有一个生成日期时间开始和日期时间结束的测试。我想计算在该时间范围之间发生了多少个“星期一”或任何给定的一天。

def day_count(week_day, start, end):
    if week_day == 1:
        day = 6
    else:
        day = week_day - 2
    num_weekdays, remainder = divmod((end - start).days, 7)
    if (day - start.weekday()) % 7 <= remainder:
        return num_weekdays + 1
    else:
        return num_weekdays

复制，这是我给它的数据：

end: 2020-08-22 00:00:00+02:00
start: 2020-08-20 22:15:55.371646+00:00
weekday: 6

我期待回馈num_weekdays = 1

但我明白了num_weekdays = 0

我不知道如何解决这个问题，num_weekdays并且remainder 在调试时也是 0。

这就是我调用函数的方式：

 total_day_count = day_count(params['dt_start__week_day'], params['dt_start__gte'], params['dt_end__lte'])

这就是我获取参数的方式：

  params = {}
    if self.request.GET.get('day') == 'Today':
        params['dt_start__week_day'] = (timezone.now().weekday() + 1) % 7 + 1
    elif is_valid_queryparam(self.request.GET.get('day')):
        day_of_the_week = self.request.GET.get('day')
        params['dt_start__week_day'] = (int(day_of_the_week) + 1) % 7 + 1
    else:
        params['dt_start__week_day'] = (timezone.now().weekday() + 1) % 7 + 1

    if is_valid_queryparam(self.request.GET.get('date_start')):
        unaware = datetime.strptime(self.request.GET.get('date_start'), '%Y-%m-%d')
        params['dt_start__gte'] = unaware.replace(tzinfo=local_tz)
    else:
        dt = datetime.now().replace(hour=0, minute=0, second=0, microsecond=0)
        first_job = Job.objects.earliest('dt_start')
        params['dt_start__gte'] = first_job.dt_start
    if is_valid_queryparam(self.request.GET.get('date_end')):
        unaware = datetime.strptime(self.request.GET.get('date_end'), '%Y-%m-%d')
        params['dt_end__lte'] = unaware.replace(tzinfo=local_tz) + timedelta(days=1)
    else:
        dt = datetime.now().replace(hour=0, minute=0, second=0, microsecond=0)
        dt_aware = local_tz.localize(dt)
        params['dt_end__lte'] = dt_aware + timedelta(days=1)

我不知道这是否有帮助，但这是测试：

class MachineDurationViewTestCase(CustomAPITestCase):
"""This testcase contains tests for the API endpoint MachineDurationView """

# todo once views and some of its variables were renamed, also apply these name changes here
fake = Faker()

def setUp(self) -> None:
    self.fake.seed()
    super(MachineDurationViewTestCase, self).setUp()

def test_a_single_job_that_occurs_in_the_space_of_one_hour(self):
    """
    In this test a single job of 30 minutes is created occuring on the first hour on the current date. This means
    that we should expect the endpoint to return an array of 24 value where the first item in the array has a value
    of 0.5 (Because the first hour only occurs 1 time, and job lasts 30 minutes, machine occupancy is 50%)
    """

    dt = timezone.localtime()
    dt_start = dt.replace(hour=0, minute=15)
    dt_end = dt.replace(hour=0, minute=45)
    # dt_end = timezone.datetime(d.year, d.month, d.day, hour=0, minute=45)
    # dt_start = timezone.make_aware(timezone.datetime.now().replace(hour=0, minute=0, second=0, microsecond=0))
    print(dt_start)
    print(dt_end)
    job = JobFactory(dt_start=dt_start, dt_end=dt_end)
    print(job.dt_start)

    # Query the dbase to ensure the job is created
    self.assertEqual(Job.objects.all().count(), 1)

    # Force create/login a user to access the http mocking client
    # (I don't care about authentication or permissions in this test)
    self.client.force_login(user=UserFactory(is_superuser=True))
    response = self.client.get(reverse('data_app:machine_avg'))
    self.assertEqual(response.status_code, 200)
    print(response.data['machines'])
    self.assertNotEqual(response.data['machines'][0], 0.0, "Since we create a job on the first hour, we expect the"
                                                        "first item of the array not to be zero.")
    self.assertEqual(response.data['machines'][0], 0.5, "Since we create a single job of 30 minutes that occurs on "
                                                        "the current date, the value on the given hour should be"
                                                        "0.5 (50% occupancy)")

    # job = JobFactory(dt_start=)
4

2 回答 2

0

我相信您的输入数据有问题。这似乎对我有用：

from datetime import date

start = date(2020, 8, 20)
end = date(2020, 8, 22)

day_count(6, start, end)

这将返回所需的输出 1。我的日期类型是：<class 'datetime.date'>

于 2020-08-21T08:50:13.220 回答
0

可以通过使用date 对象的weekday()方法来实现计算日期范围之间的具体工作日总数的函数：

from datetime import date, timedelta

def days_count(weekdays: list, start: date, end: date):
    dates_diff = end-start
    days = [start + timedelta(days=i) for i in range(dates_diff.days)]
    return len([day for day in days if day.weekday() in weekdays])


start = date(2020, 8, 20)
end = date(2020, 8, 22)
# total of weekend days between that range (end date is excluded)
print(days_count([5, 6], start, end)) # outputs: 0
于 2020-08-21T09:20:07.117 回答