mirror of
https://gh.llkk.cc/https://github.com/WeblateOrg/hosted.git
synced 2026-07-28 06:30:11 +08:00
92 lines
3.4 KiB
Python
92 lines
3.4 KiB
Python
#
|
|
# Copyright © Michal Čihař <michal@weblate.org>
|
|
#
|
|
# This file is part of Weblate <https://weblate.org/>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
from django import forms
|
|
from django.core.exceptions import ValidationError
|
|
from django.utils.translation import gettext_lazy as _
|
|
from weblate.billing.models import Billing, Plan
|
|
|
|
from wlhosted.integrations.utils import get_origin
|
|
from wlhosted.payments.models import Customer, Payment
|
|
|
|
|
|
class ChooseBillingForm(forms.Form):
|
|
billing = forms.ModelChoiceField(
|
|
queryset=Billing.objects.none(),
|
|
label=_("Billing"),
|
|
help_text=_("Choose the billing plan you want to update"),
|
|
empty_label=_("Create new billing plan"),
|
|
required=False,
|
|
)
|
|
plan = forms.ModelChoiceField(
|
|
queryset=Plan.objects.public(), widget=forms.HiddenInput, required=False
|
|
)
|
|
|
|
def __init__(self, user, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.fields["billing"].queryset = Billing.objects.for_user(user)
|
|
|
|
|
|
class BillingForm(ChooseBillingForm):
|
|
plan = forms.ModelChoiceField(
|
|
queryset=Plan.objects.public(), widget=forms.HiddenInput
|
|
)
|
|
period = forms.ChoiceField(
|
|
choices=[("y", "y"), ("m", "m")], widget=forms.HiddenInput
|
|
)
|
|
extra_domain = forms.BooleanField(required=False, widget=forms.HiddenInput)
|
|
|
|
def __init__(self, user, *args, **kwargs):
|
|
super().__init__(user, *args, **kwargs)
|
|
self.fields["plan"].queryset = Plan.objects.public(user)
|
|
|
|
def clean(self):
|
|
plan = self.cleaned_data.get("plan")
|
|
period = self.cleaned_data.get("period")
|
|
if not plan or not period:
|
|
return
|
|
if not plan.yearly_price and period == "y":
|
|
raise ValidationError("Plan does not support yearly billing!")
|
|
if not plan.price and period == "m":
|
|
raise ValidationError("Plan does not support monthly billing!")
|
|
|
|
def create_payment(self, user):
|
|
customer = Customer.objects.get_or_create(
|
|
origin=get_origin(), user_id=user.id, defaults={"email": user.email}
|
|
)[0]
|
|
|
|
plan = self.cleaned_data["plan"]
|
|
period = self.cleaned_data["period"]
|
|
description = "Weblate hosting ({}, {})".format(
|
|
plan.name, "Monthly" if period == "m" else "Yearly"
|
|
)
|
|
amount = plan.price if period == "m" else plan.yearly_price
|
|
if self.cleaned_data["extra_domain"]:
|
|
amount += 100
|
|
description += " + Custom domain"
|
|
extra = {"plan": plan.pk, "period": period}
|
|
if self.cleaned_data["billing"]:
|
|
extra["billing"] = self.cleaned_data["billing"].pk
|
|
return Payment.objects.create(
|
|
amount=amount,
|
|
description=description,
|
|
recurring=self.cleaned_data["period"],
|
|
customer=customer,
|
|
extra=extra,
|
|
)
|
