from django import forms
from .models import Wallet, Transaction


class WalletForm(forms.ModelForm):
    class Meta:
        model = Wallet
        fields = [
            "principal",
            "balance",
            "deposit",
            "earnings",
            "coins",
            "withdrawal_balance",
        ]
        widgets = {
            "principal": forms.Select(attrs={"class": "form-control"}),
            "balance": forms.NumberInput(attrs={"class": "form-control"}),
            "deposit": forms.NumberInput(attrs={"class": "form-control"}),
            "earnings": forms.NumberInput(attrs={"class": "form-control"}),
            "coins": forms.NumberInput(attrs={"class": "form-control"}),
            "withdrawal_balance": forms.NumberInput(attrs={"class": "form-control"}),
        }


class TransactionForm(forms.ModelForm):
    class Meta:
        model = Transaction
        fields = [
            "principal",
            "principal_subscription",
            "transaction_type",
            "payment_method",
            "transaction_status",
            "amount",
            "comment",
            "order_id",
            "product_id",
            "reference_id",
        ]
        widgets = {
            "principal": forms.Select(attrs={"class": "form-control"}),
            "principal_subscription": forms.Select(attrs={"class": "form-control"}),
            "transaction_type": forms.Select(attrs={"class": "form-control"}),
            "payment_method": forms.Select(attrs={"class": "form-control"}),
            "transaction_status": forms.Select(attrs={"class": "form-control"}),
            "amount": forms.NumberInput(attrs={"class": "form-control"}),
            "comment": forms.TextInput(attrs={"class": "form-control"}),
            "order_id": forms.TextInput(attrs={"class": "form-control"}),
            "product_id": forms.TextInput(attrs={"class": "form-control"}),
            "reference_id": forms.TextInput(attrs={"class": "form-control"}),
        }