From 52a911f9d31de31bcc1858b96f1f8c44289338d1 Mon Sep 17 00:00:00 2001 From: Son Date: Wed, 2 Mar 2022 19:04:45 +0100 Subject: [PATCH] add extend subscription for 1 month to admin --- app/admin_model.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/app/admin_model.py b/app/admin_model.py index 2ea0eb95..5bc7a2d9 100644 --- a/app/admin_model.py +++ b/app/admin_model.py @@ -226,7 +226,19 @@ class ManualSubscriptionAdmin(SLModelView): def extend_1y(self, ids): for ms in ManualSubscription.filter(ManualSubscription.id.in_(ids)): ms.end_at = ms.end_at.shift(years=1) - flash(f"Extend subscription for {ms.user}", "success") + flash(f"Extend subscription for 1 year for {ms.user}", "success") + + Session.commit() + + @action( + "extend_1m", + "Extend for 1 month", + "Extend 1 month more?", + ) + def extend_1m(self, ids): + for ms in ManualSubscription.filter(ManualSubscription.id.in_(ids)): + ms.end_at = ms.end_at.shift(months=1) + flash(f"Extend subscription for 1 month for {ms.user}", "success") Session.commit()