mirror of
https://github.com/eitchtee/WYGIWYH.git
synced 2026-03-21 17:09:24 +01:00
53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
import calendar
|
|
from datetime import date
|
|
|
|
from apps.transactions.models import Transaction
|
|
|
|
|
|
def get_transactions_by_day(year, month):
|
|
# Configure calendar to start on Monday
|
|
calendar.setfirstweekday(calendar.MONDAY)
|
|
|
|
# Get the first and last day of the month
|
|
first_day = date(year, month, 1)
|
|
last_day = date(year, month, calendar.monthrange(year, month)[1])
|
|
|
|
# Get all transactions for the month
|
|
transactions = Transaction.objects.filter(
|
|
date__year=year, date__month=month
|
|
).order_by(
|
|
"date",
|
|
"-type",
|
|
"-is_paid",
|
|
"id",
|
|
)
|
|
|
|
# Calculate padding days needed at start
|
|
start_padding = first_day.weekday() # Monday is 0, Sunday is 6
|
|
|
|
# Calculate padding days needed at end
|
|
end_padding = (7 - last_day.weekday() - 1) % 7
|
|
|
|
# Create padding days as empty dicts
|
|
start_padding_dates = [{}] * start_padding
|
|
end_padding_dates = [{}] * end_padding
|
|
|
|
# Create current month days
|
|
current_month_dates = [
|
|
{"day": day, "date": date(year, month, day), "transactions": []}
|
|
for day in range(1, calendar.monthrange(year, month)[1] + 1)
|
|
]
|
|
|
|
# Group transactions by day
|
|
for transaction in transactions:
|
|
day = transaction.date.day
|
|
for day_data in current_month_dates:
|
|
if day_data["day"] == day:
|
|
day_data["transactions"].append(transaction)
|
|
break
|
|
|
|
# Combine all dates
|
|
result = start_padding_dates + current_month_dates + end_padding_dates
|
|
|
|
return result
|