fix: add check if donation ticket has payment method enabled (#6789)

fossasia · Jan 26, 2020 · ac1ae07 · ac1ae07
1 parent 1559c4c
commit ac1ae07
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/app/api/tickets.py b/app/api/tickets.py
@@ -70,7 +70,7 @@ def before_create_object(self, data, view_kwargs):
                     {'event_id': data['event']}, "Event does not exist"
                 )
 
-            if data.get('type') == 'paid':
+            if data.get('type') == 'paid' or data.get('type') == 'donation':
                 if not event.is_payment_enabled():
                     raise UnprocessableEntity(
                         {'event_id': data['event']},
@@ -231,7 +231,7 @@ def before_update_object(self, ticket, data, view_kwargs):
         :param view_kwargs:
         :return:
         """
-        if ticket.type == 'paid':
+        if ticket.type == 'paid' or ticket.type == 'donation':
             try:
                 event = (
                     db.session.query(Event)