diff --git a/mywebdav/billing/usage_tracker.py b/mywebdav/billing/usage_tracker.py index 22eb2ec..9433a1d 100644 --- a/mywebdav/billing/usage_tracker.py +++ b/mywebdav/billing/usage_tracker.py @@ -1,6 +1,8 @@ import uuid -from datetime import datetime, date, timezone +from datetime import datetime, date, timezone, timedelta + from tortoise.transactions import in_transaction + from .models import UsageRecord, UsageAggregate from ..models import User @@ -54,7 +56,7 @@ class UsageTracker: target_date = date.today() start_of_day = datetime.combine(target_date, datetime.min.time()) - end_of_day = datetime.combine(target_date, datetime.max.time()) + end_of_day = datetime.combine(target_date + timedelta(days=1), datetime.min.time()) - timedelta(microseconds=1) storage_records = await UsageRecord.filter( user=user,