Allow rrulesets for repeating transaction schedules

budgeapp · Dec 26, 2024 · bb6a3df · bb6a3df
1 parent 781d335
commit bb6a3df
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/budge/transaction.py b/src/budge/transaction.py
@@ -2,7 +2,7 @@
 from datetime import date as _date
 from typing import Self
 
-from dateutil.rrule import rrule
+from dateutil.rrule import rrule, rruleset
 from stockholm import Money
 
 from . import account
@@ -30,10 +30,10 @@ def __lt__(self, other: Self):
 class RepeatingTransaction(Transaction):
     """
     A transaction that repeats on a schedule described by a
-    `dateutil.rrule.rrule`.
+    `dateutil.rrule.rrule` or `dateutil.rrule.rruleset`.
     """
 
-    schedule: rrule
+    schedule: rrule | rruleset
 
     def __hash__(self):
         return hash((self.amount, self.description, self.schedule))

diff --git a/src/budge/transfer.py b/src/budge/transfer.py
@@ -31,7 +31,7 @@ def __post_init__(self, from_account: Account, to_account: Account):
 class RepeatingTransfer(Transfer, RepeatingTransaction):
     """
     A transfer between two accounts that repeats on a schedule described by a
-    `dateutil.rrule.rrule`.
+    `dateutil.rrule.rrule` or `dateutil.rrule.rruleset`.
     """
 
     def __post_init__(self, from_account: Account, to_account: Account):