@@ -65,22 +65,37 @@ def toggle_debt_payment(debt_id):
65
65
raise ValueError ("Debt not found" )
66
66
67
67
68
- def get_unpaid_debts_by_week (user_id , piggy_id ):
68
+ def get_unpaid_debts_by_week (user_id , other_user_id ):
69
69
now = timezone .now ()
70
70
start_of_week = (now - timedelta (days = now .weekday ())).replace (hour = 0 , minute = 0 , second = 0 , microsecond = 0 )
71
71
start_of_last_week = start_of_week - timedelta (weeks = 1 )
72
72
73
73
present_week_debts = Debt .objects .filter (
74
- (Q (user__user_id = user_id ) | Q (user__user_id = piggy_id )), is_paid = False , created_at__gte = start_of_week
74
+ (
75
+ Q (user__user_id = user_id ) & Q (debtor__user_id = other_user_id )
76
+ | Q (user__user_id = other_user_id ) & Q (debtor__user_id = user_id )
77
+ ),
78
+ is_paid = False ,
79
+ created_at__gte = start_of_week ,
75
80
)
81
+
76
82
last_week_debts = Debt .objects .filter (
77
- (Q (user__user_id = user_id ) | Q (user__user_id = piggy_id )),
83
+ (
84
+ Q (user__user_id = user_id ) & Q (debtor__user_id = other_user_id )
85
+ | Q (user__user_id = other_user_id ) & Q (debtor__user_id = user_id )
86
+ ),
78
87
is_paid = False ,
79
88
created_at__gte = start_of_last_week ,
80
89
created_at__lt = start_of_week ,
81
90
)
91
+
82
92
previous_debts = Debt .objects .filter (
83
- (Q (user__user_id = user_id ) | Q (user__user_id = piggy_id )), is_paid = False , created_at__lt = start_of_last_week
93
+ (
94
+ Q (user__user_id = user_id ) & Q (debtor__user_id = other_user_id )
95
+ | Q (user__user_id = other_user_id ) & Q (debtor__user_id = user_id )
96
+ ),
97
+ is_paid = False ,
98
+ created_at__lt = start_of_last_week ,
84
99
)
85
100
86
101
return present_week_debts , last_week_debts , previous_debts
0 commit comments