How to Mock Daily and Monthly Quota Counters in Python

import random
from datetime import datetime, timedelta


class QuotaCounter:
    def __init__(self, daily_limit=1000, monthly_limit=20000):
        self.daily_limit = daily_limit
        self.monthly_limit = monthly_limit
        self.daily_usage = 0
        self.monthly_usage = 0
        self.current_day = datetime.now().date()

    def track_usage(self, amount=1):
        today = datetime.now().date()
        if today != self.current_day:
            self.daily_usage = 0
            self.current_day = today
        self.daily_usage += amount
        self.monthly_usage += amount

    def check_quota(self):
        return {
            "daily_used": self.daily_usage,
            "daily_limit": self.daily_limit,
            "daily_remaining": max(0, self.daily_limit - self.daily_usage),
            "monthly_used": self.monthly_usage,
            "monthly_limit": self.monthly_limit,
            "monthly_remaining": max(0, self.monthly_limit - self.monthly_usage),
        }

    def is_within_daily(self):
        return self.daily_usage <= self.daily_limit

    def is_within_monthly(self):
        return self.monthly_usage <= self.monthly_limit


if __name__ == "__main__":
    quota = QuotaCounter(daily_limit=10, monthly_limit=100)
    for _ in range(random.randint(5, 12)):
        quota.track_usage()
    print(quota.check_quota())
    print("Daily OK:", quota.is_within_daily())
    print("Monthly OK:", quota.is_within_monthly())