Python : その日のタスク全てが完了したかどうか判別し、その連続した日数をカウントする方法

Question

teratailの皆様、いつもお世話になっております ????????

Djangoでタスク管理ウェブサービスを作っているのですが、
○日連続達成のような仕組みをどうやって作ればよいのか悩んでいます。
やりたいことの詳細は以下のようなものです。

やりたいこと

DBにタスクを作成した日（created_at）と、完了した日(done_at）を保存しています。
これらの日付から、タスクを完了した日を判別し、それを使って連続タスク達成記録を作りたいと考えています。
タスクは一日に何度も作れて、その日のタスクを全て完了したら、その日はタスク完了したものとします。

タスク完了をあとから記録することもあるため、タスクの完了日は作成日と同じである必要はありません。
おそらく期間も絞る必要があると思うのですが、ここではDBに入っている全期間を対象にするものとします。

使用するデータ

class Task(models.Model):

    title = models.CharField(max_length=128)
    done_at = models.DateField(null=True, blank=True)
    created_at = models.DateField(default=timezone.now)
    created_by = models.ForeignKey(User, on_delete=models.CASCADE)

中には以下のようなデータが入っているとします。
日付にはPythonのdate型が入っているとお考えください。

title	done_at	created_at	created_by
タスク1	2021-08-10	2021-08-10	user1
タスク2	None	2021-08-10	user1
タスク3	2021-08-11	2021-08-11	user1
タスク4	2021-08-13	2021-08-12	user1
タスク5	None	2021-08-12	user1

上記の例だと、 8/10と8/12のタスクは２つありますが、そのうち一つだけが完了しています。
そのため、8/11のみタスク完了日となります。連続達成は発生しません。

連続達成が発生するデータ

title	done_at	created_at	created_by
タスク1	2021-08-10	2021-08-10	user1
タスク2	2021-08-11	2021-08-10	user1
タスク3	2021-08-11	2021-08-11	user1
タスク4	2021-08-13	2021-08-12	user1
タスク5	2021-08-14	2021-08-12	user1

この場合は、8/10 - 8/11 - 8-/12 の３日間連続ですべてのタスクを完了しています。
そのため、タスク連続達成記録３日とカウントされなければなりません。

説明は以上となります。
そもそものDBの構造がおかしい、などのご指摘も頂けると助かります。
どうぞよろしくお願い致します????????

Answer 1

Python側に寄せた実装ですが、set クラスを使えばどうにかなるでしょうか。
（テストしていません）

done_dates = set(Task.objects.filter(done_at__isnull=False).values_list('created_at', flat=True))
not_done_dates = set(Task.objects.filter(done_at__isnull=True).values_list('created_at', flat=True))
# 差集合ですべて終わっている日付setのみにする
all_done_dates = done_dates - not_done_dates
# list化してsort
sorted_dates = sorted(list(all_done_dates))

result = []
cur_dic = None
for i, cur_date in enumerate(sorted_dates):
  if i == 0:
    cur_dic = dict(
      start_date=cur_date,
      end_date=cur_date,
      continuous_days=1
    )
    continue
  if cur_date - datetime.timedelta(days=1) == sorted_dates[i - 1]:
    # 連続
    cur_dic['end_date'] = cur_date
    cur_dic['continuous_days'] += 1
  else:
    # 途切れた
    result.append(cur_dict)
    cur_dic = dict(
      start_date=cur_date,
      end_date=cur_date,
      continuous_days=1
    )
# 最後のcur_dicを追加
result.append(cur_dic)

print(result)

Comments

[greenisland]

ありがとうございます???????? す、凄いっ！！！！

連続した完了日がきちんと取得できました！！
連続した完了日を何個も作っても、別々にきちんとカウントされます！
これで最長記録や現在の記録など分けることもできそうです。

Djangoのvalues_listという取得方法も知りませんでした、勉強になります。