diff --git a/nummi/main/views.py b/nummi/main/views.py index d8cb5ee..75ff656 100644 --- a/nummi/main/views.py +++ b/nummi/main/views.py @@ -96,7 +96,10 @@ class TransactionCreateView(NummiCreateView): def get_form_kwargs(self): return super().get_form_kwargs() | { "initial": { - "snapshot": self.kwargs.get("snapshot"), + "snapshot": ( + self.kwargs.get("snapshot") + or Snapshot.objects.filter(user=self.request.user).first() + ), }, } @@ -128,7 +131,10 @@ class SnapshotCreateView(NummiCreateView): def get_form_kwargs(self): return super().get_form_kwargs() | { "initial": { - "account": self.kwargs.get("account"), + "account": ( + self.kwargs.get("account") + or Account.objects.filter(user=self.request.user).first() + ), }, }